From e749e8f80d1edbf120bf76547cdf6a99c3fa8bb4 Mon Sep 17 00:00:00 2001 From: Jonathan McCrohan Date: Tue, 12 Jun 2012 22:21:48 +0100 Subject: Imported Upstream version 1.3.1 --- transmission-remote-cli | 3191 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 3191 insertions(+) create mode 100755 transmission-remote-cli (limited to 'transmission-remote-cli') diff --git a/transmission-remote-cli b/transmission-remote-cli new file mode 100755 index 0000000..d9896e1 --- /dev/null +++ b/transmission-remote-cli @@ -0,0 +1,3191 @@ +#!/usr/bin/env python +######################################################################## +# This is transmission-remote-cli, whereas 'cli' stands for 'Curses # +# Luminous Interface', a client for the daemon of the BitTorrent # +# client Transmission. # +# # +# This program is free software: you can redistribute it and/or modify # +# it under the terms of the GNU General Public License as published by # +# the Free Software Foundation, either version 3 of the License, or # +# (at your option) any later version. # +# # +# This program is distributed in the hope that it will be useful, # +# but WITHOUT ANY WARRANTY; without even the implied warranty of # +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # +# GNU General Public License for more details: # +# http://www.gnu.org/licenses/gpl-3.0.txt # +######################################################################## + +VERSION = '1.3.1' + +TRNSM_VERSION_MIN = '1.90' +TRNSM_VERSION_MAX = '2.52' +RPC_VERSION_MIN = 8 +RPC_VERSION_MAX = 14 + +# error codes +CONNECTION_ERROR = 1 +JSON_ERROR = 2 +CONFIGFILE_ERROR = 3 + + +# use simplejson if available because it seems to be faster +try: + import simplejson as json +except ImportError: + try: + # Python 2.6 comes with a json module ... + import json + # ...but there is also an old json module that doesn't support .loads/.dumps. + json.dumps ; json.dumps + except (ImportError,AttributeError): + quit("Please install simplejson or Python 2.6 or higher.") + +import time +import re +import base64 +import httplib +import urllib2 +import socket +socket.setdefaulttimeout(None) +import ConfigParser +from optparse import OptionParser, SUPPRESS_HELP +import sys +import os +import signal +import unicodedata +import locale +locale.setlocale(locale.LC_ALL, '') +import curses +import curses.ascii +from textwrap import wrap +from subprocess import call +import netrc + + +# optional features provided by non-standard modules +features = {'dns':False, 'geoip':False, 'ipy':False} +try: import adns; features['dns'] = True # resolve IP to host name +except ImportError: features['dns'] = False + +try: import GeoIP; features['geoip'] = True # show country peer seems to be in +except ImportError: features['geoip'] = False + +try: import IPy; features['ipy'] = True # extract ipv4 from ipv6 addresses +except ImportError: features['ipy'] = False + + +if features['ipy']: + IPV6_RANGE_6TO4 = IPy.IP('2002::/16') + IPV6_RANGE_TEREDO = IPy.IP('2001::/32') + IPV4_ONES = 0xffffffff + +if features['geoip']: + def country_code_by_addr_vany(geo_ip, geo_ip6, addr): + if '.' in addr: + return geo_ip.country_code_by_addr(addr) + if not ':' in addr: + return None + if features['ipy']: + ip = IPy.IP(addr) + if ip in IPV6_RANGE_6TO4: + addr = str(IPy.IP(ip.int() >> 80 & IPV4_ONES)) + return geo_ip.country_code_by_addr(addr) + elif ip in IPV6_RANGE_TEREDO: + addr = str(IPy.IP(ip.int() & IPV4_ONES ^ IPV4_ONES)) + return geo_ip.country_code_by_addr(addr) + if hasattr(geo_ip6, 'country_code_by_addr_v6'): + return geo_ip6.country_code_by_addr_v6(addr) + + +# define config defaults +config = ConfigParser.SafeConfigParser() +config.add_section('Connection') +config.set('Connection', 'password', '') +config.set('Connection', 'username', '') +config.set('Connection', 'port', '9091') +config.set('Connection', 'host', 'localhost') +config.set('Connection', 'path', '/transmission/rpc') +config.set('Connection', 'ssl', 'False') +config.add_section('Sorting') +config.set('Sorting', 'order', 'name') +config.add_section('Filtering') +config.set('Filtering', 'filter', '') +config.set('Filtering', 'invert', 'False') +config.add_section('Misc') +config.set('Misc', 'compact_list', 'False') +config.set('Misc', 'torrentname_is_progressbar', 'True') +config.add_section('Colors') +config.set('Colors', 'title_seed', 'bg:green,fg:black') +config.set('Colors', 'title_download', 'bg:blue,fg:black') +config.set('Colors', 'title_idle', 'bg:cyan,fg:black') +config.set('Colors', 'title_verify', 'bg:magenta,fg:black') +config.set('Colors', 'title_paused', 'bg:black,fg:white') +config.set('Colors', 'download_rate', 'bg:black,fg:blue') +config.set('Colors', 'upload_rate', 'bg:black,fg:red') +config.set('Colors', 'eta+ratio', 'bg:black,fg:white') +config.set('Colors', 'filter_status', 'bg:red,fg:black') +config.set('Colors', 'dialog', 'bg:black,fg:white') +config.set('Colors', 'dialog_important', 'bg:red,fg:black') +config.set('Colors', 'button', 'bg:white,fg:black') +config.set('Colors', 'button_focused', 'bg:black,fg:white') +config.set('Colors', 'file_prio_high', 'bg:red,fg:black') +config.set('Colors', 'file_prio_normal', 'bg:white,fg:black') +config.set('Colors', 'file_prio_low', 'bg:yellow,fg:black') +config.set('Colors', 'file_prio_off', 'bg:blue,fg:black') + + +class ColorManager: + def __init__(self, config): + self.config = dict() + for name in config.keys(): + self.config[name] = self._parse_color_pair(config[name]) + + def _parse_color_pair(self, pair): + # BG and FG are intentionally switched here because colors are always + # used with curses.A_REVERSE. (To be honest, I forgot why, probably + # has something to do with how highlighting focus works.) + bg_name = pair.split(',')[1].split(':')[1].upper() + fg_name = pair.split(',')[0].split(':')[1].upper() + return { 'id': len(self.config.keys()) + 1, + 'bg': eval('curses.COLOR_' + bg_name), + 'fg': eval('curses.COLOR_' + fg_name) } + + def get_id(self, name): return self.config[name]['id'] + def get_bg(self, name): return self.config[name]['bg'] + def get_fg(self, name): return self.config[name]['fg'] + def get_names(self): return self.config.keys() + + +class Normalizer: + def __init__(self): + self.values = {} + + def add(self, id, value, max): + if not id in self.values.keys(): + self.values[id] = [ float(value) ] + else: + if len(self.values[id]) >= max: + self.values[id].pop(0) + self.values[id].append(float(value)) + return self.get(id) + + def get(self, id): + if not id in self.values.keys(): + return 0.0 + return sum(self.values[id]) / len(self.values[id]) + + + +authhandler = None +session_id = 0 + +# Handle communication with Transmission server. +class TransmissionRequest: + def __init__(self, host, port, path, method=None, tag=None, arguments=None): + self.url = create_url(host, port, path) + self.open_request = None + self.last_update = 0 + if method and tag: + self.set_request_data(method, tag, arguments) + + def set_request_data(self, method, tag, arguments=None): + request_data = {'method':method, 'tag':tag} + if arguments: request_data['arguments'] = arguments + self.http_request = urllib2.Request(url=self.url, data=json.dumps(request_data)) + + def send_request(self): + """Ask for information from server OR submit command.""" + + global session_id + try: + if session_id: + self.http_request.add_header('X-Transmission-Session-Id', session_id) + self.open_request = urllib2.urlopen(self.http_request) + except AttributeError: + # request data (http_request) isn't specified yet -- data will be available on next call + pass + + # authentication + except urllib2.HTTPError, e: + try: + msg = html2text(str(e.read())) + except: + msg = str(e) + + # extract session id and send request again + m = re.search('X-Transmission-Session-Id:\s*(\w+)', msg) + try: + session_id = m.group(1) + self.send_request() + except AttributeError: + quit(str(msg) + "\n", CONNECTION_ERROR) + + except urllib2.URLError, msg: + try: + reason = msg.reason[1] + except IndexError: + reason = str(msg.reason) + quit("Cannot connect to %s: %s\n" % (self.http_request.host, reason), CONNECTION_ERROR) + + def get_response(self): + """Get response to previously sent request.""" + + if self.open_request == None: + return {'result': 'no open request'} + response = self.open_request.read() + # work around regression in Python 2.6.5, caused by http://bugs.python.org/issue8797 + if authhandler: + authhandler.retried = 0 + try: + data = json.loads(unicode(response)) + except ValueError: + quit("Cannot parse response: %s\n" % response, JSON_ERROR) + self.open_request = None + return data + + +# End of Class TransmissionRequest + + +# Higher level of data exchange +class Transmission: + STATUS_STOPPED = 0 # Torrent is stopped + STATUS_CHECK_WAIT = 1 # Queued to check files + STATUS_CHECK = 2 # Checking files + STATUS_DOWNLOAD_WAIT = 3 # Queued to download + STATUS_DOWNLOAD = 4 # Downloading + STATUS_SEED_WAIT = 5 # Queued to seed + STATUS_SEED = 6 # Seeding + + TAG_TORRENT_LIST = 7 + TAG_TORRENT_DETAILS = 77 + TAG_SESSION_STATS = 21 + TAG_SESSION_GET = 22 + + LIST_FIELDS = [ 'id', 'name', 'downloadDir', 'status', 'trackerStats', 'desiredAvailable', + 'rateDownload', 'rateUpload', 'eta', 'uploadRatio', + 'sizeWhenDone', 'haveValid', 'haveUnchecked', 'addedDate', + 'uploadedEver', 'errorString', 'recheckProgress', + 'peersConnected', 'uploadLimit', 'downloadLimit', + 'uploadLimited', 'downloadLimited', 'bandwidthPriority', + 'peersSendingToUs', 'peersGettingFromUs', + 'seedRatioLimit', 'seedRatioMode', 'isPrivate' ] + + DETAIL_FIELDS = [ 'files', 'priorities', 'wanted', 'peers', 'trackers', + 'activityDate', 'dateCreated', 'startDate', 'doneDate', + 'totalSize', 'leftUntilDone', 'comment', + 'hashString', 'pieceCount', 'pieceSize', 'pieces', + 'downloadedEver', 'corruptEver', 'peersFrom' ] + LIST_FIELDS + + def __init__(self, host, port, path, username, password): + self.host = host + self.port = port + self.path = path + + if username and password: + password_mgr = urllib2.HTTPPasswordMgrWithDefaultRealm() + password_mgr.add_password(None, create_url(host, port, path), username, password) + global authhandler + authhandler = urllib2.HTTPBasicAuthHandler(password_mgr) + opener = urllib2.build_opener(authhandler) + urllib2.install_opener(opener) + + # check rpc version + request = TransmissionRequest(host, port, path, 'session-get', self.TAG_SESSION_GET) + request.send_request() + response = request.get_response() + + self.rpc_version = response['arguments']['rpc-version'] + + # rpc version too old? + version_error = "Unsupported Transmission version: " + str(response['arguments']['version']) + \ + " -- RPC protocol version: " + str(response['arguments']['rpc-version']) + "\n" + + min_msg = "Please install Transmission version " + TRNSM_VERSION_MIN + " or higher.\n" + try: + if response['arguments']['rpc-version'] < RPC_VERSION_MIN: + quit(version_error + min_msg) + except KeyError: + quit(version_error + min_msg) + + # rpc version too new? + if response['arguments']['rpc-version'] > RPC_VERSION_MAX: + quit(version_error + "Please install Transmission version " + TRNSM_VERSION_MAX + " or lower.\n") + + # setup compatibility to Transmission <2.40 + if self.rpc_version < 14: + Transmission.STATUS_CHECK_WAIT = 1 << 0 + Transmission.STATUS_CHECK = 1 << 1 + Transmission.STATUS_DOWNLOAD_WAIT = 1 << 2 + Transmission.STATUS_DOWNLOAD = 1 << 2 + Transmission.STATUS_SEED_WAIT = 1 << 3 + Transmission.STATUS_SEED = 1 << 3 + Transmission.STATUS_STOPPED = 1 << 4 + + # set up request list + self.requests = {'torrent-list': + TransmissionRequest(host, port, path, 'torrent-get', self.TAG_TORRENT_LIST, {'fields': self.LIST_FIELDS}), + 'session-stats': + TransmissionRequest(host, port, path, 'session-stats', self.TAG_SESSION_STATS, 21), + 'session-get': + TransmissionRequest(host, port, path, 'session-get', self.TAG_SESSION_GET), + 'torrent-details': + TransmissionRequest(host, port, path)} + + self.torrent_cache = [] + self.status_cache = dict() + self.torrent_details_cache = dict() + self.peer_progress_cache = dict() + self.hosts_cache = dict() + self.geo_ips_cache = dict() + if features['dns']: self.resolver = adns.init() + if features['geoip']: + self.geo_ip = GeoIP.new(GeoIP.GEOIP_MEMORY_CACHE) + try: + self.geo_ip6 = GeoIP.open_type(GeoIP.GEOIP_COUNTRY_EDITION_V6, GeoIP.GEOIP_MEMORY_CACHE); + except AttributeError: self.geo_ip6 = None + except GeoIP.error: self.geo_ip6 = None + + # make sure there are no undefined values + self.wait_for_torrentlist_update() + self.requests['torrent-details'] = TransmissionRequest(self.host, self.port, self.path) + + + def update(self, delay, tag_waiting_for=0): + """Maintain up-to-date data.""" + + tag_waiting_for_occurred = False + + for request in self.requests.values(): + if time.time() - request.last_update >= delay: + request.last_update = time.time() + response = request.get_response() + + if response['result'] == 'no open request': + request.send_request() + + elif response['result'] == 'success': + tag = self.parse_response(response) + if tag == tag_waiting_for: + tag_waiting_for_occurred = True + + if tag_waiting_for: + return tag_waiting_for_occurred + else: + return None + + + + def parse_response(self, response): + # response is a reply to torrent-get + if response['tag'] == self.TAG_TORRENT_LIST or response['tag'] == self.TAG_TORRENT_DETAILS: + for t in response['arguments']['torrents']: + t['uploadRatio'] = round(float(t['uploadRatio']), 2) + t['percentDone'] = percent(float(t['sizeWhenDone']), + float(t['haveValid'] + t['haveUnchecked'])) + t['available'] = t['desiredAvailable'] + t['haveValid'] + t['haveUnchecked'] + if t['downloadDir'][-1] != '/': + t['downloadDir'] += '/' + try: + t['seeders'] = max(map(lambda x: x['seederCount'], t['trackerStats'])) + t['leechers'] = max(map(lambda x: x['leecherCount'], t['trackerStats'])) + except ValueError: + t['seeders'] = t['leechers'] = -1 + + if response['tag'] == self.TAG_TORRENT_LIST: + self.torrent_cache = response['arguments']['torrents'] + + elif response['tag'] == self.TAG_TORRENT_DETAILS: + # torrent list may be empty sometimes after deleting + # torrents. no idea why and why the server sends us + # TAG_TORRENT_DETAILS, but just passing seems to help.(?) + try: + torrent_details = response['arguments']['torrents'][0] + torrent_details['pieces'] = base64.decodestring(torrent_details['pieces']) + self.torrent_details_cache = torrent_details + self.upgrade_peerlist() + except IndexError: + pass + + elif response['tag'] == self.TAG_SESSION_STATS: + self.status_cache.update(response['arguments']) + + elif response['tag'] == self.TAG_SESSION_GET: + self.status_cache.update(response['arguments']) + + return response['tag'] + + def upgrade_peerlist(self): + for index,peer in enumerate(self.torrent_details_cache['peers']): + ip = peer['address'] + peerid = ip + self.torrent_details_cache['hashString'] + + # make sure peer cache exists + if not self.peer_progress_cache.has_key(peerid): + self.peer_progress_cache[peerid] = {'last_progress':peer['progress'], 'last_update':time.time(), + 'download_speed':0, 'time_left':0} + + # estimate how fast a peer is downloading + if peer['progress'] < 1: + this_time = time.time() + time_diff = this_time - self.peer_progress_cache[peerid]['last_update'] + progress_diff = peer['progress'] - self.peer_progress_cache[peerid]['last_progress'] + if self.peer_progress_cache[peerid]['last_progress'] and progress_diff > 0 and time_diff > 5: + downloaded = self.torrent_details_cache['totalSize'] * progress_diff + avg_speed = downloaded / time_diff + + if self.peer_progress_cache[peerid]['download_speed'] > 0: # make it less jumpy + avg_speed = (self.peer_progress_cache[peerid]['download_speed'] + avg_speed) /2 + + download_left = self.torrent_details_cache['totalSize'] - \ + (self.torrent_details_cache['totalSize']*peer['progress']) + time_left = download_left / avg_speed + + self.peer_progress_cache[peerid]['last_update'] = this_time # remember update time + self.peer_progress_cache[peerid]['download_speed'] = avg_speed + self.peer_progress_cache[peerid]['time_left'] = time_left + + self.peer_progress_cache[peerid]['last_progress'] = peer['progress'] # remember progress + self.torrent_details_cache['peers'][index].update(self.peer_progress_cache[peerid]) + + # resolve and locate peer's ip + if features['dns'] and not self.hosts_cache.has_key(ip): + try: + self.hosts_cache[ip] = self.resolver.submit_reverse(ip, adns.rr.PTR) + except adns.Error: + pass + if features['geoip'] and not self.geo_ips_cache.has_key(ip): + self.geo_ips_cache[ip] = country_code_by_addr_vany(self.geo_ip, self.geo_ip6, ip) + if self.geo_ips_cache[ip] == None: + self.geo_ips_cache[ip] = '?' + + def get_rpc_version(self): + return self.rpc_version + + def get_global_stats(self): + return self.status_cache + + def get_torrent_list(self, sort_orders): + try: + for sort_order in sort_orders: + if isinstance(self.torrent_cache[0][sort_order['name']], (str, unicode)): + self.torrent_cache.sort(key=lambda x: x[sort_order['name']].lower(), + reverse=sort_order['reverse']) + else: + self.torrent_cache.sort(key=lambda x: x[sort_order['name']], + reverse=sort_order['reverse']) + except IndexError: + return [] + return self.torrent_cache + + def get_torrent_by_id(self, id): + i = 0 + while self.torrent_cache[i]['id'] != id: i += 1 + if self.torrent_cache[i]['id'] == id: + return self.torrent_cache[i] + else: + return None + + + def get_torrent_details(self): + return self.torrent_details_cache + def set_torrent_details_id(self, id): + if id < 0: + self.requests['torrent-details'] = TransmissionRequest(self.host, self.port, self.path) + else: + self.requests['torrent-details'].set_request_data('torrent-get', self.TAG_TORRENT_DETAILS, + {'ids':id, 'fields': self.DETAIL_FIELDS}) + + def get_hosts(self): + return self.hosts_cache + + def get_geo_ips(self): + return self.geo_ips_cache + + + def set_option(self, option_name, option_value): + request = TransmissionRequest(self.host, self.port, self.path, 'session-set', 1, {option_name: option_value}) + request.send_request() + self.wait_for_status_update() + + + def set_rate_limit(self, direction, new_limit, torrent_id=-1): + data = dict() + if new_limit <= -1: + new_limit = None + limit_enabled = False + else: + limit_enabled = True + + if torrent_id < 0: + type = 'session-set' + data['speed-limit-'+direction] = new_limit + data['speed-limit-'+direction+'-enabled'] = limit_enabled + else: + type = 'torrent-set' + data['ids'] = [torrent_id] + data[direction+'loadLimit'] = new_limit + data[direction+'loadLimited'] = limit_enabled + + request = TransmissionRequest(self.host, self.port, self.path, type, 1, data) + request.send_request() + self.wait_for_torrentlist_update() + + + def set_seed_ratio(self, ratio, torrent_id=-1): + data = dict() + if ratio == -1: + ratio = None + mode = 0 # Use global settings + elif ratio == 0: + ratio = None + mode = 2 # Seed regardless of ratio + elif ratio >= 0: + mode = 1 # Stop seeding at seedRatioLimit + else: + return + + data['ids'] = [torrent_id] + data['seedRatioLimit'] = ratio + data['seedRatioMode'] = mode + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-set', 1, data) + request.send_request() + self.wait_for_torrentlist_update() + + + def increase_bandwidth_priority(self, torrent_id): + torrent = self.get_torrent_by_id(torrent_id) + if torrent == None or torrent['bandwidthPriority'] >= 1: + return False + else: + new_priority = torrent['bandwidthPriority'] + 1 + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-set', 1, + {'ids': [torrent_id], 'bandwidthPriority':new_priority}) + request.send_request() + self.wait_for_torrentlist_update() + + def decrease_bandwidth_priority(self, torrent_id): + torrent = self.get_torrent_by_id(torrent_id) + if torrent == None or torrent['bandwidthPriority'] <= -1: + return False + else: + new_priority = torrent['bandwidthPriority'] - 1 + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-set', 1, + {'ids': [torrent_id], 'bandwidthPriority':new_priority}) + request.send_request() + self.wait_for_torrentlist_update() + + + def toggle_turtle_mode(self): + self.set_option('alt-speed-enabled', not self.status_cache['alt-speed-enabled']) + + + def add_torrent(self, location): + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-add', 1, {'filename': location}) + request.send_request() + response = request.get_response() + if response['result'] != 'success': + return response['result'] + else: + return '' + + def stop_torrent(self, id): + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-stop', 1, {'ids': [id]}) + request.send_request() + self.wait_for_torrentlist_update() + + def start_torrent(self, id): + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-start', 1, {'ids': [id]}) + request.send_request() + self.wait_for_torrentlist_update() + + def verify_torrent(self, id): + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-verify', 1, {'ids': [id]}) + request.send_request() + self.wait_for_torrentlist_update() + + def reannounce_torrent(self, id): + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-reannounce', 1, {'ids': [id]}) + request.send_request() + self.wait_for_torrentlist_update() + + def move_torrent(self, torrent_id, new_location): + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-set-location', 1, + {'ids': torrent_id, 'location': new_location, 'move': True}) + request.send_request() + self.wait_for_torrentlist_update() + + def remove_torrent(self, id): + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-remove', 1, {'ids': [id]}) + request.send_request() + self.wait_for_torrentlist_update() + + def remove_torrent_local_data(self, id): + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-remove', 1, {'ids': [id], 'delete-local-data':True}) + request.send_request() + self.wait_for_torrentlist_update() + + def add_torrent_tracker(self, id, tracker): + data = { 'ids' : [id], + 'trackerAdd' : [tracker] } + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-set', 1, data) + request.send_request() + response = request.get_response() + return response['result'] if response['result'] != 'success' else '' + + def remove_torrent_tracker(self, id, tracker): + data = { 'ids' : [id], + 'trackerRemove' : [tracker] } + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-set', 1, data) + request.send_request() + response = request.get_response() + self.wait_for_torrentlist_update() + return response['result'] if response['result'] != 'success' else '' + + def increase_file_priority(self, file_nums): + file_nums = list(file_nums) + ref_num = file_nums[0] + for num in file_nums: + if not self.torrent_details_cache['wanted'][num]: + ref_num = num + break + elif self.torrent_details_cache['priorities'][num] < \ + self.torrent_details_cache['priorities'][ref_num]: + ref_num = num + current_priority = self.torrent_details_cache['priorities'][ref_num] + if not self.torrent_details_cache['wanted'][ref_num]: + self.set_file_priority(self.torrent_details_cache['id'], file_nums, 'low') + elif current_priority == -1: + self.set_file_priority(self.torrent_details_cache['id'], file_nums, 'normal') + elif current_priority == 0: + self.set_file_priority(self.torrent_details_cache['id'], file_nums, 'high') + + def decrease_file_priority(self, file_nums): + file_nums = list(file_nums) + ref_num = file_nums[0] + for num in file_nums: + if self.torrent_details_cache['priorities'][num] > \ + self.torrent_details_cache['priorities'][ref_num]: + ref_num = num + current_priority = self.torrent_details_cache['priorities'][ref_num] + if current_priority >= 1: + self.set_file_priority(self.torrent_details_cache['id'], file_nums, 'normal') + elif current_priority == 0: + self.set_file_priority(self.torrent_details_cache['id'], file_nums, 'low') + elif current_priority == -1: + self.set_file_priority(self.torrent_details_cache['id'], file_nums, 'off') + + + def set_file_priority(self, torrent_id, file_nums, priority): + request_data = {'ids': [torrent_id]} + if priority == 'off': + request_data['files-unwanted'] = file_nums + else: + request_data['files-wanted'] = file_nums + request_data['priority-' + priority] = file_nums + request = TransmissionRequest(self.host, self.port, self.path, 'torrent-set', 1, request_data) + request.send_request() + self.wait_for_details_update() + + def get_file_priority(self, torrent_id, file_num): + priority = self.torrent_details_cache['priorities'][file_num] + if not self.torrent_details_cache['wanted'][file_num]: return 'off' + elif priority <= -1: return 'low' + elif priority == 0: return 'normal' + elif priority >= 1: return 'high' + return '?' + + def wait_for_torrentlist_update(self): + self.wait_for_update(7) + def wait_for_details_update(self): + self.wait_for_update(77) + def wait_for_status_update(self): + self.wait_for_update(22) + def wait_for_update(self, update_id): + self.update(0) # send request + while True: # wait for response + if self.update(0, update_id): break + time.sleep(0.1) + + def get_status(self, torrent): + if torrent['status'] == Transmission.STATUS_STOPPED: + status = 'paused' + elif torrent['status'] == Transmission.STATUS_CHECK: + status = 'verifying' + elif torrent['status'] == Transmission.STATUS_CHECK_WAIT: + status = 'will verify' + elif torrent['status'] == Transmission.STATUS_DOWNLOAD: + status = ('idle','downloading')[torrent['rateDownload'] > 0] + elif torrent['status'] == Transmission.STATUS_DOWNLOAD_WAIT: + status = 'will download' + elif torrent['status'] == Transmission.STATUS_SEED: + status = 'seeding' + elif torrent['status'] == Transmission.STATUS_SEED_WAIT: + status = 'will seed' + else: + status = 'unknown state' + return status + + def get_bandwidth_priority(self, torrent): + if torrent['bandwidthPriority'] == -1: + return '-' + elif torrent['bandwidthPriority'] == 0: + return ' ' + elif torrent['bandwidthPriority'] == 1: + return '+' + else: + return '?' + +# End of Class Transmission + + + + + +# User Interface +class Interface: + TRACKER_ITEM_HEIGHT = 6 + + def __init__(self): + self.filter_list = config.get('Filtering', 'filter') + self.filter_inverse = config.getboolean('Filtering', 'invert') + self.sort_orders = parse_sort_str(config.get('Sorting', 'order')) + self.compact_list = config.getboolean('Misc', 'compact_list') + self.torrentname_is_progressbar = config.getboolean('Misc', 'torrentname_is_progressbar') + + self.torrents = server.get_torrent_list(self.sort_orders) + self.stats = server.get_global_stats() + self.torrent_details = [] + self.selected_torrent = -1 # changes to >-1 when focus >-1 & user hits return + self.all_paused = False + self.highlight_dialog = False + self.search_focus = 0 # like self.focus but for searches in torrent list + self.focused_id = -1 # the id (provided by Transmission) of self.torrents[self.focus] + self.focus = -1 # -1: nothing focused; 0: top of list; <# of torrents>-1: bottom of list + self.scrollpos = 0 # start of torrentlist + self.torrents_per_page = 0 # will be set by manage_layout() + self.rateDownload_width = self.rateUpload_width = 2 + + self.details_category_focus = 0 # overview/files/peers/tracker in details + self.focus_detaillist = -1 # same as focus but for details + self.selected_files = [] # marked files in details + self.scrollpos_detaillist = 0 # same as scrollpos but for details + self.compact_torrentlist = False # draw only one line for each torrent in compact mode + self.exit_now = False + + self.keybindings = { + ord('?'): self.call_list_key_bindings, + curses.KEY_F1: self.call_list_key_bindings, + 27: self.go_back_or_unfocus, + curses.KEY_BREAK: self.go_back_or_unfocus, + 12: self.go_back_or_unfocus, + curses.KEY_BACKSPACE: self.leave_details, + ord('q'): self.go_back_or_quit, + ord('o'): self.o_key, + ord('\n'): self.select_torrent_detail_view, + curses.KEY_RIGHT: self.right_key, + ord('l'): self.l_key, + ord('s'): self.show_sort_order_menu, + ord('f'): self.f_key, + ord('u'): self.global_upload, + ord('d'): self.global_download, + ord('U'): self.torrent_upload, + ord('D'): self.torrent_download, + ord('L'): self.seed_ratio, + ord('t'): self.t_key, + ord('+'): self.bandwidth_priority, + ord('-'): self.bandwidth_priority, + ord('p'): self.pause_unpause_torrent, + ord('P'): self.pause_unpause_all_torrent, + ord('v'): self.verify_torrent, + ord('y'): self.verify_torrent, + ord('r'): self.r_key, + curses.KEY_DC: self.r_key, + ord('R'): self.remove_torrent_local_data, + curses.KEY_SDC: self.remove_torrent_local_data, + curses.KEY_UP: self.movement_keys, + ord('k'): self.movement_keys, + curses.KEY_DOWN: self.movement_keys, + ord('j'): self.movement_keys, + curses.KEY_PPAGE: self.movement_keys, + curses.KEY_NPAGE: self.movement_keys, + curses.KEY_HOME: self.movement_keys, + curses.KEY_END: self.movement_keys, + ord('g'): self.movement_keys, + ord('G'): self.movement_keys, + curses.ascii.ctrl(ord('f')): self.movement_keys, + curses.ascii.ctrl(ord('b')): self.movement_keys, + curses.ascii.ctrl(ord('n')): self.movement_keys, + curses.ascii.ctrl(ord('p')): self.movement_keys, + ord("\t"): self.move_in_details, + curses.KEY_BTAB: self.move_in_details, + ord('e'): self.move_in_details, + ord('c'): self.move_in_details, + ord('C'): self.toggle_compact_torrentlist, + ord('h'): self.file_pritority_or_switch_details, + curses.KEY_LEFT: self.file_pritority_or_switch_details, + ord(' '): self.space_key, + ord('a'): self.a_key, + ord('m'): self.move_torrent, + ord('n'): self.reannounce_torrent, + ord('/'): self.dialog_search_torrentlist + } + + self.sort_options = [ + ('name','_Name'), ('addedDate','_Age'), ('percentDone','_Progress'), + ('seeders','_Seeds'), ('leechers','Lee_ches'), ('sizeWhenDone', 'Si_ze'), + ('status','S_tatus'), ('uploadedEver','Up_loaded'), + ('rateUpload','_Upload Speed'), ('rateDownload','_Download Speed'), + ('uploadRatio','_Ratio'), ('peersConnected','P_eers'), + ('downloadDir', 'L_ocation'), ('reverse','Re_verse') + ] + + + try: + self.init_screen() + self.run() + except: + self.restore_screen() + (exc_type, exc_value, exc_traceback) = sys.exc_info() + raise exc_type, exc_value, exc_traceback + else: + self.restore_screen() + + + def init_screen(self): + os.environ['ESCDELAY'] = '0' # make escape usable + self.screen = curses.initscr() + curses.noecho() ; curses.cbreak() ; self.screen.keypad(1) + curses.halfdelay(10) # STDIN timeout + + hide_cursor() + + # enable colors if available + try: + curses.start_color() + self.colors = ColorManager(dict(config.items('Colors'))) + for name in sorted(self.colors.get_names()): + curses.init_pair(self.colors.get_id(name), + self.colors.get_fg(name), + self.colors.get_bg(name)) + except: + pass + + # http://bugs.python.org/issue2675 + try: + del os.environ['LINES'] + del os.environ['COLUMNS'] + except: + pass + + # http://bugs.python.org/issue2675 + try: + del os.environ['LINES'] + del os.environ['COLUMNS'] + except: + pass + + signal.signal(signal.SIGWINCH, lambda y,frame: self.get_screen_size()) + self.get_screen_size() + + def restore_screen(self): + curses.endwin() + + def get_screen_size(self): + time.sleep(0.1) # prevents curses.error on rapid resizing + while True: + curses.endwin() + self.screen.refresh() + self.height, self.width = self.screen.getmaxyx() + # Tracker list breaks if width smaller than 73 + if self.width < 73 or self.height < 16: + self.screen.erase() + self.screen.addstr(0,0, "Terminal too small", curses.A_REVERSE + curses.A_BOLD) + time.sleep(1) + else: + break + self.manage_layout() + + def manage_layout(self): + self.recalculate_torrents_per_page() + self.pad_height = max((len(self.torrents)+1) * self.tlist_item_height, self.height) + self.pad = curses.newpad(self.pad_height, self.width) + self.detaillistitems_per_page = self.height - 8 + + if self.selected_torrent > -1: + self.rateDownload_width = self.get_rateDownload_width([self.torrent_details]) + self.rateUpload_width = self.get_rateUpload_width([self.torrent_details]) + self.torrent_title_width = self.width - self.rateUpload_width - 2 + # show downloading column only if torrents is downloading + if self.torrent_details['status'] == Transmission.STATUS_DOWNLOAD: + self.torrent_title_width -= self.rateDownload_width + 2 + + elif self.torrents: + visible_torrents = self.torrents[self.scrollpos/self.tlist_item_height : self.scrollpos/self.tlist_item_height + self.torrents_per_page + 1] + self.rateDownload_width = self.get_rateDownload_width(visible_torrents) + self.rateUpload_width = self.get_rateUpload_width(visible_torrents) + + self.torrent_title_width = self.width - self.rateUpload_width - 2 + # show downloading column only if any downloading torrents are visible + if filter(lambda x: x['status']==Transmission.STATUS_DOWNLOAD, visible_torrents): + self.torrent_title_width -= self.rateDownload_width + 2 + else: + self.torrent_title_width = 80 + + def get_rateDownload_width(self, torrents): + new_width = max(map(lambda x: len(scale_bytes(x['rateDownload'])), torrents)) + new_width = max(max(map(lambda x: len(scale_time(x['eta'])), torrents)), new_width) + new_width = max(len(scale_bytes(self.stats['downloadSpeed'])), new_width) + new_width = max(self.rateDownload_width, new_width) # don't shrink + return new_width + + def get_rateUpload_width(self, torrents): + new_width = max(map(lambda x: len(scale_bytes(x['rateUpload'])), torrents)) + new_width = max(max(map(lambda x: len(num2str(x['uploadRatio'], '%.02f')), torrents)), new_width) + new_width = max(len(scale_bytes(self.stats['uploadSpeed'])), new_width) + new_width = max(self.rateUpload_width, new_width) # don't shrink + return new_width + + def recalculate_torrents_per_page(self): + self.tlist_item_height = 3 if not self.compact_list else 1 + self.mainview_height = self.height - 2 + self.torrents_per_page = self.mainview_height / self.tlist_item_height + + def run(self): + self.draw_title_bar() + self.draw_stats() + self.draw_torrent_list() + + while True: + server.update(1) + + # display torrentlist + if self.selected_torrent == -1: + self.draw_torrent_list() + + # display some torrent's details + else: + self.draw_details() + + self.stats = server.get_global_stats() + self.draw_title_bar() # show shortcuts and stuff + self.draw_stats() # show global states + self.screen.move(0,0) # in case cursor can't be invisible + self.handle_user_input() + if self.exit_now: + sort_str = ','.join(map(lambda x: ('','reverse:')[x['reverse']] + x['name'], self.sort_orders)) + config.set('Sorting', 'order', sort_str) + config.set('Filtering', 'filter', self.filter_list) + config.set('Filtering', 'invert', str(self.filter_inverse)) + config.set('Misc', 'compact_list', str(self.compact_list)) + config.set('Misc', 'torrentname_is_progressbar', str(self.torrentname_is_progressbar)) + save_config(cmd_args.configfile) + return + + def go_back_or_unfocus(self, c): + if self.focus_detaillist > -1: # unfocus and deselect file + self.focus_detaillist = -1 + self.scrollpos_detaillist = 0 + self.selected_files = [] + elif self.selected_torrent > -1: # return from details + self.details_category_focus = 0 + self.selected_torrent = -1 + self.selected_files = [] + else: + if self.focus > -1: + self.scrollpos = 0 # unfocus main list + self.focus = -1 + elif self.filter_list: + self.filter_list = '' # reset filter + + def leave_details(self, c): + if self.selected_torrent > -1: + server.set_torrent_details_id(-1) + self.selected_torrent = -1 + self.details_category_focus = 0 + self.scrollpos_detaillist = 0 + self.selected_files = [] + + def go_back_or_quit(self, c): + if self.selected_torrent == -1: + self.exit_now = True + else: # return to list view + server.set_torrent_details_id(-1) + self.selected_torrent = -1 + self.details_category_focus = 0 + self.focus_detaillist = -1 + self.scrollpos_detaillist = 0 + self.selected_files = [] + + def space_key(self, c): + # File list + if self.selected_torrent > -1 and self.details_category_focus == 1: + self.select_unselect_file(c) + # Torrent list + elif self.selected_torrent == -1: + self.select_torrent_detail_view(c) + + def a_key(self, c): + # File list + if self.selected_torrent > -1 and self.details_category_focus == 1: + self.select_unselect_file(c) + # Trackers + elif self.selected_torrent > -1 and self.details_category_focus == 3: + self.add_tracker() + + # Do nothing in other detail tabs + elif self.selected_torrent > -1: + pass + else: + self.add_torrent() + + def o_key(self, c): + if self.selected_torrent == -1: + self.draw_options_dialog() + elif self.selected_torrent > -1: + self.details_category_focus = 0 + + def l_key(self, c): + if self.focus > -1 and self.selected_torrent == -1: + self.select_torrent_detail_view(c) + elif self.selected_torrent > -1: + self.file_pritority_or_switch_details(c) + + def t_key(self, c): + if self.selected_torrent == -1: + server.toggle_turtle_mode() + elif self.selected_torrent > -1: + self.details_category_focus = 3 + + def f_key(self, c): + if self.selected_torrent == -1: + self.show_state_filter_menu(c) + elif self.selected_torrent > -1: + self.details_category_focus = 1 + + def r_key(self, c): + # Torrent list + if self.selected_torrent == -1: + self.remove_torrent(c) + # Trackers + elif self.selected_torrent > -1 and self.details_category_focus == 3: + self.remove_tracker() + + def right_key(self, c): + if self.focus > -1 and self.selected_torrent == -1: + self.select_torrent_detail_view(c) + else: + self.file_pritority_or_switch_details(c) + + def add_torrent(self): + location = self.dialog_input_text("Add torrent from file or URL", os.getcwd()) + if location: + error = server.add_torrent(location) + if error: + msg = wrap("Couldn't add torrent \"%s\":" % location) + msg.extend(wrap(error, self.width-4)) + self.dialog_ok("\n".join(msg)) + + def select_torrent_detail_view(self, c): + if self.focus > -1 and self.selected_torrent == -1: + self.screen.clear() + self.selected_torrent = self.focus + server.set_torrent_details_id(self.torrents[self.focus]['id']) + server.wait_for_details_update() + + def show_sort_order_menu(self, c): + if self.selected_torrent == -1: + choice = self.dialog_menu('Sort order', self.sort_options, + map(lambda x: x[0]==self.sort_orders[-1]['name'], self.sort_options).index(True)+1) + if choice != -128: + if choice == 'reverse': + self.sort_orders[-1]['reverse'] = not self.sort_orders[-1]['reverse'] + else: + self.sort_orders.append({'name':choice, 'reverse':False}) + while len(self.sort_orders) > 2: + self.sort_orders.pop(0) + + def show_state_filter_menu(self, c): + if self.selected_torrent == -1: + options = [('uploading','_Uploading'), ('downloading','_Downloading'), + ('active','Ac_tive'), ('paused','_Paused'), ('seeding','_Seeding'), + ('incomplete','In_complete'), ('verifying','Verif_ying'), ('private','P_rivate'), + ('invert','In_vert'), ('','_All')] + choice = self.dialog_menu(('Show only','Filter all')[self.filter_inverse], options, + map(lambda x: x[0]==self.filter_list, options).index(True)+1) + if choice != -128: + if choice == 'invert': + self.filter_inverse = not self.filter_inverse + else: + if choice == '': + self.filter_inverse = False + self.filter_list = choice + + def global_upload(self, c): + current_limit = (-1,self.stats['speed-limit-up'])[self.stats['speed-limit-up-enabled']] + limit = self.dialog_input_number("Global upload limit in kilobytes per second", current_limit) + if limit == -128: + return + server.set_rate_limit('up', limit) + + def global_download(self, c): + current_limit = (-1,self.stats['speed-limit-down'])[self.stats['speed-limit-down-enabled']] + limit = self.dialog_input_number("Global download limit in kilobytes per second", current_limit) + if limit == -128: + return + server.set_rate_limit('down', limit) + + def torrent_upload(self, c): + if self.focus > -1: + current_limit = (-1,self.torrents[self.focus]['uploadLimit'])[self.torrents[self.focus]['uploadLimited']] + limit = self.dialog_input_number("Upload limit in kilobytes per second for\n%s" % \ + self.torrents[self.focus]['name'], current_limit) + if limit == -128: + return + server.set_rate_limit('up', limit, self.torrents[self.focus]['id']) + + def torrent_download(self, c): + if self.focus > -1: + current_limit = (-1,self.torrents[self.focus]['downloadLimit'])[self.torrents[self.focus]['downloadLimited']] + limit = self.dialog_input_number("Download limit in Kilobytes per second for\n%s" % \ + self.torrents[self.focus]['name'], current_limit) + if limit == -128: + return + server.set_rate_limit('down', limit, self.torrents[self.focus]['id']) + + def seed_ratio(self, c): + if self.focus > -1: + if self.torrents[self.focus]['seedRatioMode'] == 0: # Use global settings + current_limit = '' + elif self.torrents[self.focus]['seedRatioMode'] == 1: # Stop seeding at seedRatioLimit + current_limit = self.torrents[self.focus]['seedRatioLimit'] + elif self.torrents[self.focus]['seedRatioMode'] == 2: # Seed regardless of ratio + current_limit = -1 + limit = self.dialog_input_number("Seed ratio limit for\n%s" % self.torrents[self.focus]['name'], + current_limit, floating_point=True, allow_empty=True) + if limit == -1: + limit = 0 + if limit == -2: # -2 means 'empty' in dialog_input_number return codes + limit = -1 + server.set_seed_ratio(float(limit), self.torrents[self.focus]['id']) + + def bandwidth_priority(self, c): + if c == ord('-') and self.focus > -1: + server.decrease_bandwidth_priority(self.torrents[self.focus]['id']) + elif c == ord('+') and self.focus > -1: + server.increase_bandwidth_priority(self.torrents[self.focus]['id']) + + def pause_unpause_torrent(self, c): + if self.focus > -1: + if self.selected_torrent > -1: + t = self.torrent_details + else: + t = self.torrents[self.focus] + if t['status'] == Transmission.STATUS_STOPPED: + server.start_torrent(t['id']) + else: + server.stop_torrent(t['id']) + + def pause_unpause_all_torrent(self, c): + if self.all_paused: + for t in self.torrents: + server.start_torrent(t['id']) + self.all_paused = False + else: + for t in self.torrents: + server.stop_torrent(t['id']) + self.all_paused = True + + def verify_torrent(self, c): + if self.focus > -1: + if self.torrents[self.focus]['status'] != Transmission.STATUS_CHECK \ + and self.torrents[self.focus]['status'] != Transmission.STATUS_CHECK_WAIT: + server.verify_torrent(self.torrents[self.focus]['id']) + + def reannounce_torrent(self, c): + if self.focus > -1: + server.reannounce_torrent(self.torrents[self.focus]['id']) + + def remove_torrent(self, c): + if self.focus > -1: + name = self.torrents[self.focus]['name'][0:self.width - 15] + if self.dialog_yesno("Remove %s?" % name) == True: + if self.selected_torrent > -1: # leave details + server.set_torrent_details_id(-1) + self.selected_torrent = -1 + self.details_category_focus = 0 + server.remove_torrent(self.torrents[self.focus]['id']) + self.focus_next_after_delete() + + def remove_torrent_local_data(self, c): + if self.focus > -1: + name = self.torrents[self.focus]['name'][0:self.width - 15] + if self.dialog_yesno("Remove and delete %s?" % name, important=True) == True: + if self.selected_torrent > -1: # leave details + server.set_torrent_details_id(-1) + self.selected_torrent = -1 + self.details_category_focus = 0 + server.remove_torrent_local_data(self.torrents[self.focus]['id']) + self.focus_next_after_delete() + + def focus_next_after_delete(self): + """ Focus next torrent after user deletes torrent """ + new_focus = min(self.focus + 1, len(self.torrents) - 2) + self.focused_id = self.torrents[new_focus]['id'] + + def add_tracker(self): + if server.get_rpc_version() < 10: + self.dialog_ok("You need Transmission v2.10 or higher to add trackers.") + return + + tracker = self.dialog_input_text('Add tracker URL:') + if tracker: + t = self.torrent_details + response = server.add_torrent_tracker(t['id'], tracker) + + if response: + msg = wrap("Couldn't add tracker: %s" % response) + self.dialog_ok("\n".join(msg)) + + def remove_tracker(self): + if server.get_rpc_version() < 10: + self.dialog_ok("You need Transmission v2.10 or higher to remove trackers.") + return + + t = self.torrent_details + if (self.scrollpos_detaillist >= 0 and \ + self.scrollpos_detaillist < len(t['trackerStats']) and \ + self.dialog_yesno("Do you want to remove this tracker?") is True): + + tracker = t['trackerStats'][self.scrollpos_detaillist] + response = server.remove_torrent_tracker(t['id'], tracker['id']) + + if response: + msg = wrap("Couldn't remove tracker: %s" % response) + self.dialog_ok("\n".join(msg)) + + def movement_keys(self, c): + if self.selected_torrent == -1 and len(self.torrents) > 0: + if c == curses.KEY_UP or c == ord('k') or c == curses.ascii.ctrl(ord('p')): + self.focus, self.scrollpos = self.move_up(self.focus, self.scrollpos, self.tlist_item_height) + elif c == curses.KEY_DOWN or c == ord('j') or c == curses.ascii.ctrl(ord('n')): + self.focus, self.scrollpos = self.move_down(self.focus, self.scrollpos, self.tlist_item_height, + self.torrents_per_page, len(self.torrents)) + elif c == curses.KEY_PPAGE or c == curses.ascii.ctrl(ord('b')): + self.focus, self.scrollpos = self.move_page_up(self.focus, self.scrollpos, self.tlist_item_height, + self.torrents_per_page) + elif c == curses.KEY_NPAGE or c == curses.ascii.ctrl(ord('f')): + self.focus, self.scrollpos = self.move_page_down(self.focus, self.scrollpos, self.tlist_item_height, + self.torrents_per_page, len(self.torrents)) + elif c == curses.KEY_HOME or c == ord('g'): + self.focus, self.scrollpos = self.move_to_top() + elif c == curses.KEY_END or c == ord('G'): + self.focus, self.scrollpos = self.move_to_end(self.tlist_item_height, self.torrents_per_page, len(self.torrents)) + self.focused_id = self.torrents[self.focus]['id'] + elif self.selected_torrent > -1: + # file list + if self.details_category_focus == 1: + # focus/movement + if c == curses.KEY_UP or c == ord('k') or c == curses.ascii.ctrl(ord('p')): + self.focus_detaillist, self.scrollpos_detaillist = \ + self.move_up(self.focus_detaillist, self.scrollpos_detaillist, 1) + elif c == curses.KEY_DOWN or c == ord('j') or c == curses.ascii.ctrl(ord('n')): + self.focus_detaillist, self.scrollpos_detaillist = \ + self.move_down(self.focus_detaillist, self.scrollpos_detaillist, 1, + self.detaillistitems_per_page, len(self.torrent_details['files'])) + elif c == curses.KEY_PPAGE or c == curses.ascii.ctrl(ord('b')): + self.focus_detaillist, self.scrollpos_detaillist = \ + self.move_page_up(self.focus_detaillist, self.scrollpos_detaillist, 1, + self.detaillistitems_per_page) + elif c == curses.KEY_NPAGE or c == curses.ascii.ctrl(ord('f')): + self.focus_detaillist, self.scrollpos_detaillist = \ + self.move_page_down(self.focus_detaillist, self.scrollpos_detaillist, 1, + self.detaillistitems_per_page, len(self.torrent_details['files'])) + elif c == curses.KEY_HOME or c == ord('g'): + self.focus_detaillist, self.scrollpos_detaillist = self.move_to_top() + elif c == curses.KEY_END or c == ord('G'): + self.focus_detaillist, self.scrollpos_detaillist = \ + self.move_to_end(1, self.detaillistitems_per_page, len(self.torrent_details['files'])) + list_len = 0 + + # peer list movement + if self.details_category_focus == 2: + list_len = len(self.torrent_details['peers']) + + # tracker list movement + elif self.details_category_focus == 3: + list_len = len(self.torrent_details['trackerStats']) + + # pieces list movement + elif self.details_category_focus == 4: + piece_count = self.torrent_details['pieceCount'] + margin = len(str(piece_count)) + 2 + map_width = int(str(self.width-margin-1)[0:-1] + '0') + list_len = int(piece_count / map_width) + 1 + + if list_len: + if c == curses.KEY_UP or c == ord('k') or c == curses.ascii.ctrl(ord('p')): + if self.scrollpos_detaillist > 0: + self.scrollpos_detaillist -= 1 + elif c == curses.KEY_DOWN or c == ord('j') or c == curses.ascii.ctrl(ord('n')): + if self.scrollpos_detaillist < list_len - 1: + self.scrollpos_detaillist += 1 + elif c == curses.KEY_PPAGE or c == curses.ascii.ctrl(ord('b')): + self.scrollpos_detaillist = \ + max(self.scrollpos_detaillist - self.detaillistitems_per_page - 1, 0) + elif c == curses.KEY_NPAGE or c == curses.ascii.ctrl(ord('f')): + if self.scrollpos_detaillist + self.detaillistitems_per_page >= list_len: + self.scrollpos_detaillist = list_len - 1 + else: + self.scrollpos_detaillist += self.detaillistitems_per_page + elif c == curses.KEY_HOME or c == ord('g'): + self.scrollpos_detaillist = 0 + elif c == curses.KEY_END or c == ord('G'): + self.scrollpos_detaillist = list_len - 1 + + # Disallow scrolling past the last item that would cause blank + # space to be displayed in pieces and peer lists. + if self.details_category_focus in (2, 4): + self.scrollpos_detaillist = min(self.scrollpos_detaillist, + max(0, list_len - self.detaillistitems_per_page)) + + def file_pritority_or_switch_details(self, c): + if self.selected_torrent > -1: + # file priority OR walk through details + if c == curses.KEY_RIGHT or c == ord('l'): + if self.details_category_focus == 1 and \ + (self.selected_files or self.focus_detaillist > -1): + if self.selected_files: + files = set(self.selected_files) + server.increase_file_priority(files) + elif self.focus_detaillist > -1: + server.increase_file_priority([self.focus_detaillist]) + else: + self.scrollpos_detaillist = 0 + self.next_details() + elif c == curses.KEY_LEFT or c == ord('h'): + if self.details_category_focus == 1 and \ + (self.selected_files or self.focus_detaillist > -1): + if self.selected_files: + files = set(self.selected_files) + server.decrease_file_priority(files) + elif self.focus_detaillist > -1: + server.decrease_file_priority([self.focus_detaillist]) + else: + self.scrollpos_detaillist = 0 + self.prev_details() + + def select_unselect_file(self, c): + if self.selected_torrent > -1 and self.details_category_focus == 1 and self.focus_detaillist >= 0: + # file selection with space + if c == ord(' '): + try: + self.selected_files.pop(self.selected_files.index(self.focus_detaillist)) + except ValueError: + self.selected_files.append(self.focus_detaillist) + curses.ungetch(curses.KEY_DOWN) # move down + # (un)select all files + elif c == ord('a'): + if self.selected_files: + self.selected_files = [] + else: + self.selected_files = range(0, len(self.torrent_details['files'])) + + def move_in_details(self, c): + if self.selected_torrent > -1: + if c == ord("\t"): + self.next_details() + elif c == curses.KEY_BTAB: + self.prev_details() + elif c == ord('e'): + self.details_category_focus = 2 + elif c == ord('c'): + self.details_category_focus = 4 + + def call_list_key_bindings(self, c): + self.list_key_bindings() + + def toggle_compact_torrentlist(self, c): + self.compact_list = not self.compact_list + self.recalculate_torrents_per_page() + self.follow_list_focus() + + def move_torrent(self, c): + if self.focus > -1: + location = homedir2tilde(self.torrents[self.focus]['downloadDir']) + msg = 'Move "%s" from\n%s to' % (self.torrents[self.focus]['name'], location) + path = self.dialog_input_text(msg, location) + if path: + server.move_torrent(self.torrents[self.focus]['id'], tilde2homedir(path)) + + def handle_user_input(self): + c = self.screen.getch() + if c == -1: + return 0 + + f = self.keybindings.get(c, None) + if f: + f(c) + + # update view + if self.selected_torrent == -1: + self.draw_torrent_list() + else: + self.draw_details() + + def filter_torrent_list(self): + unfiltered = self.torrents + if self.filter_list == 'downloading': + self.torrents = [t for t in self.torrents if t['rateDownload'] > 0] + elif self.filter_list == 'uploading': + self.torrents = [t for t in self.torrents if t['rateUpload'] > 0] + elif self.filter_list == 'paused': + self.torrents = [t for t in self.torrents if t['status'] == Transmission.STATUS_STOPPED] + elif self.filter_list == 'seeding': + self.torrents = [t for t in self.torrents if t['status'] == Transmission.STATUS_SEED \ + or t['status'] == Transmission.STATUS_SEED_WAIT] + elif self.filter_list == 'incomplete': + self.torrents = [t for t in self.torrents if t['percentDone'] < 100] + elif self.filter_list == 'private': + self.torrents = [t for t in self.torrents if t['isPrivate']] + elif self.filter_list == 'active': + self.torrents = [t for t in self.torrents if t['peersGettingFromUs'] > 0 \ + or t['peersSendingToUs'] > 0 \ + or t['status'] == Transmission.STATUS_CHECK] + elif self.filter_list == 'verifying': + self.torrents = [t for t in self.torrents if t['status'] == Transmission.STATUS_CHECK \ + or t['status'] == Transmission.STATUS_CHECK_WAIT] + # invert list? + if self.filter_inverse: + self.torrents = [t for t in unfiltered if t not in self.torrents] + + def follow_list_focus(self): + if self.focus == -1: + return + + # check if list is empty or id to look for isn't in list + ids = [t['id'] for t in self.torrents] + if len(self.torrents) == 0 or self.focused_id not in ids: + self.focus, self.scrollpos = -1, 0 + return + + # find focused_id + self.focus = min(self.focus, len(self.torrents)-1) + if self.torrents[self.focus]['id'] != self.focused_id: + for i,t in enumerate(self.torrents): + if t['id'] == self.focused_id: + self.focus = i + break + + # make sure the focus is not above the visible area + while self.focus < (self.scrollpos/self.tlist_item_height): + self.scrollpos -= self.tlist_item_height + # make sure the focus is not below the visible area + while self.focus > (self.scrollpos/self.tlist_item_height) + self.torrents_per_page-1: + self.scrollpos += self.tlist_item_height + # keep min and max bounds + self.scrollpos = min(self.scrollpos, (len(self.torrents) - self.torrents_per_page) * self.tlist_item_height) + self.scrollpos = max(0, self.scrollpos) + + def draw_torrent_list(self, search_keyword=''): + self.torrents = server.get_torrent_list(self.sort_orders) + self.filter_torrent_list() + + if search_keyword: + matched_torrents = [t for t in self.torrents if search_keyword.lower() in t['name'].lower()] + if matched_torrents: + self.focus = 0 + if self.search_focus >= len(matched_torrents): + self.search_focus = 0 + self.focused_id = matched_torrents[self.search_focus]['id'] + self.highlight_dialog = False + else: + self.highlight_dialog = True + curses.beep() + else: + self.search_focus = 0 + + self.follow_list_focus() + self.manage_layout() + + ypos = 0 + for i in range(len(self.torrents)): + ypos += self.draw_torrentlist_item(self.torrents[i], + (i == self.focus), + self.compact_list, + ypos) + + self.pad.refresh(self.scrollpos,0, 1,0, self.mainview_height,self.width-1) + self.screen.refresh() + + + def draw_torrentlist_item(self, torrent, focused, compact, y): + # the torrent name is also a progress bar + self.draw_torrentlist_title(torrent, focused, self.torrent_title_width, y) + + rates = '' + if torrent['status'] == Transmission.STATUS_DOWNLOAD: + self.draw_downloadrate(torrent, y) + if torrent['status'] == Transmission.STATUS_DOWNLOAD or torrent['status'] == Transmission.STATUS_SEED: + self.draw_uploadrate(torrent, y) + + if not compact: + # the line below the title/progress + if torrent['percentDone'] < 100 and torrent['status'] == Transmission.STATUS_DOWNLOAD: + self.draw_eta(torrent, y) + + self.draw_ratio(torrent, y) + self.draw_torrentlist_status(torrent, focused, y) + + return 3 # number of lines that were used for drawing the list item + else: + # Draw ratio in place of upload rate if upload rate = 0 + if not torrent['rateUpload']: + self.draw_ratio(torrent, y - 1) + + return 1 + + def draw_downloadrate(self, torrent, ypos): + self.pad.move(ypos, self.width-self.rateDownload_width-self.rateUpload_width-3) + self.pad.addch(curses.ACS_DARROW, (0,curses.A_BOLD)[torrent['downloadLimited']]) + rate = ('',scale_bytes(torrent['rateDownload']))[torrent['rateDownload']>0] + self.pad.addstr(rate.rjust(self.rateDownload_width), + curses.color_pair(self.colors.get_id('download_rate')) + curses.A_BOLD + curses.A_REVERSE) + def draw_uploadrate(self, torrent, ypos): + self.pad.move(ypos, self.width-self.rateUpload_width-1) + self.pad.addch(curses.ACS_UARROW, (0,curses.A_BOLD)[torrent['uploadLimited']]) + rate = ('',scale_bytes(torrent['rateUpload']))[torrent['rateUpload']>0] + self.pad.addstr(rate.rjust(self.rateUpload_width), + curses.color_pair(self.colors.get_id('upload_rate')) + curses.A_BOLD + curses.A_REVERSE) + def draw_ratio(self, torrent, ypos): + self.pad.addch(ypos+1, self.width-self.rateUpload_width-1, curses.ACS_DIAMOND, + (0,curses.A_BOLD)[torrent['uploadRatio'] < 1 and torrent['uploadRatio'] >= 0]) + self.pad.addstr(ypos+1, self.width-self.rateUpload_width, + num2str(torrent['uploadRatio'], '%.02f').rjust(self.rateUpload_width), + curses.color_pair(self.colors.get_id('eta+ratio')) + curses.A_BOLD + curses.A_REVERSE) + def draw_eta(self, torrent, ypos): + self.pad.addch(ypos+1, self.width-self.rateDownload_width-self.rateUpload_width-3, curses.ACS_PLMINUS) + self.pad.addstr(ypos+1, self.width-self.rateDownload_width-self.rateUpload_width-2, + scale_time(torrent['eta']).rjust(self.rateDownload_width), + curses.color_pair(self.colors.get_id('eta+ratio')) + curses.A_BOLD + curses.A_REVERSE) + + + def draw_torrentlist_title(self, torrent, focused, width, ypos): + if torrent['status'] == Transmission.STATUS_CHECK: + percentDone = float(torrent['recheckProgress']) * 100 + else: + percentDone = torrent['percentDone'] + + bar_width = int(float(width) * (float(percentDone)/100)) + + size = "%6s" % scale_bytes(torrent['sizeWhenDone']) + if torrent['percentDone'] < 100: + if torrent['seeders'] <= 0 and torrent['status'] != Transmission.STATUS_CHECK: + size = "%6s / " % scale_bytes(torrent['available']) + size + size = "%6s / " % scale_bytes(torrent['haveValid'] + torrent['haveUnchecked']) + size + size = '| ' + size + title = ljust_columns(torrent['name'], width - len(size)) + size + + if torrent['status'] == Transmission.STATUS_SEED \ + or torrent['status'] == Transmission.STATUS_SEED_WAIT: + color = curses.color_pair(self.colors.get_id('title_seed')) + elif torrent['status'] == Transmission.STATUS_STOPPED: + color = curses.color_pair(self.colors.get_id('title_paused')) + elif torrent['status'] == Transmission.STATUS_CHECK \ + or torrent['status'] == Transmission.STATUS_CHECK_WAIT: + color = curses.color_pair(self.colors.get_id('title_verify')) + elif torrent['rateDownload'] == 0: + color = curses.color_pair(self.colors.get_id('title_idle')) + elif torrent['percentDone'] < 100: + color = curses.color_pair(self.colors.get_id('title_download')) + else: + color = 0 + + tag = curses.A_REVERSE + tag_done = tag + color + if focused: + tag += curses.A_BOLD + tag_done += curses.A_BOLD + + if self.torrentname_is_progressbar: + # addstr() dies when you tell it to draw on the last column of the + # terminal, so we have to catch this exception. + try: + self.pad.addstr(ypos, 0, title[0:bar_width].encode('utf-8'), tag_done) + self.pad.addstr(ypos, len_columns(title[0:bar_width]), title[bar_width:].encode('utf-8'), tag) + except: + pass + else: + self.pad.addstr(ypos, 0, title.encode('utf-8'), tag_done) + + + def draw_torrentlist_status(self, torrent, focused, ypos): + peers = '' + parts = [server.get_status(torrent)] + + # show tracker error if appropriate + if torrent['errorString'] and \ + not torrent['seeders'] and not torrent['leechers'] and \ + not torrent['status'] == Transmission.STATUS_STOPPED: + parts[0] = torrent['errorString'].encode('utf-8') + + else: + if torrent['status'] == Transmission.STATUS_CHECK: + parts[0] += " (%d%%)" % int(float(torrent['recheckProgress']) * 100) + elif torrent['status'] == Transmission.STATUS_DOWNLOAD: + parts[0] += " (%d%%)" % torrent['percentDone'] + parts[0] = parts[0].ljust(20) + + # seeds and leeches will be appended right justified later + peers = "%5s seed%s " % (num2str(torrent['seeders']), ('s', ' ')[torrent['seeders']==1]) + peers += "%5s leech%s" % (num2str(torrent['leechers']), ('es', ' ')[torrent['leechers']==1]) + + # show additional information if enough room + if self.torrent_title_width - sum(map(lambda x: len(x), parts)) - len(peers) > 18: + uploaded = scale_bytes(torrent['uploadedEver']) + parts.append("%7s uploaded" % ('nothing',uploaded)[uploaded != '0B']) + + if self.torrent_title_width - sum(map(lambda x: len(x), parts)) - len(peers) > 22: + parts.append("%4s peer%s connected" % (torrent['peersConnected'], + ('s',' ')[torrent['peersConnected'] == 1])) + + if focused: tags = curses.A_REVERSE + curses.A_BOLD + else: tags = 0 + + remaining_space = self.torrent_title_width - sum(map(lambda x: len(x), parts), len(peers)) - 2 + delimiter = ' ' * int(remaining_space / (len(parts))) + + line = server.get_bandwidth_priority(torrent) + ' ' + delimiter.join(parts) + + # make sure the peers element is always right justified + line += ' ' * int(self.torrent_title_width - len(line) - len(peers)) + peers + self.pad.addstr(ypos+1, 0, line, tags) + + + def draw_details(self): + self.torrent_details = server.get_torrent_details() + self.manage_layout() + + # details could need more space than the torrent list + self.pad_height = max(50, len(self.torrent_details['files'])+10, (len(self.torrents)+1)*3, self.height) + self.pad = curses.newpad(self.pad_height, self.width) + + # torrent name + progress bar + self.draw_torrentlist_item(self.torrent_details, False, False, 0) + + # divider + menu + menu_items = ['_Overview', "_Files", 'P_eers', '_Trackers', 'Pie_ces' ] + xpos = int((self.width - sum(map(lambda x: len(x), menu_items))-len(menu_items)) / 2) + for item in menu_items: + self.pad.move(3, xpos) + tags = curses.A_BOLD + if menu_items.index(item) == self.details_category_focus: + tags += curses.A_REVERSE + title = item.split('_') + self.pad.addstr(title[0], tags) + self.pad.addstr(title[1][0], tags + curses.A_UNDERLINE) + self.pad.addstr(title[1][1:], tags) + xpos += len(item)+1 + + # which details to display + if self.details_category_focus == 0: + self.draw_details_overview(5) + elif self.details_category_focus == 1: + self.draw_filelist(5) + elif self.details_category_focus == 2: + self.draw_peerlist(5) + elif self.details_category_focus == 3: + self.draw_trackerlist(5) + elif self.details_category_focus == 4: + self.draw_pieces_map(5) + + self.pad.refresh(0,0, 1,0, self.height-2,self.width) + self.screen.refresh() + + + def draw_details_overview(self, ypos): + t = self.torrent_details + info = [] + info.append(['Hash: ', "%s" % t['hashString']]) + info.append(['ID: ', "%s" % t['id']]) + + wanted = 0 + for i, file_info in enumerate(t['files']): + if t['wanted'][i] == True: wanted += t['files'][i]['length'] + + sizes = ['Size: ', "%s; " % scale_bytes(t['totalSize'], 'long'), + "%s wanted; " % (scale_bytes(wanted, 'long'),'everything') [t['totalSize'] == wanted]] + if t['available'] < t['totalSize']: + sizes.append("%s available; " % scale_bytes(t['available'], 'long')) + sizes.extend(["%s left" % scale_bytes(t['leftUntilDone'], 'long')]) + info.append(sizes) + + info.append(['Files: ', "%d; " % len(t['files'])]) + complete = map(lambda x: x['bytesCompleted'] == x['length'], t['files']).count(True) + not_complete = filter(lambda x: x['bytesCompleted'] != x['length'], t['files']) + partial = map(lambda x: x['bytesCompleted'] > 0, not_complete).count(True) + if complete == len(t['files']): + info[-1].append("all complete") + else: + info[-1].append("%d complete; " % complete) + info[-1].append("%d commenced" % partial) + + info.append(['Pieces: ', "%s; " % t['pieceCount'], + "%s each" % scale_bytes(t['pieceSize'], 'long')]) + + info.append(['Download: ']) + info[-1].append("%s" % scale_bytes(t['downloadedEver'], 'long') + \ + " (%d%%) received; " % int(percent(t['sizeWhenDone'], t['downloadedEver']))) + info[-1].append("%s" % scale_bytes(t['haveValid'], 'long') + \ + " (%d%%) verified; " % int(percent(t['sizeWhenDone'], t['haveValid']))) + info[-1].append("%s corrupt" % scale_bytes(t['corruptEver'], 'long')) + if t['percentDone'] < 100: + info[-1][-1] += '; ' + if t['rateDownload']: + info[-1].append("receiving %s per second" % scale_bytes(t['rateDownload'], 'long')) + if t['downloadLimited']: + info[-1][-1] += " (throttled to %s)" % scale_bytes(t['downloadLimit']*1024, 'long') + else: + info[-1].append("no reception in progress") + + try: + copies_distributed = (float(t['uploadedEver']) / float(t['sizeWhenDone'])) + except ZeroDivisionError: + copies_distributed = 0 + info.append(['Upload: ', "%s (%d%%) transmitted; " % + (scale_bytes(t['uploadedEver'], 'long'), t['uploadRatio']*100)]) + if t['rateUpload']: + info[-1].append("sending %s per second" % scale_bytes(t['rateUpload'], 'long')) + if t['uploadLimited']: + info[-1][-1] += " (throttled to %s)" % scale_bytes(t['uploadLimit']*1024, 'long') + else: + info[-1].append("no transmission in progress") + + info.append(['Ratio: ', '%.2f copies distributed' % copies_distributed]) + norm_upload_rate = norm.add('%s:rateUpload' % t['id'], t['rateUpload'], 15) + if norm_upload_rate > 0: + target_ratio = self.get_target_ratio() + bytes_left = (max(t['downloadedEver'],t['sizeWhenDone']) * target_ratio) - t['uploadedEver'] + time_left = bytes_left / norm_upload_rate + info[-1][-1] += '; ' + if time_left < 86400: # 1 day + info[-1].append('approaching %.2f at %s' % \ + (target_ratio, timestamp(time.time() + time_left, "%R"))) + else: + info[-1].append('approaching %.2f on %s' % \ + (target_ratio, timestamp(time.time() + time_left, "%x"))) + + info.append(['Seed limit: ']) + if t['seedRatioMode'] == 0: + if self.stats['seedRatioLimited']: + info[-1].append('default (pause torrent after distributing %s copies)' % self.stats['seedRatioLimit']) + else: + info[-1].append('default (unlimited)') + elif t['seedRatioMode'] == 1: + info[-1].append('pause torrent after distributing %s copies' % t['seedRatioLimit']) + elif t['seedRatioMode'] == 2: + info[-1].append('unlimited (ignore global limits)') + + info.append(['Peers: ', + "connected to %d; " % t['peersConnected'], + "downloading from %d; " % t['peersSendingToUs'], + "uploading to %d" % t['peersGettingFromUs']]) + + # average peer speed + incomplete_peers = [peer for peer in self.torrent_details['peers'] if peer['progress'] < 1] + if incomplete_peers: + # use at least 2/3 or 10 of incomplete peers to make an estimation + active_peers = [peer for peer in incomplete_peers if peer['download_speed']] + min_active_peers = min(10, max(1, round(len(incomplete_peers)*0.666))) + if 1 <= len(active_peers) >= min_active_peers: + swarm_speed = sum([peer['download_speed'] for peer in active_peers]) / len(active_peers) + info.append(['Swarm speed: ', "%s on average; " % scale_bytes(swarm_speed), + "distribution of 1 copy takes %s" % \ + scale_time(int(t['totalSize'] / swarm_speed), 'long')]) + else: + info.append(['Swarm speed: ', "" % \ + (min_active_peers, len(active_peers))]) + else: + info.append(['Swarm speed: ', ""]) + + + info.append(['Privacy: ']) + if t['isPrivate']: + info[-1].append('Private to this tracker -- DHT and PEX disabled') + else: + info[-1].append('Public torrent') + + info.append(['Location: ',"%s" % homedir2tilde(t['downloadDir'])]) + + ypos = self.draw_details_list(ypos, info) + + self.draw_details_eventdates(ypos+1) + return ypos+1 + + def get_target_ratio(self): + t = self.torrent_details + if t['seedRatioMode'] == 1: + return t['seedRatioLimit'] # individual limit + elif t['seedRatioMode'] == 0 and self.stats['seedRatioLimited']: + return self.stats['seedRatioLimit'] # global limit + else: + # round up to next 10/5/1 + if t['uploadRatio'] >= 100: + step_size = 10.0 + elif t['uploadRatio'] >= 10: + step_size = 5.0 + else: + step_size = 1.0 + return int(round((t['uploadRatio'] + step_size/2) / step_size) * step_size) + + def draw_details_eventdates(self, ypos): + t = self.torrent_details + + self.pad.addstr(ypos, 1, ' Created: ' + timestamp(t['dateCreated'])) + self.pad.addstr(ypos+1, 1, ' Added: ' + timestamp(t['addedDate'])) + self.pad.addstr(ypos+2, 1, ' Started: ' + timestamp(t['startDate'])) + self.pad.addstr(ypos+3, 1, ' Activity: ' + timestamp(t['activityDate'])) + + if t['percentDone'] < 100 and t['eta'] > 0: + self.pad.addstr(ypos+4, 1, 'Finishing: ' + timestamp(time.time() + t['eta'])) + elif t['doneDate'] <= 0: + self.pad.addstr(ypos+4, 1, 'Finishing: sometime') + else: + self.pad.addstr(ypos+4, 1, ' Finished: ' + timestamp(t['doneDate'])) + + if t['comment']: + if self.width >= 90: + width = self.width - 50 + comment = wrap_multiline(t['comment'], width, initial_indent='Comment: ') + for i, line in enumerate(comment): + if(ypos+i > self.height-1): + break + self.pad.addstr(ypos+i, 50, line.encode('utf8')) + else: + width = self.width - 2 + comment = wrap_multiline(t['comment'], width, initial_indent='Comment: ') + for i, line in enumerate(comment): + self.pad.addstr(ypos+6+i, 2, line.encode('utf8')) + + def draw_filelist(self, ypos): + column_names = ' # Progress Size Priority Filename' + self.pad.addstr(ypos, 0, column_names.ljust(self.width), curses.A_UNDERLINE) + ypos += 1 + + for line in self.create_filelist(): + curses_tags = 0 + # highlight focused/selected line(s) + while line.startswith('_'): + if line[1] == 'S': + curses_tags = curses.A_BOLD + line = line[2:] + if line[1] == 'F': + curses_tags += curses.A_REVERSE + line = line[2:] + try: + self.pad.addstr(ypos, 0, ' '*self.width, curses_tags) + except: pass + + # colored priority (only in the first 30 chars, the rest is filename) + xpos = 0 + for part in re.split('(high|normal|low|off)', line[0:30], 1): + if part == 'high': + self.pad.addstr(ypos, xpos, part, + curses_tags + curses.color_pair(self.colors.get_id('file_prio_high'))) + elif part == 'normal': + self.pad.addstr(ypos, xpos, part, + curses_tags + curses.color_pair(self.colors.get_id('file_prio_normal'))) + elif part == 'low': + self.pad.addstr(ypos, xpos, part, + curses_tags + curses.color_pair(self.colors.get_id('file_prio_low'))) + elif part == 'off': + self.pad.addstr(ypos, xpos, part, + curses_tags + curses.color_pair(self.colors.get_id('file_prio_off'))) + else: + self.pad.addstr(ypos, xpos, part.encode('utf-8'), curses_tags) + xpos += len(part) + self.pad.addstr(ypos, xpos, line[30:].encode('utf-8'), curses_tags) + ypos += 1 + if ypos > self.height: + break + + def create_filelist(self): + filelist = [] + files = self.torrent_details['files'] + current_folder = [] + current_depth = 0 + index = 0 + pos = 0 + pos_before_focus = 0 + for file in files: + f = file['name'].split('/') + f_len = len(f) - 1 + if f[:f_len] != current_folder: + [current_depth, pos] = self.create_filelist_transition(f, current_folder, filelist, current_depth, pos) + current_folder = f[:f_len] + filelist.append(self.create_filelist_line(f[-1], index, percent(file['length'], file['bytesCompleted']), + file['length'], current_depth)) + index += 1 + if self.focus_detaillist == index - 1: + pos_before_focus = pos + if index + pos >= self.focus_detaillist + 1 + pos + self.detaillistitems_per_page/2 \ + and index + pos >= self.detaillistitems_per_page: + if self.focus_detaillist + 1 + pos_before_focus < self.detaillistitems_per_page / 2: + return filelist + return filelist[self.focus_detaillist + 1 + pos_before_focus - self.detaillistitems_per_page / 2 + : self.focus_detaillist + 1 + pos_before_focus + self.detaillistitems_per_page / 2] + begin = len(filelist) - self.detaillistitems_per_page + return filelist[begin > 0 and begin or 0:] + + def create_filelist_transition(self, f, current_folder, filelist, current_depth, pos): + f_len = len(f) - 1 + current_folder_len = len(current_folder) + same = 0 + while same < current_folder_len and same < f_len and f[same] == current_folder[same]: + same += 1 + for i in range(current_folder_len - same): + current_depth -= 1 + filelist.append(' '*current_depth + ' '*31 + '/') + pos += 1 + if f_len < current_folder_len: + return [current_depth, pos] + while current_depth < f_len: + filelist.append('%s\\ %s' % (' '*current_depth + ' '*31 , f[current_depth])) + current_depth += 1 + pos += 1 + return [current_depth, pos] + + def create_filelist_line(self, name, index, percent, length, current_depth): + line = "%s %6.1f%%" % (str(index+1).rjust(3), percent) + \ + ' '+scale_bytes(length).rjust(5) + \ + ' '+server.get_file_priority(self.torrent_details['id'], index).center(8) + \ + " %s| %s" % (' '*current_depth, name[0:self.width-31-current_depth]) + if index == self.focus_detaillist: + line = '_F' + line + if index in self.selected_files: + line = '_S' + line + return line + + def draw_peerlist(self, ypos): + # Start drawing list either at the "selected" index, or at the index + # that is required to display all remaining items without further scrolling. + last_possible_index = max(0, len(self.torrent_details['peers']) - self.detaillistitems_per_page) + start = min(self.scrollpos_detaillist, last_possible_index) + end = start + self.detaillistitems_per_page + peers = self.torrent_details['peers'][start:end] + + # Find width of columns + clientname_width = 0 + address_width = 0 + for peer in peers: + if len(peer['clientName']) > clientname_width: clientname_width = len(peer['clientName']) + if len(peer['address']) > address_width: address_width = len(peer['address']) + + # Column names + column_names = "Flags %3d Down %3d Up Progress ETA " % \ + (self.torrent_details['peersSendingToUs'], self.torrent_details['peersGettingFromUs']) + column_names += ' Client'.ljust(clientname_width + 2) \ + + " Address".ljust(address_width + 2) + if features['geoip']: column_names += " Country" + if features['dns']: column_names += " Host" + + self.pad.addstr(ypos, 0, column_names.ljust(self.width), curses.A_UNDERLINE) + ypos += 1 + + # Peers + hosts = server.get_hosts() + geo_ips = server.get_geo_ips() + for index, peer in enumerate(peers): + if features['dns']: + try: + try: + host = hosts[peer['address']].check() + host_name = host[3][0] + except (IndexError, KeyError): + host_name = "" + except adns.NotReady: + host_name = "" + except adns.Error, msg: + host_name = msg + + upload_tag = download_tag = line_tag = 0 + if peer['rateToPeer']: upload_tag = curses.A_BOLD + if peer['rateToClient']: download_tag = curses.A_BOLD + + self.pad.move(ypos, 0) + # Flags + self.pad.addstr("%-6s " % peer['flagStr']) + # Down + self.pad.addstr("%5s " % scale_bytes(peer['rateToClient']), download_tag) + # Up + self.pad.addstr("%5s " % scale_bytes(peer['rateToPeer']), upload_tag) + + # Progress + if peer['progress'] < 1: self.pad.addstr("%3d%%" % (float(peer['progress'])*100)) + else: self.pad.addstr("%3d%%" % (float(peer['progress'])*100), curses.A_BOLD) + + # ETA + if peer['progress'] < 1 and peer['download_speed'] > 1024: + self.pad.addstr(" @ ") + self.pad.addch(curses.ACS_PLMINUS) + self.pad.addstr("%-5s " % scale_bytes(peer['download_speed'])) + self.pad.addch(curses.ACS_PLMINUS) + self.pad.addstr("%-4s " % scale_time(peer['time_left'])) + else: + self.pad.addstr(" ") + # Client + self.pad.addstr(peer['clientName'].ljust(clientname_width + 2).encode('utf-8')) + # Address + self.pad.addstr(peer['address'].ljust(address_width + 2)) + # Country + if features['geoip']: self.pad.addstr(" %2s " % geo_ips[peer['address']]) + # Host + if features['dns']: self.pad.addstr(host_name.encode('utf-8'), curses.A_DIM) + ypos += 1 + +#TODO +# 1. Issue #14 on GitHub is asking for feature to be able to modify trackers. + def draw_trackerlist(self, ypos): + top = ypos - 1 + def addstr(ypos, xpos, *args): + if ypos > top and ypos < self.height - 2: + self.pad.addstr(ypos, xpos, *args) + + tracker_per_page = self.detaillistitems_per_page // self.TRACKER_ITEM_HEIGHT + page = self.scrollpos_detaillist // tracker_per_page + start = tracker_per_page * page + end = tracker_per_page * (page + 1) + tlist = self.torrent_details['trackerStats'][start:end] + + # keep position in range when last tracker gets deleted + self.scrollpos_detaillist = min(self.scrollpos_detaillist, + len(self.torrent_details['trackerStats'])-1) + # show newly added tracker when list was empty before + if self.torrent_details['trackerStats']: + self.scrollpos_detaillist = max(0, self.scrollpos_detaillist) + + current_tier = -1 + for index, t in enumerate(tlist): + announce_msg_size = scrape_msg_size = 0 + selected = t == self.torrent_details['trackerStats'][self.scrollpos_detaillist] + + if current_tier != t['tier']: + current_tier = t['tier'] + + tiercolor = curses.A_BOLD + curses.A_REVERSE \ + if selected else curses.A_REVERSE + addstr(ypos, 0, ("Tier %d" % (current_tier+1)).ljust(self.width), tiercolor) + ypos += 1 + + if selected: + for i in range(4): + addstr(ypos+i, 0, ' ', curses.A_BOLD + curses.A_REVERSE) + + addstr(ypos+1, 4, "Last announce: %s" % timestamp(t['lastAnnounceTime'])) + addstr(ypos+1, 57, " Last scrape: %s" % timestamp(t['lastScrapeTime'])) + + if t['lastAnnounceSucceeded']: + peers = "%s peer%s" % (num2str(t['lastAnnouncePeerCount']), ('s', '')[t['lastAnnouncePeerCount']==1]) + addstr(ypos, 2, t['announce'], curses.A_BOLD + curses.A_UNDERLINE) + addstr(ypos+2, 11, "Result: ") + addstr(ypos+2, 19, "%s received" % peers, curses.A_BOLD) + else: + addstr(ypos, 2, t['announce'], curses.A_UNDERLINE) + addstr(ypos+2, 9, "Response:") + announce_msg_size = self.wrap_and_draw_result(top, ypos+2, 19, t['lastAnnounceResult'].encode('utf-8')) + + if t['lastScrapeSucceeded']: + seeds = "%s seed%s" % (num2str(t['seederCount']), ('s', '')[t['seederCount']==1]) + leeches = "%s leech%s" % (num2str(t['leecherCount']), ('es', '')[t['leecherCount']==1]) + addstr(ypos+2, 57, "Tracker knows: ") + addstr(ypos+2, 72, "%s and %s" % (seeds, leeches), curses.A_BOLD) + else: + addstr(ypos+2, 62, "Response:") + scrape_msg_size += self.wrap_and_draw_result(top, ypos+2, 72, t['lastScrapeResult']) + + ypos += max(announce_msg_size, scrape_msg_size) + + addstr(ypos+3, 4, "Next announce: %s" % timestamp(t['nextAnnounceTime'])) + addstr(ypos+3, 57, " Next scrape: %s" % timestamp(t['nextScrapeTime'])) + + ypos += 5 + + def wrap_and_draw_result(self, top, ypos, xpos, result): + result = wrap(result, 30) + i = 0 + for i, line in enumerate(result): + if ypos+i > top and ypos+i < self.height - 2: + self.pad.addstr(ypos+i, xpos, line, curses.A_UNDERLINE) + return i + + + def draw_pieces_map(self, ypos): + pieces = self.torrent_details['pieces'] + piece_count = self.torrent_details['pieceCount'] + margin = len(str(piece_count)) + 2 + + map_width = int(str(self.width-margin-1)[0:-1] + '0') + for x in range(10, map_width, 10): + self.pad.addstr(ypos, x+margin-1, str(x), curses.A_BOLD) + + start = self.scrollpos_detaillist * map_width + end = min(start + (self.height - ypos - 3) * map_width, piece_count) + if end <= start: return + block = ord(pieces[start >> 3]) << (start & 7) + + format = "%%%dd" % (margin - 2) + for counter in xrange(start, end): + if counter % map_width == 0: + ypos += 1 ; xpos = margin + self.pad.addstr(ypos, 1, format % counter, curses.A_BOLD) + else: + xpos += 1 + + if counter & 7 == 0: + block = ord(pieces[counter >> 3]) + piece = block & 0x80 + if piece: self.pad.addch(ypos, xpos, ' ', curses.A_REVERSE) + else: self.pad.addch(ypos, xpos, '_') + block <<= 1 + + missing_pieces = piece_count - counter - 1 + if missing_pieces: + line = "%d further piece%s" % (missing_pieces, ('','s')[missing_pieces>1]) + xpos = (self.width - len(line)) / 2 + self.pad.addstr(self.height-3, xpos, line, curses.A_REVERSE) + + def draw_details_list(self, ypos, info): + key_width = max(map(lambda x: len(x[0]), info)) + for i in info: + self.pad.addstr(ypos, 1, i[0].rjust(key_width).encode('utf-8')) # key + # value part may be wrapped if it gets too long + for v in i[1:]: + y, x = self.pad.getyx() + if x + len(v) >= self.width: + ypos += 1 + self.pad.move(ypos, key_width+1) + self.pad.addstr(v.encode('utf-8')) + ypos += 1 + return ypos + + def next_details(self): + if self.details_category_focus >= 4: + self.details_category_focus = 0 + else: + self.details_category_focus += 1 + self.focus_detaillist = -1 + self.scrollpos_detaillist = 0 + self.pad.erase() + + def prev_details(self): + if self.details_category_focus <= 0: + self.details_category_focus = 4 + else: + self.details_category_focus -= 1 + self.pad.erase() + + + + + def move_up(self, focus, scrollpos, step_size): + if focus < 0: focus = -1 + else: + focus -= 1 + if scrollpos/step_size - focus > 0: + scrollpos -= step_size + scrollpos = max(0, scrollpos) + while scrollpos % step_size: + scrollpos -= 1 + return focus, scrollpos + + def move_down(self, focus, scrollpos, step_size, elements_per_page, list_height): + if focus < list_height - 1: + focus += 1 + if focus+1 - scrollpos/step_size > elements_per_page: + scrollpos += step_size + return focus, scrollpos + + def move_page_up(self, focus, scrollpos, step_size, elements_per_page): + for x in range(elements_per_page - 1): + focus, scrollpos = self.move_up(focus, scrollpos, step_size) + if focus < 0: focus = 0 + return focus, scrollpos + + def move_page_down(self, focus, scrollpos, step_size, elements_per_page, list_height): + if focus < 0: focus = 0 + for x in range(elements_per_page - 1): + focus, scrollpos = self.move_down(focus, scrollpos, step_size, elements_per_page, list_height) + return focus, scrollpos + + def move_to_top(self): + return 0, 0 + + def move_to_end(self, step_size, elements_per_page, list_height): + focus = list_height - 1 + scrollpos = max(0, (list_height - elements_per_page) * step_size) + return focus, scrollpos + + + + + + def draw_stats(self): + self.screen.insstr(self.height-1, 0, ' '.center(self.width), curses.A_REVERSE) + self.draw_torrents_stats() + self.draw_global_rates() + + def draw_torrents_stats(self): + if self.selected_torrent > -1 and self.details_category_focus == 2: + self.screen.insstr((self.height-1), 0, + "%d peer%s connected (" % (self.torrent_details['peersConnected'], + ('s','')[self.torrent_details['peersConnected'] == 1]) + \ + "Trackers:%d " % self.torrent_details['peersFrom']['fromTracker'] + \ + "DHT:%d " % self.torrent_details['peersFrom']['fromDht'] + \ + "LTEP:%d " % self.torrent_details['peersFrom']['fromLtep'] + \ + "PEX:%d " % self.torrent_details['peersFrom']['fromPex'] + \ + "Incoming:%d " % self.torrent_details['peersFrom']['fromIncoming'] + \ + "Cache:%d)" % self.torrent_details['peersFrom']['fromCache'], + curses.A_REVERSE) + else: + self.screen.addstr((self.height-1), 0, "Torrent%s:" % ('s','')[len(self.torrents) == 1], + curses.A_REVERSE) + self.screen.addstr("%d (" % len(self.torrents), curses.A_REVERSE) + + downloading = len(filter(lambda x: x['status']==Transmission.STATUS_DOWNLOAD, self.torrents)) + seeding = len(filter(lambda x: x['status']==Transmission.STATUS_SEED, self.torrents)) + paused = self.stats['pausedTorrentCount'] + + self.screen.addstr("Downloading:", curses.A_REVERSE) + self.screen.addstr("%d " % downloading, curses.A_REVERSE) + self.screen.addstr("Seeding:", curses.A_REVERSE) + self.screen.addstr("%d " % seeding, curses.A_REVERSE) + self.screen.addstr("Paused:", curses.A_REVERSE) + self.screen.addstr("%d) " % paused, curses.A_REVERSE) + + if self.filter_list: + self.screen.addstr("Filter:", curses.A_REVERSE) + self.screen.addstr("%s%s" % (('','not ')[self.filter_inverse], self.filter_list), + curses.color_pair(self.colors.get_id('filter_status')) + + curses.A_REVERSE) + + # show last sort order (if terminal size permits it) + curpos_y, curpos_x = self.screen.getyx() + if self.sort_orders and self.width - curpos_x > 20: + self.screen.addstr(" Sort by:", curses.A_REVERSE) + name = [name[1] for name in self.sort_options if name[0] == self.sort_orders[-1]['name']][0] + name = name.replace('_', '').lower() + curses_tags = curses.color_pair(self.colors.get_id('filter_status')) + curses.A_REVERSE + if self.sort_orders[-1]['reverse']: + self.screen.addch(curses.ACS_DARROW, curses_tags) + else: + self.screen.addch(curses.ACS_UARROW, curses_tags) + try: # 'name' may be too long + self.screen.addstr(name, curses_tags) + except curses.error: + pass + + def draw_global_rates(self): + rates_width = self.rateDownload_width + self.rateUpload_width + 3 + + if self.stats['alt-speed-enabled']: + upload_limit = "/%dK" % self.stats['alt-speed-up'] + download_limit = "/%dK" % self.stats['alt-speed-down'] + else: + upload_limit = ('', "/%dK" % self.stats['speed-limit-up'])[self.stats['speed-limit-up-enabled']] + download_limit = ('', "/%dK" % self.stats['speed-limit-down'])[self.stats['speed-limit-down-enabled']] + + limits = {'dn_limit' : download_limit, 'up_limit' : upload_limit} + limits_width = len(limits['dn_limit']) + len(limits['up_limit']) + + if self.stats['alt-speed-enabled']: + self.screen.move(self.height-1, self.width-rates_width - limits_width - len('Turtle mode ')) + self.screen.addstr('Turtle mode', curses.A_REVERSE + curses.A_BOLD) + self.screen.addch(' ', curses.A_REVERSE) + + self.screen.move(self.height - 1, self.width - rates_width - limits_width) + self.screen.addch(curses.ACS_DARROW, curses.A_REVERSE) + self.screen.addstr(scale_bytes(self.stats['downloadSpeed']).rjust(self.rateDownload_width), + curses.color_pair(self.colors.get_id('download_rate')) + + curses.A_REVERSE + curses.A_BOLD) + self.screen.addstr(limits['dn_limit'], curses.A_REVERSE) + self.screen.addch(' ', curses.A_REVERSE) + self.screen.addch(curses.ACS_UARROW, curses.A_REVERSE) + self.screen.insstr(limits['up_limit'], curses.A_REVERSE) + self.screen.insstr(scale_bytes(self.stats['uploadSpeed']).rjust(self.rateUpload_width), + curses.color_pair(self.colors.get_id('upload_rate')) + + curses.A_REVERSE + curses.A_BOLD) + + + + def draw_title_bar(self): + self.screen.insstr(0, 0, ' '.center(self.width), curses.A_REVERSE) + self.draw_connection_status() + self.draw_quick_help() + def draw_connection_status(self): + status = "Transmission @ %s:%s" % (server.host, server.port) + if cmd_args.DEBUG: + status = "%d x %d " % (self.width, self.height) + status + self.screen.addstr(0, 0, status.encode('utf-8'), curses.A_REVERSE) + + def draw_quick_help(self): + help = [('?','Show Keybindings')] + + if self.selected_torrent == -1: + if self.focus >= 0: + help = [('enter','View Details'), ('p','Pause/Unpause'), ('r','Remove'), ('v','Verify')] + else: + help = [('/','Search'), ('f','Filter'), ('s','Sort')] + help + [('o','Options'), ('q','Quit')] + else: + help = [('Move with','cursor keys'), ('q','Back to List')] + if self.details_category_focus == 1 and self.focus_detaillist > -1: + help = [('space','(De)Select File'), + ('left/right','De-/Increase Priority'), + ('escape','Unfocus/-select')] + help + elif self.details_category_focus == 2: + help = [('F1/?','Explain flags')] + help + elif self.details_category_focus == 3: + help = [('a','Add Tracker'),('r','Remove Tracker')] + help + + line = ' | '.join(map(lambda x: "%s %s" % (x[0], x[1]), help)) + line = line[0:self.width] + self.screen.insstr(0, self.width-len(line), line, curses.A_REVERSE) + + + def list_key_bindings(self): + title = 'Help Menu' + message = " F1/? Show this help\n" + \ + " u/d Adjust maximum global upload/download rate\n" + \ + " U/D Adjust maximum upload/download rate for focused torrent\n" + \ + " L Set seed ratio limit for focused torrent\n" + \ + " +/- Adjust bandwidth priority for focused torrent\n" + \ + " p Pause/Unpause torrent\n" + \ + " P Pause/Unpause all torrents\n" + \ + " v/y Verify torrent\n" + \ + " m Move torrent\n" + \ + " n Reannounce torrent\n" + \ + " a Add torrent\n" + \ + " Del/r Remove torrent and keep content\n" + \ + " Shift+Del/R Remove torrent and delete content\n" + # Torrent list + if self.selected_torrent == -1: + message += " / Search in torrent list\n" + \ + " f Filter torrent list\n" + \ + " s Sort torrent list\n" \ + " Enter/Right View torrent's details\n" + \ + " o Configuration options\n" + \ + " t Toggle turtle mode\n" + \ + " C Toggle compact list mode\n" + \ + " Esc Unfocus\n" + \ + " q Quit" + else: + # Peer list + if self.details_category_focus == 2: + title = 'Peer status flags' + message = " O Optimistic unchoke\n" + \ + " D Downloading from this peer\n" + \ + " d We would download from this peer if they'd let us\n" + \ + " U Uploading to peer\n" + \ + " u We would upload to this peer if they'd ask\n" + \ + " K Peer has unchoked us, but we're not interested\n" + \ + " ? We unchoked this peer, but they're not interested\n" + \ + " E Encrypted Connection\n" + \ + " H Peer was discovered through DHT\n" + \ + " X Peer was discovered through Peer Exchange (PEX)\n" + \ + " I Peer is an incoming connection\n" + \ + " T Peer is connected via uTP" + else: + # Viewing torrent details + message += " o Jump to overview\n" + \ + " f Jump to file list\n" + \ + " e Jump to peer list\n" + \ + " t Jump to tracker information\n" + \ + " Tab/Right Jump to next view\n" + \ + " Shift+Tab/Left Jump to previous view\n" + if self.details_category_focus == 1: # files + if self.focus_detaillist > -1: + message += " Left/Right Decrease/Increase file priority\n" + message += " Up/Down Select file\n" + \ + " Space Select/Deselect focused file\n" + \ + " a Select/Deselect all files\n" + \ + " Esc Unfocus+Unselect or Back to torrent list\n" + \ + " q/Backspace Back to torrent list" + else: + message += "q/Backspace/Esc Back to torrent list" + + width = max(map(lambda x: len(x), message.split("\n"))) + 4 + width = min(self.width, width) + height = min(self.height, message.count("\n")+3) + win = self.window(height, width, message=message, title=title) + while True: + if win.getch() >= 0: return + + + def window(self, height, width, message='', title=''): + height = min(self.height, height) + width = min(self.width, width) + ypos = int( (self.height - height) / 2 ) + xpos = int( (self.width - width) / 2 ) + win = curses.newwin(height, width, ypos, xpos) + win.box() + win.bkgd(' ', curses.A_REVERSE + curses.A_BOLD) + + if width >= 20: + win.addch(height-1, width-19, curses.ACS_RTEE) + win.addstr(height-1, width-18, " Close with Esc ") + win.addch(height-1, width-2, curses.ACS_LTEE) + + if width >= (len(title) + 6) and title != '': + win.addch(0, 1, curses.ACS_RTEE) + win.addstr(0, 2, " " + title + " ") + win.addch(0, len(title) + 4 , curses.ACS_LTEE) + + ypos = 1 + for line in message.split("\n"): + if len(line) > width: + line = line[0:width-7] + '...' + win.addstr(ypos, 2, line.encode('utf-8')) + ypos += 1 + return win + + + def dialog_ok(self, message): + height = 3 + message.count("\n") + width = max(max(map(lambda x: len(x), message.split("\n"))), 40) + 4 + win = self.window(height, width, message=message) + while True: + if win.getch() >= 0: return + + def dialog_yesno(self, message, important=False): + height = 5 + message.count("\n") + width = max(len(message), 8) + 4 + win = self.window(height, width, message=message) + win.keypad(True) + + if important: + win.bkgd(' ', curses.color_pair(self.colors.get_id('dialog_important')) + + curses.A_REVERSE) + + focus_tags = curses.color_pair(self.colors.get_id('button_focused')) + unfocus_tags = 0 + + input = False + while True: + win.move(height-2, (width/2)-4) + if input: + win.addstr('Y', focus_tags + curses.A_UNDERLINE) + win.addstr('es', focus_tags) + win.addstr(' ') + win.addstr('N', curses.A_UNDERLINE) + win.addstr('o') + else: + win.addstr('Y', curses.A_UNDERLINE) + win.addstr('es') + win.addstr(' ') + win.addstr('N', focus_tags + curses.A_UNDERLINE) + win.addstr('o', focus_tags) + + c = win.getch() + if c == ord('y'): + return True + elif c == ord('n'): + return False + elif c == ord("\t"): + input = not input + elif c == curses.KEY_LEFT or c == ord('h'): + input = True + elif c == curses.KEY_RIGHT or c == ord('l'): + input = False + elif c == ord("\n") or c == ord(' '): + return input + elif c == 27 or c == curses.KEY_BREAK: + return -1 + + def dialog_input_text(self, message, input='', on_change=None, on_enter=None): + width = self.width - 4 + textwidth = self.width - 8 + height = message.count("\n") + 4 + + win = self.window(height, width, message=message) + win.keypad(True) + show_cursor() + index = len(input) + while True: + # Cut the text into pages, each as long as the text field + # The current page is determined by index position + page = index // textwidth + displaytext = input[textwidth*page:textwidth*(page + 1)] + displayindex = index - textwidth*page + + color = (curses.color_pair(self.colors.get_id('dialog_important')) if self.highlight_dialog + else curses.color_pair(self.colors.get_id('dialog'))) + win.addstr(height - 2, 2, displaytext.ljust(textwidth), color) + win.move(height - 2, displayindex + 2) + c = win.getch() + if c == 27 or c == curses.KEY_BREAK: + hide_cursor() + return '' + elif index < len(input) and ( c == curses.KEY_RIGHT or c == curses.ascii.ctrl(ord('f')) ): + index += 1 + elif index > 0 and ( c == curses.KEY_LEFT or c == curses.ascii.ctrl(ord('b')) ): + index -= 1 + elif (c == curses.KEY_BACKSPACE or c == 127) and index > 0: + input = input[:index - 1] + (index < len(input) and input[index:] or '') + index -= 1 + if on_change: on_change(input) + elif index < len(input) and ( c == curses.KEY_DC or c == curses.ascii.ctrl(ord('d')) ): + input = input[:index] + input[index + 1:] + if on_change: on_change(input) + elif index < len(input) and c == curses.ascii.ctrl(ord('k')): + input = input[:index] + if on_change: on_change(input) + elif c == curses.KEY_HOME or c == curses.ascii.ctrl(ord('a')): + index = 0 + elif c == curses.KEY_END or c == curses.ascii.ctrl(ord('e')): + index = len(input) + elif c == ord('\n'): + if on_enter: + on_enter(input) + else: + hide_cursor() + return input + elif c >= 32 and c < 127: + input = input[:index] + chr(c) + (index < len(input) and input[index:] or '') + index += 1 + if on_change: on_change(input) + if on_change: win.redrawwin() + + def dialog_search_torrentlist(self, c): + self.dialog_input_text('Search torrent by title:', + on_change=self.draw_torrent_list, + on_enter=self.increment_search) + + def increment_search(self, input): + self.search_focus += 1 + self.draw_torrent_list(input) + + + def dialog_input_number(self, message, current_value, + cursorkeys=True, floating_point=False, allow_empty=False, + allow_zero=True, allow_negative_one=True): + if not allow_zero: + allow_negative_one = False + + width = max(max(map(lambda x: len(x), message.split("\n"))), 40) + 4 + width = min(self.width, width) + height = message.count("\n") + (4,6)[cursorkeys] + + show_cursor() + win = self.window(height, width, message=message) + win.keypad(True) + input = str(current_value) + if cursorkeys: + if floating_point: + bigstep = 1 + smallstep = 0.1 + else: + bigstep = 100 + smallstep = 10 + win.addstr(height-4, 2, (" up/down +/- %-3s" % bigstep).rjust(width-4)) + win.addstr(height-3, 2, ("left/right +/- %3s" % smallstep).rjust(width-4)) + if allow_negative_one: + win.addstr(height-3, 2, "-1 means unlimited") + if allow_empty: + win.addstr(height-4, 2, "leave empty for default") + + while True: + win.addstr(height-2, 2, input.ljust(width-4), curses.color_pair(self.colors.get_id('dialog'))) + win.move(height - 2, len(input) + 2) + c = win.getch() + if c == 27 or c == ord('q') or c == curses.KEY_BREAK: + hide_cursor() + return -128 + elif c == ord("\n"): + try: + if allow_empty and len(input) <= 0: + hide_cursor() + return -2 + elif floating_point: + hide_cursor() + return float(input) + else: + hide_cursor() + return int(input) + except ValueError: + hide_cursor() + return -1 + + elif c == curses.KEY_BACKSPACE or c == curses.KEY_DC or c == 127 or c == 8: + input = input[:-1] + elif len(input) >= width-5: + curses.beep() + elif c >= ord('1') and c <= ord('9'): + input += chr(c) + elif allow_zero and c == ord('0') and input != '-' and not input.startswith('0'): + input += chr(c) + elif allow_negative_one and c == ord('-') and len(input) == 0: + input += chr(c) + elif floating_point and c == ord('.') and not '.' in input: + input += chr(c) + + elif cursorkeys and c != -1: + try: + if input == '': input = 0 + if floating_point: number = float(input) + else: number = int(input) + if c == curses.KEY_LEFT or c == ord('h'): number -= smallstep + elif c == curses.KEY_RIGHT or c == ord('l'): number += smallstep + elif c == curses.KEY_DOWN or c == ord('j'): number -= bigstep + elif c == curses.KEY_UP or c == ord('k'): number += bigstep + if not allow_zero and number <= 0: + number = 1 + elif not allow_negative_one and number < 0: + number = 0 + elif number < 0: # input like -0.6 isn't useful + number = -1 + input = str(number) + except ValueError: + pass + + def dialog_menu(self, title, options, focus=1): + height = len(options) + 2 + width = max(max(map(lambda x: len(x[1])+3, options)), len(title)+3) + win = self.window(height, width) + + win.addstr(0,1, title) + win.keypad(True) + + old_focus = focus + while True: + keymap = self.dialog_list_menu_options(win, width, options, focus) + c = win.getch() + + if c > 96 and c < 123 and chr(c) in keymap: + return options[keymap[chr(c)]][0] + elif c == 27 or c == ord('q'): + return -128 + elif c == ord("\n"): + return options[focus-1][0] + elif c == curses.KEY_DOWN or c == ord('j') or c == curses.ascii.ctrl(ord('n')): + focus += 1 + if focus > len(options): focus = 1 + elif c == curses.KEY_UP or c == ord('k') or c == curses.ascii.ctrl(ord('p')): + focus -= 1 + if focus < 1: focus = len(options) + elif c == curses.KEY_HOME or c == ord('g'): + focus = 1 + elif c == curses.KEY_END or c == ord('G'): + focus = len(options) + + def dialog_list_menu_options(self, win, width, options, focus): + keys = dict() + i = 1 + for option in options: + title = option[1].split('_') + if i == focus: tag = curses.color_pair(self.colors.get_id('dialog')) + else: tag = 0 + win.addstr(i,2, title[0], tag) + win.addstr(title[1][0], tag + curses.A_UNDERLINE) + win.addstr(title[1][1:], tag) + win.addstr(''.ljust(width - len(option[1]) - 3), tag) + + keys[title[1][0].lower()] = i-1 + i+=1 + return keys + + def draw_options_dialog(self): + enc_options = [('required','_required'), ('preferred','_preferred'), ('tolerated','_tolerated')] + seed_ratio = self.stats['seedRatioLimit'] + while True: + options = [] + options.append(('Peer _Port', "%d" % self.stats['peer-port'])) + options.append(('UP_nP/NAT-PMP', ('disabled','enabled ')[self.stats['port-forwarding-enabled']])) + options.append(('Peer E_xchange', ('disabled','enabled ')[self.stats['pex-enabled']])) + options.append(('_Distributed Hash Table', ('disabled','enabled ')[self.stats['dht-enabled']])) + options.append(('_Local Peer Discovery', ('disabled','enabled ')[self.stats['lpd-enabled']])) + options.append(('Protocol En_cryption', "%s" % self.stats['encryption'])) + # uTP support was added in Transmission v2.3 + if server.get_rpc_version() >= 13: + options.append(('_Micro Transport Protocol', ('disabled','enabled')[self.stats['utp-enabled']])) + options.append(('_Global Peer Limit', "%d" % self.stats['peer-limit-global'])) + options.append(('Peer Limit per _Torrent', "%d" % self.stats['peer-limit-per-torrent'])) + options.append(('_Seed Ratio Limit', "%s" % ('unlimited',self.stats['seedRatioLimit'])[self.stats['seedRatioLimited']])) + options.append(('T_urtle Mode UL Limit', "%dK" % self.stats['alt-speed-up'])) + options.append(('Tu_rtle Mode DL Limit', "%dK" % self.stats['alt-speed-down'])) + options.append(('Title is Progress _Bar', ('no','yes')[self.torrentname_is_progressbar])) + + max_len = max([sum([len(re.sub('_', '', x)) for x in y[0]]) for y in options]) + win = self.window(len(options)+2, max_len+15, '', "Global Options") + + line_num = 1 + for option in options: + parts = re.split('_', option[0]) + parts_len = sum([len(x) for x in parts]) + + win.addstr(line_num, max_len-parts_len+2, parts.pop(0)) + for part in parts: + win.addstr(part[0], curses.A_UNDERLINE) + win.addstr(part[1:] + ': ' + option[1]) + line_num += 1 + + c = win.getch() + if c == 27 or c == ord('q') or c == ord("\n"): + return + elif c == ord('p'): + port = self.dialog_input_number("Port for incoming connections", + self.stats['peer-port'], + cursorkeys=False) + if port >= 0 and port <= 65535: + server.set_option('peer-port', port) + elif port != -128: # user hit ESC + self.dialog_ok('Port must be in the range of 0 - 65535') + elif c == ord('n'): + server.set_option('port-forwarding-enabled', + (1,0)[self.stats['port-forwarding-enabled']]) + elif c == ord('x'): + server.set_option('pex-enabled', (1,0)[self.stats['pex-enabled']]) + elif c == ord('d'): + server.set_option('dht-enabled', (1,0)[self.stats['dht-enabled']]) + elif c == ord('l'): + server.set_option('lpd-enabled', (1,0)[self.stats['lpd-enabled']]) + # uTP support was added in Transmission v2.3 + elif c == ord('m') and server.get_rpc_version() >= 13: + server.set_option('utp-enabled', (1,0)[self.stats['utp-enabled']]) + elif c == ord('g'): + limit = self.dialog_input_number("Maximum number of connected peers", + self.stats['peer-limit-global'], + allow_negative_one=False) + if limit >= 0: + server.set_option('peer-limit-global', limit) + elif c == ord('t'): + limit = self.dialog_input_number("Maximum number of connected peers per torrent", + self.stats['peer-limit-per-torrent'], + allow_negative_one=False) + if limit >= 0: + server.set_option('peer-limit-per-torrent', limit) + elif c == ord('s'): + limit = self.dialog_input_number('Stop seeding with upload/download ratio', + (-1,self.stats['seedRatioLimit'])[self.stats['seedRatioLimited']], + floating_point=True) + if limit >= 0: + server.set_option('seedRatioLimit', limit) + server.set_option('seedRatioLimited', True) + elif limit < 0 and limit != -128: + server.set_option('seedRatioLimited', False) + elif c == ord('c'): + choice = self.dialog_menu('Encryption', enc_options, + map(lambda x: x[0]==self.stats['encryption'], enc_options).index(True)+1) + if choice != -128: + server.set_option('encryption', choice) + elif c == ord('u'): + limit = self.dialog_input_number('Upload limit for Turtle Mode in kilobytes per second', + self.stats['alt-speed-up'], + allow_negative_one=False) + if limit != -128: + server.set_option('alt-speed-up', limit) + elif c == ord('r'): + limit = self.dialog_input_number('Download limit for Turtle Mode in kilobytes per second', + self.stats['alt-speed-down'], + allow_negative_one=False) + if limit != -128: + server.set_option('alt-speed-down', limit) + elif c == ord('b'): + self.torrentname_is_progressbar = not self.torrentname_is_progressbar + + self.draw_torrent_list() + +# End of class Interface + + + +def percent(full, part): + try: percent = 100/(float(full) / float(part)) + except ZeroDivisionError: percent = 0.0 + return percent + + +def scale_time(seconds, type='short'): + minute_in_sec = float(60) + hour_in_sec = float(3600) + day_in_sec = float(86400) + month_in_sec = 27.321661 * day_in_sec # from wikipedia + year_in_sec = 365.25 * day_in_sec # from wikipedia + + if seconds < 0: + return ('?', 'some time')[type=='long'] + + elif seconds < minute_in_sec: + if type == 'long': + if seconds < 5: + return 'now' + else: + return "%d second%s" % (seconds, ('', 's')[seconds>1]) + else: + return "%ds" % seconds + + elif seconds < hour_in_sec: + minutes = round(seconds / minute_in_sec, 0) + if type == 'long': + return "%d minute%s" % (minutes, ('', 's')[minutes>1]) + else: + return "%dm" % minutes + + elif seconds < day_in_sec: + hours = round(seconds / hour_in_sec, 0) + if type == 'long': + return "%d hour%s" % (hours, ('', 's')[hours>1]) + else: + return "%dh" % hours + + elif seconds < month_in_sec: + days = round(seconds / day_in_sec, 0) + if type == 'long': + return "%d day%s" % (days, ('', 's')[days>1]) + else: + return "%dd" % days + + elif seconds < year_in_sec: + months = round(seconds / month_in_sec, 0) + if type == 'long': + return "%d month%s" % (months, ('', 's')[months>1]) + else: + return "%dM" % months + + else: + years = round(seconds / year_in_sec, 0) + if type == 'long': + return "%d year%s" % (years, ('', 's')[years>1]) + else: + return "%dy" % years + + +def timestamp(timestamp, format="%x %X"): + if timestamp < 1: + return 'never' + + absolute = time.strftime(format, time.localtime(timestamp)) + if timestamp > time.time(): + relative = 'in ' + scale_time(int(timestamp - time.time()), 'long') + else: + relative = scale_time(int(time.time() - timestamp), 'long') + ' ago' + + if relative.startswith('now') or relative.endswith('now'): + relative = 'now' + return "%s (%s)" % (absolute, relative) + + +def scale_bytes(bytes, type='short'): + if bytes >= 1073741824: + scaled_bytes = round((bytes / 1073741824.0), 2) + unit = 'G' + elif bytes >= 1048576: + scaled_bytes = round((bytes / 1048576.0), 1) + if scaled_bytes >= 100: + scaled_bytes = int(scaled_bytes) + unit = 'M' + elif bytes >= 1024: + scaled_bytes = int(bytes / 1024) + unit = 'K' + else: + scaled_bytes = round((bytes / 1024.0), 1) + unit = 'K' + + + # handle 0 bytes special + if bytes == 0 and type == 'long': + return 'nothing' + + # convert to integer if .0 + if int(scaled_bytes) == float(scaled_bytes): + scaled_bytes = str(int(scaled_bytes)) + else: + scaled_bytes = str(scaled_bytes).rstrip('0') + + if type == 'long': + return num2str(bytes) + ' [' + scaled_bytes + unit + ']' + else: + return scaled_bytes + unit + + +def homedir2tilde(path): + return re.sub(r'^'+os.environ['HOME'], '~', path) +def tilde2homedir(path): + return re.sub(r'^~', os.environ['HOME'], path) + +def html2text(str): + str = re.sub(r'', "\n", str) + str = re.sub(r'

