"""
    @Author:Neo
    @Date:2020/1/16
    @Filename:students_info.py
    @Software:Pycharm
"""


class NotArgError(Exception):
    def __init__(self, message):
        self.message = message


class StudentInfo(object):
    def __init__(self, students):
        self.students = students

    def get_by_id(self, student_id):
        return self.students.get(student_id)

    def get_all_students(self):
        for id_, value in self.students.items():
            print('學(xué)號(hào)：{}, 姓名:{}, 年齡:{}, 性別:{}, 班級(jí):{}'.format(
                id_, value['name'], value['age'], value['sex'], value['class_number']
            ))
        return self.students

    def add(self, **student):
        try:
            self.check_user_info(**student)
        except Exception as e:
            raise e
        self.__add(**student)

    def adds(self, new_students):
        for student in new_students:
            try:
                self.check_user_info(**student)
            except Exception as e:
                print(e, student.get('name'))
                continue
            self.__add(**student)

    def __add(self, **student):
        new_id = max(self.students) + 1
        self.students[new_id] = student

    def delete(self, student_id):
        if student_id not in self.students:
            print('{} 并不存在'.format(student_id))
        else:
            user_info = self.students.pop(student_id)
            print('學(xué)號(hào)是{}, {}同學(xué)的信息已經(jīng)被刪除了'.format(student_id, user_info['name']))

    def deletes(self, ids):
        for id_ in ids:
            if id_ not in self.students:
                print(f'{id_} 不存在學(xué)生庫(kù)中')
                continue
            student_info = self.students.pop(id_)
            print(f'學(xué)號(hào){id_} 學(xué)生{student_info["name"]} 已被移除')

    def update(self, student_id, **kwargs):
        if student_id not in self.students:
            print('并不存在這個(gè)學(xué)號(hào):{}'.format(student_id))
        try:
            self.check_user_info(**kwargs)
        except Exception as e:
            raise e

        self.students[student_id] = kwargs
        print('同學(xué)信息更新完畢')

    def updates(self, update_students):
        for student in update_students:
            try:
                id_ = list(student.keys())[0]
            except IndexError as e:
                print(e)
                continue
            if id_ not in self.students:
                print(f'學(xué)號(hào){id_} 不存在')
                continue
            user_info = student[id_]
            try:
                self.check_user_info(**user_info)
            except Exception as e:
                print(e)
                continue
            self.students[id_] = user_info
        print('所有用戶(hù)信息更新完成')

    def search_users(self, **kwargs):

        assert len(kwargs) == 1, '參數(shù)數(shù)量傳遞錯(cuò)誤'

        values = list(self.students.values())
        key = None
        value = None
        result = []

        if 'name' in kwargs:
            key = 'name'
            value = kwargs[key]
        elif 'sex' in kwargs:
            key = 'sex'
            value = kwargs['sex']
        elif 'class_number' in kwargs:
            key = 'class_number'
            value = kwargs[key]
        elif 'age' in kwargs:
            key = 'age'
            value = kwargs[key]
        else:
            raise NotArgError('沒(méi)有發(fā)現(xiàn)搜索的關(guān)鍵字')

        for user in values:  # [{name, sex, age, class_number}, {}]
            if value in user[key]:
                result.append(user)
        return result

    def check_user_info(self, **kwargs):
        assert len(kwargs) == 4, '參數(shù)必須是4個(gè)'

        if 'name' not in kwargs:
            raise NotArgError('沒(méi)有發(fā)現(xiàn)學(xué)生姓名參數(shù)')
        if 'age' not in kwargs:
            raise NotArgError('缺少學(xué)生年齡參數(shù)')
        if 'sex' not in kwargs:
            raise NotArgError('缺少學(xué)生性別參數(shù)')
        if 'class_number' not in kwargs:
            raise NotArgError('缺少學(xué)生班級(jí)參數(shù)')

        name_value = kwargs['name']  # type(name_value)
        age_value = kwargs['age']
        sex_value = kwargs['sex']
        class_number_value = kwargs['class_number']
        # isinstace(對(duì)比的數(shù)據(jù), 目標(biāo)類(lèi)型） isinstance(1, str)

        if not isinstance(name_value, str):
            raise TypeError('name應(yīng)該是字符串類(lèi)型')
        if not isinstance(age_value, int):
            raise TypeError('age 應(yīng)該是整型')
        if not isinstance(sex_value, str):
            raise TypeError('sex應(yīng)該是字符串類(lèi)型')
        if not isinstance(class_number_value, str):
            raise TypeError('class_number應(yīng)該是字符串類(lèi)型')


students = {
    1: {
        'name': 'Neo',
        'age': 18,
        'class_number': 'A',
        'sex': 'boy'
    },
    2: {
        'name': 'Jack',
        'age': 16,
        'class_number': 'B',
        'sex': 'boy'
    },
    3: {
        'name': 'Lily',
        'age': 18,
        'class_number': 'A',
        'sex': 'girl'
    },
    4: {
        'name': 'Adem',
        'age': 18,
        'class_number': 'C',
        'sex': 'boy'
    },
    5: {
        'name': 'HanMeiMei',
        'age': 18,
        'class_number': 'B',
        'sex': 'girl'
    }
}

