[VNC] update db_export.py for Task#10544
This commit is contained in:
parent
0a9c8f8245
commit
16d7e80133
|
@ -1,5 +1,14 @@
|
||||||
import psycopg2 as pg2
|
import psycopg2 as pg2
|
||||||
from config import config
|
from config import config
|
||||||
|
import logging
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.setLevel(logging.INFO)
|
||||||
|
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||||||
|
file_handler = logging.FileHandler('/var/log/desktop.log')
|
||||||
|
file_handler.setFormatter(formatter)
|
||||||
|
logger.addHandler(file_handler)
|
||||||
|
|
||||||
|
|
||||||
db_name = config['db']['db_name']
|
db_name = config['db']['db_name']
|
||||||
db_user = config['db']['db_user']
|
db_user = config['db']['db_user']
|
||||||
|
@ -10,15 +19,15 @@ db_port = config['db']['db_port']
|
||||||
def setconn(u_id, vm_num, vm_port,vm_host):
|
def setconn(u_id, vm_num, vm_port,vm_host):
|
||||||
conn = pg2.connect("host = localhost dbname={} user={} password={} port={}".format(db_name,db_user,db_password,db_port))
|
conn = pg2.connect("host = localhost dbname={} user={} password={} port={}".format(db_name,db_user,db_password,db_port))
|
||||||
conn.autocommit = True
|
conn.autocommit = True
|
||||||
cur = conn.cursor()
|
cur = conn.cursor()
|
||||||
cur.execute("SELECT entity_id FROM guacamole_entity WHERE name = '{}'".format(u_id))
|
cur.execute("SELECT entity_id FROM guacamole_entity WHERE name = '{}'".format(u_id))
|
||||||
row = cur.fetchone()
|
row = cur.fetchone()
|
||||||
if row == None:
|
if row == None:
|
||||||
cur.execute("INSERT INTO guacamole_entity (name, type) VALUES ('{}','USER')".format(u_id))
|
cur.execute("INSERT INTO guacamole_entity (name, type) VALUES ('{}','USER')".format(u_id))
|
||||||
cur.execute("SELECT entity_id FROM guacamole_entity WHERE name = '{}'".format(u_id))
|
cur.execute("SELECT entity_id FROM guacamole_entity WHERE name = '{}'".format(u_id))
|
||||||
row = cur.fetchone()
|
row = cur.fetchone()
|
||||||
en_id = row[0]
|
en_id = row[0]
|
||||||
cur.execute("INSERT INTO guacamole_user(entity_id, password_hash, password_date) VALUES ('{}', '\x74657374', now())".format(en_id))
|
cur.execute("INSERT INTO guacamole_user(entity_id, password_hash, password_date) VALUES ('{}', '\x74657374', now())".format(en_id))
|
||||||
print("create user : " , u_id)
|
print("create user : " , u_id)
|
||||||
else:
|
else:
|
||||||
en_id = row[0]
|
en_id = row[0]
|
||||||
|
@ -43,7 +52,9 @@ def setconn(u_id, vm_num, vm_port,vm_host):
|
||||||
#clipboard-encoding
|
#clipboard-encoding
|
||||||
cur.execute("INSERT INTO guacamole_connection_parameter VALUES ('{}','clipboard-encoding','UTF-8')".format(cn_id))
|
cur.execute("INSERT INTO guacamole_connection_parameter VALUES ('{}','clipboard-encoding','UTF-8')".format(cn_id))
|
||||||
print("create connection")
|
print("create connection")
|
||||||
else:
|
log = "create connection : " + cn
|
||||||
|
logging.info(log)
|
||||||
|
else:
|
||||||
cur.execute("SELECT MAX(connection_id) FROM guacamole_connection WHERE connection_name = '{}' AND parent_id IS NULL".format(cn))
|
cur.execute("SELECT MAX(connection_id) FROM guacamole_connection WHERE connection_name = '{}' AND parent_id IS NULL".format(cn))
|
||||||
temp_cn_id = cur.fetchone()
|
temp_cn_id = cur.fetchone()
|
||||||
cn_id = temp_cn_id[0]
|
cn_id = temp_cn_id[0]
|
||||||
|
@ -52,4 +63,30 @@ def setconn(u_id, vm_num, vm_port,vm_host):
|
||||||
#cur.execute("UPDATE guacamole_connection_parameter SET parameter_value='UTF-8' where connection_id='{}' and parameter_name='clipboard-encoding'".format(cn_id))
|
#cur.execute("UPDATE guacamole_connection_parameter SET parameter_value='UTF-8' where connection_id='{}' and parameter_name='clipboard-encoding'".format(cn_id))
|
||||||
print("no connection")
|
print("no connection")
|
||||||
conn.close()
|
conn.close()
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def delconn(u_id, vm_num):
|
||||||
|
conn2 = pg2.connect("host = localhost dbname={} user={} password={} port={}".format(db_name,db_user,db_password,db_port))
|
||||||
|
conn2.autocommit = True
|
||||||
|
cur2 = conn2.cursor()
|
||||||
|
cur2.execute("SELECT entity_id FROM guacamole_entity WHERE name = '{}'".format(u_id))
|
||||||
|
row2 = cur2.fetchone()
|
||||||
|
if row2 == None:
|
||||||
|
print("no user : " , u_id)
|
||||||
|
else:
|
||||||
|
cn2 = "{}{}".format(u_id,vm_num)
|
||||||
|
cur2.execute("SELECT connection_id FROM guacamole_connection WHERE connection_name = '{}'".format(cn2))
|
||||||
|
row2 = cur2.fetchone()
|
||||||
|
if row2 != None:
|
||||||
|
print("cn_id : ", row2[0])
|
||||||
|
#delete connection
|
||||||
|
cur2.execute("SELECT connection_id from guacamole_connection_permission where connection_id = '{}'".format(row2[0]))
|
||||||
|
row2 = cur2.fetchone()
|
||||||
|
if row2 != None:
|
||||||
|
print("delete connection : ",row2[0])
|
||||||
|
cur2.execute("delete from guacamole_connection_permission where connection_id = '{}'".format(row2[0]))
|
||||||
|
log = "delete connection : " + cn2
|
||||||
|
logging.info(log)
|
||||||
|
conn2.close()
|
||||||
|
|
Loading…
Reference in New Issue