"""
學(xué)生管理系統(tǒng)（文件存儲(chǔ)版）
再原有功能的基礎(chǔ)上添加了如下功能：
  1.保存學(xué)生數(shù)據(jù)，保存到與python文件同目錄下 保存的時(shí)候設(shè)置編碼UTF-8 防止亂碼
  2.刪除所有學(xué)生信息
  3.添加了函數(shù)recover_data運(yùn)行程序先獲取數(shù)據(jù) 文件不存在的話會(huì)自行創(chuàng)建并存儲(chǔ)一個(gè)空列表
"""
import os
 
name_list = [] # 存儲(chǔ)學(xué)生信息字典，學(xué)生信息用字典存，再用列表存儲(chǔ)字典
 
 
# 菜單
def display_menu():
  print("-" * 30)
  print("  學(xué)生管理系統(tǒng) v8.8 ")
  print("1.添加學(xué)生信息")
  print("2.刪除學(xué)生信息")
  print("3.修改學(xué)生信息")
  print("4.查詢單個(gè)學(xué)生信息")
  print("5.查詢所有學(xué)生信息")
  print("6.保存學(xué)生信息")
  print("7.刪除所有學(xué)生")
  print("0.退出系統(tǒng)")
  print("-" * 30)
 
 
# 選擇序號的獲得
def get_choice():
  selected_key = input("請輸入選擇的序號：")
  return selected_key
 
 
# 檢查性別是否合法
def check_sex(new_sex):
  flag = True
  while flag:
    if new_sex == '男' or new_sex == '女':
      flag = False
    else:
      new_sex = input("輸入性別有誤，請重新輸入(男/女)：")
  return new_sex
 
 
# 檢查電話號碼是否合法
def check_phone(new_phone):
  flag = True
  while flag:
    if new_phone.isdigit():
      flag = False
    else:
      new_phone = input("您輸入的電話號碼有誤，請重新輸入：")
  return new_phone
 
 
# 檢查學(xué)號是否重復(fù)或者有誤
def check_id(new_id):
  flag = True
  while flag:
    # 先檢查是不是純數(shù)字再去考慮是否重復(fù)的事情，如果不是純數(shù)字直接pass
    if new_id.isdigit():
      for i in range(len(name_list)):
        if name_list[i]['id'] == new_id:
          new_id = check_id(input("您輸入的學(xué)號重復(fù)，請重新輸入："))
      flag = False
    else:
      new_id = input("您輸入的學(xué)號有誤，請重新輸入：")
  return new_id
 
 
# 添加學(xué)生信息
def add_name():
  new_info = {}
  new_id = check_id(input("請輸入學(xué)號："))
  new_info['id'] = new_id
  new_name = input("請輸入姓名：")
  new_info['name'] = new_name
  new_sex = check_sex(input("請輸入性別(男/女)："))
  new_info['sex'] = new_sex
  new_phone = check_phone(input("請輸入電話號碼："))
  new_info['phone'] = new_phone
  name_list.append(new_info)
  print("添加成功！")
 
 
# 查詢所有學(xué)生信息
def find_all():
  print("=" * 30)
  for name in name_list:
    print(name['id'], name['name'], name['sex'], name['phone'])
  print("=" * 30)
 
 
# 刪除學(xué)生信息
def del_name():
  del_id_is = input("請輸入要?jiǎng)h除的學(xué)生學(xué)號：")
  flag = False
  index = 0
  for i in range(len(name_list)):
    if name_list[i]['id'] == del_id_is:
      flag = True
      index = i
      break
  if flag:
    name_list.pop(index)
    print("刪除成功！")
  else:
    print("學(xué)生未找到！請檢查學(xué)號輸入是否有誤！")
 
 
# 刪除所有學(xué)生
def del_all():
  global name_list
  flag = input("確定要?jiǎng)h除嗎?(Y刪除/輸入其他字符返回菜單):")
  if flag == 'Y' or flag == 'y':
    name_list = []
    print("刪除成功！")
  else:
    return
 
 
# 名字修改細(xì)節(jié)函數(shù)
def choice_of_name(index):
  while True:
    choice = input("請輸入要修改學(xué)生的(1.id 2.姓名 3.性別 4.電話號碼，5.全部修改)：")
    if choice == '5':
      new_id = input("請輸入新的學(xué)號：")
      name_list[index]['id'] = new_id
      new_name = input("請輸入新的姓名：")
      name_list[index]['name'] = new_name
      new_sex = check_sex(input("請輸入性別(男/女)："))
      name_list[index]['sex'] = new_sex
      new_phone = check_phone(input("請輸入電話號碼："))
      name_list[index]['phone'] = new_phone
      break
    elif choice == '1':
      new_id = input("請輸入新的學(xué)號：")
      name_list[index]['id'] = new_id
      break
    elif choice == '2':
      new_name = input("請輸入新的姓名：")
      name_list[index]['name'] = new_name
      break
    elif choice == '3':
      new_sex = check_sex(input("請輸入性別(男/女)："))
      name_list[index]['sex'] = new_sex
      break
    elif choice == '4':
      new_phone = check_phone(input("請輸入電話號碼："))
      name_list[index]['phone'] = new_phone
      break
    else:
      print("輸入有誤，請重新輸入！")
 
 
# 修改學(xué)生信息
def re_name():
  id_is = input("請輸入要修改的學(xué)生學(xué)號：")
  flag = False
  index = 0
  # 先找到要修改的學(xué)生的下標(biāo)
  for i in range(len(name_list)):
    if name_list[i]['id'] == id_is:
      flag = True
      index = i
      break
  if flag:
    choice_of_name(index)
    print("修改成功！")
  else:
    print("修改失敗，學(xué)生信息未找到！")
 
 
# 查詢單個(gè)學(xué)生信息
def find_name():
  find_id_is = input("請輸入要查詢的學(xué)生學(xué)號：")
  flag = False
  index = 0
  for i in range(len(name_list)):
    if name_list[i]['id'] == find_id_is:
      flag = True
      index = i
      break
  if flag:
    print("學(xué)生查詢到，學(xué)生信息為：")
    print(name_list[index]['id'], name_list[index]['name'], name_list[index]['sex'], name_list[index]['phone'])
  else:
    print("學(xué)生未找到！")
 
 
# 保存學(xué)生信息
def save_stu():
  students = str(name_list)
  with open("students.txt", "w", encoding="utf-8") as f:
    f.write(students)
  print("保存成功！文件位置在"+os.getcwd())
 
 
# 恢復(fù)數(shù)據(jù)
def recover_data():
  global name_list
  try:
    with open("students.txt", "r", encoding="utf-8") as f:
      content = f.read()
      if content != '':
        name_list = eval(content)
  except:
    f = open("students.txt", "w")
    f.write("[]")
 
 
def main():
  recover_data()
  exit_name = True
  while exit_name:
    display_menu()
    key = get_choice()
    if key == '1':
      add_name()
    elif key == '2':
      del_name()
    elif key == '3':
      re_name()
    elif key == '4':
      find_name()
    elif key == '5':
      find_all()
    elif key == '6':
      save_stu()
    elif key == '7':
      del_all()
    elif key == '0':
      exit_name = False
    else:
      print("請輸入正確的數(shù)值！")
 
 
main()