Run the async cmdloop.
This commit is contained in:
parent
3315413087
commit
2194175d95
1 changed files with 7 additions and 4 deletions
|
@ -20,19 +20,22 @@ def main():
|
|||
arg_parser.add_argument('server_port', metavar='SERVER_PORT', type=int, nargs='?', default=8888)
|
||||
results = arg_parser.parse_args()
|
||||
|
||||
loop = asyncio.get_event_loop()
|
||||
terminal = None
|
||||
if results.option[0] == 'server':
|
||||
tracker = Tracker(results.host, results.host_port)
|
||||
tracker.start()
|
||||
loop.run_until_complete(tracker.start())
|
||||
terminal = TrackerTerminal(tracker)
|
||||
terminal.cmdloop()
|
||||
elif results.option[0] == 'peer':
|
||||
peer = Peer(results.host, results.host_port, results.server, results.server_port)
|
||||
peer.start()
|
||||
loop.run_until_complete(peer.start())
|
||||
terminal = PeerTerminal(peer)
|
||||
terminal.cmdloop()
|
||||
else:
|
||||
logging.error('Option must either be \'server\' or \'peer\'')
|
||||
|
||||
loop.run_until_complete(terminal.cmdloop())
|
||||
loop.close()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
Loading…
Reference in a new issue