mirror of https://github.com/Aidaho12/haproxy-wi
				
				
				
			
		
			
				
	
	
		
			102 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			Python
		
	
	
			
		
		
	
	
			102 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			Python
		
	
	
#!/usr/bin/env python3
 | 
						|
import cgi
 | 
						|
import os
 | 
						|
import http.cookies
 | 
						|
import funct
 | 
						|
import sql
 | 
						|
from jinja2 import Environment, FileSystemLoader
 | 
						|
env = Environment(loader=FileSystemLoader('templates/'), autoescape=True,extensions=['jinja2.ext.loopcontrols'])
 | 
						|
template = env.get_template('sections.html')
 | 
						|
 | 
						|
print('Content-type: text/html\n')
 | 
						|
funct.check_login()
 | 
						|
 | 
						|
form = funct.form
 | 
						|
serv = form.getvalue('serv')
 | 
						|
section = form.getvalue('section')
 | 
						|
sections = ""
 | 
						|
config_read = ""
 | 
						|
cfg = ""
 | 
						|
stderr = ""
 | 
						|
error = ""
 | 
						|
aftersave = ""
 | 
						|
start_line = ""
 | 
						|
end_line = ""
 | 
						|
 | 
						|
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)
 | 
						|
	role = sql.get_user_role_by_uuid(user_id.value)
 | 
						|
except:
 | 
						|
	pass
 | 
						|
 | 
						|
hap_configs_dir = funct.get_config_var('configs', 'haproxy_save_configs_dir')
 | 
						|
 | 
						|
if serv is not None and open is not None:
 | 
						|
	cfg = hap_configs_dir + serv + "-" + funct.get_data('config') + ".cfg"
 | 
						|
	error = funct.get_config(serv, cfg)
 | 
						|
	sections = funct.get_sections(cfg)
 | 
						|
 | 
						|
if serv is not None and section is not None :
 | 
						|
	
 | 
						|
	try:
 | 
						|
		funct.logging(serv, "sections.py open config")
 | 
						|
	except:
 | 
						|
		pass
 | 
						|
	
 | 
						|
	start_line, end_line, config_read = funct.get_section_from_config(cfg, section)
 | 
						|
 | 
						|
	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')
 | 
						|
	start_line = form.getvalue('start_line')
 | 
						|
	end_line = form.getvalue('end_line')
 | 
						|
	aftersave = 1
 | 
						|
	
 | 
						|
	config = funct.rewrite_section(start_line, end_line, oldcfg, config)
 | 
						|
	
 | 
						|
	try:
 | 
						|
		with open(cfg, "w") as conf:
 | 
						|
			conf.write(config)
 | 
						|
	except IOError:
 | 
						|
		error = "Can't read import config file"
 | 
						|
	
 | 
						|
	stderr = funct.master_slave_upload_and_restart(serv, cfg, just_save=save)
 | 
						|
		
 | 
						|
	funct.diff_config(oldcfg, cfg)
 | 
						|
	
 | 
						|
	os.system("/bin/rm -f " + hap_configs_dir + "*.old")
 | 
						|
 | 
						|
 | 
						|
template = template.render(h2 = 1, title = "Working with HAProxy configs",
 | 
						|
							role = role,
 | 
						|
							action = "sections.py",
 | 
						|
							user = user,
 | 
						|
							select_id = "serv",
 | 
						|
							serv = serv,
 | 
						|
							aftersave = aftersave,
 | 
						|
							config = config_read,
 | 
						|
							cfg = cfg,
 | 
						|
							selects = servers,
 | 
						|
							stderr = stderr,
 | 
						|
							error = error,
 | 
						|
							start_line = start_line,
 | 
						|
							end_line = end_line,
 | 
						|
							section = section,
 | 
						|
							sections = sections,
 | 
						|
							note = 1,
 | 
						|
							versions = funct.versions(),
 | 
						|
							token = token)
 | 
						|
print(template)
 |