github
/
haproxy-wi
mirror of https://github.com/Aidaho12/haproxy-wi
1
0
Fork 0
Code Issues Releases Wiki Activity

v5.1.4.0 

Changelog: https://haproxy-wi.org/changelog.py#5_1_4
pull/304/head
Pavel Loginov 2021-06-08 13:37:42 +06:00
parent ac8a22669e
commit 7d4e504337
5 changed files with 142 additions and 130 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

223
app/funct.py
View File

@ -33,17 +33,20 @@ def get_data(log_type):
from datetime import datetime from datetime import datetime
from pytz import timezone from pytz import timezone
import sql import sql
fmt = "%Y-%m-%d.%H:%M:%S"
try: try:
now_utc = datetime.now(timezone(sql.get_setting('time_zone'))) now_utc = datetime.now(timezone(sql.get_setting('time_zone')))
except Exception: except Exception:
now_utc = datetime.now(timezone('UTC')) now_utc = datetime.now(timezone('UTC'))
if log_type == 'config': if log_type == 'config':
fmt = "%Y-%m-%d.%H:%M:%S" fmt = "%Y-%m-%d.%H:%M:%S"
if log_type == 'logs': elif log_type == 'logs':
fmt = '%Y%m%d' fmt = '%Y%m%d'
if log_type == "date_in_log": elif log_type == "date_in_log":
fmt = "%b %d %H:%M:%S" fmt = "%b %d %H:%M:%S"
return now_utc.strftime(fmt) return now_utc.strftime(fmt)
@ -80,7 +83,7 @@ def logging(serv, action, **kwargs):
if not os.path.exists(log_path): if not os.path.exists(log_path):
os.makedirs(log_path) os.makedirs(log_path)
try: try:
ip = cgi.escape(os.environ["REMOTE_ADDR"]) ip = cgi.escape(os.environ["REMOTE_ADDR"])
except Exception: except Exception:
@ -91,7 +94,7 @@ def logging(serv, action, **kwargs):
login = sql.get_user_name_by_uuid(user_uuid.value) login = sql.get_user_name_by_uuid(user_uuid.value)
except Exception: except Exception:
login = '' login = ''
if kwargs.get('alerting') == 1: if kwargs.get('alerting') == 1:
mess = get_data('date_in_log') + action + "\n" mess = get_data('date_in_log') + action + "\n"
log = open(log_path + "/checker-"+get_data('logs')+".log", "a") log = open(log_path + "/checker-"+get_data('logs')+".log", "a")
@ -124,13 +127,14 @@ def logging(serv, action, **kwargs):
log.close() log.close()
except IOError as e: except IOError as e:
print('<center><div class="alert alert-danger">Can\'t write log. Please check log_path in config %e</div></center>' % e) print('<center><div class="alert alert-danger">Can\'t write log. Please check log_path in config %e</div></center>' % e)
pass
def telegram_send_mess(mess, **kwargs): def telegram_send_mess(mess, **kwargs):
import telebot import telebot
from telebot import apihelper from telebot import apihelper
import sql import sql
token_bot = ''
channel_name = ''
if kwargs.get('telegram_channel_id'): if kwargs.get('telegram_channel_id'):
telegrams = sql.get_telegram_by_id(kwargs.get('telegram_channel_id')) telegrams = sql.get_telegram_by_id(kwargs.get('telegram_channel_id'))
@ -139,11 +143,11 @@ def telegram_send_mess(mess, **kwargs):
slack_send_mess(mess, ip=kwargs.get('ip')) slack_send_mess(mess, ip=kwargs.get('ip'))
proxy = sql.get_setting('proxy') proxy = sql.get_setting('proxy')
for telegram in telegrams: for telegram in telegrams:
token_bot = telegram[1] token_bot = telegram[1]
channel_name = telegram[2] channel_name = telegram[2]
if token_bot == '' or channel_name == '': if token_bot == '' or channel_name == '':
mess = " error: Can't send message. Add Telegram channel before use alerting at this servers group" mess = " error: Can't send message. Add Telegram channel before use alerting at this servers group"
print(mess) print(mess)
@ -164,6 +168,8 @@ def slack_send_mess(mess, **kwargs):
import sql import sql
from slack_sdk import WebClient from slack_sdk import WebClient
from slack_sdk.errors import SlackApiError from slack_sdk.errors import SlackApiError
slack_token = ''
channel_name = ''
if kwargs.get('slack_channel_id'): if kwargs.get('slack_channel_id'):
slacks = sql.get_slack_by_id(kwargs.get('slack_channel_id')) slacks = sql.get_slack_by_id(kwargs.get('slack_channel_id'))
@ -197,7 +203,7 @@ def check_login():
ref = os.environ.get("REQUEST_URI") ref = os.environ.get("REQUEST_URI")
sql.delete_old_uuid() sql.delete_old_uuid()
if user_uuid is not None: if user_uuid is not None:
sql.update_last_act_user(user_uuid.value) sql.update_last_act_user(user_uuid.value)
if sql.get_user_name_by_uuid(user_uuid.value) is None: if sql.get_user_name_by_uuid(user_uuid.value) is None:
@ -206,8 +212,8 @@ def check_login():
else: else:
print('<meta http-equiv="refresh" content="0; url=login.py?ref=%s">' % ref) print('<meta http-equiv="refresh" content="0; url=login.py?ref=%s">' % ref)
return False return False
def is_admin(**kwargs): def is_admin(**kwargs):
import sql import sql
import http.cookies import http.cookies
@ -219,10 +225,10 @@ def is_admin(**kwargs):
role = 4 role = 4
pass pass
level = kwargs.get("level") level = kwargs.get("level")
if level is None: if level is None:
level = 1 level = 1
try: try:
return True if role <= level else False return True if role <= level else False
except Exception: except Exception:
@ -234,20 +240,21 @@ def page_for_admin(**kwargs):
give_level = kwargs.get("level") give_level = kwargs.get("level")
else: else:
give_level = 1 give_level = 1
if not is_admin(level=give_level): if not is_admin(level=give_level):
print('<center><h3 style="color: red">How did you get here?! O_o You do not have need permissions</h>') print('<center><h3 style="color: red">How did you get here?! O_o You do not have need permissions</h>')
print('<meta http-equiv="refresh" content="5; url=/">') print('<meta http-equiv="refresh" content="5; url=/">')
import sys import sys
sys.exit() sys.exit()
def return_ssh_keys_path(serv, **kwargs): def return_ssh_keys_path(serv, **kwargs):
import sql import sql
full_path = get_config_var('main', 'fullpath') full_path = get_config_var('main', 'fullpath')
ssh_enable = '' ssh_enable = ''
ssh_user_name = '' ssh_user_name = ''
ssh_user_password = '' ssh_user_password = ''
ssh_key_name = ''
if kwargs.get('id'): if kwargs.get('id'):
for sshs in sql.select_ssh(id=kwargs.get('id')): for sshs in sql.select_ssh(id=kwargs.get('id')):
@ -261,25 +268,26 @@ def return_ssh_keys_path(serv, **kwargs):
ssh_user_name = sshs[4] ssh_user_name = sshs[4]
ssh_user_password = sshs[5] ssh_user_password = sshs[5]
ssh_key_name = full_path+'/keys/%s.pem' % sshs[2] ssh_key_name = full_path+'/keys/%s.pem' % sshs[2]
return ssh_enable, ssh_user_name, ssh_user_password, ssh_key_name return ssh_enable, ssh_user_name, ssh_user_password, ssh_key_name
def ssh_connect(serv, **kwargs): def ssh_connect(serv, **kwargs):
import paramiko import paramiko
from paramiko import SSHClient from paramiko import SSHClient
import sql import sql
ssh_enable, ssh_user_name, ssh_user_password, ssh_key_name = return_ssh_keys_path(serv)
ssh_enable, ssh_user_name, ssh_user_password, ssh_key_name = return_ssh_keys_path(serv)
servers = sql.select_servers(server=serv) servers = sql.select_servers(server=serv)
ssh_port = 22 ssh_port = 22
for server in servers: for server in servers:
ssh_port = server[10] ssh_port = server[10]
ssh = SSHClient() ssh = SSHClient()
ssh.load_system_host_keys() ssh.load_system_host_keys()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try: try:
if ssh_enable == 1: if ssh_enable == 1:
cloud = sql.is_cloud() cloud = sql.is_cloud()
@ -293,39 +301,31 @@ def ssh_connect(serv, **kwargs):
return ssh return ssh
except paramiko.AuthenticationException: except paramiko.AuthenticationException:
return 'Authentication failed, please verify your credentials' return 'Authentication failed, please verify your credentials'
pass
except paramiko.SSHException as sshException: except paramiko.SSHException as sshException:
return 'error: Unable to establish SSH connection: %s ' % sshException return 'error: Unable to establish SSH connection: %s ' % sshException
pass
except paramiko.PasswordRequiredException as e: except paramiko.PasswordRequiredException as e:
return 'error: %s ' % e return 'error: %s ' % e
pass
except paramiko.BadHostKeyException as badHostKeyException: except paramiko.BadHostKeyException as badHostKeyException:
return 'error: Unable to verify server\'s host key: %s ' % badHostKeyException return 'error: Unable to verify server\'s host key: %s ' % badHostKeyException
pass
except Exception as e: except Exception as e:
if e == "No such file or directory": if e == "No such file or directory":
return 'error: %s. Check ssh key' % e return 'error: %s. Check ssh key' % e
pass
elif e == "Invalid argument": elif e == "Invalid argument":
error = 'error: Check the IP of the server' return 'error: Check the IP of the server'
pass
else: else:
error = e return str(e)
pass
return str(error)
def get_config(serv, cfg, **kwargs): def get_config(serv, cfg, **kwargs):
import sql import sql
if kwargs.get("keepalived"): if kwargs.get("keepalived"):
config_path = "/etc/keepalived/keepalived.conf" config_path = "/etc/keepalived/keepalived.conf"
elif kwargs.get("nginx"): elif kwargs.get("nginx"):
config_path = sql.get_setting('nginx_config_path') config_path = sql.get_setting('nginx_config_path')
else: else:
config_path = sql.get_setting('haproxy_config_path') config_path = sql.get_setting('haproxy_config_path')
ssh = ssh_connect(serv) ssh = ssh_connect(serv)
try: try:
sftp = ssh.open_sftp() sftp = ssh.open_sftp()
@ -342,8 +342,8 @@ def get_config(serv, cfg, **kwargs):
ssh = str(e) ssh = str(e)
logging('localhost', ssh, haproxywi=1) logging('localhost', ssh, haproxywi=1)
return ssh return ssh
def diff_config(oldcfg, cfg): def diff_config(oldcfg, cfg):
import http.cookies import http.cookies
import sql import sql
@ -361,7 +361,7 @@ def diff_config(oldcfg, cfg):
login = '' login = ''
output, stderr = subprocess_execute(cmd) output, stderr = subprocess_execute(cmd)
for line in output: for line in output:
diff += date + " user: " + login + ", group: " + user_group + " " + line + "\n" diff += date + " user: " + login + ", group: " + user_group + " " + line + "\n"
try: try:
@ -372,7 +372,7 @@ def diff_config(oldcfg, cfg):
print('<center><div class="alert alert-danger">Can\'t read write change to log. %s</div></center>' % stderr) print('<center><div class="alert alert-danger">Can\'t read write change to log. %s</div></center>' % stderr)
pass pass
def get_sections(config, **kwargs): def get_sections(config, **kwargs):
return_config = list() return_config = list()
with open(config, 'r') as f: with open(config, 'r') as f:
@ -400,9 +400,9 @@ def get_sections(config, **kwargs):
): ):
line = line.strip() line = line.strip()
return_config.append(line) return_config.append(line)
return return_config return return_config
def get_section_from_config(config, section): def get_section_from_config(config, section):
record = False record = False
@ -417,7 +417,6 @@ def get_section_from_config(config, section):
record = True record = True
continue continue
if record: if record:
if ( if (
line.startswith('listen') or line.startswith('listen') or
line.startswith('frontend') or line.startswith('frontend') or
@ -437,15 +436,15 @@ def get_section_from_config(config, section):
end_line = end_line - 1 end_line = end_line - 1
else: else:
return_config += line return_config += line
if end_line == "": if end_line == "":
f = open(config, "r") f = open(config, "r")
line_list = f.readlines() line_list = f.readlines()
end_line = len(line_list) end_line = len(line_list)
return start_line, end_line, return_config return start_line, end_line, return_config
def rewrite_section(start_line, end_line, config, section): def rewrite_section(start_line, end_line, config, section):
record = False record = False
start_line = int(start_line) start_line = int(start_line)
@ -466,11 +465,11 @@ def rewrite_section(start_line, end_line, config, section):
continue continue
return_config += line return_config += line
return return_config return return_config
def get_userlists(config, **kwargs): def get_userlists(config):
return_config = '' return_config = ''
with open(config, 'r') as f: with open(config, 'r') as f:
for line in f: for line in f:
@ -479,11 +478,12 @@ def get_userlists(config, **kwargs):
return_config += line + ',' return_config += line + ','
return return_config return return_config
def get_backends_from_config(serv, backends='', **kwargs):
def get_backends_from_config(serv, backends=''):
configs_dir = get_config_var('configs', 'haproxy_save_configs_dir') configs_dir = get_config_var('configs', 'haproxy_save_configs_dir')
format_cfg = 'cfg' format_cfg = 'cfg'
try: try:
cfg = configs_dir+get_files(dir=configs_dir, format=format_cfg)[0] cfg = configs_dir+get_files(dir=configs_dir, format=format_cfg)[0]
except Exception as e: except Exception as e:
@ -505,7 +505,7 @@ def get_backends_from_config(serv, backends='', **kwargs):
if (line.startswith('listen') or line.startswith('frontend')) and 'stats' not in line: if (line.startswith('listen') or line.startswith('frontend')) and 'stats' not in line:
line = line.strip() line = line.strip()
print(line.split(' ')[1], end="<br>") print(line.split(' ')[1], end="<br>")
def get_all_stick_table(): def get_all_stick_table():
import sql import sql
@ -514,7 +514,7 @@ def get_all_stick_table():
output, stderr = subprocess_execute(cmd) output, stderr = subprocess_execute(cmd)
return output[0] return output[0]
def get_stick_table(table): def get_stick_table(table):
import sql import sql
hap_sock_p = sql.get_setting('haproxy_sock_port') hap_sock_p = sql.get_setting('haproxy_sock_port')
@ -524,10 +524,10 @@ def get_stick_table(table):
for i in output[0].split(','): for i in output[0].split(','):
i = i.split(':')[1] i = i.split(':')[1]
tables_head.append(i) tables_head.append(i)
cmd = 'echo "show table %s"|nc %s %s |grep -v "#"' % (table, serv, hap_sock_p) cmd = 'echo "show table %s"|nc %s %s |grep -v "#"' % (table, serv, hap_sock_p)
output, stderr = subprocess_execute(cmd) output, stderr = subprocess_execute(cmd)
return tables_head, output return tables_head, output
@ -557,36 +557,36 @@ def install_haproxy(serv, **kwargs):
stats_user = sql.get_setting('stats_user') stats_user = sql.get_setting('stats_user')
stats_password = sql.get_setting('stats_password') stats_password = sql.get_setting('stats_password')
proxy = sql.get_setting('proxy') proxy = sql.get_setting('proxy')
hapver = kwargs.get('hapver') haproxy_ver = kwargs.get('hapver')
server_for_installing = kwargs.get('server') server_for_installing = kwargs.get('server')
ssh_port = 22 ssh_port = 22
ssh_enable, ssh_user_name, ssh_user_password, ssh_key_name = return_ssh_keys_path(serv) ssh_enable, ssh_user_name, ssh_user_password, ssh_key_name = return_ssh_keys_path(serv)
if ssh_enable == 0: if ssh_enable == 0:
ssh_key_name = '' ssh_key_name = ''
servers = sql.select_servers(server=serv) servers = sql.select_servers(server=serv)
for server in servers: for server in servers:
ssh_port = str(server[10]) ssh_port = str(server[10])
os.system("cp scripts/%s ." % script) os.system("cp scripts/%s ." % script)
if hapver is None: if haproxy_ver is None:
hapver = '2.3.0-1' haproxy_ver = '2.3.0-1'
if proxy is not None and proxy != '' and proxy != 'None': if proxy is not None and proxy != '' and proxy != 'None':
proxy_serv = proxy proxy_serv = proxy
else: else:
proxy_serv = '' proxy_serv = ''
syn_flood_protect = '1' if kwargs.get('syn_flood') == "1" else '' syn_flood_protect = '1' if kwargs.get('syn_flood') == "1" else ''
commands = ["chmod +x " + script + " && ./" + script + " PROXY=" + proxy_serv + commands = ["chmod +x " + script + " && ./" + script + " PROXY=" + proxy_serv +
" SOCK_PORT=" + hap_sock_p + " STAT_PORT=" + stats_port + " STAT_FILE="+server_state_file + " SOCK_PORT=" + hap_sock_p + " STAT_PORT=" + stats_port + " STAT_FILE="+server_state_file +
" SSH_PORT=" + ssh_port + " STATS_USER=" + stats_user + " SSH_PORT=" + ssh_port + " STATS_USER=" + stats_user +
" STATS_PASS=" + stats_password + " HAPVER=" + hapver + " SYN_FLOOD=" + syn_flood_protect + " STATS_PASS=" + stats_password + " HAPVER=" + haproxy_ver + " SYN_FLOOD=" + syn_flood_protect +
" HOST=" + serv + " USER=" + ssh_user_name + " PASS=" + ssh_user_password + " KEY=" + ssh_key_name] " HOST=" + serv + " USER=" + ssh_user_name + " PASS=" + ssh_user_password + " KEY=" + ssh_key_name]
output, error = subprocess_execute(commands[0]) output, error = subprocess_execute(commands[0])
if server_for_installing: if server_for_installing:
service = server_for_installing + ' HAProxy' service = server_for_installing + ' HAProxy'
@ -596,8 +596,8 @@ def install_haproxy(serv, **kwargs):
os.system("rm -f %s" % script) os.system("rm -f %s" % script)
sql.update_haproxy(serv) sql.update_haproxy(serv)
def waf_install(serv): def waf_install(serv):
import sql import sql
script = "waf.sh" script = "waf.sh"
@ -607,26 +607,26 @@ def waf_install(serv):
ver = check_haproxy_version(serv) ver = check_haproxy_version(serv)
os.system("cp scripts/%s ." % script) os.system("cp scripts/%s ." % script)
if proxy is not None and proxy != '' and proxy != 'None': if proxy is not None and proxy != '' and proxy != 'None':
proxy_serv = proxy proxy_serv = proxy
else: else:
proxy_serv = '' proxy_serv = ''
commands = ["sudo chmod +x " + tmp_config_path+script + " && " + tmp_config_path+script + " PROXY=" + proxy_serv + commands = ["sudo chmod +x " + tmp_config_path+script + " && " + tmp_config_path+script + " PROXY=" + proxy_serv +
" HAPROXY_PATH=" + haproxy_dir + " VERSION=" + ver] " HAPROXY_PATH=" + haproxy_dir + " VERSION=" + ver]
error = str(upload(serv, tmp_config_path, script)) error = str(upload(serv, tmp_config_path, script))
if error: if error:
print('error: '+error) print('error: '+error)
logging('localhost', error, haproxywi=1) logging('localhost', error, haproxywi=1)
os.system("rm -f %s" % script) os.system("rm -f %s" % script)
stderr = ssh_command(serv, commands, print_out="1") stderr = ssh_command(serv, commands, print_out="1")
sql.insert_waf_metrics_enable(serv, "0") sql.insert_waf_metrics_enable(serv, "0")
sql.insert_waf_rules(serv) sql.insert_waf_rules(serv)
def install_nginx(serv, **kwargs): def install_nginx(serv, **kwargs):
import sql import sql
@ -639,39 +639,39 @@ def install_nginx(serv, **kwargs):
server_for_installing = kwargs.get('server') server_for_installing = kwargs.get('server')
proxy = sql.get_setting('proxy') proxy = sql.get_setting('proxy')
ssh_enable, ssh_user_name, ssh_user_password, ssh_key_name = return_ssh_keys_path(serv) ssh_enable, ssh_user_name, ssh_user_password, ssh_key_name = return_ssh_keys_path(serv)
if ssh_enable == 0: if ssh_enable == 0:
ssh_key_name = '' ssh_key_name = ''
os.system("cp scripts/%s ." % script) os.system("cp scripts/%s ." % script)
if proxy is not None and proxy != '' and proxy != 'None': if proxy is not None and proxy != '' and proxy != 'None':
proxy_serv = proxy proxy_serv = proxy
else: else:
proxy_serv = '' proxy_serv = ''
servers = sql.select_servers(server=serv) servers = sql.select_servers(server=serv)
for server in servers: for server in servers:
ssh_port = str(server[10]) ssh_port = str(server[10])
syn_flood_protect = '1' if form.getvalue('syn_flood') == "1" else '' syn_flood_protect = '1' if form.getvalue('syn_flood') == "1" else ''
commands = ["chmod +x " + script + " && ./" + script + " PROXY=" + proxy_serv + " STATS_USER=" + stats_user + commands = ["chmod +x " + script + " && ./" + script + " PROXY=" + proxy_serv + " STATS_USER=" + stats_user +
" STATS_PASS=" + stats_password + " SSH_PORT=" + ssh_port + " CONFIG_PATH=" + config_path + " STATS_PASS=" + stats_password + " SSH_PORT=" + ssh_port + " CONFIG_PATH=" + config_path +
" STAT_PORT=" + stats_port + " STAT_PAGE=" + stats_page+" SYN_FLOOD=" + syn_flood_protect + " STAT_PORT=" + stats_port + " STAT_PAGE=" + stats_page+" SYN_FLOOD=" + syn_flood_protect +
" HOST=" + serv + " USER=" + ssh_user_name + " PASS=" + ssh_user_password + " KEY=" + ssh_key_name] " HOST=" + serv + " USER=" + ssh_user_name + " PASS=" + ssh_user_password + " KEY=" + ssh_key_name]
output, error = subprocess_execute(commands[0]) output, error = subprocess_execute(commands[0])
if server_for_installing: if server_for_installing:
service = server_for_installing + ' Nginx' service = server_for_installing + ' Nginx'
else: else:
service = ' Nginx' service = ' Nginx'
show_installation_output(error, output, service) show_installation_output(error, output, service)
os.system("rm -f %s" % script) os.system("rm -f %s" % script)
sql.update_nginx(serv) sql.update_nginx(serv)
def update_haproxy_wi(service): def update_haproxy_wi(service):
if service != 'haproxy-wi': if service != 'haproxy-wi':
try: try:
@ -684,7 +684,7 @@ def update_haproxy_wi(service):
output, stderr = subprocess_execute(cmd) output, stderr = subprocess_execute(cmd)
print(output) print(output)
print(stderr) print(stderr)
def check_haproxy_version(serv): def check_haproxy_version(serv):
import sql import sql
@ -694,9 +694,10 @@ def check_haproxy_version(serv):
output, stderr = subprocess_execute(cmd) output, stderr = subprocess_execute(cmd)
for line in output: for line in output:
ver = line ver = line
return ver return ver
def upload(serv, path, file, **kwargs): def upload(serv, path, file, **kwargs):
error = "" error = ""
full_path = path + file full_path = path + file
@ -709,8 +710,7 @@ def upload(serv, path, file, **kwargs):
error = e.args error = e.args
logging('localhost', str(e.args[0]), haproxywi=1) logging('localhost', str(e.args[0]), haproxywi=1)
print(' Cannot upload '+file+' to '+full_path+' to server: '+serv+' error: '+str(e.args)) print(' Cannot upload '+file+' to '+full_path+' to server: '+serv+' error: '+str(e.args))
pass
try: try:
sftp = ssh.open_sftp() sftp = ssh.open_sftp()
except Exception as e: except Exception as e:
@ -724,8 +724,7 @@ def upload(serv, path, file, **kwargs):
error = e.args error = e.args
print('Cannot upload '+file+' to '+full_path+' to server: '+serv+' error: '+str(e.args)) print('Cannot upload '+file+' to '+full_path+' to server: '+serv+' error: '+str(e.args))
logging('localhost', ' Cannot upload '+file+' to '+full_path+' to server: '+serv+' Error: '+str(e.args), haproxywi=1) logging('localhost', ' Cannot upload '+file+' to '+full_path+' to server: '+serv+' Error: '+str(e.args), haproxywi=1)
pass
try: try:
sftp.close() sftp.close()
ssh.close() ssh.close()
@ -733,28 +732,26 @@ def upload(serv, path, file, **kwargs):
error = e.args error = e.args
logging('localhost', str(error[0]), haproxywi=1) logging('localhost', str(error[0]), haproxywi=1)
print('Cannot upload '+file+' to '+full_path+' to server: '+serv+' error: '+str(e.args)) print('Cannot upload '+file+' to '+full_path+' to server: '+serv+' error: '+str(e.args))
pass
return str(error) return str(error)
def upload_and_restart(serv, cfg, **kwargs): def upload_and_restart(serv, cfg, **kwargs):
import sql import sql
error = ""
if kwargs.get("nginx"): if kwargs.get("nginx"):
config_path = sql.get_setting('nginx_config_path') config_path = sql.get_setting('nginx_config_path')
tmp_file = sql.get_setting('tmp_config_path') + "/" + get_data('config') + ".conf" tmp_file = sql.get_setting('tmp_config_path') + "/" + get_data('config') + ".conf"
else: else:
config_path = sql.get_setting('haproxy_config_path') config_path = sql.get_setting('haproxy_config_path')
tmp_file = sql.get_setting('tmp_config_path') + "/" + get_data('config') + ".cfg" tmp_file = sql.get_setting('tmp_config_path') + "/" + get_data('config') + ".cfg"
error = ""
try: try:
os.system("dos2unix "+cfg) os.system("dos2unix "+cfg)
except OSError: except OSError:
return 'Please install dos2unix' return 'Please install dos2unix'
pass
if kwargs.get("keepalived") == 1: if kwargs.get("keepalived") == 1:
if kwargs.get("just_save") == "save": if kwargs.get("just_save") == "save":
commands = ["sudo mv -f " + tmp_file + " /etc/keepalived/keepalived.conf"] commands = ["sudo mv -f " + tmp_file + " /etc/keepalived/keepalived.conf"]
@ -795,10 +792,10 @@ def upload_and_restart(serv, cfg, **kwargs):
error += e error += e
if error: if error:
logging('localhost', error, haproxywi=1) logging('localhost', error, haproxywi=1)
return error return error
def master_slave_upload_and_restart(serv, cfg, just_save, **kwargs): def master_slave_upload_and_restart(serv, cfg, just_save, **kwargs):
import sql import sql
masters = sql.is_master(serv) masters = sql.is_master(serv)
@ -806,21 +803,21 @@ def master_slave_upload_and_restart(serv, cfg, just_save, **kwargs):
for master in masters: for master in masters:
if master[0] is not None: if master[0] is not None:
error += upload_and_restart(master[0], cfg, just_save=just_save, nginx=kwargs.get('nginx')) error += upload_and_restart(master[0], cfg, just_save=just_save, nginx=kwargs.get('nginx'))
error += upload_and_restart(serv, cfg, just_save=just_save, nginx=kwargs.get('nginx')) error += upload_and_restart(serv, cfg, just_save=just_save, nginx=kwargs.get('nginx'))
return error return error
def open_port_firewalld(cfg, serv, **kwargs): def open_port_firewalld(cfg, serv, **kwargs):
try: try:
conf = open(cfg, "r") conf = open(cfg, "r")
except IOError: except IOError:
print('<div class="alert alert-danger">Can\'t read export config file</div>') print('<div class="alert alert-danger">Cannot read exported config file</div>')
firewalld_commands = ' &&' firewalld_commands = ' &&'
ports = '' ports = ''
for line in conf: for line in conf:
if kwargs.get('service') == 'nginx': if kwargs.get('service') == 'nginx':
if "listen " in line and '#' not in line: if "listen " in line and '#' not in line:
@ -851,12 +848,12 @@ def open_port_firewalld(cfg, serv, **kwargs):
pass pass
except Exception: except Exception:
pass pass
firewalld_commands += 'sudo firewall-cmd --reload -q' firewalld_commands += 'sudo firewall-cmd --reload -q'
logging(serv, ' Next ports have been opened: ' + ports + ' has opened ') logging(serv, ' Next ports have been opened: ' + ports + ' has opened ')
return firewalld_commands return firewalld_commands
def check_haproxy_config(serv): def check_haproxy_config(serv):
import sql import sql
commands = ["haproxy -q -c -f %s" % sql.get_setting('haproxy_config_path')] commands = ["haproxy -q -c -f %s" % sql.get_setting('haproxy_config_path')]
@ -873,6 +870,8 @@ def check_haproxy_config(serv):
def show_log(stdout, **kwargs): def show_log(stdout, **kwargs):
i = 0 i = 0
out = '' out = ''
grep = ''
if kwargs.get('grep'): if kwargs.get('grep'):
import re import re
grep = kwargs.get('grep') grep = kwargs.get('grep')
@ -1022,8 +1021,8 @@ def ssh_command(serv, commands, **kwargs):
for command in commands: for command in commands:
try: try:
stdin, stdout, stderr = ssh.exec_command(command, get_pty=True) stdin, stdout, stderr = ssh.exec_command(command, get_pty=True)
except Exception: except Exception as e:
continue logging('localhost', ' ' + str(e), haproxywi=1)
if kwargs.get("ip") == "1": if kwargs.get("ip") == "1":
show_ip(stdout) show_ip(stdout)
@ -1045,12 +1044,12 @@ def ssh_command(serv, commands, **kwargs):
if line: if line:
print("<div class='alert alert-warning'>"+line+"</div>") print("<div class='alert alert-warning'>"+line+"</div>")
logging('localhost', ' '+line, haproxywi=1) logging('localhost', ' '+line, haproxywi=1)
try: try:
ssh.close() ssh.close()
except Exception: except Exception:
logging('localhost', ' '+str(ssh), haproxywi=1) logging('localhost', ' '+str(ssh), haproxywi=1)
return "error: "+str(ssh) return "error: "+str(ssh)
pass
def subprocess_execute(cmd): def subprocess_execute(cmd):

33
app/templates/metrics.html
View File

@ -1,28 +1,27 @@
{% extends "base.html" %} {% extends "base.html" %}
{% block content %} {% block content %}
<style> <style>
table, th, tr, td { table, th, tr, td {
border: 1px solid #ddd; border: 1px solid #ddd;
align: center; align: center;
text-align: center; text-align: center;
} }
th, tr, td { th, tr, td {
width: 6%; width: 6%;
min-width: 3%; min-width: 3%;
padding: 10px; padding: 10px;
} }
</style> </style>
<link href="/inc/chart.min.css" rel="stylesheet"> <link href="/inc/chart.min.css" rel="stylesheet">
<script src="/inc/metrics.js"></script> <script src="/inc/metrics.js"></script>
<script src="/inc/chart.min.js"></script> <script src="/inc/chart.min.js"></script>
{% if services == '0' %} {% if services == '0' %}
<center> <div style="text-align: center;">
<br /> <br />
<h3>You do not have installed Metrics service. <h3>You do not have installed Metrics service.
Read <a href="https://haproxy-wi.org/services.py?service=metrics#installation" title="Metrics installation" style="color: #5d9ceb;" target="_blank">hear</a> Read <a href="https://haproxy-wi.org/services.py?service=metrics#installation" title="Metrics installation" style="color: #5d9ceb;" target="_blank">hear</a>
how to install Metrics service</h3> how to install Metrics service</h3>
</center> </div>
{% else %} {% else %}
{% if servers|length == 0 %} {% if servers|length == 0 %}
{% include 'include/getstarted.html' %} {% include 'include/getstarted.html' %}
@ -31,8 +30,8 @@ th, tr, td {
<tr class="overviewHead"> <tr class="overviewHead">
<th colspan=13 style="background-color: #d1ecf1"> <th colspan=13 style="background-color: #d1ecf1">
<span id="table_metrics_head" style="position: absolute;margin-left: 43%;">Metrics</span> <span id="table_metrics_head" style="position: absolute;margin-left: 43%;">Metrics</span>
<span id="en_table_metric" class="add switcher_table_metric" title="Enable showing Average table"></span> <span id="en_table_metric" class="plus switcher_table_metric" title="Enable display of the table of averages"></span>
<span id="dis_table_metric" class="minus switcher_table_metric" title="Disable showing Average table"></span> <span id="dis_table_metric" class="minus switcher_table_metric" title="Disable display of the table of averages"></span>
</th> </th>
</tr> </tr>
</table> </table>
@ -54,13 +53,18 @@ th, tr, td {
<div class="chart-container"> <div class="chart-container">
<canvas id="{{s.0}}" role="img"></canvas> <canvas id="{{s.0}}" role="img"></canvas>
</div> </div>
<div id="http_metrics_div" class="chart-container">
<canvas id="http_{{s.0}}" role="img"></canvas>
</div>
{% endfor %} {% endfor %}
<script> <script>
function showMetrics() { function showMetrics() {
let metrics = new Promise( let metrics = new Promise(
(resolve, reject) => { (resolve, reject) => {
removeData();
{% for s in servers %} {% for s in servers %}
getChartData('{{s.0}}') getChartData('{{s.0}}')
getHttpChartData('{{s.0}}')
{% endfor %} {% endfor %}
if (localStorage.getItem('table_metrics') == 0 || localStorage.getItem('table_metrics') === null) { if (localStorage.getItem('table_metrics') == 0 || localStorage.getItem('table_metrics') === null) {
$('#dis_table_metric').css('display', 'none'); $('#dis_table_metric').css('display', 'none');
@ -79,6 +83,7 @@ th, tr, td {
(resolve, reject) => { (resolve, reject) => {
{% for s in servers %} {% for s in servers %}
getChartData('{{s.0}}') getChartData('{{s.0}}')
getHttpChartData('{{s.0}}')
{% endfor %} {% endfor %}
}); });
metrics.then(); metrics.then();

3
app/templates/ovw.html
View File

@ -253,10 +253,11 @@
<td> <td>
{% if prometheus|int() >= 1 %} {% if prometheus|int() >= 1 %}
<span class="serverUp server-status" title="running {{ prometheus }} process"></span> <span class="serverUp server-status" title="running {{ prometheus }} process"></span>
<a href="http://{{host}}:9090" target="_blank" title="Open Prometheus" class="logs_link">Prometheus</a>
{% else %} {% else %}
<span class="serverNone server-status" title="The service does not started or does not installed"></span> <span class="serverNone server-status" title="The service does not started or does not installed"></span>
Prometheus
{% endif %} {% endif %}
Prometheus
</td> </td>
{% endif %} {% endif %}
<td {% if role != 1 %}class="padding10 first-collumn-wi"{%endif%}> <td {% if role != 1 %}class="padding10 first-collumn-wi"{%endif%}>

7
inc/awesome.css
View File

@ -70,7 +70,7 @@
font-family: "Font Awesome 5 Solid"; font-family: "Font Awesome 5 Solid";
content: "\f24d"; content: "\f24d";
} }
.add .fa-clone { .add .fa-clone, .plus .fa-plus {
color: var(--green-color); color: var(--green-color);
} }
.add-proxy::before { .add-proxy::before {
@ -273,6 +273,11 @@
font-family: "Font Awesome 5 Solid"; font-family: "Font Awesome 5 Solid";
content: "\f068"; content: "\f068";
} }
.plus::before {
display: none;
font-family: "Font Awesome 5 Solid";
content: "\f067";
}
.row-down::after { .row-down::after {
display: none; display: none;
font-family: "Font Awesome 5 Solid"; font-family: "Font Awesome 5 Solid";

6
inc/style.css
View File

@ -500,7 +500,6 @@ ul{
} }
.menu li:last-child a, .menu li .v_menu li:last-child a{ .menu li:last-child a, .menu li .v_menu li:last-child a{
border-radius: 0 0 3px 3px; border-radius: 0 0 3px 3px;
border-bottom: 0;
} }
.menu li .v_menu li a{ .menu li .v_menu li a{
border-radius: 0; border-radius: 0;
@ -1200,7 +1199,7 @@ label {
} }
.switcher_table_metric { .switcher_table_metric {
margin-left: 99%; margin-left: 99%;
color: #a0d100; color: var(--green-color);
cursor: pointer; cursor: pointer;
} }
.geoip_country_code { .geoip_country_code {
@ -1210,3 +1209,6 @@ label {
width: 110px; width: 110px;
height: 60px; height: 60px;
} }
.help_cursor {
cursor: help;
}