作业-学生管理系统

作业需求：

# 1. 员工信息表程序，实现增删改查操作：
# 2. 可进行模糊查询，语法至少支持下面3种:
# 　　select name,age from staff_table where age > 22
# 　　select * from staff_table where dept = "IT"
# 　　select * from staff_table where enroll_date like "2013"
# 　　查到的信息，打印后，最后面还要显示查到的条数
# 3. 可创建新员工纪录，以phone做唯一键，staff_id需自增
# 4. 可删除指定员工信息纪录，输入员工id，即可删除
# 5. 可修改员工信息，语法如下:
# 　　UPDATE staff_table SET dept="Market" where dept = "IT"
# 6. 注意：以上需求，要充分使用函数，请尽你的最大限度来减少重复代码


import json
import os

#打印框架
def kuang_jia(dress):
    print('员工信息表'.center(90))
    print('输入相应的序号获得相应功能：  ', '1.查看'.center(5), '2.增加'.center(5), '3.删除'.center(5), '4.修改'.center(5),
          '5.查找'.center(5), '6.退出'.center(5))
    print('-'*90)
    print('|', 'Number'.center(10), 'Id'.center(10), 'Name'.center(10), 'Age'.center(10),
          'Phone'.center(20), 'Dept'.center(10), 'Enorll_date'.center(10), '|')
    try:
        with open(dress, 'r', encoding='utf-8') as f:
            for line in f:
                ret = json.loads(line)
                print('|', ret[0].center(10), ret[1].center(10), ret[2].center(10),
                      ret[3].center(10), ret[4].center(20), ret[5].center(10), ret[6].center(10), '|')
    except FileNotFoundError:
        f = open(dress, 'w', encoding='utf-8')
        f.close()
    print('='*90)
    print('输入相应的序号获得相应功能：  ', '1.查看'.center(5), '2.增加'.center(5), '3.删除'.center(5), '4.修改'.center(5),
          '5.查找'.center(5), '6.退出'.center(5))
    print('=' * 90)

