python--员工信息表修改

# -*- 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

猜你喜欢

转载自blog.csdn.net/qq_42721694/article/details/85252539