mirror of https://github.com/Aidaho12/haproxy-wi
parent
54f9bf6577
commit
24229750b7
|
@ -55,7 +55,7 @@ def get_token():
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
def check_login(required_service=0):
|
def check_login(required_service=0) -> bool:
|
||||||
try:
|
try:
|
||||||
user_status, user_plan = funct.return_user_status()
|
user_status, user_plan = funct.return_user_status()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
@ -73,7 +73,10 @@ def check_login(required_service=0):
|
||||||
if sql.get_api_token(token):
|
if sql.get_api_token(token):
|
||||||
if required_service != 0:
|
if required_service != 0:
|
||||||
user_id = sql.get_user_id_by_api_token(token)
|
user_id = sql.get_user_id_by_api_token(token)
|
||||||
user_services = sql.select_user_services(user_id)
|
try:
|
||||||
|
user_services = sql.select_user_services(user_id)
|
||||||
|
except Exception:
|
||||||
|
return False
|
||||||
|
|
||||||
if str(required_service) in user_services:
|
if str(required_service) in user_services:
|
||||||
return True
|
return True
|
||||||
|
|
|
@ -75,6 +75,7 @@ if serv is not None and form.getvalue('config') is not None:
|
||||||
|
|
||||||
if save == 'delete':
|
if save == 'delete':
|
||||||
config = ''
|
config = ''
|
||||||
|
save = 'reload'
|
||||||
|
|
||||||
config = funct.rewrite_section(start_line, end_line, oldcfg, config)
|
config = funct.rewrite_section(start_line, end_line, oldcfg, config)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue