def register_zhuce():
stu_data = {}
print("欢迎".center(50, "-"))
print("请注册")
name = input("姓名").strip()
age = input("年龄").strip()
phone = input("手机").strip()
if phone in phone_list:
exit("手机号已注册")
id_num = input("身份号").strip()
if id_num in id_num_list:
exit("身份证已存在")
course_list = ["JAVA", "PYTHON", "HTML", "PHP"]
for index, course in enumerate(course_list):
print(f"{index}.{course}")
select_course = input("选择什么课:")
if select_course.isdigit():
select_course = int(select_course)
if select_course >= 0 and select_course <= len(course_list):
picked_course = course_list[select_course]
else:
exit("不合法")
else:
exit("非法输入")
stu_data["name"] = name
stu_data["age"] = age
stu_data["phone"] = phone
stu_data["id_num"] = id_num
stu_data["course"] = picked_course
return stu_data
db_file = "student_data.db"
def register_zhuce(): stu_data = {} print("欢迎".center(50, "-")) print("请注册") name = input("姓名").strip() age = input("年龄").strip() phone = input("手机").strip() if phone in phone_list: exit("手机号已注册") id_num = input("身份号").strip() if id_num in id_num_list: exit("身份证已存在")
def commit_to_db(filename, stu_data): f = open(filename, "a") row = f"{stu_data['name']},{stu_data['age']},{stu_data['phone']},{stu_data['id_num']},{stu_data['course']}\n" f.write(row) f.close()
def load_validated_data(filename): f=open(filename) phone_list = [] id_num_list = [] for line in f: line = line.split(",") phone = line[2] id_num = line[3] phone_list.append(phone) id_num_list.append(id_num)
phone_list,id_num_list = load_validated_data(db_file)
student_data = register_zhuce() print(student_data) commit_to_db(db_file, student_data)