#!/usr/bin/env python
# coding=utf8
import os, re
#这里我把查询这块分为3个函数了,纠结了很久是放一起还是分开，最后还是分开了，容易写一些
def search_age_above_22():   #定义年纪大于22岁函数
    new = []  # 把读取到的文件写入至此列表
    temp = []  # 记录查询年龄大于22岁的总数列表

    with open("Information.txt", "r", encoding="utf8") as file:
        for line in file.readlines():
            new.append(line.strip().split(","))
        for each_line in new:
            if each_line[2] > "22":
                temp.append(each_line[2])
                print(each_line[1:3])
        print("查到年龄大于22岁的总数为:%s" % len(temp))

def search_depot_of_IT():  #定义部门为IT的函数
    new = []  # 把读取到的文件写入至此列表
    temp = []  # 记录查询部门为IT的总数列表
    with open("Information.txt", "r", encoding="utf8") as file:
        for line in file.readlines():
            new.append(line.strip().split(","))
        for each_line in new:
            if each_line[4] == "IT":
                temp.append(each_line[4])
                print(str(each_line[::]))
        print("查到部门为IT的总数为:%s" % len(temp))

def search_date_of_2013():  #定义开始工作日期为2013年的函数
    new = []  # 把读取到的文件写入至此列表
    temp = []  # 记录查询开始工作日期为2013年的总数列表
    with open("Information.txt", "r", encoding="utf8") as file:
        for line in file.readlines():
            new.append(line.strip().split(","))
    for each_line in new:
        if re.search("2013", each_line[-1]):
            temp.append(each_line[5])
            print(each_line[::])
    print("查到工作时间在2013年名单总数为:%s" % len(temp))

def add():   #定义增加函数
    new = []  # 把读取到的文件写入至此列表
    temp = [] #此列表为了自增而建，代码写的很low
    user_input = str(input("请输入类似于[吴东杭,21,17710890829,运维,1995-08-29]的一串东西:"))
    with open("Information.txt", "r", encoding="utf8") as read_file, open("NewInformation.txt", "w+",
                                                                          encoding="utf8")as write_file:
        for line in read_file:
            new.append(line.strip().split(","))
        for new_line in new:
            # i.append(j)
            temp.append(new_line)
            # print(i[-1][0])
        Added_Data = "%d,%s" % (int(temp[-1][0]) + 1, user_input)
        temp.append(Added_Data.split(','))
        for each_line in temp:
            write_file.write(','.join(each_line).strip(""))  # 通过内置方法join把列表转换为字符串
            write_file.write("\n")  # 写入这行，可使文件换行
    os.rename("Information.txt", "Information.bak")
    os.rename("NewInformation.txt", "Information.txt")
    os.remove("Information.bak")

def delete():  #定义删除函数
    ReadFile = []
    User_Choice = int(input("请输入staff_id号:"))
    with open("Information.txt", "r", encoding="utf8") as read_file, open("NewInformation.txt", "w+",
                                                                          encoding="utf8") as write_file:
        for line in read_file:
            ReadFile.append(line.strip().split(","))
        for each_line in ReadFile:
            if User_Choice == int(each_line[0]):
                ReadFile.remove(ReadFile[User_Choice - 1])
                for new_line in ReadFile:
                    write_file.write(','.join(new_line).strip(""))  # 通过内置方法join把列表转换为字符串
                    write_file.write("\n")  # 写入这行，可使文件换行
    os.rename("Information.txt", "Information.bak")
    os.rename("NewInformation.txt", "Information.txt")
    os.remove("Information.bak")

def modify():  #定义更改函数
    if not os.path.exists("Information.txt"):
        exit(-1)
    User_Input = str(input("请输入:"))
    with open("Information.txt", "r", encoding="utf8") as read_file, open("NewInformation.txt", "w+",
                                                                          encoding="utf8") as write_file:
        for line in read_file.readlines():
            write_file.write(line.replace("IT", User_Input))
    os.rename("Information.txt", "Information.bak")
    os.rename("NewInformation.txt", "Information.txt")
    os.remove("Information.bak")

if __name__ == '__main__':
    msg = '''
       1：查询年龄大于22岁学生信息
       2：查询部门为IT学生信息
       3：查询工作开始时间在2013年学生信息
       4：添加
       5：删除
       6：修改
       7：退出
       '''
    menu_dic = {

        '1': search_age_above_22,
        '2': search_depot_of_IT,
        '3': search_date_of_2013,
        '4': add,
        '5': delete,
        '6': modify,
        '7': exit,
    }
    while True:
        print(msg)
        choice = input("操作>>: ").strip()
        if len(choice) == 0 or choice not in menu_dic or choice.isalpha():
            print("温馨提示:请输入1-7之间的数字")
            continue
        if choice == '7': break
        menu_dic[choice]()