python--员工信息表修改

雨点打透心脏的1/2处 2022-04-02 07:24 268阅读 0赞

# -*- coding: utf-8 -*-
    from Emp import Emp
    import json
    arr = []
    with open("d://123.txt", "r") as empfile:
        if empfile.read() != "":
            empfile.seek(0)
            arr = json.load(empfile)
    while True:
        print ("""
    1、员工录入
    2、查询员工信息
    3、修改员工信息
    4、删除
    5、根据工号查看
    6、退出
        """)
        cho = int(raw_input("请输入您的选择"))
        if cho == 1:
            num = raw_input("请输入工号")
            name = raw_input("请输入姓名")
            age = raw_input("请输入年龄")
            sex = raw_input("请输入性别")
            sal = raw_input("请输入工资")
            # emp = Emp(num, name, age, sex, sal)
            # emp = {"num": num, "name": name, "age": age, "sex": sex, "sal": sal}
            arr.append(Emp(num, name, age, sex, sal))
        elif cho == 2:
            for x in arr:
                print x
        elif cho == 3:
            num = raw_input("请输入工号")
            name = raw_input("请输入姓名")
            age = raw_input("请输入年龄")
            sex = raw_input("请输入性别")
            sal = raw_input("请输入工资")
            for x in arr:
                if x["num"] == num:
                    x["name"] = name
                    x["sex"] = sex
                    x["age"] = age
                    x["sal"] = sal
        elif cho == 4:
            num = raw_input("请输入工号")
            for x in arr:
                if x["num"] == num:
                    arr.remove(x)
                    break
        elif cho == 5:
            num = raw_input("请输入工号")
            for x in arr:
                if x["num"] == num:
                    print x
        elif cho == 6:
            with open("d://123.txt", "w") as empfile:
                json.dump(arr, empfile, default=lambda obj: obj.__dict__)
            break