def execute_sql(sql):
# 连接数据库
conn, cur = mysql_util.conn_db()
# 根据sql更新用户
result = mysql_util.operate(conn, cur, sql)
# 关闭数据库连接
mysql_util.close_db(conn, cur)
return result
def delete_user(self):
# 连接数据库
conn, cur = mysql_util.conn_db()
# 从数据库删除
sql = "DELETE FROM t_user WHERE id = {_id}".format(_id=self.id)
result = mysql_util.operate(conn, cur, sql)
# 关闭数据库连接
mysql_util.close_db(conn, cur)
return result
def save_user(self):
# 连接数据库
conn, cur = mysql_util.conn_db()
# 保存至数据库
sql = "INSERT INTO t_user (name, age, pass) VALUES ('{_name}','{_age}','{_password}')".format(_name=self.name,
_age=self.age,
_password=md5(
self.password))
result = mysql_util.operate(conn, cur, sql)
# 关闭数据库连接
mysql_util.close_db(conn, cur)
return result
def modify_user(self):
# 连接数据库
conn, cur = mysql_util.conn_db()
# 修改用户属性
sql = "UPDATE t_user SET name = '{_name}', age = '{_age}', password = '******' WHERE id = {_id}".format(
_name=str(self.name),
_age=int(self.age),
_id=int(self.id),
_password=str(md5(self.password)))
result = mysql_util.operate(conn, cur, sql)
# 关闭数据库连接
mysql_util.close_db(conn, cur)
return result
def batch_insert(users):
# 连接数据库
conn, cur = mysql_util.conn_db()
# 批量添加用户
sql = "INSERT INTO t_user (name, age, password) VALUES "
for i in range(len(users)):
u_name = users[i].name
u_age = users[i].age
u_password = md5(users[i].password)
sql += "('{}','{}','{}')".format(u_name, u_age, u_password) if i == len(
users) - 1 else "('{}','{}','{}'),".format(u_name, u_age, u_password)
result = mysql_util.operate(conn, cur, sql)
# 关闭数据库连接
mysql_util.close_db(conn, cur)
return result
