From 62ec44ee23513a0cdaf968edd28a79f05bc85f04 Mon Sep 17 00:00:00 2001 From: Sarina Date: Mon, 22 Jul 2024 16:13:17 +0330 Subject: [PATCH] Fix bug --- core/cli.py | 4 ++-- core/validator.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/core/cli.py b/core/cli.py index 5573e9c..3ba5e4a 100644 --- a/core/cli.py +++ b/core/cli.py @@ -59,7 +59,7 @@ def cli(): # region hysteria2 menu options @cli.command('install-hysteria2') -@click.option('--port','-p', required=True, help='New port for Hysteria2',type=int,validate=validator.validate_port) +@click.option('--port','-p', required=True, help='New port for Hysteria2',type=int,callback=validator.validate_port) def install_hysteria2(port:int): run_cmd(['bash', Command.INSTALL_HYSTERIA2.value, str(port)]) @@ -78,7 +78,7 @@ def restart_hysteria2(): @cli.command('change-hysteria2-port') -@click.option('--port','-p', required=True, help='New port for Hysteria2',type=int,validate=validator.validate_port) +@click.option('--port','-p', required=True, help='New port for Hysteria2',type=int,callback=validator.validate_port) def change_hysteria2_port(port:int): run_cmd(['bash', Command.CHANGE_PORT_HYSTERIA2.value, str(port)]) diff --git a/core/validator.py b/core/validator.py index 1ede3d4..bd4f8b0 100644 --- a/core/validator.py +++ b/core/validator.py @@ -1,10 +1,10 @@ import os -def validate_port(p:int)-> bool: - if p < 1 or p > 65535: +def validate_port(ctx,param,value)-> bool: + if value < 1 or value > 65535: return False # check if port is in use - if os.system(f'lsof -i:{p}') == 0: + if os.system(f'lsof -i:{value}') == 0: return False return True \ No newline at end of file