phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix award add

Browse Source
This commit is contained in:
Виктор Гладких 2019-11-14 16:13:44 +03:00
parent 8ef31fa560
commit 2a9c7bf28d
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
apps/main/management/commands/add_award.py
View File

@ -27,14 +27,17 @@ class Command(BaseCommand):
def handle(self, *args, **kwargs):
objects =[]
for a in tqdm(self.award_sql(), desc='Add award to profile'):
profile = Employee.objects.filter(old_id=a.profile_id).first()
type = AwardType.objects.filter(old_id=a.award_type).first()
profiles = Employee.objects.filter(old_id=a.profile_id)
type = AwardType.objects.filter(old_id=a.award_type)
state = Award.PUBLISHED if a.state == 'published' else Award.WAITING
if profile and type:
award = Award(award_type=type, vintage_year=a.vintage_year,
title={"en-GB": a.title}, state=state,
content_object=profile, old_id=a.id)
objects.append(award)
if profiles.exists() and type.exists():
for profile in profiles:
print(f'PROFILE {profile}')
award = Award(award_type=type.first(), vintage_year=a.vintage_year,
title={"en-GB": a.title}, state=state,
content_object=profile, old_id=a.id)
objects.append(award)
awards = Award.objects.bulk_create(objects)
self.stdout.write(self.style.WARNING(f'Created awards objects.'))