mirror of https://github.com/Aidaho12/haproxy-wi
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
2.4 KiB
102 lines
2.4 KiB
#!/usr/bin/env python3
|
|
import html
|
|
import cgi
|
|
import os
|
|
import http.cookies
|
|
import funct
|
|
import sql
|
|
from jinja2 import Environment, FileSystemLoader
|
|
env = Environment(loader=FileSystemLoader('templates/'))
|
|
template = env.get_template('config.html')
|
|
|
|
|
|
funct.check_login()
|
|
|
|
form = cgi.FieldStorage()
|
|
serv = form.getvalue('serv')
|
|
config_read = ""
|
|
cfg = ""
|
|
stderr = ""
|
|
error = ""
|
|
aftersave = ""
|
|
|
|
try:
|
|
cookie = http.cookies.SimpleCookie(os.environ.get("HTTP_COOKIE"))
|
|
user_id = cookie.get('uuid')
|
|
user = sql.get_user_name_by_uuid(user_id.value)
|
|
servers = sql.get_dick_permit()
|
|
token = sql.get_token(user_id.value)
|
|
except:
|
|
pass
|
|
|
|
hap_configs_dir = funct.get_config_var('configs', 'haproxy_save_configs_dir')
|
|
|
|
if serv is not None:
|
|
cfg = hap_configs_dir + serv + "-" + funct.get_data('config') + ".cfg"
|
|
|
|
if serv is not None and form.getvalue('open') is not None :
|
|
|
|
try:
|
|
funct.logging(serv, "config.py open config")
|
|
except:
|
|
pass
|
|
|
|
error = funct.get_config(serv, cfg)
|
|
|
|
try:
|
|
conf = open(cfg, "r")
|
|
config_read = conf.read()
|
|
conf.close
|
|
except IOError:
|
|
error += '<br />Can\'t read import config file'
|
|
|
|
os.system("/bin/mv %s %s.old" % (cfg, cfg))
|
|
|
|
if serv is not None and form.getvalue('config') is not None:
|
|
try:
|
|
funct.logging(serv, "config.py edited config")
|
|
except:
|
|
pass
|
|
|
|
config = form.getvalue('config')
|
|
oldcfg = form.getvalue('oldconfig')
|
|
save = form.getvalue('save')
|
|
aftersave = 1
|
|
try:
|
|
with open(cfg, "a") as conf:
|
|
conf.write(config)
|
|
except IOError:
|
|
error = "Can't read import config file"
|
|
|
|
MASTERS = sql.is_master(serv)
|
|
for master in MASTERS:
|
|
if master[0] != None:
|
|
funct.upload_and_restart(master[0], cfg, just_save=save)
|
|
|
|
stderr = funct.upload_and_restart(serv, cfg, just_save=save)
|
|
|
|
funct.diff_config(oldcfg, cfg)
|
|
|
|
if save:
|
|
c = http.cookies.SimpleCookie(os.environ.get("HTTP_COOKIE"))
|
|
c["restart"] = form.getvalue('serv')
|
|
print(c)
|
|
|
|
os.system("/bin/rm -f " + hap_configs_dir + "*.old")
|
|
|
|
print('Content-type: text/html\n')
|
|
template = template.render(h2 = 1, title = "Working with HAProxy configs",
|
|
role = sql.get_user_role_by_uuid(user_id.value),
|
|
action = "config.py",
|
|
user = user,
|
|
select_id = "serv",
|
|
serv = serv,
|
|
aftersave = aftersave,
|
|
config = config_read,
|
|
cfg = cfg,
|
|
selects = servers,
|
|
stderr = stderr,
|
|
error = error,
|
|
note = 1,
|
|
token = token)
|
|
print(template) |