Add list_peers command for Peer.

This commit is contained in:
Yuxin Wang 2018-09-26 23:16:33 -04:00
parent f0740851d5
commit e2d130e687
2 changed files with 6 additions and 0 deletions

View file

@ -67,6 +67,9 @@ class Peer(MessageServer):
del self._publish_results[filename]
return is_success, message
def peers(self):
return tuple(self._peers.values())
def list_file(self):
self._write_message(self._server_sock, {
'type': MessageType.REQUEST_FILE_LIST,

View file

@ -43,6 +43,9 @@ class PeerTerminal(cmd.Cmd):
assert isinstance(peer, Peer)
self._peer = peer
def do_list_peers(self, arg):
print(self._peer.peers())
def do_publish(self, arg):
arg = arg.split(' ')[0]
_, message = self._peer.publish(arg)