Python Day 03 Ps:今天的是真滴难！

# list 函数 保存本地

import os
def add_student():
    name = input('请输入学员姓名:')
    age = input('请输入学员年龄:')
    phone = input('请输入学员电话:')
    student = [name,age,phone]
    student_list.append(student)
    print('-----添加学员成功-----')

# 查询学员函数
def query_student():
    print('--------------学员信息列表---------------')
    for x in range(0,len(student_list)):
        student = student_list[x]
        name = student[0]
        age = student[1]
        phone = student[2]
        print('序号:%s 姓名:%s 年龄:%s 电话:%s'%(x,name,age,phone))

#


# 修改学员函数
def update_student():
# 判断是否有学员信息，如果没有，直接结束函数执行
    if len(student_list) == 0:
        print('**************未找到学员信息**************')
        return
    query_student()
    num = input('请选择要修改的学员序号:')
    num = int(num)
    while num not in range(0, len(student_list)):
        num = input('您选择的学员序号不存在,请重新选择:')
        num = int(num)
    student = student_list[num]
    new_name = input('请输入修改后的姓名(%s):'%student[0])
    new_age = input('请输入修改后的年龄(%s):'%student[1])
    new_phone = input('请输入修改后的电话(%s):'%student[2])
    student[0] = new_name
    student[1] = new_age
    student[2] = new_phone
    print('--------------修改数据完成--------------')

# 删除学员函数
def delete_student():
    if len(student_list) == 0:
        print('**************未找到学员信息**************')
        return
    print('1.根据学员序号删除')
    print('2.删除所有学员')
    print('3.根据学员姓名删除')
    num1 = input('请选择删除方式:')
    num1 = int(num1)
    while num1 not in range(1,4):
        num1 = input('您选择的删除方式不存在,请重新选择:')
        num1 = int(num1)
    if num1 == 1:
        query_student()
        num = input('请选择要删除的学员序号:')
        num = int(num)
        while num not in range(0,len(student_list)):
            num = input('您选择的序号不存在,请重选:')
            num = int(num)
        del student_list[num]
        print('删除成功！')
    else:
        rs = input('确认删除所有信息？Y(确定)/N(取消):')
        if rs =='Y':
            student_list.clear()
            print('删除成功！')
        else:
            print('操作已取消！')
def save():
    file_handle = open('student_2.0.txt', mode='w')
    # split() 通过某个字符分割字符串,返回的是分割完成后的列表
    for student in student_list:
        s = ' '.join(student)
        file_handle.write(s)
        file_handle.write('\n')
    file_handle.close()

def read():
    rs = os.path.exists('student_2.0.txt')
    if rs == True:
        file_handle = open('student_2.0.txt',mode='r')
        contents = file_handle.readlines()
        student = []
        for msg in contents:
            msg = msg.strip('\n')
            student = msg.split(' ')
            student_list.append(student)
        file_handle.close()
        print(student)

student_list = []
read()
while True:
    print('----------欢迎使用学员管理系统！-----------')
    print('1.添加学员')       
    print('2.查询学员')
    print('3.修改学员')
    print('4.删除学员')
    print('0.退出程序')
    print('-----------------------------------------')
    num = input('请选择您的操作：')
    num = int(num)
    while num not in range(0,5):
        num = input('没有该选项，请重选：')
        num = int(num)
    if num == 1:
        save()
        add_student()
    elif num == 2:
        query_student()
    elif num == 3:
        save()
        update_student()
    elif num == 4:
        save()
        delete_student()
    else:
        print('程序结束!')
        break