mirror of https://github.com/Aidaho12/haproxy-wi
parent
c312f1290a
commit
36960cfc5f
33
app/sql.py
33
app/sql.py
|
@ -445,9 +445,8 @@ def get_user_name_by_uuid(uuid):
|
||||||
|
|
||||||
def get_user_role_by_uuid(uuid):
|
def get_user_role_by_uuid(uuid):
|
||||||
con, cur = get_cur()
|
con, cur = get_cur()
|
||||||
sql = """ select role.id from user left join uuid as uuid on user.id = uuid.user_id left join role on role.name = user.role where uuid.uuid = '%s' """ % uuid
|
|
||||||
try:
|
try:
|
||||||
cur.execute(sql)
|
cur.execute("select role.id from user left join uuid as uuid on user.id = uuid.user_id left join role on role.name = user.role where uuid.uuid = '%s'" % uuid)
|
||||||
except sqltool.Error as e:
|
except sqltool.Error as e:
|
||||||
out_error(e)
|
out_error(e)
|
||||||
else:
|
else:
|
||||||
|
@ -1450,6 +1449,7 @@ def select_alert(**kwargs):
|
||||||
cur.close()
|
cur.close()
|
||||||
con.close()
|
con.close()
|
||||||
|
|
||||||
|
|
||||||
def select_keep_alive(**kwargs):
|
def select_keep_alive(**kwargs):
|
||||||
con, cur = get_cur()
|
con, cur = get_cur()
|
||||||
sql = """select ip from servers where active = 1 """
|
sql = """select ip from servers where active = 1 """
|
||||||
|
@ -1463,6 +1463,35 @@ def select_keep_alive(**kwargs):
|
||||||
con.close()
|
con.close()
|
||||||
|
|
||||||
|
|
||||||
|
def select_keealived(serv, **kwargs):
|
||||||
|
con, cur = get_cur()
|
||||||
|
sql = """select keepalived from `servers` where ip='%s' """ % serv
|
||||||
|
try:
|
||||||
|
cur.execute(sql)
|
||||||
|
except sqltool.Error as e:
|
||||||
|
out_error(e)
|
||||||
|
else:
|
||||||
|
for value in cur.fetchone():
|
||||||
|
return value
|
||||||
|
cur.close()
|
||||||
|
con.close()
|
||||||
|
|
||||||
|
|
||||||
|
def update_keepalived(serv):
|
||||||
|
con, cur = get_cur()
|
||||||
|
sql = """update `servers` set `keepalived` = '1' where ip = '%s' """ % serv
|
||||||
|
try:
|
||||||
|
cur.execute(sql)
|
||||||
|
con.commit()
|
||||||
|
return True
|
||||||
|
except sqltool.Error as e:
|
||||||
|
out_error(e)
|
||||||
|
con.rollback()
|
||||||
|
return False
|
||||||
|
cur.close()
|
||||||
|
con.close()
|
||||||
|
|
||||||
|
|
||||||
def check_token_exists(token):
|
def check_token_exists(token):
|
||||||
try:
|
try:
|
||||||
import http.cookies
|
import http.cookies
|
||||||
|
|
Loading…
Reference in New Issue