weegee: cli: fix some type issues

This commit is contained in:
Shiz 2021-12-14 18:00:32 +01:00
parent 666f2a56b1
commit e68316f250
1 changed files with 5 additions and 3 deletions

View File

@ -329,7 +329,7 @@ def main():
interface_name = args.interface or 'wg0'
interface = WeegeeInterface.create(ctx, args.name, interface_name,
private_key=args.private_key, public_key=args.public_key,
addresses=args.address, port=args.port, hosts=args.hosts
addresses=args.address, port=args.port, hosts=[WeegeeHost.load(ctx, h) for h in args.host],
)
interface.save()
@ -455,7 +455,7 @@ def main():
)
else:
connection = WeegeeConnection.load(ctx, args.name)
connections.peers.extend(WeegeePeer.load(ctx, peer).item for peer in args.peer)
connection.peers += [WeegeePeer.load(ctx, peer).item for peer in args.peer]
if args.preshared_key:
connection.preshared_key = preshared_key
connection.save()
@ -468,8 +468,10 @@ def main():
def do_disconnect_peer(parser: argparse.ArgumentParser, args: argparse.Namespace, ctx: WeegeeContext) -> None:
connection = WeegeeConnection.load(ctx, args.name)
peers = connection.peers
for peer in args.peer:
connection.peers.remove(WeegeePeer.load(ctx, peer).item)
peers.remove(WeegeePeer.load(ctx, peer).item)
connection.peers = peers
if connection.peers:
connection.save()
else: