def get(self, id=None): if id: try: faculty = Faculty.objects.get(id=id) except (DoesNotExist, VE): return 'No cuch faculty' return FacultySchema().dump(faculty) else: faculties = Faculty.objects() return FacultySchema().dump(faculties, many=True)
def get_student_data(): return_ = {} faculty = random.choice(Faculty.objects()) return_['curator'] = random.choice(Curator.objects(faculty=faculty)) return_['academic_group'] = random.choice( AcademicGroup.objects(faculty=faculty)) return_['student_card'] = random.randint(0, 100000000000000) return_['student_name'] = get_random_word(random.randint(2, 10)).title() return_['student_surname'] = get_random_word(random.randint(2, 15)).title() return_['marks'] = [] lower_mark = random.randint(2, 5) for i in range(random.randint(5, 10)): mark = Mark() mark.mark = random.randint(lower_mark, 5) return_['marks'].append(mark) return return_
def create_curator(**kwargs): requirede_fields = ('faculty', 'curator_name', 'curator_surname') [kwargs[arg] for arg in requirede_fields] return Curator.objects.create(**kwargs) def create_student(**kwargs): requirede_fields = ('academic_group', 'student_name', 'student_surname', 'student_card') [kwargs[arg] for arg in requirede_fields] return Student.objects.create(**kwargs) if __name__ == '__main__': for i in range(5): print(create_faculty(**get_faculty_data())) for fac in Faculty.objects(): for i in range(random.randint(1, 3)): print(create_academic_group(**get_academic_group_data(fac))) for fac in Faculty.objects(): for i in range(random.randint(2, 5)): print(create_curator(**get_curator_data(fac))) for i in range(100): print(create_student(**get_student_data()))