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.py | 3191 -------------------------------------------- 1 file changed, 3191 deletions(-) delete mode 100755 transmission-remote-cli.py (limited to 'transmission-remote-cli.py') diff --git a/transmission-remote-cli.py b/transmission-remote-cli.py deleted file mode 100755 index abcca78..0000000 --- a/transmission-remote-cli.py +++ /dev/null @@ -1,3191 +0,0 @@ -#!/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' - -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, 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