79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
import pymysql
|
|
|
|
# 数据库定义参数
|
|
database_ip = "localhost"
|
|
database_user = "guest"
|
|
database_passwd = "guest1234"
|
|
database_name = "GUESTDB"
|
|
table_name = "user_info"
|
|
|
|
DEBUG = False
|
|
|
|
# 打开数据库连接
|
|
db = pymysql.connect(host=database_ip, user=database_user, password=database_passwd, database=database_name )
|
|
print('open mysql success !!!')
|
|
# 使用cursor()方法获取操作游标
|
|
cursor = db.cursor()
|
|
|
|
def DB_Check_byName(user_name):
|
|
sql = "SELECT * FROM %s WHERE user_name='%s' " % (table_name,user_name)
|
|
try:
|
|
# 执行SQL语句
|
|
cursor.execute(sql)
|
|
# 获取所有记录列表
|
|
results = cursor.fetchall()
|
|
passwd = results[0][2]
|
|
return passwd
|
|
except:
|
|
if DEBUG:
|
|
print("There is no the user's info !")
|
|
return False
|
|
|
|
|
|
def DB_Check_byEmail(email):
|
|
sql = "SELECT * FROM %s WHERE user_email='%s' " % (table_name,email)
|
|
try:
|
|
# 执行SQL语句
|
|
cursor.execute(sql)
|
|
# 获取所有记录列表
|
|
results = cursor.fetchall()
|
|
passwd = results[0][2]
|
|
return passwd
|
|
except:
|
|
if DEBUG:
|
|
print("There is no the user's info !")
|
|
return False
|
|
|
|
|
|
|
|
def DB_Insert_User(user_info):
|
|
user_name = user_info["username"]
|
|
user_passwd = user_info["password"]
|
|
user_email = user_info["email"]
|
|
user_date = user_info["date"]
|
|
sql = "INSERT INTO user_info(user_name,user_passwd,user_email,user_register_date) VALUES('%s', '%s', '%s', '%s' )"% (user_name, user_passwd, user_email, user_date)
|
|
try:
|
|
# 执行sql语句
|
|
cursor.execute(sql)
|
|
# 执行sql语句
|
|
db.commit()
|
|
if DEBUG:
|
|
print("Insert user_info successfully!")
|
|
return True
|
|
except:
|
|
# 发生错误时回滚
|
|
db.rollback()
|
|
if DEBUG:
|
|
print("Insert user_info error!")
|
|
return False
|
|
|
|
|
|
# if __name__ == "__main__":
|
|
# res = authenticate('user5','1234')
|
|
# print("authenticate result: ",res)
|
|
# for i in range(10):
|
|
# s = str(i)
|
|
# res = add_user('user'+s,'1234','user'+s+'@qq.com')
|
|
# print("add user", s ," res: ",res)
|
|
# res = authenticate('user5','1234')
|
|
# print("authenticate result: ",res) |