def assets_add(data):
print data
sql = '''INSERT INTO assets(hostname,i_ip,e_ip,ram,hd,cpu,cpu_model,address) values('%s', '%s', '%s', %d, %d, %d, '%s','%s')''' % (
data['hostname'], data['private_ip'], data['public_ip'],
int(data["mem_total"]), int(data["disk"]), int(
data["cpu_num"]), data["cpu_model"], data["machine_room"])
return execute_sql(sql)
def registerUser(data):
#cry_passwd = encryption(data['password'])
#sql = '''INSERT INTO users(username, email, password, role) values('%s', '%s', '%s', '%s')''' % (data['username'], data['email'], cry_passwd, data['role'])
fields, values = fmt_data(data)
sql = '''INSERT INTO users (%s) VALUES (%s);''' % (fields, values)
print sql
return execute_sql(sql)
def updateUserinfo(data, uid):
#sql = '''UPDATE users SET username='******', email='%s', password='******', role='%s' WHERE id=%d; ''' % (data['username'], data['email'], data['password'], data['role'], int(data['uid']) )
#print sql
#return execute_sql(sql)
values = fmt_update_data(data)
sql = '''UPDATE users SET %s WHERE id = %d; ''' % (values, int(uid))
print sql
return execute_sql(sql)
def add_asset(data):
fields, values = [], []
for k, v in data.items():
fields.append(k)
values.append("'%s'" % v)
format_fields = ', '.join(fields)
format_values = ', '.join(values)
sql = '''INSERT INTO assets (%s) VALUES (%s);''' % (format_fields, format_values)
print sql
return execute_sql(sql)
def userEdit(**kwargs):
uid = kwargs['id']
username = kwargs['username']
tmp_password = kwargs['password']
email = kwargs['email']
role = kwargs['role']
db_password = get_pwd(username)
if db_password == tmp_password:
passowrd = tmp_password
sql = ''' UPDATE users SET username="******",password="******",email="%s",role="%s" WHERE id = %s; ''' % (
username, password, email, role, uid)
else:
password = encryption(tmp_password)
sql = ''' UPDATE users SET username="******",password="******",email="%s",role="%s" WHERE id = %s; ''' % (
username, password, email, role, uid)
return execute_sql(sql)
def userDel(uid):
sql = '''DELETE FROM users WHERE id = %s; ''' % uid
return execute_sql(sql)
def register(data):
md5passwd = encryption(data['password'])
sql = '''INSERT INTO users(username, email, password,role) values('%s', '%s', '%s','%s')''' % (
data['username'], data['email'], md5passwd, data['role'])
return execute_sql(sql)
def validate_hostname_unique(hostname):
sql = '''SELECT * FROM assets WHERE hostname='%s'; ''' % hostname
return execute_sql(sql)
def assetDel(pk):
sql = '''DELETE FROM assets WHERE id = %s; ''' % int(pk)
print sql
return execute_sql(sql)
def assets_edit(**kwargs):
sql = ''' UPDATE assets SET hostname="%s",i_ip="%s",e_ip="%s",ram=%d,hd=%d,cpu=%d,cpu_model="%s",address="%s" where id=%d;''' % (
kwargs['hostname'], kwargs['private_ip'], kwargs['public_ip'],
int(kwargs['mem_total']), int(kwargs['disk']), int(kwargs['cpu_num']),
kwargs['cpu_model'], kwargs['machine_room'], int(kwargs['id']))
return execute_sql(sql)
def assets_del(uid):
sql = '''DELETE FROM assets WHERE id = %s; ''' % int(uid)
return execute_sql(sql)
def userUpdate(data):
md5passwd = encrption(data['password'])
sql = '''UPDATE users set username='******',password='******',email='%s',role='%s' WHERE id = %d;''' % (
data['username'], md5passwd, data['email'], data['id'], data['role'])
logging.debug(sql)
return execute_sql(sql)
def userDel(uid):
sql = '''DELETE FROM users WHERE id = %s; ''' % uid
logging.debug(sql)
return execute_sql(sql)
