西北乱跑娃 --- python列表写的学生管理信息系统

注意：个人成果，转发请标记出处！

# 列表学生信息管理系统
import os

def get_data():
    with open('data.txt', 'r', encoding='utf8') as f:
        return f.readlines()

def input_data():
    name = input("请输入姓名：")
    sex = input("请输入性别：")
    age = input("请输入年龄：")
    return [name, sex, age]

def manege_file(cur,data=None):
    if cur == 0:
        with open('data.txt', 'w', encoding='utf8') as f1:
            pass
    else:
        with open('data.txt', 'w', encoding='utf8') as f2:
            f2.write(data)

def add_change():
    all_stu = get_data()
    stu_data = input_data()
    if stu_data[0] in [i.split(',')[0] for i in all_stu]:
        new_data = []
        for i in all_stu:
            if i.split(',')[0] == stu_data[0]:
                i = f'{stu_data[0]},{stu_data[1]},{stu_data[2]}\n'
            else:
                pass
            new_data.append(i)
        data = ''.join(new_data)
        manege_file(1,data)
    else:
        all_stu.append(f'{stu_data[0]},{stu_data[1]},{stu_data[2]}\n')
        data = ''.join(all_stu)
        manege_file(1,data)

def delete():
    all_stu = get_data()
    name = input("请输入删除姓名：")
    data = ''
    for i in all_stu:
        if i.split(',')[0] != name:
            data += i
        else:
            print(i)
    manege_file(1,data)

def index():
    all_stu = get_data()
    data = ''.join(all_stu)
    print("姓名   性别  年龄")
    print(data)

if __name__ == "__main__":
    if os.path.exists('data.txt'):
        pass
    else:
        manege_file(0)
    while True:
        print('-------------------------')
        print("1、查看\n2、删除\n3、添加修改")
        choose = int(input('请选择：'))
        if choose == 1:
            index()
        elif choose == 2:
            delete()
        else:
            add_change()