|
| 1 | +import datetime |
| 2 | +from ennead.config import Config |
| 3 | + |
| 4 | +from ennead.models.base import database |
| 5 | +from ennead.models.user import User, UserGroup |
| 6 | +from ennead.models.task import TaskSet, Task |
| 7 | +from ennead.models.thread import Thread, Post |
| 8 | + |
| 9 | +config_path = 'ennead.json' |
| 10 | +if config_path: |
| 11 | + config = Config.from_filename(config_path) |
| 12 | +else: |
| 13 | + config = Config() |
| 14 | + |
| 15 | +database.initialize(config.DB_CLASS(config.DB_NAME, **config.DB_PARAMS)) |
| 16 | +database.create_tables([User, Task, TaskSet, Thread, Post]) |
| 17 | + |
| 18 | +student = User( |
| 19 | + username = 'test', |
| 20 | + email = 'test@test.com', |
| 21 | + registered_at = datetime.datetime.now(), |
| 22 | + first_name = 'Иван', |
| 23 | + surname = 'Петров', |
| 24 | + patronym = 'Иванович', |
| 25 | + group = UserGroup.student) |
| 26 | +student.set_password('password') |
| 27 | +student.save() |
| 28 | + |
| 29 | +teacher = User( |
| 30 | + username = 'prep', |
| 31 | + email = 'prep@test.com', |
| 32 | + registered_at = datetime.datetime.now(), |
| 33 | + first_name = 'Препод', |
| 34 | + surname = 'Злой', |
| 35 | + patronym = '', |
| 36 | + group = UserGroup.teacher) |
| 37 | +teacher.set_password('password') |
| 38 | +teacher.save() |
| 39 | + |
| 40 | + |
| 41 | +prev_task_set = TaskSet.create(name='Старая заочка', active=True) |
| 42 | +task_set = TaskSet.create(name='Текущая заочка', active=True) |
| 43 | + |
| 44 | +task_1_1 = Task.create(order_num=1, name='Очень старая задача #1', description='Когда трава была зеленее', base_score=1, task_set=prev_task_set) |
| 45 | +task_1_2 = Task.create(order_num=2, name='Очень старая задача #2', description='И задачи были забористей', base_score=1, task_set=prev_task_set) |
| 46 | + |
| 47 | +task_2_1 = Task.create(order_num=1, name='Задача первая', description='Самая первая', base_score=1, task_set=task_set) |
| 48 | +task_2_2 = Task.create(order_num=3, name='Задача последняя', description='Хардкор', base_score=42, task_set=task_set) |
| 49 | +task_2_3 = Task.create(order_num=2, name='Задача два', description='Посложнее', base_score=5, task_set=task_set) |
| 50 | + |
| 51 | +thread_1 = Thread.create(student=student, task=task_2_1) |
| 52 | +thread_2 = Thread.create(student=student, task=task_2_2) |
| 53 | + |
| 54 | +post_1_1 = Post.create(thread=thread_1, text='Первый нах', date = datetime.datetime.now(), author=student) |
| 55 | +post_1_2 = Post.create(thread=thread_1, text='Так себе решение. Пока 0 баллов', date = datetime.datetime.now(), author=teacher) |
| 56 | +post_1_3 = Post.create(thread=thread_1, text='Ну ладно, 2+2=3', date = datetime.datetime.now(), author=student) |
| 57 | +post_1_3 = Post.create(thread=thread_1, text='Ой, 2+2=4', date = datetime.datetime.now(), author=student) |
| 58 | +post_1_4 = Post.create(thread=thread_1, text='Ок, угадал. 1 балл', date = datetime.datetime.now(), author=teacher) |
0 commit comments