def add_user():
name = request.form.get('name', default=None)
password = request.form.get('password', default=None)
email = request.form.get('email', default=None)
sex = request.form.get('sex', default=None)
db = DBSession()
u = db.query(User).filter(User.name == name).first()
if u is not None:
db.close()
error_message = '用户名' + name + '已存在,请重新输入'
res = {
'code': 100,
'message': error_message,
'data': {
'email': email,
'name': name
}
}
return render_template('user/register.html', result=res)
else:
user = User(id=get_uuid(),
name=name,
password=password,
email=email,
sex=sex,
create_time=datetime.now())
db.add(user)
db.commit()
db.close()
res = {'code': 200, 'message': 'success'}
return render_template('login.html', result=res)
def read_config(config_file):
session = DBSession()
tree = ElementTree()
tree.parse(config_file)
root = tree.getroot()
server = root.attrib['name']
server_vm = root.attrib['virtual_machine']
protocals = root.getchildren()
acnt = [0, 0, 0, 0]
cur_datetime = datetime.now()
t = []
for protocal in protocals:
pro_name = protocal.attrib['name']
clients = protocal.getchildren()
cnt = 0
for client in clients:
cnt += 1
client_name = client.attrib['name']
client_host = client[0].text
client_vm = client[1].text
guacamoleClientInfo = GuacamoleClientInfo('', '', server,
client_name, pro_name,
client_host, client_vm,
0, cur_datetime)
#t.append(guacamoleClientInfo)
session.add(guacamoleClientInfo)
#session.commit()
if pro_name == 'vnc':
acnt[0] = cnt
elif pro_name == 'vnc-read-only':
acnt[1] = cnt
elif pro_name == 'ssh':
acnt[2] = cnt
else:
acnt[3] = cnt
guacamoleServerLoad = GuacamoleServerLoad(server, server_vm, acnt[0],
acnt[1], acnt[2], acnt[3],
sum(acnt), cur_datetime, 0)
#guacamoleServerLoad.guacamole_client_info = t
session.add(guacamoleServerLoad)
session.commit()
session.close()
def read_config(config_file):
session = DBSession()
tree = ElementTree()
tree.parse(config_file)
root = tree.getroot()
server = root.attrib["name"]
server_vm = root.attrib["virtual_machine"]
protocals = root.getchildren()
acnt = [0, 0, 0, 0]
cur_datetime = datetime.now()
t = []
for protocal in protocals:
pro_name = protocal.attrib["name"]
clients = protocal.getchildren()
cnt = 0
for client in clients:
cnt += 1
client_name = client.attrib["name"]
client_host = client[0].text
client_vm = client[1].text
guacamoleClientInfo = GuacamoleClientInfo(
"", "", server, client_name, pro_name, client_host, client_vm, 0, cur_datetime
)
# t.append(guacamoleClientInfo)
session.add(guacamoleClientInfo)
# session.commit()
if pro_name == "vnc":
acnt[0] = cnt
elif pro_name == "vnc-read-only":
acnt[1] = cnt
elif pro_name == "ssh":
acnt[2] = cnt
else:
acnt[3] = cnt
guacamoleServerLoad = GuacamoleServerLoad(
server, server_vm, acnt[0], acnt[1], acnt[2], acnt[3], sum(acnt), cur_datetime, 0
)
# guacamoleServerLoad.guacamole_client_info = t
session.add(guacamoleServerLoad)
session.commit()
session.close()
