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