一、基础功能需求
版本一：
版本二
二、功能代码实现
1.结构设计
2.代码实现
# from exercise.common import check_username, check_password, check_phone
from exercise.common import * # 可以一口气全部引用，但是不推荐
# username = ''
# password = ''
# phone = ''
def input_username():
# global username
username = input("请输入用户名：")
if check_username(username):
print("用户名正确.")
return username
else:
print("用户名错误.")
input_username() # 递归调用函数自身，完成循环的功能
def input_password():
# global password
password = input("请输入密码：")
if check_password(password):
print("密码正确.")
return password
else:
print("密码错误.")
input_password()
def input_phone():
# global phone
phone = input("请输入手机号：")
if check_phone(phone):
print("手机号正确.")
return phone
else:
print("手机号错误.")
input_phone()
def do_reg():
username = input_username()
password = input_password()
phone = input_phone()
user_list = []
user = {'username': username, 'password': password, 'phone': phone}
user_list.append(user)
print(user_list)
if __name__ == '__main__':
do_reg()三、Python文件读写操作
1.common.py
import re, pymysql
from pymysql.cursors import DictCursor
'''
长度的判断5-12位，使用len函数进行判断
不能以数字开头：username[0]不能是0-9的范围
如何判断只能是大小写或数字：
返回值：True: 用户名正确， False: 用户名错误
'''
def check_username(username):
if len(username) < 5 or len(username) > 12:
return False
if username[0] >= '0' and username[0] <= '9':
# if not (username[0] < '0' or username[0] > '9'):
return False
for char in username:
if not (ord(char) in range(65, 91) or ord(char) in range(97, 123) or ord(char) in range (48, 58)):
return False
return True
'''
检查密码：由大写、小写和数字构成，且必须是6~15位
只要确保密码中至少有1位是大写，至少有1位是小写，至少有1位是数字
'''
def check_password(password):
if len(password) < 6 or len(password) > 12:
return False
lower = 0
upper = 0
digit = 0
other = 0
for char in password:
if char >= 'A' and char <= 'Z':
upper += 1
elif char >= 'a' and char <= 'z':
lower += 1
elif char >= '0' and char <= '9':
digit += 1
else:
other += 1
if upper < 1 or lower < 1 or digit < 1 or other > 0:
return False
return True
# 检查电话号码
def check_phone(phone):
pattern = "^1[3-9]\d{9}$"
if re.match(pattern, phone):
return True
else:
return False
# 读取CSV，并动态将第一行处理为字典的Key，返回[{},{}]的格式
def read_csv(csvfile, has_column=True):
with open(csvfile) as f:
line_list = f.readlines()
if not has_column:
raise Exception('CSV文件必须要使用第一行作为列名')
# return None
key_list = line_list[0].strip().split(',')
list = []
for i in range(1, len(line_list)):
temp_list = line_list[i].strip().split(',')
dict = {}
for j in range(len(temp_list)):
dict[key_list[j]] = temp_list[j]
list.append(dict)
return list
# 读取csv文件，并检查用户名是否存在
def check_user_exists(username):
user_list = read_csv('./userpass.csv')
for user in user_list:
if user['username'] == username:
return True
return False
# 根据用户名取得CSV文件中对应的一行用户数据，以字典的形式返回
def check_get_user(username):
user_list = read_csv('./userpass.csv')
for user in user_list:
if user['username'] == username:
return user
return None
# 修改用户的密码，修改一个CSV文件中的某一行中的某一列，不能直接修改（Python中不存在文件内容部分修改的操作）
# 将CSV整体读入到内存中，形成列表+字典，然后修改字典的某一项，再整体写入到CSV（覆盖写入）
def change_password(username, newpass):
csv_list = read_csv('./userpass.csv')
for user in csv_list:
if user['username'] == username:
index = csv_list.index(user)
break
csv_list[index]['password'] = newpass
# 将列表+字典还原成行+逗号分隔数据
with open('./userpass.csv', mode='w') as f:
f.write("username,password,phone\n")
for user in csv_list:
line = f"{user['username']},{user['password']},{user['phone']}\n"
# print(line)
# line = line.replace("'","")
f.write(line)
# 针对数据库连接进行封装操作
def query_mysql(sql):
conn = pymysql.connect(host='localhost', user='root', password='', database='learn', charset='utf8')
cursor = conn.cursor(DictCursor)
cursor.execute(sql)
result =cursor.fetchall()
conn.close()
return result
def update_mysql(sql):
conn = pymysql.connect(host='localhost', user='root', password='', database='learn', charset='utf8')
cursor = conn.cursor(DictCursor)
cursor.execute(sql)
conn.commit()
conn.close()
# 利用异常处理改造查询函数
def query_mysql_2(sql):
try:
conn = pymysql.connect(host='localhost', user='root', password='123456', database='learn', charset='utf8')
cursor = conn.cursor(DictCursor)
cursor.execute(sql)
result =cursor.fetchall()
return result
except:
# print("数据库处理出错.")
raise Exception("数据库处理出错.") # 主动抛出异常，让程序停止
# pass
finally:
conn.close()
# 最基本最简单的单元测试代码：半自动化测试代码
print(check_username('qiang'))
print(check_username('132457'))
print(check_username('Q132457'))
# 全自动的单元测试代码，编写一个测试驱动程序
def test_driver(func, expect, *args):
actual = func(*args)
if actual == expect:
print("测试 %s: 成功" % func.__name__)
else:
print("测试 %s: 失败" % func.__name__)
if __name__ == '__main__':
pass
#测试用户名的规则
test_driver(check_username, False, 'hi')
test_driver(check_username, False, 'welcomeToWoniu123')
test_driver(check_username, True, 'qiang')
test_driver(check_username, False, '132457')
test_driver(check_username, True, 'Q132457abc')
test_driver(check_username, False, 'Q13#@%$bc')
# 测试密码的规则，TDD：Test-Driven Development
test_driver(check_password, False, 'qiang')
test_driver(check_password, False, 'welcomeToWoniu1231235')
test_driver(check_password, False, 'WELCOME')
test_driver(check_password, False, '12349234')
test_driver(check_password, False, 'jojojoju1')
test_driver(check_password, False, 'WILL12432')
test_driver(check_password, True, 'Woniu123')
test_driver(check_password, False, 'Woniu@#123')
test_driver(check_phone, False, '1881234567')
test_driver(check_phone, False, '188123456799')
test_driver(check_phone, False, '988123456799')
test_driver(check_phone, False, '128123456799')
test_driver(check_phone, True, '13812345679')2.verson1.py
# from exercise.common import check_username, check_password, check_phone
from exercise.common import * # 可以一口气全部引用，但是不推荐
# username = ''
# password = ''
# phone = ''
def input_username():
# global username
username = input("请输入用户名：")
if check_username(username):
print("用户名正确.")
return username
else:
print("用户名错误.")
input_username() # 递归调用函数自身，完成循环的功能
def input_password():
# global password
password = input("请输入密码：")
if check_password(password):
print("密码正确.")
return password
else:
print("密码错误.")
input_password()
def input_phone():
# global phone
phone = input("请输入手机号：")
if check_phone(phone):
print("手机号正确.")
return phone
else:
print("手机号错误.")
input_phone()
def do_reg():
username = input_username()
password = input_password()
phone = input_phone()
user_list = []
user = {'username': username, 'password': password, 'phone': phone}
user_list.append(user)
print(user_list)
if __name__ == '__main__':
do_reg()3.verson2.py
from exercise.common import * # 可以一口气全部引用，但是不推荐
def input_username():
username = input("请输入用户名：")
if check_username(username):
if check_user_exists(username):
print("用户名已存在.")
return input_username()
else:
print("用户名正确.")
return username
else:
print("用户名错误.")
return input_username() # 递归调用函数自身，完成循环的功能
def input_password():
password = input("请输入密码：")
if check_password(password):
print("密码正确.")
return password
else:
print("密码错误.")
return input_password()
def input_phone():
phone = input("请输入手机号：")
if check_phone(phone):
print("手机号正确.")
return phone
else:
print("手机号错误.")
return input_phone()
def do_reg():
username = input_username()
password = input_password()
phone = input_phone()
with open('./userpass.csv', mode='a') as f:
f.write(f"\n{username},{password},{phone}")
print("恭喜你，注册成功.")
draw_menu()
def do_login():
username = input("请输入用户名：")
password = input("请输入密码：")
user = check_get_user(username)
if user is None:
print("用户名不存在.")
exit(0)
elif user['password'] == password:
print("用户名密码正确，登录成功.")
else:
print("登录失败.")
draw_menu()
def do_change():
username = input("请输入用户名：")
password = input("请输入旧密码：")
user = check_get_user(username)
if user is None:
print("用户名不存在.")
draw_menu()
elif user['password'] == password:
newpass = input("请输入新密码：")
change_password(username, newpass)
else:
print("旧密码验证不通过.")
def draw_menu():
print("========= 欢迎使用用户管理系统 ===========")
print("1、注册 2、登录 3、修改密码 4、退出")
option = input("请选择菜单项：[1 2 3 4]：")
if option == '1':
do_reg()
elif option == '2':
do_login()
elif option == '3':
do_change()
elif option == '4':
exit(0)
else:
print("请输入正确的菜单编号.")
draw_menu()
if __name__ == '__main__':
# do_reg()
# do_login()
do_change()
# draw_menu()4.verson3.py
from exercise.common import *
def do_reg():
username = input("请输入用户名：")
result = query_mysql(f"select username from user where username='{username}'")
if len(result) == 0:
password = input("请输入密码：")
phone = input("请输入手机号码：")
update_mysql(f"insert into user(username, password, role, createtime) values('{username}', '{password}', 'user', now())")
else:
print("用户名已经存在，不允许注册.")
if __name__ == '__main__':
do_reg()本文章为转载内容，我们尊重原作者对文章享有的著作权。如有内容错误或侵权问题，欢迎原作者联系我们进行内容更正或删除文章。
