diff --git a/apps/main/management/commands/add_award.py b/apps/main/management/commands/add_award.py index b7abd20d..319dc54e 100644 --- a/apps/main/management/commands/add_award.py +++ b/apps/main/management/commands/add_award.py @@ -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.'))