DjangoAPI/setup_data.py at master · FUBARS-SIH/DjangoAPI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
from api.models import *
from datetime import date

district1 = District.objects.create(name="chennai")
district2 = District.objects.create(name="trichy")
district3 = District.objects.create(name="madurai")
district4 = District.objects.create(name="salem")
district5 = District.objects.create(name="theni")

schedules = []

district=district1
items1 = ['idly', 'sidedish', 'rice', 'egg', 'chappati']
for day in range(5):
    schedules.append(Schedule.objects.create(district=district, day=day, item=items1[day]))

items2 = ['dosa', 'roti', 'dal', 'naan', 'bokra']
for day in range(5):
    schedules.append(Schedule.objects.create(district=district, day=day, item=items2[day]))

user1 = CustomUser.objects.create_user(username="auth", email="jyuvaraj000@gmail.com", password="teddybear", is_authority=True)
user2 = CustomUser.objects.create_user(username="school1", email="school1@gmail.com", password="leomessilm10")
user3 = CustomUser.objects.create_user(username="school2", email="school2@gmail.com", password="ronaldocr7")
user4 = CustomUser.objects.create_user(username="school3", email="school3@gmail.com", password="neymarjunior")
user5 = CustomUser.objects.create_user(username="school4", email="school5@gmail.com", password="neymarjunior")
authority = Authority.objects.create(user=user1, district=district1)

school1 = School.objects.create(user=user2, name="SBOI" ,district=district1)
school2 = School.objects.create(user=user3, name="PSBB" ,district=district1)
school3 = School.objects.create(user=user4, name="DAV" ,district=district1)
school4 = School.objects.create(user=user5, name='VSSS', district=district1)

report1 = Report.objects.create(school=school1, student_count=45, for_date=date(2020,8,3), added_by_school=True)
report2 = Report.objects.create(school=school2, student_count=45, for_date=date(2020,8,3), added_by_school=True)
report3 = Report.objects.create(school=school3, student_count=50, for_date=date(2020,8,3), added_by_school=True)


day = date(2020,8,3).weekday()
items = Schedule.objects.filter(district=district1, day=day)
for item in items:
    report1.items.create(item=item.item)
    report2.items.create(item=item.item)
    report3.items.create(item=item.item)


estimate_report1 = Report.objects.create(school=school1, student_count=45, for_date=date(2020,8,3), added_by_school=False, actual_report=report1)
estimate_report2 = Report.objects.create(school=school2, student_count=20, for_date=date(2020,8,3), added_by_school=False, actual_report=report2)
estimate_report3 = Report.objects.create(school=school3, student_count=45, for_date=date(2020,8,3), added_by_school=False, actual_report=report3)

for item in items:
    estimate_report1.items.create(item=item.item)
    estimate_report3.items.create(item=item.item)