', ' ', str) + str = re.sub(r'<[^>]*?>', '', str) + return str + +def hide_cursor(): + try: curses.curs_set(0) # hide cursor if possible + except curses.error: pass # some terminals seem to have problems with that +def show_cursor(): + try: curses.curs_set(1) + except curses.error: pass + +def wrap_multiline(text, width, initial_indent='', subsequent_indent=None): + if subsequent_indent is None: + subsequent_indent = ' ' * len(initial_indent) + for line in text.splitlines(): + # this is required because wrap() strips empty lines + if not line.strip(): + yield line + continue + for line in wrap(line, width, replace_whitespace=False, + initial_indent=initial_indent, subsequent_indent=subsequent_indent): + yield line + initial_indent = subsequent_indent + +def ljust_columns(text, max_width, padchar=' '): + """ Returns a string that is exactly display columns wide, + padded with if necessary. Accounts for characters that are + displayed two columns wide, i.e. kanji. """ + + chars = [] + columns = 0 + max_width = max(0, max_width) + for character in text: + width = len_columns(character) + if columns + width <= max_width: + chars.append(character) + columns += width + else: + break + + # Fill up any remaining space + while columns < max_width: + assert len(padchar) == 1 + chars.append(padchar) + columns += 1 + return ''.join(chars) + +def len_columns(text): + """ Returns the amount of columns that would occupy. """ + columns = 0 + for character in text: + columns += 2 if unicodedata.east_asian_width(character) in ('W', 'F') else 1 + return columns + + +def num2str(num, format='%s'): + if int(num) == -1: + return '?' + elif int(num) == -2: + return 'oo' + else: + if num > 999: + return (re.sub(r'(\d{3})', '\g<1>,', str(num)[::-1])[::-1]).lstrip(',') + else: + return format % num + + +def debug(data): + if cmd_args.DEBUG: + file = open("debug.log", 'a') + if type(data) == type(str()): + file.write(data.encode('utf-8')) + else: + import pprint + pp = pprint.PrettyPrinter(indent=4) + file.write("\n====================\n" + pp.pformat(data) + "\n====================\n\n") + file.close + +def quit(msg='', exitcode=0): + try: + curses.endwin() + except curses.error: + pass + + # if this is a graceful exit and config file is present + if not msg and not exitcode: + save_config(cmd_args.configfile) + else: + print >> sys.stderr, msg, + os._exit(exitcode) + + +def explode_connection_string(connection): + host, port, path = \ + config.get('Connection', 'host'), \ + config.getint('Connection', 'port'), \ + config.get('Connection', 'path') + username, password = \ + config.get('Connection', 'username'), \ + config.get('Connection', 'password') + try: + if connection.count('@') == 1: + auth, connection = connection.split('@') + if auth.count(':') == 1: + username, password = auth.split(':') + if connection.count(':') == 1: + host, port = connection.split(':') + if port.count('/') >= 1: + port, path = port.split('/', 1) + port = int(port) + else: + host = connection + except ValueError: + quit("Wrong connection pattern: %s\n" % connection) + return host, port, path, username, password + +def create_url(host, port, path): + url = '%s:%d/%s' % (host, port, path) + url = url.replace('//', '/') # double-/ doesn't work for some reason + if config.getboolean('Connection', 'ssl'): + return 'https://%s' % url + else: + return 'http://%s' % url + +def read_netrc(file=os.environ['HOME'] + '/.netrc', hostname=None): + try: + login = password = '' + try: + login, account, password = netrc.netrc(file).authenticators(hostname) + except TypeError: + pass + try: + netrc.netrc(file).hosts[hostname] + except KeyError: + if hostname != 'localhost': + print "Unknown machine in %s: %s" % (file, hostname) + if login and password: + print "Using default login: %s" % login + else: + exit(CONFIGFILE_ERROR) + except netrc.NetrcParseError, e: + quit("Error in %s at line %s: %s\n" % (e.filename, e.lineno, e.msg)) + except IOError, msg: + quit("Cannot read %s: %s\n" % (file, msg)) + return login, password + + +# create initial config file +def create_config(option, opt_str, value, parser): + configfile = parser.values.configfile + config.read(configfile) + if parser.values.connection: + host, port, path, username, password = explode_connection_string(parser.values.connection) + config.set('Connection', 'host', host) + config.set('Connection', 'port', str(port)) + config.set('Connection', 'path', path) + config.set('Connection', 'username', username) + config.set('Connection', 'password', password) + + # create directory if necessary + dir = os.path.dirname(configfile) + if dir != '' and not os.path.isdir(dir): + try: + os.makedirs(dir) + except OSError, msg: + print msg + exit(CONFIGFILE_ERROR) + + # write file + if not save_config(configfile, force=True): + exit(CONFIGFILE_ERROR) + print "Wrote config file: %s" % configfile + exit(0) + +def save_config(filepath, force=False): + if force or os.path.isfile(filepath): + try: + config.write(open(filepath, 'w')) + os.chmod(filepath, 0600) # config may contain password + return 1 + except IOError, msg: + print >> sys.stderr, "Cannot write config file %s:\n%s" % (filepath, msg) + return 0 + return -1 + +def parse_sort_str(sort_str): + sort_orders = [] + for i in sort_str.split(','): + x = i.split(':') + if len(x) > 1: + sort_orders.append( { 'name':x[1], 'reverse':True } ) + else: + sort_orders.append( { 'name':x[0], 'reverse':False } ) + return sort_orders + +def show_version(option, opt_str, value, parser): + quit("transmission-remote-cli %s (supports Transmission %s-%s)\n" % \ + (VERSION, TRNSM_VERSION_MIN, TRNSM_VERSION_MAX)) + + +if __name__ == '__main__': + # command line parameters + default_config_path = os.environ['HOME'] + '/.config/transmission-remote-cli/settings.cfg' + parser = OptionParser(usage="%prog [options] [-- transmission-remote options]", + description="%%prog %s" % VERSION) + parser.add_option("-v", "--version", action="callback", callback=show_version, + help="Show version number and supported Transmission versions.") + parser.add_option("-c", "--connect", action="store", dest="connection", default="", + help="Point to the server using pattern [username:password@]host[:port]/[path]") + parser.add_option("-s", "--ssl", action="store_true", dest="ssl", default=False, + help="Connect to Transmission using SSL.") + parser.add_option("-f", "--config", action="store", dest="configfile", default=default_config_path, + help="Path to configuration file.") + parser.add_option("--create-config", action="callback", callback=create_config, + help="Create configuration file CONFIGFILE with default values.") + parser.add_option("-n", "--netrc", action="store_true", dest="use_netrc", default=False, + help="Get authentication info from your ~/.netrc file.") + parser.add_option("--debug", action="store_true", dest="DEBUG", default=False, + help="Everything passed to the debug() function will be added to the file debug.log.") + (cmd_args, transmissionremote_args) = parser.parse_args() + + + # read config from config file + config.read(cmd_args.configfile) + + # command line connection data can override config file + if cmd_args.connection: + host, port, path, username, password = explode_connection_string(cmd_args.connection) + config.set('Connection', 'host', host) + config.set('Connection', 'port', str(port)) + config.set('Connection', 'path', path) + config.set('Connection', 'username', username) + config.set('Connection', 'password', password) + if cmd_args.use_netrc: + username, password = read_netrc(hostname=config.get('Connection','host')) + config.set('Connection', 'username', username) + config.set('Connection', 'password', password) + if cmd_args.ssl: + config.set('Connection', 'ssl', 'True') + + + + # forward arguments after '--' to transmission-remote + if transmissionremote_args: + cmd = ['transmission-remote', '%s:%s' % + (config.get('Connection', 'host'), config.get('Connection', 'port'))] + + # one argument and it doesn't start with '-' --> treat it like it's a torrent link/url + if len(transmissionremote_args) == 1 and not transmissionremote_args[0].startswith('-'): + cmd.extend(['-a', transmissionremote_args[0]]) + + if config.get('Connection', 'username') and config.get('Connection', 'password'): + cmd_print = cmd + cmd_print.extend(['--auth', '%s:PASSWORD' % config.get('Connection', 'username')]) + print "EXECUTING:\n%s\nRESPONSE:" % ' '.join(cmd_print) + cmd.extend(['--auth', '%s:%s' % (config.get('Connection', 'username'), config.get('Connection', 'password'))]) + cmd.extend(transmissionremote_args) + else: + print "EXECUTING:\n%s\nRESPONSE:" % ' '.join(cmd) + + try: + retcode = call(cmd) + except OSError, msg: + quit("Could not execute the above command: %s\n" % msg, 128) + quit('', retcode) + + + norm = Normalizer() + server = Transmission(config.get('Connection', 'host'), + config.getint('Connection', 'port'), + config.get('Connection', 'path'), + config.get('Connection', 'username'), + config.get('Connection', 'password')) + Interface() + -- cgit v1.2.3