if __name__ == '__main__':
    student_info = StudentInfo(students)
    user = student_info.get_by_id(1)
    student_info.add(name='Marry', age=16, class_number='A', sex='girl')
    users = [
        {'name': 'Atom', 'age': 17, 'class_number': 'B', 'sex': 'boy'},
        {'name': 'Lucy', 'age': 18, 'class_number': 'C', 'sex': 'girl'}
    ]
    student_info.adds(users)
    student_info.get_all_students()
    print('------------------------------------------------------')

# >>> 執(zhí)行結(jié)果如下：
# >>> 學(xué)號(hào)：1, 姓名:Neo, 年齡:18, 性別:boy, 班級(jí):A
# >>> 學(xué)號(hào)：2, 姓名:Jack, 年齡:16, 性別:boy, 班級(jí):B
# >>> 學(xué)號(hào)：3, 姓名:Lily, 年齡:18, 性別:girl, 班級(jí):A
# >>> 學(xué)號(hào)：4, 姓名:Adem, 年齡:18, 性別:boy, 班級(jí):C
# >>> 學(xué)號(hào)：5, 姓名:HanMeiMei, 年齡:18, 性別:girl, 班級(jí):B
# >>> 學(xué)號(hào)：6, 姓名:Marry, 年齡:16, 性別:girl, 班級(jí):A
# >>> 學(xué)號(hào)：7, 姓名:Atom, 年齡:17, 性別:boy, 班級(jí):B
# >>> 學(xué)號(hào)：8, 姓名:Lucy, 年齡:18, 性別:girl, 班級(jí):C
# >>> ------------------------------------------------------

    student_info.deletes([7, 8])
    student_info.get_all_students()
    print('------------------------------------------------------')

# >>> 執(zhí)行結(jié)果如下：
# >>> ------------------------------------------------------
# >>> 學(xué)號(hào)7 學(xué)生Atom 已被移除
# >>> 學(xué)號(hào)8 學(xué)生Lucy 已被移除
# >>> 學(xué)號(hào)：1, 姓名:Neo, 年齡:18, 性別:boy, 班級(jí):A
# >>> 學(xué)號(hào)：2, 姓名:Jack, 年齡:16, 性別:boy, 班級(jí):B
# >>> 學(xué)號(hào)：3, 姓名:Lily, 年齡:18, 性別:girl, 班級(jí):A
# >>> 學(xué)號(hào)：4, 姓名:Adem, 年齡:18, 性別:boy, 班級(jí):C
# >>> 學(xué)號(hào)：5, 姓名:HanMeiMei, 年齡:18, 性別:girl, 班級(jí):B
# >>> 學(xué)號(hào)：6, 姓名:Marry, 年齡:16, 性別:girl, 班級(jí):A
# >>> ------------------------------------------------------


    student_info.updates([
        {1: {'name': 'Jone', 'age': 18, 'class_number': 'A', 'sex': 'boy'}},
        {2: {'name': 'Nike', 'age': 18, 'class_number': 'A', 'sex': 'boy'}}
    ])
    student_info.get_all_students()
    print('------------------------------------------------------')

# >>> 執(zhí)行結(jié)果如下：
# >>> ------------------------------------------------------
# >>> 所有用戶(hù)信息更新完成
# >>> 學(xué)號(hào)：1, 姓名:Jone, 年齡:18, 性別:boy, 班級(jí):A
# >>> 學(xué)號(hào)：2, 姓名:Nike, 年齡:18, 性別:boy, 班級(jí):A
# >>> 學(xué)號(hào)：3, 姓名:Lily, 年齡:18, 性別:girl, 班級(jí):A
# >>> 學(xué)號(hào)：4, 姓名:Adem, 年齡:18, 性別:boy, 班級(jí):C
# >>> 學(xué)號(hào)：5, 姓名:HanMeiMei, 年齡:18, 性別:girl, 班級(jí):B
# >>> 學(xué)號(hào)：6, 姓名:Marry, 年齡:16, 性別:girl, 班級(jí):A
# >>> ------------------------------------------------------


    result = student_info.search_users(name='d')
    print(result)
    print('------------------------------------------------------')

# >>> 執(zhí)行結(jié)果如下：
# >>> [{'name': 'Adem', 'age': 18, 'class_number': 'C', 'sex': 'boy'}]


    result = student_info.search_users(name='小')
    print(result)
    print('------------------------------------------------------')

# >>> 執(zhí)行結(jié)果如下：
# >>> ------------------------------------------------------
# >>> []
# >>> ------------------------------------------------------

    result = student_info.search_users(name='')
    print(result)
    result = student_info.search_users(name='小')
    print(result)
    print('------------------------------------------------------')
# >>> 執(zhí)行結(jié)果如下：
# >>> ------------------------------------------------------
# >>> [{'name': 'Jone', 'age': 18, 'class_number': 'A', 'sex': 'boy'}, {'name': 'Nike', 'age': 18, 'class_number': 'A', 'sex': 'boy'}, {'name': 'Lily', 'age': 18, 'class_number': 'A', 'sex': 'girl'}, {'name': 'Adem', 'age': 18, 'class_number': 'C', 'sex': 'boy'}, {'name': 'HanMeiMei', 'age': 18, 'class_number': 'B', 'sex': 'girl'}, {'name': 'Marry', 'age': 16, 'class_number': 'A', 'sex': 'girl'}]
# >>> []

    result = student_info.search_users(name='')
    print(result)