2018-08-01day-13作业

k_name='name'
k_age='age'
k_job_number='job_number'
k_salary='salary'
k_position='position'
k_department='department'
k_all_worker='workers'
Workers={}

class Staff:
    def __init__(self):
        self.name=''
        self.age=0
        self.job_number=''
        self.salary=0
        self.position=''
        self.department=''

class Company:
    def __init__(self,name):
        self.name=name

    def add_staff(self):
        self.name=input('name:')
        self.age=input('age:')
        self.job_number=input('job_number:')
        self.salary=input('salary:')
        self.position=input('position:')
        self.department=input('department:')
        add_worker={k_name:self.name,k_age:self.age,k_job_number:self.job_number,k_salary:self.salary,k_position:self.position,k_department:self.department}
        all_workers = Workers.get(k_all_worker)
        if all_workers == None:
            all_workers = []
        # 将员工添加到容器中
        all_workers.append(add_worker)
        Workers[k_all_worker] = all_workers
        print('添加员工 %s 的信息成功' % self.name)
    def view(self):
        print(Workers)
    def delete(self):
        all_workers = Workers.get(k_all_worker)
        del_name=input('name:')
        del_workers=[]
        for worker in all_workers:
            if del_name==worker[k_name]:
                del_workers.append(worker)
            if not len(del_workers):
                print('没有找到相关的员工信息!')
                continue
        index = 0
        for worker in del_workers:
            print(index, worker)
            index += 1

        del_num = input('请选择要删除的员工的编号:')
        del_worker = del_workers[int(del_num)]
        Workers[k_all_worker].remove(del_worker)
        print('删除成功!')
c1=Company('阿里')
print('添加员工')
while True:
    c1.add_staff()
    print('1.继续添加\n其他：返回')
    prompt=input('==')
    if prompt!='1':
        break

print('查看员工信息')
c1.view()
print('删除员工')
c1.delete()