bivashy/Blitz-Proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix bug

Browse Source
This commit is contained in:
Sarina
2024-07-22 16:13:17 +03:30
parent a9fe44c771
commit 62ec44ee23
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
core/cli.py
View File

@ -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)])

6
core/validator.py
View File

@ -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