创建MysqlHelper.py文件,定义类
import pymysql
import hashlib
class MysqlHelper():
def __init__(self, host, database, user, password, port=3306, charset='utf8'):
self.host = host
self.database = database
self.user = user
self.password = password
self.port = port
self.charset = charset
def connect(self):
self.conn = pymysql.connect(host=self.host, database=self.database, user=self.user, password=self.password,
port=self.port, charset=self.charset)
self.cursor = self.conn.cursor()
def close(self):
self.cursor.close()
self.conn.close()
def select_one(self, sql, params=[]):
result = None
try:
self.connect()
self.cursor.execute(sql, params)
result = self.cursor.fetchone()
self.close()
except Exception as ex:
print(ex)
return result
def select_all(self, sql, params):
result = ()
try:
self.connect()
self.cursor.execute(sql, params)
result = self.cursor.fetchall()
self.close()
except Exception as ex:
print(ex)
return result
def __edit(self, sql, params):
count = 0
try:
self.connect()
count = self.cursor.execute(sql, params)
self.conn.commit()
self.close()
except Exception as ex:
print(ex)
return count
def insert(self, sql, params=[]):
return self.__edit(sql, params)
def update(self, sql, params=[]):
return self.__edit(sql, params)
def delete(self, sql, params=[]):
return self.__edit(sql, params)
def my_md5(self, pwd):
my_md5 = hashlib.md5()
my_md5.update(pwd.encode('utf-8'))
return my_md5.hexdigest()
创建testLogin.py文件,引入MysqlHelper模块
from MysqlHelper import MysqlHelper
def register():
name = input('请输入用户名:')
pwd = input('请输入密码:')
helper = MysqlHelper(host='localhost', database='bank', user='root', password='55555')
ret = helper.insert('insert into user(name,pwd) values(%s,%s)', [name, helper.my_md5(pwd)])
if ret > 0:
print('注册成功!')
else:
print('注册失败,请重新注册!')
def login():
name = input('请输入用户名:')
pwd = input('请输入密码:')
helper = MysqlHelper(host='localhost', database='bank', user='root', password='55555')
ret = helper.select_one('select count(*) from user where name=%s and pwd=%s', [name, helper.my_md5(pwd)])
if ret[0] > 0:
print('登录成功!')
else:
print('登录失败,请重新登录!')
def main():
while True:
choice = input('1、注册 2、登录\n请输入要进行的操作:')
if choice == '1':
register()
elif choice == '2':
login()
else:
print('请等待版本更新!')
if __name__ == '__main__':
main()
运行结果: