mirror of https://github.com/Aidaho12/haproxy-wi
131 lines
4.5 KiB
Python
131 lines
4.5 KiB
Python
import os
|
|
import sys
|
|
|
|
from flask import render_template, request, redirect, url_for
|
|
from flask_login import login_required
|
|
|
|
from app import app, login_manager
|
|
from app.routes.checker import bp
|
|
|
|
sys.path.append(os.path.join(sys.path[0], '/var/www/haproxy-wi/app'))
|
|
|
|
import modules.db.sql as sql
|
|
import modules.common.common as common
|
|
import modules.roxywi.common as roxywi_common
|
|
import modules.tools.alerting as alerting
|
|
import modules.tools.checker as checker_mod
|
|
|
|
|
|
@bp.before_request
|
|
@login_required
|
|
def before_request():
|
|
""" Protect all of the admin endpoints. """
|
|
pass
|
|
|
|
|
|
@bp.route('/settings')
|
|
def checker_settings():
|
|
roxywi_common.check_user_group_for_flask()
|
|
|
|
try:
|
|
user_params = roxywi_common.get_users_params()
|
|
user = user_params['user']
|
|
except Exception:
|
|
return redirect(url_for('login_page'))
|
|
|
|
return render_template(
|
|
'checker.html', h2=1, role=user_params['role'], user=user, lang=user_params['lang'],
|
|
token=user_params['token'], user_services=user_params['user_services']
|
|
)
|
|
|
|
|
|
@bp.post('/settings/update')
|
|
def update_settings():
|
|
service = request.form.get('service')
|
|
setting_id = int(request.form.get('setting_id'))
|
|
email = int(request.form.get('email'))
|
|
service_alert = int(request.form.get('server'))
|
|
backend_alert = int(request.form.get('backend'))
|
|
maxconn_alert = int(request.form.get('maxconn'))
|
|
telegram_id = int(request.form.get('telegram_id'))
|
|
slack_id = int(request.form.get('slack_id'))
|
|
pd_id = int(request.form.get('pd_id'))
|
|
|
|
if service == 'haproxy':
|
|
return checker_mod.update_haproxy_settings(
|
|
setting_id, email, service_alert, backend_alert, maxconn_alert, telegram_id, slack_id, pd_id
|
|
)
|
|
elif service in ('nginx', 'apache'):
|
|
return checker_mod.update_service_settings(setting_id, email, service_alert, telegram_id, slack_id, pd_id)
|
|
else:
|
|
return checker_mod.update_keepalived_settings(setting_id, email, service_alert, backend_alert, telegram_id, slack_id, pd_id)
|
|
|
|
|
|
@bp.route('/settings/load')
|
|
def load_checker():
|
|
return checker_mod.load_checker()
|
|
|
|
|
|
@bp.route('/history')
|
|
def checker_history():
|
|
roxywi_common.check_user_group_for_flask()
|
|
|
|
user_group = roxywi_common.get_user_group(id=1)
|
|
smon_status, stderr = smon_mod.return_smon_status()
|
|
smon = sql.alerts_history('Checker', user_group)
|
|
user_subscription = roxywi_common.return_user_subscription()
|
|
|
|
try:
|
|
user_params = roxywi_common.get_users_params()
|
|
user = user_params['user']
|
|
except Exception:
|
|
return redirect(url_for('login_page'))
|
|
|
|
return render_template(
|
|
'smon/checker_history.html', h2=1, autorefresh=0, role=user_params['role'], user=user, smon=smon,
|
|
lang=user_params['lang'], user_status=user_subscription['user_status'], user_plan=user_subscription['user_plan'],
|
|
token=user_params['token'], smon_status=smon_status, smon_error=stderr, user_services=user_params['user_services']
|
|
)
|
|
|
|
|
|
@bp.route('/check/<channel_id>/<receiver_name>')
|
|
def check_receiver(channel_id, receiver_name):
|
|
channel_id = common.checkAjaxInput(channel_id)
|
|
receiver_name = common.checkAjaxInput(receiver_name)
|
|
|
|
return alerting.check_receiver(channel_id, receiver_name)
|
|
|
|
|
|
@bp.route('/check/rabbit')
|
|
def check_rabbit():
|
|
return alerting.check_rabbit_alert()
|
|
|
|
|
|
@bp.route('/check/email')
|
|
def check_email():
|
|
return alerting.check_email_alert()
|
|
|
|
|
|
@bp.route('/receiver/<receiver_name>', methods=['PUT', 'POST', 'DELETE'])
|
|
def receiver(receiver_name):
|
|
if request.method == 'POST':
|
|
token = common.checkAjaxInput(request.form.get('receiver'))
|
|
channel = common.checkAjaxInput(request.form.get('channel'))
|
|
group = common.checkAjaxInput(request.form.get('group'))
|
|
page = common.checkAjaxInput(request.form.get('page'))
|
|
page = page.split("#")[0]
|
|
|
|
return alerting.add_receiver_channel(receiver_name, token, channel, group, page)
|
|
elif request.method == 'PUT':
|
|
token = common.checkAjaxInput(request.form.get('receiver_token'))
|
|
channel = common.checkAjaxInput(request.form.get('channel'))
|
|
group = common.checkAjaxInput(request.form.get('group'))
|
|
user_id = common.checkAjaxInput(request.form.get('id'))
|
|
|
|
return alerting.update_receiver_channel(receiver_name, token, channel, group, user_id)
|
|
elif request.method == 'DELETE':
|
|
channel_id = common.checkAjaxInput(request.form.get('channel_id'))
|
|
|
|
return alerting.delete_receiver_channel(channel_id, receiver_name)
|
|
|