学籍注册程序.py

jb_filename = "student_db.txt"


# 1.用户输入注册信息
def register_api():
    student_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 = ["python", "linux", "网络安全", "web前端", "数据分析&&机器学习"]
    for index, course in enumerate(course_list):
        print(f"{index}.{course}")
    select_list = input("想选择的学科：")
    if select_list.isdigit():
        select_list = int(select_list)
        if select_list >= 0 and select_list < len(course_list):
            picked_course = course_list[select_list]
        else:
            exit("不合法的输出")
    else:
        exit("非法输入")

    student_data["name"] = name
    student_data["age"] = age
    student_data["phone"] = phone
    student_data["id_num"] = id_num
    student_data["course"] = picked_course
    return student_data


def commit_to_db(filename, student_data):
    f = open(filename, "a")
    row = f"{student_data['name']},{student_data['age']},{student_data['phone']},{student_data['id_num']},{student_data['course']}"
    f.write(row)
    f.close()


def load_validated(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)

    return phone_list, id_num_list


# 2.判断手机号、身份证号唯一性
phone_list, id_num_list = load_validated(jb_filename)
stu_data = register_api()
# 3.写入文件
commit_to_db(jb_filename, stu_data)