#增加
def add(dress):
    while 1:
        print('输入q退出[增加]功能！')
        l = []
        id = input('Id: ')
        if id == 'q':
            break

        name = input('Name: ')
        if name == 'q':
            break
        age = input('Age: ')
        if age == 'q':
            break

        phone = input('Phone: ')
        if phone == 'q':
            break

        dept = input('Dept: ')
        if dept == 'q':
            break

        enorll_date = input('Enorll_date: ')
        if enorll_date == 'q':
            break

        with open(dress, 'r', encoding='utf-8') as f:
            lines = f.readlines()

        l.append(str(len(lines)))
        l.append(id)
        l.append(name)
        l.append(age)
        l.append(phone)
        l.append(dept)
        l.append(enorll_date)


        with open(dress, 'a', encoding='utf-8') as f:
            ret = json.dumps(l)
            f.write(ret + '
')



#删除
def dele(dress):
    number = input('Number: ')
    with open(dress, 'r', encoding='utf-8') as f, open('file.bak', 'w', encoding='utf-8') as f1:
        for line in f:
            ret = json.loads(line)
            if number != ret[0]:
                f1.write(line)

    os.remove(dress)
    os.rename('file.bak', dress)
#修改
def changes(dress):
    with open(dress, 'r', encoding='utf-8') as f, open('file.bak', 'w', encoding='utf-8') as f1:
        number = input('Number: ')
        for line in f:
            ret = json.loads(line)
            if number == ret[0]:
                print('|', 'Number'.center(10), 'Id'.center(10), 'Name'.center(10), 'Age'.center(10),
                      'Phone'.center(20), 'Dept'.center(10), 'Enorll_date'.center(10), '|')
                print('|', ret[0].center(10), ret[1].center(10), ret[2].center(10),
                      ret[3].center(10), ret[4].center(20), ret[5].center(10), ret[6].center(10), '|')
                while 1:
                    print('输入q退出[修改]')
                    can_shu = input('输入要修改的参数： ').lower().strip()
                    info = input('您希望修改成： ')
                    if can_shu == 'id':
                        ret[1] = info
                        break
                    elif can_shu == 'name':
                        ret[2] = info
                        break
                    elif can_shu == 'age':
                        ret[3] = info
                        break
                    elif can_shu == 'phone':
                        ret[4] = info
                        break
                    elif can_shu == 'dept':
                        ret[5] = info
                        break
                    elif can_shu == 'enorll_date':
                        ret[6] = info
                        break
                    elif can_shu == 'q':
                        break
                    else:
                        print('输入有误，请重新输入！')
                        continue
                f1.write(json.dumps(ret))
            else:
                f1.write(line)

    os.remove(dress)
    os.rename('file.bak', dress)

#查找
def find(dress):
    while 1:
        can_shu = input('输入您要查询的参数： ').lower().strip()
        if can_shu not in 'nameageidphonedeptenroll_date':
            print('输入有误，请重新输入！')
            continue
        else:
            break
    info = input('输入您要查询的参数值： ')
    with open(dress, 'r', encoding='utf-8') as f:
        num = 0
        print('|', 'Number'.center(10), 'Id'.center(10), 'Name'.center(10), 'Age'.center(10),
              'Phone'.center(20), 'Dept'.center(10), 'Enorll_date'.center(10), '|')
        for line in f:
            ret = json.loads(line)
            if can_shu == 'id':
                if info == ret[1]:
                    print('|', ret[0].center(10), ret[1].center(10), ret[2].center(10),
                          ret[3].center(10), ret[4].center(20), ret[5].center(10), ret[6].center(10), '|')
                    num += 1
            elif can_shu == 'name':
                if info == ret[2]:
                    print('|', ret[0].center(10), ret[1].center(10), ret[2].center(10),
                          ret[3].center(10), ret[4].center(20), ret[5].center(10), ret[6].center(10), '|')
                    num += 1
            elif can_shu == 'age':
                if info == ret[3]:
                    print('|', ret[0].center(10), ret[1].center(10), ret[2].center(10),
                          ret[3].center(10), ret[4].center(20), ret[5].center(10), ret[6].center(10), '|')
                    num += 1
            elif can_shu == 'phone':
                if info == ret[4]:
                    print('|', ret[0].center(10), ret[1].center(10), ret[2].center(10),
                          ret[3].center(10), ret[4].center(20), ret[5].center(10), ret[6].center(10), '|')
                    num += 1
            elif can_shu == 'dept':
                if info == ret[5]:
                    print('|', ret[0].center(10), ret[1].center(10), ret[2].center(10),
                          ret[3].center(10), ret[4].center(20), ret[5].center(10), ret[6].center(10), '|')
                    num += 1
            elif can_shu == 'enorll_date':
                if info == ret[6]:
                    print('|', ret[0].center(10), ret[1].center(10), ret[2].center(10),
                          ret[3].center(10), ret[4].center(20), ret[5].center(10), ret[6].center(10), '|')
                    num += 1
    print('共找到{}条相关信息。'.format(num))
    print('输入相应的序号获得相应功能：  ', '1.查看'.center(5), '2.增加'.center(5), '3.删除'.center(5), '4.修改'.center(5),
          '5.查找'.center(5), '6.退出'.center(5))

#自动编号
def bian_hao(dress):
    with open(dress, 'r', encoding='utf-8') as f, open('file.bak', 'w', encoding='utf-8') as f1:
        lines = f.readlines()
        f.seek(0)
        for line in f:
            ret = json.loads(line)
            ret[0] = str(lines.index(line))
            f1.write(json.dumps(ret) + '
')

    os.remove(dress)
    os.rename('file.bak', dress)

#主程序
def main():
    dress = input('输入相对位置下需要导入的文件名： ').strip()
    bian_hao(dress)
    kuang_jia(dress)
    while 1:
        want = input('输入你想要进行的操作： ').strip()
        if want == '1':
            kuang_jia(dress)
        elif want == '2':
            add(dress)
            kuang_jia(dress)
        elif want == '3':
            dele(dress)
            bian_hao(dress)
            kuang_jia(dress)
        elif want == '4':
            changes(dress)
            kuang_jia(dress)
        elif want == '5':
            find(dress)
        elif want == '6':
            break

main()