class Torrent_Downloader():
''' Manages download logic:
- Creation and removal of peers.
- Book keeping of pieces downloaded and in progress.
- Checking completed pieces and writing to file.
'''
def __init__(self, torrent, start_listener_callback):
self.torrent = torrent
self.message_handler = MessageHandler(self.torrent, self)
self.start_listener_callback = start_listener_callback
self.ip = self.get_IP_address()
self.tracker = Tracker(self.torrent.announce,
self.torrent.get_params())
self.peers = self.create_peers()
self.io_loop = get_event_loop()
self.index = 0
self.callback_dict = {
'check_piece': self.torrent.check_piece_callback,
'pieces_changed': self.pieces_changed_callback,
'start_listener': self.start_listener_callback,
}
self.pieces_needed = []
def get_IP_address(self):
response = get('http://api.ipify.org?format=json')
ip_object = loads(response.text)
return ip_object["ip"]
def create_peers(self):
peers = []
for p in self.tracker.parse_peer_address():
if p[0] == self.ip:
continue
peers.append(Peer(p[0], p[1], self))
return peers
def pieces_changed_callback(self, peer):
''' Check if connected peer has pieces I need. Send interested message.
Call choose_piece.
If peer has no pieces I need, disconnect and remove from peers list.
'''
self.torrent.update_pieces_needed()
for i in self.torrent.pieces_needed:
if peer.has_pieces[i]:
self.io_loop.create_task(
self.message_handler.send_message(peer=peer, message_id=2))
self.choose_piece(peer=peer)
break
else:
self.peers.remove(peer)
def choose_piece(self, peer):
''' Finds the next needed piece, updates self.have and self.pieces_needed.
calls construct_request_payload.
'''
piece_index = self.torrent.pieces_needed[0]
self.torrent.have[piece_index] = True
self.torrent.update_pieces_needed()
self.message_handler.construct_request_payload(peer=peer,
piece_index=piece_index)
class Torrent_Downloader():
''' Manages download logic:
- Creation and removal of peers.
- Book keeping of pieces downloaded and in progress.
- Checking completed pieces and writing to file.
'''
def __init__(self, torrent, start_listener_callback):
self.torrent = torrent
self.message_handler = MessageHandler(self.torrent, self)
self.start_listener_callback = start_listener_callback
self.ip = self.get_IP_address()
self.tracker = Tracker(self.torrent.announce, self.torrent.get_params())
self.peers = self.create_peers()
self.io_loop = get_event_loop()
self.index = 0
self.callback_dict = {
'check_piece' : self.torrent.check_piece_callback,
'pieces_changed' : self.pieces_changed_callback,
'start_listener' : self.start_listener_callback,
}
self.pieces_needed = []
def get_IP_address(self):
response = get('http://api.ipify.org?format=json')
ip_object = loads(response.text)
return ip_object["ip"]
def create_peers(self):
peers = []
for p in self.tracker.parse_peer_address():
if p[0] == self.ip:
continue
peers.append(Peer(p[0], p[1], self))
return peers
def pieces_changed_callback(self, peer):
''' Check if connected peer has pieces I need. Send interested message.
Call choose_piece.
If peer has no pieces I need, disconnect and remove from peers list.
'''
self.torrent.update_pieces_needed()
for i in self.torrent.pieces_needed:
if peer.has_pieces[i]:
self.io_loop.create_task(self.message_handler.send_message(peer=peer, message_id=2))
self.choose_piece(peer=peer)
break
else:
self.peers.remove(peer)
def choose_piece(self, peer):
''' Finds the next needed piece, updates self.have and self.pieces_needed.
calls construct_request_payload.
'''
piece_index = self.torrent.pieces_needed[0]
self.torrent.have[piece_index] = True
self.torrent.update_pieces_needed()
self.message_handler.construct_request_payload(peer=peer, piece_index=piece_index)
def run(self):
sock = socket(AF_INET, SOCK_STREAM)
sock.connect(self.server_adr)
print("Client started")
mh = MessageHandler(sock, self.bsize, self.pubkey, self.privkey)
print("Enter command after the PocketDB > prompt. Enter .exit to exit")
while True:
print("PocketDB > ", end='')
command = input().strip()
if command == ".q":
mh.send_message("\n")
break
if command.startswith("filter"):
mh.send_message("select")
data = mh.receive_message()
lines = data.split("\n")
lines = lines[1:-1]
words = [i.strip() for i in lines]
target = command.split()[1:]
target = ' '.join(target)
rst = []
words.pop()
for word in words:
if match(target, word.split(',')[1].strip()):
rst.append(word)
print(">Matched results:")
print(rst)
continue
mh.send_message(command)
data = mh.receive_message()
if data == "\n\n\n":
break
print(data)
print("Database connection closed")
class Torrent_Downloader:
def __init__(self, torrent):
self.torrent = torrent
self.tracker = Tracker(self.torrent.announce, self.torrent.get_params())
self.peers = self.create_peers()
self.message_handler = MessageHandler(self.torrent, self)
self.io_loop = get_event_loop()
self.visualizer = Visualizer()
def create_peers(self):
peers = []
for p in self.tracker.peer_list:
if p[0] == self.torrent.ip:
continue
peers.append(Peer(p[0], p[1], self))
return peers
def pieces_changed_callback(self, peer):
''' Check if connected peer has pieces I need. Send interested message.
Call choose_piece.
If peer has no pieces I need, disconnect and remove from peers list.
'''
self.torrent.update_pieces_needed()
for i in self.torrent.pieces_needed:
if peer.has_pieces[i]: # created in bitfield function in message_handler
self.io_loop.create_task(self.message_handler.send_message(peer=peer, message_id=2))
self.choose_piece(peer=peer)
break
else:
self.peers.remove(peer)
def choose_piece(self, peer):
''' Finds the next needed piece, updates self.have and self.pieces_needed.
calls construct_request_payload.
'''
piece_index = self.torrent.pieces_needed[0]
self.torrent.have[piece_index] = True
self.torrent.update_pieces_needed()
self.message_handler.construct_request_payload(peer=peer, piece_index=piece_index, piece_length=self.torrent.piece_length)
