mirror of https://github.com/Aidaho12/haproxy-wi
parent
695bd7155f
commit
40fa9c06ef
|
@ -2279,13 +2279,13 @@ if form.getvalue('newsmon') is not None:
|
||||||
print('SMON error: Cannot be HTTP with 443 port')
|
print('SMON error: Cannot be HTTP with 443 port')
|
||||||
sys.exit()
|
sys.exit()
|
||||||
|
|
||||||
if sql.insert_smon(server, port, enable, http, uri, body, group, desc, telegram, user_group):
|
last_id = sql.insert_smon(server, port, enable, http, uri, body, group, desc, telegram, user_group)
|
||||||
|
if last_id:
|
||||||
from jinja2 import Environment, FileSystemLoader
|
from jinja2 import Environment, FileSystemLoader
|
||||||
|
|
||||||
env = Environment(loader=FileSystemLoader('templates'), autoescape=True)
|
env = Environment(loader=FileSystemLoader('templates'), autoescape=True)
|
||||||
template = env.get_template('ajax/show_new_smon.html')
|
template = env.get_template('ajax/show_new_smon.html')
|
||||||
template = template.render(
|
template = template.render(
|
||||||
smon=sql.select_smon(user_group, ip=server, port=port, proto=http, uri=uri, body=body),
|
smon=sql.select_smon_by_id(last_id),
|
||||||
telegrams=sql.get_user_telegram_by_group(user_group))
|
telegrams=sql.get_user_telegram_by_group(user_group))
|
||||||
print(template)
|
print(template)
|
||||||
funct.logging('SMON', ' Has been add a new server ' + server + ' to SMON ', haproxywi=1, login=1)
|
funct.logging('SMON', ' Has been add a new server ' + server + ' to SMON ', haproxywi=1, login=1)
|
||||||
|
|
24
app/sql.py
24
app/sql.py
|
@ -1936,13 +1936,13 @@ def insert_smon(server, port, enable, proto, uri, body, group, desc, telegram, u
|
||||||
http = ''
|
http = ''
|
||||||
|
|
||||||
try:
|
try:
|
||||||
SMON.insert(ip=server, port=port, en=enable, desc=desc, group=group, http=http, body=body,
|
last_id = SMON.insert(ip=server, port=port, en=enable, desc=desc, group=group, http=http, body=body,
|
||||||
telegram_channel_id=telegram, user_group=user_group, status='3').execute()
|
telegram_channel_id=telegram, user_group=user_group, status='3').execute()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
out_error(e)
|
out_error(e)
|
||||||
return False
|
return False
|
||||||
else:
|
else:
|
||||||
return True
|
return last_id
|
||||||
|
|
||||||
|
|
||||||
def select_smon(user_group, **kwargs):
|
def select_smon(user_group, **kwargs):
|
||||||
|
@ -1958,11 +1958,6 @@ def select_smon(user_group, **kwargs):
|
||||||
else:
|
else:
|
||||||
user_group = "where user_group='%s'" % user_group
|
user_group = "where user_group='%s'" % user_group
|
||||||
|
|
||||||
if kwargs.get('body') is None:
|
|
||||||
body = ''
|
|
||||||
else:
|
|
||||||
body = kwargs.get('body')
|
|
||||||
|
|
||||||
if kwargs.get('ip'):
|
if kwargs.get('ip'):
|
||||||
try:
|
try:
|
||||||
http = kwargs.get('proto')+':'+kwargs.get('uri')
|
http = kwargs.get('proto')+':'+kwargs.get('uri')
|
||||||
|
@ -1976,6 +1971,21 @@ def select_smon(user_group, **kwargs):
|
||||||
%s order by `group`""" % user_group
|
%s order by `group`""" % user_group
|
||||||
else:
|
else:
|
||||||
sql = """select * from `smon` %s """ % user_group
|
sql = """select * from `smon` %s """ % user_group
|
||||||
|
|
||||||
|
try:
|
||||||
|
cursor.execute(sql)
|
||||||
|
except Exception as e:
|
||||||
|
out_error(e)
|
||||||
|
else:
|
||||||
|
return cursor.fetchall()
|
||||||
|
|
||||||
|
|
||||||
|
def select_smon_by_id(last_id):
|
||||||
|
cursor = conn.cursor()
|
||||||
|
|
||||||
|
sql = """select id, ip, port, en, http, body, telegram_channel_id, `desc`, `group`, user_group
|
||||||
|
from `smon` where id = {} """.format(last_id)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
cursor.execute(sql)
|
cursor.execute(sql)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|
Loading…
Reference in New Issue