stu-system-02（批量）

'''
学生信息库，调整面向对象，添加批量功能
'''

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

    def get_all_students(self):
        for id_, value in self.students.items():
            print('学号:{}, 姓名:{}, 年龄:{}, 班级:{}, 性别:{}'.format(
                id_, value['name'], value['age'], value['class_number'], value['sex']))
        return self.students

    def check_user_info(self,**kwargs):
        if 'name' not in kwargs:
            return '缺少学生姓名'  # 需要使用return退出函数
        if 'age' not in kwargs:
            return '缺少学生年龄'
        if 'class_number' not in kwargs:
            return '缺少学生班级'
        if 'sex' not in kwargs:
            return '缺少学生性别'
        return True

    # 添加学生信息
    def add(self,**kwargs):
        check = self.check_user_info(**kwargs)
        if check != True:
            print(check)
            return
        else:
            self.__add(**kwargs)

    # 批量添加
    def adds(self,new_students):
        for student in new_students:
            check = self.check_user_info(**student)
            if check != True:
                print(check,student.get['name'])
                return
            else:
                self.__add(**student)

    def __add(self, **kwargs):
        new_id = max(self.students) + 1
        self.students[new_id] = {'name': kwargs['name'],
                              'age': kwargs['age'],
                              'class_number': kwargs['class_number'],
                              'sex': kwargs['sex']}

    # 学生信息的删除
    def delete(self,id_):
        if id_ not in self.students.keys():
            print('需要删除的学号%s 学生不存在' % id_)
        else:
            result = self.students.pop(id_)
            print('学号为%s 的 %s 同学已被删除' % (id_, result['name']))
        # return students

    # 批量删除
    def deletes(self,*ids):
        for id_ in ids:
            if id_ not in self.students.keys():
                print('需要删除的学号%s 学生不存在' % id_)
            else:
                result = self.students.pop(id_)
                print('学号为%s 的 %s 同学已被删除' % (id_, result['name']))

    # 学生信息的修改/更新,student_id作为关键参数传入
    def update(self,student_id, **kwargs):
        if student_id not in self.students:
            print('需要更改的学号%s 学生不存在' % student_id)
        check = self.check_user_info(**kwargs)
        if check != True:
            print(check)
            return
        else:
            self.students[student_id] = kwargs
            print('学号为{} {}同学信息更新完成'.format(student_id, kwargs['name']))

    #批量更新，传入的是列表，列表里头是字典{唯一的学号+学生信息(又是一个字典)}
    def updates(self,students):
        for stu in students:
            id_ = list(stu.keys())[0] # 重要，list(伪列表（key）)[0]
            user_info = stu[id_]
            self.update(id_,**user_info)
        print('所有同学已更新完毕')

    def get_by_id(self,student_id):
        if student_id not in self.students:
            print('需要更改的学号%s 学生不存在' % student_id)
        else:
            result = self.students[student_id]
            print('查询结果：学号为{} 的学生：姓名:{}, 年龄:{}, 班级:{}, 性别:{}'.format(
                student_id, result['name'], result['age'], result['class_number'], result['sex']))
        # return students.get(student_id) 也可以只用一行代码获取，但是得到的是字典

    # 根据关键词查询==>模糊查找匹配
    def search_user(self,**kwargs):
        users = list(self.students.values())
        key = None
        value = None
        result = []
        if 'name' in kwargs:
            key = 'name'
            value = kwargs[key]
        elif 'age' in kwargs:
            key = 'age'
            value = str(kwargs[key])
        elif 'class_number' in kwargs:
            key = 'class_number'
            value = kwargs[key]
        elif 'sex' in kwargs:
            key = 'sex'
            value = kwargs[key]
        else:
            return '您所查找的关键词不存在'

        for user in users:
            if value in str(user[key]):
                result.append(user)
        if len(result) == 0:
            print('关键词为%s=%s的同学不存在' % (key, value))
            return
        else:
            return result

students = {1: {'name': 'mark', 'age': 18, 'class_number': 'B', 'sex': 'boy'},
                2: {'name': 'nancy', 'age': 17, 'class_number': 'A', 'sex': 'girl'},
                3: {'name': 'pall', 'age': 19, 'class_number': 'A', 'sex': 'boy'},
                4: {'name': 'Lily', 'age': 18, 'class_number': 'B', 'sex': 'girl'}
                }

if __name__ == '__main__':
    stu = StudentInfo(students)
    re = stu.search_user(name='mark')
    print(re)
    stu.get_by_id(4)
    stu.add(name='lucy',sex='girl',class_number='B',age=18)
    stu.get_all_students()
    users = [{'name':'Nick','sex':'boy','class_number':'B','age':18},{'name':'Nana','sex':'girl','class_number':'A','age':17}]
    stu.adds(users)
    stu.get_all_students()
    # stu.deletes(2,4)
    # stu.get_all_students()
    print('===============')
    update_users = [{6:{'name': 'Nick.li', 'sex': 'boy', 'class_number': 'C', 'age': 18}},
             {7:{'name': 'Nana.zhang', 'sex': 'girl', 'class_number': 'A','age': 17}}]
    stu.updates(update_users)
    stu.get_all_students()
    result = stu.search_user(name='a')
    print(result)
    result = stu.search_user(age=18)
    print(result)
    # print(stu_all)