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):
|
||||
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:
|
||||
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:
|
||||
out_error(e)
|
||||
else:
|
||||
|
@ -1450,6 +1449,7 @@ def select_alert(**kwargs):
|
|||
cur.close()
|
||||
con.close()
|
||||
|
||||
|
||||
def select_keep_alive(**kwargs):
|
||||
con, cur = get_cur()
|
||||
sql = """select ip from servers where active = 1 """
|
||||
|
@ -1463,6 +1463,35 @@ def select_keep_alive(**kwargs):
|
|||
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):
|
||||
try:
|
||||
import http.cookies
|
||||
|
|
Loading…
Reference in New Issue