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