def parse_args(args):
parser = argparse.ArgumentParser()
parser.add_argument("--master", action="store_true")
parser.add_argument("--slave", action="store_true")
parser.add_argument("port", type=int)
return parser.parse_args(args)
if __name__ == '__main__':
args = parse_args(sys.argv[1:])
if not (args.master or args.slave):
print('wrong args')
if args.master:
print(f'starting master on {get_ip()}:{args.port}')
master = Master(args.port)
master.set_size(1200, 750)
addr, port = input('slave port? ').split(':')
master.set_slave((addr, int(port)))
master.run()
if args.slave:
print(f'starting slave on {get_ip()}:{args.port}')
slave = Slave(args.port)
addr, port = input('master port? ').split(':')
slave.set_master((addr, int(port)))
slave.run()
