def __init__(self, app, config):
self.app = app
self.config = config
self.metainfo = config.metainfo
self.info_hash = self.metainfo.info_hash
self.downloadSpeedMonitor = SpeedMonitor(5)
self.uploadSpeedMonitor = SpeedMonitor(5)
self.my_peer_id = generate_peer_id()
self.connectionManager = ConnectionManager(self)
self.pieceManager = BTPieceManager(self)
if not self.config.trackerless:
if len(self.metainfo.announce_list) > 0:
self.bttrackerclient = BTTrackerClient(self)
else:
raise Exception("Need at least one tracker if torrent is "+\
"not trackerless.")
self.status = None
class BTManager (object):
def __init__(self, app, config):
self.app = app
self.config = config
self.metainfo = config.metainfo
self.info_hash = self.metainfo.info_hash
self.downloadSpeedMonitor = SpeedMonitor(5)
self.uploadSpeedMonitor = SpeedMonitor(5)
self.my_peer_id = generate_peer_id()
self.connectionManager = ConnectionManager(self)
self.pieceManager = BTPieceManager(self)
if not self.config.trackerless:
if len(self.metainfo.announce_list) > 0:
self.bttrackerclient = BTTrackerClient(self)
else:
raise Exception("Need at least one tracker if torrent is "+\
"not trackerless.")
self.status = None
def startDownload(self):
self.pieceManager.start()
self.connectionManager.start()
self.downloadSpeedMonitor.start()
self.uploadSpeedMonitor.start()
if not self.config.trackerless:
self.bttrackerclient.start()
self.status = 'running'
def stopDownload(self):
self.pieceManager.stop()
self.connectionManager.stop()
self.downloadSpeedMonitor.stop()
self.uploadSpeedMonitor.stop()
if not self.config.trackerless:
self.bttrackerclient.stop()
self.status = 'stopped'
def get_speed(self):
"""Returns the speed in kibibit per second (Kibit/s).
"""
return {
"down": self.downloadSpeedMonitor.get_speed(),
"up": self.uploadSpeedMonitor.get_speed() }
def get_num_connections(self):
return {
"client": len(self.connectionManager.clientFactory.active_connection),
"server": len(self.connectionManager.serverFactory.active_connection)}
def add_peers(self, peers):
"""Adds peers to the torrent for downloading pieces.
@param peers list of tuples e.g. [('173.248.194.166', 12005),
('192.166.145.8', 13915)]
"""
self.connectionManager.clientFactory.updateTrackerPeers(peers)
def exit(self):
if self.status == 'running' :
self.stopDownload()
for i in self.__dict__ :
del self.__dict__[i]