人力资源管理系统1.0

"""__author__= xxx"""
"""
面向对象的人力资源管理系统：
功能：
    1.能存多个员工信息（姓名、年龄、工号、薪资、职位、部门）
    2.新员工入职：添加员工
    3.员工离职：删除员工
    4.查看某个员工信息
    5.查询薪资最高的员工
    6.查询指定部门中所有员工的平均薪资
    7.求整个公司的员工的平均年龄

"""
import json


def read_json():
    """获取json文件"""
    try:
        with open('./company.json', 'r', encoding='utf-8') as f:
            return json.load(f)
    except (FileNotFoundError, json.decoder.JSONDecodeError):
        with open('./company.json', 'w', encoding='utf-8') as f:
            json.dump({}, f)
        with open('./company.json', 'r', encoding='utf-8') as f:
            return json.load(f)


def write_json(content):
    """写json文件"""
    with open('./company.json', 'w', encoding='utf-8') as f:
        json.dump(content, f)


class Staff:
    """员工类"""

    def __init__(self, name='', age=0, number='', salary=0, position='', sect=''):
        self.name = name
        self.age = age
        self.number = number
        self.salary = salary
        self.position = position
        self.sect = sect

    def __str__(self):
        return '姓名:%s, 年龄:%d, 工号:%s, 薪资:%d, 职位:%s, 部门:%s' % \
               (self.name, self.age, self.number, self.salary, self.position, self.sect)


# class Sect:
#     """部门类"""
#     def __init__(self, name=''):
#         self.name = name
#         self.count = 0
#         self.staffs = []
#
#     def add_staffs(self):
#         """添加员工"""
#         self.count += 1
#         count = self.count
#         print('请输入员工的信息')
#         add_name = input('姓名:')
#         add_age = int(input('年龄:'))
#         add_number = self.name + str(count).rjust(3, '0')
#         add_salary = int(input('薪资:'))
#         add_position = input('职位:')
#         sta = Staff(add_name, add_age, add_number, add_salary, add_position, self.name)
#         self.staffs.append(sta)
#
#     def del_staffs(self):
#         """删除学生"""

class Company:
    """公司类"""

    def __init__(self, name='', boss=''):
        self.com_name = name
        self.com_boss = boss
        self.sects = []
        # self.staffs = []

    def add_staff(self):
        """添加员工"""
        sect = input('请输入添加员工所属部门：')
        count = 0
        flag = True
        # 判断部门是否存在
        for item in self.sects[:]:
            if sect in item['sect_name']:
                flag = False
                count = item['count']
                break
        if flag:
            sect_list = {'sect_name': sect, 'staffs': [], 'count': 0}
            self.sects.append(sect_list)
            count = 0

        print('请输入员工的信息')
        add_name = input('姓名:')
        add_age = int(input('年龄:'))
        count += 1
        add_number = sect + str(count).rjust(3, '0')
        add_salary = int(input('薪资:'))
        add_position = input('职位:')
        sta = {'name': add_name, 'age': add_age, 'number': add_number, 'salary': add_salary, 'position': add_position,
               'sect': sect}
        # sta = Staff(add_name, add_age, add_number, add_salary, add_position, sect)

        # 将员工添加进去
        for item in self.sects[:]:
            if sect == item['sect_name']:
                item['staffs'].append(sta)
                item['count'] = count

        return self.sects

    def del_staff(self):
        """删除员工"""
        # 根据姓名删除员工
        del_name = input('请输入要删除的员工姓名：')
        flag = True
        del_list = []
        index = 0
        for index1 in range(0, len(self.sects)):
            staffs = self.sects[index1]['staffs']
            for index2 in range(0, len(staffs)):
                if del_name == staffs[index2]['name']:
                    flag = False
                    del_list.append([index1, index2])
                    print(index, staffs[index2])
                    index += 1
        if flag:
            print('本公司没有这个人！')
        else:
            index_input = int(input('请输入要删除的员工下标：'))
            if 0 <= index_input <= index-1:
                del self.sects[del_list[index_input][0]]['staffs'][del_list[index_input][1]]
                print('删除成功')
            else:
                print('下标输入错误！')

        return self.sects

    def find_staff(self):
        """查看员工信息"""
        # 根据姓名查看员工信息
        find_name = input('请输入要查看的员工姓名：')
        flag = True
        for index1 in range(0, len(self.sects)):
            staffs = self.sects[index1]['staffs']
            for index2 in range(0, len(staffs)):
                if find_name == staffs[index2]['name']:
                    print(self.sects[index1]['staffs'][index2])
                    flag = False
        if flag:
            print('本公司没有这个人！')

    def find_max_salary(self):
        """查看公司薪资最高的员工"""
        max_salary = 0
        staff = {}
        for item in self.sects[:]:
            item1 = item['staffs']
            for item2 in item1:
                if max_salary <= item2['salary']:
                    max_salary = item2['salary']
                    staff = item2
        print(staff)

    def avg_salary(self):
        """查看公司某部门的平均薪资"""
        avg1 = None
        sum1 = 0
        sect_name_input = input('请输入部门的名字：')
        for item in self.sects[:]:
            if sect_name_input == item['sect_name']:
                for item1 in item['staffs']:
                    sum1 += item1['salary']
                avg1 = sum1 / len(item['staffs'])
        print('该部门的平均薪资为：%d' % avg1)

    def avg_age(self):
        """查看公司所有员工的平均年龄"""
        avg2 = None
        sum2 = 0
        index2 = 0
        for item in self.sects[:]:
            for item1 in item['staffs']:
                sum2 += item1['age']
                index2 += 1
        avg2 = sum2 / index2
        print('该公司的平均年龄为：%d' % avg2)


def print_main():
    print('人力资源管理系统')
    print('1.添加员工')
    print('2.删除员工')
    print('3.查看员工信息')
    print('4.查看公司薪资最高的员工')
    print('5.查看公司某部门的平均薪资')
    print('6.查看公司平均年龄')
    print('0.退出')


if __name__ == '__main__':
    content = read_json()
    com1 = Company('qianfeng', 'xxx')
    content['com_name'] = com1.com_name
    content['com_boss'] = com1.com_boss
    write_json(content)
    while True:
        print_main()
        num = input('>>>')
        if num == '0':
            break
        if num == '1':
            content = read_json()
            if not content.get('sects'):
                content['sects'] = []
            com1.sects = content['sects']
            print(com1.sects)
            sects = com1.add_staff()
            content['sects'] = sects
            write_json(content)
            print('添加成功')
            continue
        if num == '2':
            content = read_json()
            com1.sects = content['sects']
            sects = com1.del_staff()
            content['sects'] = sects
            write_json(content)
            continue
        if num == '3':
            content = read_json()
            com1.sects = content['sects']
            com1.find_staff()
            continue
        if num == '4':
            content = read_json()
            com1.sects = content['sects']
            com1.find_max_salary()
            continue
        if num == '5':
            content = read_json()
            com1.sects = content['sects']
            com1.avg_salary()
            continue
        if num == '6':
            content = read_json()
            com1.sects = content['sects']
            com1.avg_age()
            continue