mirror of
https://github.com/logos-storage/deluge.git
synced 2026-01-24 07:53:09 +00:00
97 lines
3.9 KiB
Python
97 lines
3.9 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# tab_details.py
|
|
|
|
from itertools import izip
|
|
|
|
import gobject
|
|
import gtk
|
|
|
|
import dgtk
|
|
|
|
class PeersTabManager(object):
|
|
def __init__(self, peer_view, manager):
|
|
self.peer_view = peer_view
|
|
self.manager = manager
|
|
|
|
# IP int, IP string, Client, Percent Complete, Down Speed, Up Speed
|
|
# IP int is for faster sorting
|
|
self.peer_store = gtk.ListStore(gobject.TYPE_UINT, str, str, float,
|
|
int, int)
|
|
# Stores IP -> gtk.TreeIter's iter mapping for quick look up
|
|
# in update_torrent_info_widget
|
|
self.peer_store_dict = {}
|
|
|
|
def build_peers_view(self):
|
|
def percent(column, cell, model, iter, data):
|
|
percent = float(model.get_value(iter, data))
|
|
percent_str = "%.2f%%"%percent
|
|
cell.set_property("text", percent_str)
|
|
|
|
self.peer_view.set_model(self.peer_store)
|
|
|
|
self.peer_ip_column = dgtk.add_text_column(self.peer_view,
|
|
_("IP Address"), 1)
|
|
self.peer_client_column = dgtk.add_text_column(self.peer_view,
|
|
_("Client"), 2)
|
|
self.peer_complete_column = dgtk.add_func_column(self.peer_view,
|
|
_("Percent Complete"), percent, 3)
|
|
self.peer_download_column = dgtk.add_func_column(self.peer_view,
|
|
_("Down Speed"), dgtk.cell_data_speed,
|
|
4)
|
|
self.peer_upload_column = dgtk.add_func_column(self.peer_view,
|
|
_("Up Speed"), dgtk.cell_data_speed,
|
|
5)
|
|
|
|
self.peer_ip_column.set_sort_column_id(0)
|
|
|
|
def clear_store(self):
|
|
self.peer_store.clear()
|
|
self.peer_store_dict = {}
|
|
|
|
def update(self, unique_id):
|
|
new_peer_info = self.manager.get_torrent_peer_info(unique_id)
|
|
new_ips = set()
|
|
|
|
for peer in new_peer_info:
|
|
# Update peers already in peers list
|
|
if peer['ip'] in self.peer_store_dict:
|
|
iter = self.peer_store_dict[peer['ip']]
|
|
|
|
dgtk.update_store(self.peer_store, iter, (3, 4, 5),
|
|
(round(peer["peer_has"], 2),
|
|
peer["download_speed"],
|
|
peer["upload_speed"]))
|
|
|
|
if peer['client'] != "":
|
|
new_ips.add(peer['ip'])
|
|
|
|
# Add new peers
|
|
if peer['ip'] not in self.peer_store_dict:
|
|
# convert IP to int for sorting purposes
|
|
ip_int = sum([int(byte) << shift
|
|
for byte, shift in
|
|
izip(peer["ip"].split("."),
|
|
(24, 16, 8, 0))])
|
|
|
|
client = peer["client"]
|
|
try:
|
|
client = client.decode('utf-8')
|
|
except UnicodeDecodeError:
|
|
# Fallback to latin-1 in case peer's client
|
|
# doesn't use utf-8. utorrent < 1.7 for example
|
|
client = client.decode('latin-1')
|
|
|
|
iter = self.peer_store.append([ip_int, peer["ip"],
|
|
client, round(peer["peer_has"], 2),
|
|
peer["download_speed"],
|
|
peer["upload_speed"]])
|
|
|
|
self.peer_store_dict[peer['ip']] = iter
|
|
|
|
# Remove peers that no longer exist in new_ips
|
|
for ip in set(self.peer_store_dict.keys()).difference(new_ips):
|
|
self.peer_store.remove(self.peer_store_dict[ip])
|
|
del self.peer_store_dict[ip]
|