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

v2.0.4 

MySQL support
pull/19/head
Aidaho12 2018-04-23 22:20:19 +06:00
parent 23cdc92940
commit cd462dcb7d
8 changed files with 246 additions and 159 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
README.md
View File

@ -25,9 +25,9 @@ For install just [dowload](https://github.com/Aidaho12/haproxy-wi/archive/master
$ cd /var/www/ $ cd /var/www/
$ unzip master.zip $ unzip master.zip
$ mv haproxy-wi-master/ haproxy-wi $ mv haproxy-wi-master/ haproxy-wi
$ chown -R apache:apache haproxy-wi/
$ pip install -r haproxy-wi/requirements.txt $ pip install -r haproxy-wi/requirements.txt
$ cd haproxy-wi/cgi-bin $ chmod +x haproxy-wi/cgi-bin/*.py
$ chmod +x *.py
``` ```
For Apache do virtualhost with cgi-bin. Like this: For Apache do virtualhost with cgi-bin. Like this:
@ -48,7 +48,16 @@ For Apache do virtualhost with cgi-bin. Like this:
</Directory> </Directory>
</VirtualHost> </VirtualHost>
``` ```
# Database support
Default Haproxy-WI use Sqlite, if you want use MySQL enable in config, and create database:
## For MySQL support:
```
MariaDB [(none)]> create user 'haproxy-wi'@'%';
MariaDB [(none)]> create database haproxywi;
MariaDB [(none)]> grant all on haproxywi.* to 'haproxy-wi'@'%' IDENTIFIED BY 'haproxy-wi';
```
![alt text](image/haproxy-wi-overview.jpeg "Overview page") ![alt text](image/haproxy-wi-overview.jpeg "Overview page")
# Settings # Settings
@ -64,9 +73,10 @@ Copy ssh key on all HAproxy servers
For Runtime API enable state file on HAproxt servers and need install socat on all haproxy servers: For Runtime API enable state file on HAproxt servers and need install socat on all haproxy servers:
``` ```
global global
server-state-file /etc/haproxy/haproxy/haproxy.state stats socket *:1999 level admin
server-state-file /etc/haproxy/haproxy/haproxy.state
defaults defaults
load-server-state-from-file global load-server-state-from-file global
``` ```
![alt text](image/haproxy-wi-logs.jpeg "View logs page") ![alt text](image/haproxy-wi-logs.jpeg "View logs page")

211
cgi-bin/create_db.py
View File

@ -1,116 +1,157 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import cgi import cgi
import html import html
import sqlite3 as sqlite
import os import os
import sys import sys
from configparser import ConfigParser, ExtendedInterpolation
db = "haproxy-wi.db" path_config = "haproxy-webintarface.config"
config = ConfigParser(interpolation=ExtendedInterpolation())
config.read(path_config)
mysql_enable = config.get('mysql', 'enable')
if mysql_enable == '1':
mysql_user = config.get('mysql', 'mysql_user')
mysql_password = config.get('mysql', 'mysql_password')
mysql_db = config.get('mysql', 'mysql_db')
mysql_host = config.get('mysql', 'mysql_host')
from mysql.connector import errorcode
import mysql.connector as sqltool
else:
db = "haproxy-wi.db"
import sqlite3 as sqltool
def check_db(): def check_db():
if os.path.isfile(db): if mysql_enable == '0':
if os.path.getsize(db) > 100: if os.path.isfile(db):
with open(db,'r', encoding = "ISO-8859-1") as f: if os.path.getsize(db) > 100:
header = f.read(100) with open(db,'r', encoding = "ISO-8859-1") as f:
if header.startswith('SQLite format 3'): header = f.read(100)
return False if header.startswith('SQLite format 3'):
else: return False
return True else:
return True
else:
return True
else: else:
return True con, cur = get_cur()
sql = """ select id from `groups` where id='1' """
try:
cur.execute(sql)
except sqltool.Error as err:
#print('<div class="alert alert-danger">')
#if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
# print("Something is wrong with your user name or password")
#elif err.errno == errorcode.ER_BAD_DB_ERROR:
# print("Database does not exist")
#else:
# print(err)
print('</div>')
return True
else:
return False
con.close()
def get_cur(): def get_cur():
con = sqlite.connect(db, isolation_level=None) if mysql_enable == '0':
cur = con.cursor() con = sqltool.connect(db, isolation_level=None)
else:
con = sqltool.connect(user=mysql_user, password=mysql_password,
host=mysql_host,
database=mysql_db)
cur = con.cursor()
return con, cur return con, cur
def create_table(): def create_table():
con, cur = get_cur() con, cur = get_cur()
sql = """ if mysql_enable == '0':
BEGIN TRANSACTION; sql = """
CREATE TABLE IF NOT EXISTS `user` ( CREATE TABLE IF NOT EXISTS user (
`id` INTEGER NOT NULL,
`username` VARCHAR ( 64 ) UNIQUE,
`email` VARCHAR ( 120 ) UNIQUE,
`password` VARCHAR ( 128 ),
`role` VARCHAR ( 128 ),
`groups` VARCHAR ( 120 ),
PRIMARY KEY(`id`)
);
INSERT INTO user (username, email, password, role, groups) VALUES ('admin','admin@localhost','admin','admin','1'),
('editor','editor@localhost','editor','editor','1'),
('guest','guest@localhost','guest','guest','1');
CREATE TABLE IF NOT EXISTS `servers` (
`id` INTEGER NOT NULL,
`hostname` VARCHAR ( 64 ) UNIQUE,
`ip` VARCHAR ( 64 ) UNIQUE,
`groups` VARCHAR ( 64 ),
PRIMARY KEY(`id`)
);
CREATE TABLE IF NOT EXISTS `roles_users` (
`user_id` INTEGER,
`role_id` INTEGER,
FOREIGN KEY(`user_id`) REFERENCES `user`(`id`),
FOREIGN KEY(`role_id`) REFERENCES `role`(`id`)
);
CREATE TABLE IF NOT EXISTS `role` (
`id` INTEGER NOT NULL,
`name` VARCHAR ( 80 ) UNIQUE,
`description` VARCHAR ( 255 ),
PRIMARY KEY(`id`)
);
INSERT INTO `role` (name, description) VALUES ('admin','Can do everything'),
('editor','Can edit configs'),
('guest','Read only access');
CREATE TABLE IF NOT EXISTS `groups` (
`id` INTEGER NOT NULL, `id` INTEGER NOT NULL,
`username` VARCHAR ( 64 ) UNIQUE, `name` VARCHAR ( 80 ) UNIQUE,
`email` VARCHAR ( 120 ) UNIQUE, `description` VARCHAR ( 255 ),
`password` VARCHAR ( 128 ), PRIMARY KEY(`id`)
`role` VARCHAR ( 128 ), );
`groups` VARCHAR ( 120 ), INSERT INTO `groups` (name, description) VALUES ('All','All servers enter in this group');
PRIMARY KEY(`id`)
); CREATE TABLE IF NOT EXISTS `servers` (
INSERT INTO `user` (username, email, password, role, groups) VALUES ('admin','admin@localhost','admin','admin','1'),
('editor','editor@localhost','editor','editor','1'),
('guest','guest@localhost','guest','guest','1');
CREATE TABLE IF NOT EXISTS `servers` (
`id` INTEGER NOT NULL, `id` INTEGER NOT NULL,
`hostname` VARCHAR ( 64 ) UNIQUE, `hostname` VARCHAR ( 64 ) UNIQUE,
`ip` VARCHAR ( 64 ) UNIQUE, `ip` VARCHAR ( 64 ) UNIQUE,
`groups` VARCHAR ( 64 ), `groups` VARCHAR ( 64 ),
PRIMARY KEY(`id`) PRIMARY KEY(`id`)
); );
CREATE TABLE IF NOT EXISTS `roles_users` ( """
`user_id` INTEGER, try:
`role_id` INTEGER, cur.executescript(sql)
FOREIGN KEY(`user_id`) REFERENCES `user`(`id`), except sqltool.Error as e:
FOREIGN KEY(`role_id`) REFERENCES `role`(`id`) print("An error occurred:", e)
); return False
CREATE TABLE IF NOT EXISTS `role` ( else:
`id` INTEGER NOT NULL, return True
`name` VARCHAR ( 80 ) UNIQUE,
`description` VARCHAR ( 255 ),
PRIMARY KEY(`id`)
);
INSERT INTO `role` (name, description) VALUES ('admin','Can do everything'),
('editor','Can edit configs'),
('guest','Read only access');
CREATE TABLE IF NOT EXISTS `groups` (
`id` INTEGER NOT NULL,
`name` VARCHAR ( 80 ) UNIQUE,
`description` VARCHAR ( 255 ),
PRIMARY KEY(`id`)
);
CREATE TABLE IF NOT EXISTS `groups` (
`id` INTEGER NOT NULL,
`name` VARCHAR ( 80 ) UNIQUE,
`description` VARCHAR ( 255 ),
PRIMARY KEY(`id`)
);
INSERT INTO `groups` (name, description) VALUES ('All','All servers enter in this group');
CREATE TABLE IF NOT EXISTS `servers` (
`id` INTEGER NOT NULL,
`hostname` VARCHAR ( 64 ) UNIQUE,
`ip` VARCHAR ( 64 ) UNIQUE,
`groups` VARCHAR ( 64 ),
PRIMARY KEY(`id`)
);
COMMIT;
"""
try:
cur.executescript(sql)
except sqlite.Error as e:
print("An error occurred:", e.args[0])
return False
else: else:
return True try:
for line in open("haproxy-wi.db.sql"):
cur.execute(line)
except sqltool.Error as e:
print('<div class="alert alert-danger">')
print("An error occurred:", e)
print('</div>')
return False
else:
return True
cur.close() cur.close()
con.close() con.close()
def update_db_v_2_0_1(): def update_db_v_2_0_1():
con, cur = get_cur() con, cur = get_cur()
sql = """ sql = """
ALTER TABLE servers ADD COLUMN type_ip INTEGER NOT NULL DEFAULT(0); ALTER TABLE `servers` ADD COLUMN type_ip INTEGER NOT NULL DEFAULT 0;
""" """
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
if e.args[0] == 'duplicate column name: type_ip': if e.args[0] == 'duplicate column name: type_ip':
print('Updating... go to version 2.0.1.1') print('Updating... go to version 2.0.1.1')
return False return False
else: else:
print("An error occurred:", e.args[0]) print("An error occurred:", e)
return False return False
else: else:
print("DB was update to 2.0.1") print("DB was update to 2.0.1")
@ -121,16 +162,16 @@ def update_db_v_2_0_1():
def update_db_v_2_0_1_1(): def update_db_v_2_0_1_1():
con, cur = get_cur() con, cur = get_cur()
sql = """ sql = """
ALTER TABLE `servers` ADD COLUMN enable INTEGER NOT NULL DEFAULT(1); ALTER TABLE `servers` ADD COLUMN enable INTEGER NOT NULL DEFAULT 1;
""" """
try: try:
cur.executescript(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
if e.args[0] == 'duplicate column name: enable': if e.args[0] == 'duplicate column name: enable':
print('Already updated. No run more. Thx =^.^=') print('Already updated. No run more. Thx =^.^=')
return False return False
else: else:
print("An error occurred:", e.args[0]) print("An error occurred:", e)
return False return False
else: else:
print("DB was update to 2.0.1.1") print("DB was update to 2.0.1.1")

8
cgi-bin/funct.py
View File

@ -146,7 +146,7 @@ def head(title):
'<div class="top-menu">' '<div class="top-menu">'
'<div class="LogoText">' '<div class="LogoText">'
'<span style="padding: 10px;">HAproxy-WI</span>' '<span style="padding: 10px;">HAproxy-WI</span>'
'<a href="#" id="hide_menu" title="Hide menu" style="margin-left: 16px;margin-top: -10px;position: absolute;">' '<a href="#" id="hide_menu" title="Hide menu" style="margin-left: 23px; position: absolute;">'
'<span class="ui-state-default ui-corner-all">' '<span class="ui-state-default ui-corner-all">'
'<span class="ui-icon ui-icon-arrowthick-1-w" id="arrow"></span>' '<span class="ui-icon ui-icon-arrowthick-1-w" id="arrow"></span>'
'</span>' '</span>'
@ -194,7 +194,7 @@ def links():
'</li>') '</li>')
print('</ul>' print('</ul>'
'</nav>' '</nav>'
'<div class="copyright-menu">HAproxy-WI v2.0.3</div>' '<div class="copyright-menu">HAproxy-WI v2.0.4</div>'
'</div>') '</div>')
def show_login_links(): def show_login_links():
@ -528,7 +528,7 @@ def chooseServer(formName, title, note, **kwargs):
print('</p></form>') print('</p></form>')
if note == "y": #if note == "y":
print('<p><b>Note:</b> If you reconfigure First server, second will reconfigured automatically</p>') # print('<p><b>Note:</b> If you reconfigure First server, second will reconfigured automatically</p>')
print('</center>') print('</center>')

8
cgi-bin/haproxy-webintarface.config
View File

@ -11,6 +11,14 @@ haproxy_configs_server = localhost
#Dir where configs will be save #Dir where configs will be save
haproxy_save_configs_dir = /var/www/haproxy-wi/cgi-bin/hap_config/ haproxy_save_configs_dir = /var/www/haproxy-wi/cgi-bin/hap_config/
[mysql]
#Enable MySQL DB. If default will be used Sqlite DB. Default disable
enable = 0
mysql_user = haproxy-wi
mysql_password = haproxy-wi
mysql_db = haproxywi
mysql_host = 127.0.0.1
[ssh] [ssh]
#If ssh connect disable entare password for ssh connect. Default enable #If ssh connect disable entare password for ssh connect. Default enable
ssh_keys_enable = 1 ssh_keys_enable = 1

12
cgi-bin/haproxy-wi.db.sql Normal file
View File

@ -0,0 +1,12 @@
CREATE TABLE IF NOT EXISTS `user` (`id` INTEGER NOT NULL AUTO_INCREMENT,`username` VARCHAR ( 64 ) UNIQUE,`email` VARCHAR ( 120 ) UNIQUE, `password` VARCHAR ( 128 ),`role` VARCHAR ( 128 ),`groups` VARCHAR ( 120 ),PRIMARY KEY(`id`) );
INSERT INTO `user` (username, email, password, role, groups) VALUES ('admin','admin@localhost','admin','admin','1');
INSERT INTO `user` (username, email, password, role, groups) VALUES ('editor','editor@localhost','editor','editor','1');
INSERT INTO `user` (username, email, password, role, groups) VALUES ('guest','guest@localhost','guest','guest','1');
CREATE TABLE IF NOT EXISTS `servers` (`id` INTEGER NOT NULL AUTO_INCREMENT,`hostname` VARCHAR ( 64 ) UNIQUE,`ip` VARCHAR ( 64 ) UNIQUE,`groups` VARCHAR ( 64 ),PRIMARY KEY(`id`) );
CREATE TABLE IF NOT EXISTS `role` (`id` INTEGER NOT NULL AUTO_INCREMENT,`name` VARCHAR ( 80 ) UNIQUE,`description` VARCHAR ( 255 ),PRIMARY KEY(`id`) );
INSERT INTO `role` (name, description) VALUES ('admin','Can do everything');
INSERT INTO `role` (name, description) VALUES ('editor','Can edit configs');
INSERT INTO `role` (name, description) VALUES ('guest','Read only access');
CREATE TABLE IF NOT EXISTS `groups` (`id` INTEGER NOT NULL AUTO_INCREMENT,`name` VARCHAR ( 80 ) UNIQUE,`description` VARCHAR ( 255 ),PRIMARY KEY(`id`));
INSERT INTO `groups` (name, description) VALUES ('All','All servers enter in this group');
CREATE TABLE IF NOT EXISTS `servers` (`id` INTEGER NOT NULL AUTO_INCREMENT,`hostname` VARCHAR ( 64 ) UNIQUE,`ip` VARCHAR ( 64 ) UNIQUE,`groups` VARCHAR ( 64 ), PRIMARY KEY(`id`));

5
cgi-bin/login.py
View File

@ -78,7 +78,4 @@ if login is not None and password is not None:
sys.exit() sys.exit()
login_page("error") login_page("error")
funct.footer() funct.footer()

140
cgi-bin/sql.py
View File

@ -1,21 +1,31 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# -*- coding: utf-8 -*-" # -*- coding: utf-8 -*-"
import sqlite3 as sqlite
import cgi import cgi
import create_db
def get_cur(): from configparser import ConfigParser, ExtendedInterpolation
con = sqlite.connect("haproxy-wi.db", isolation_level=None)
cur = con.cursor() path_config = "haproxy-webintarface.config"
return con, cur config = ConfigParser(interpolation=ExtendedInterpolation())
config.read(path_config)
mysql_enable = config.get('mysql', 'enable')
if mysql_enable == '1':
from mysql.connector import errorcode
import mysql.connector as sqltool
else:
db = "haproxy-wi-test.db"
import sqlite3 as sqltool
def add_user(user, email, password, role, group): def add_user(user, email, password, role, group):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """INSERT INTO user (username, email, password, role, groups) VALUES ('%s', '%s', '%s', '%s', '%s')""" % (user, email, password, role, group) sql = """INSERT INTO user (username, email, password, role, groups) VALUES ('%s', '%s', '%s', '%s', '%s')""" % (user, email, password, role, group)
try: try:
with con: cur.execute(sql)
cur.executescript(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
con.rollback()
return False return False
else: else:
return True return True
@ -23,7 +33,7 @@ def add_user(user, email, password, role, group):
con.close() con.close()
def update_user(user, email, password, role, group, id): def update_user(user, email, password, role, group, id):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """update user set username = '%s', sql = """update user set username = '%s',
email = '%s', email = '%s',
password = '%s', password = '%s',
@ -31,10 +41,11 @@ def update_user(user, email, password, role, group, id):
groups = '%s' groups = '%s'
where id = '%s'""" % (user, email, password, role, group, id) where id = '%s'""" % (user, email, password, role, group, id)
try: try:
with con: cur.execute(sql)
cur.executescript(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
con.rollback()
return False return False
else: else:
return True return True
@ -42,25 +53,27 @@ def update_user(user, email, password, role, group, id):
con.close() con.close()
def delete_user(id): def delete_user(id):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """delete from user where id = '%s'""" % (id) sql = """delete from user where id = '%s'""" % (id)
try: try:
with con: cur.execute(sql)
cur.executescript(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
con.rollback()
else: else:
return True return True
cur.close() cur.close()
def add_group(name, description): def add_group(name, description):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """INSERT INTO groups (name, description) VALUES ('%s', '%s')""" % (name, description) sql = """INSERT INTO groups (name, description) VALUES ('%s', '%s')""" % (name, description)
try: try:
with con: cur.execute(sql)
cur.execute(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
con.rollback()
return False return False
else: else:
print(cur.lastrowid) print(cur.lastrowid)
@ -69,19 +82,20 @@ def add_group(name, description):
con.close() con.close()
def delete_group(id): def delete_group(id):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """delete from groups where id = '%s'""" % (id) sql = """delete from groups where id = '%s'""" % (id)
try: try:
with con: cur.execute(sql)
cur.executescript(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
con.rollback()
else: else:
return True return True
cur.close() cur.close()
def update_group(name, descript, id): def update_group(name, descript, id):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """ sql = """
update groups set update groups set
name = '%s', name = '%s',
@ -89,10 +103,11 @@ def update_group(name, descript, id):
where id = '%s'; where id = '%s';
""" % (name, descript, id) """ % (name, descript, id)
try: try:
with con: cur.execute(sql)
cur.executescript(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
con.rollback()
return False return False
else: else:
return True return True
@ -100,13 +115,14 @@ def update_group(name, descript, id):
con.close() con.close()
def add_server(hostname, ip, group, typeip, enable): def add_server(hostname, ip, group, typeip, enable):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """INSERT INTO servers (hostname, ip, groups, type_ip, enable) VALUES ('%s', '%s', '%s', '%s', '%s')""" % (hostname, ip, group, typeip, enable) sql = """INSERT INTO servers (hostname, ip, groups, type_ip, enable) VALUES ('%s', '%s', '%s', '%s', '%s')""" % (hostname, ip, group, typeip, enable)
try: try:
with con: cur.execute(sql)
cur.executescript(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
con.rollback()
return False return False
else: else:
return True return True
@ -114,20 +130,21 @@ def add_server(hostname, ip, group, typeip, enable):
con.close() con.close()
def delete_server(id): def delete_server(id):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """delete from servers where id = '%s'""" % (id) sql = """delete from servers where id = '%s'""" % (id)
try: try:
with con: cur.execute(sql)
cur.executescript(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
con.rollback()
else: else:
return True return True
cur.close() cur.close()
con.close() con.close()
def update_server(hostname, ip, group, typeip, enable, id): def update_server(hostname, ip, group, typeip, enable, id):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """update servers set sql = """update servers set
hostname = '%s', hostname = '%s',
ip = '%s', ip = '%s',
@ -136,21 +153,22 @@ def update_server(hostname, ip, group, typeip, enable, id):
enable = '%s' enable = '%s'
where id = '%s'""" % (hostname, ip, group, typeip, enable, id) where id = '%s'""" % (hostname, ip, group, typeip, enable, id)
try: try:
with con: cur.execute(sql)
cur.executescript(sql) con.commit()
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
con.rollback()
cur.close() cur.close()
con.close() con.close()
def select_users(**kwargs): def select_users(**kwargs):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """select * from user ORDER BY id""" sql = """select * from user ORDER BY id"""
if kwargs.get("user") is not None: if kwargs.get("user") is not None:
sql = """select * from user where username='%s' """ % kwargs.get("user") sql = """select * from user where username='%s' """ % kwargs.get("user")
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
else: else:
return cur.fetchall() return cur.fetchall()
@ -158,13 +176,13 @@ def select_users(**kwargs):
con.close() con.close()
def select_groups(**kwargs): def select_groups(**kwargs):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """select * from groups ORDER BY id""" sql = """select * from groups ORDER BY id"""
if kwargs.get("group") is not None: if kwargs.get("group") is not None:
sql = """select * from groups where name='%s' """ % kwargs.get("group") sql = """select * from groups where name='%s' """ % kwargs.get("group")
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
else: else:
return cur.fetchall() return cur.fetchall()
@ -172,11 +190,11 @@ def select_groups(**kwargs):
con.close() con.close()
def select_user_name_group(id): def select_user_name_group(id):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """select name from groups where id='%s' """ % id sql = """select name from groups where id='%s' """ % id
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>') print('<span class="alert alert-danger" id="error">An error occurred: ' + e.args[0] + ' <a title="Close" id="errorMess"><b>X</b></a></span>')
else: else:
return cur.fetchone() return cur.fetchone()
@ -201,7 +219,7 @@ def get_groups_select(id, **kwargs):
print('</select>') print('</select>')
def select_servers(**kwargs): def select_servers(**kwargs):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """select * from servers where enable = '1' ORDER BY groups """ sql = """select * from servers where enable = '1' ORDER BY groups """
if kwargs.get("server") is not None: if kwargs.get("server") is not None:
sql = """select * from servers where hostname='%s' """ % kwargs.get("server") sql = """select * from servers where hostname='%s' """ % kwargs.get("server")
@ -209,7 +227,7 @@ def select_servers(**kwargs):
sql = """select * from servers ORDER BY groups """ sql = """select * from servers ORDER BY groups """
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
else: else:
return cur.fetchall() return cur.fetchall()
@ -217,11 +235,11 @@ def select_servers(**kwargs):
con.close() con.close()
def get_type_ip_checkbox(id, **kwargs): def get_type_ip_checkbox(id, **kwargs):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """select id, type_ip from servers where id='%s' """ % id sql = """select id, type_ip from servers where id='%s' """ % id
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
else: else:
for server in cur.fetchall(): for server in cur.fetchall():
@ -234,11 +252,11 @@ def get_type_ip_checkbox(id, **kwargs):
con.close() con.close()
def get_enable_checkbox(id, **kwargs): def get_enable_checkbox(id, **kwargs):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """select id, enable from servers where id='%s' """ % id sql = """select id, enable from servers where id='%s' """ % id
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
else: else:
for server in cur.fetchall(): for server in cur.fetchall():
@ -255,7 +273,7 @@ def get_dick_permit(**kwargs):
import os import os
cookie = http.cookies.SimpleCookie(os.environ.get("HTTP_COOKIE")) cookie = http.cookies.SimpleCookie(os.environ.get("HTTP_COOKIE"))
login = cookie.get('login') login = cookie.get('login')
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """ select * from user where username = '%s' """ % login.value sql = """ select * from user where username = '%s' """ % login.value
if kwargs.get('virt'): if kwargs.get('virt'):
type_ip = "" type_ip = ""
@ -263,7 +281,7 @@ def get_dick_permit(**kwargs):
type_ip = "and type_ip = 0" type_ip = "and type_ip = 0"
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
else: else:
for group in cur: for group in cur:
@ -273,7 +291,7 @@ def get_dick_permit(**kwargs):
sql = """ select * from servers where groups like '%{group}%' and enable = 1 {type_ip} """.format(group=group[5], type_ip=type_ip) sql = """ select * from servers where groups like '%{group}%' and enable = 1 {type_ip} """.format(group=group[5], type_ip=type_ip)
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
else: else:
return cur.fetchall() return cur.fetchall()
@ -354,13 +372,13 @@ def show_update_group(group):
print('</tr>') print('</tr>')
def select_roles(**kwargs): def select_roles(**kwargs):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """select * from role ORDER BY id""" sql = """select * from role ORDER BY id"""
if kwargs.get("role") is not None: if kwargs.get("role") is not None:
sql = """select * from role where name='%s' """ % kwargs.get("group") sql = """select * from role where name='%s' """ % kwargs.get("group")
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
else: else:
return cur.fetchall() return cur.fetchall()
@ -368,13 +386,13 @@ def select_roles(**kwargs):
con.close() con.close()
def select_roles(**kwargs): def select_roles(**kwargs):
con, cur = get_cur() con, cur = create_db.get_cur()
sql = """select * from role ORDER BY id""" sql = """select * from role ORDER BY id"""
if kwargs.get("roles") is not None: if kwargs.get("roles") is not None:
sql = """select * from role where name='%s' """ % kwargs.get("roles") sql = """select * from role where name='%s' """ % kwargs.get("roles")
try: try:
cur.execute(sql) cur.execute(sql)
except sqlite.Error as e: except sqltool.Error as e:
print("An error occurred:", e.args[0]) print("An error occurred:", e.args[0])
else: else:
return cur.fetchall() return cur.fetchall()

3
requirements.txt
View File

@ -9,4 +9,5 @@ networkx==2.1
numpy==1.14.0 numpy==1.14.0
matplotlib==2.1.2 matplotlib==2.1.2
urllib3==1.22 urllib3==1.22
future==0.13.1 future==0.13.1
mysqlclient==1.3.12