regluit/core/management/commands/set_campaign_editions.py

15 lines
496 B
Python

from django.core.management.base import BaseCommand
from regluit.core.models import Campaign
class Command(BaseCommand):
help = "set campaign edition for every campaign"
def handle(self, **options):
fixed = 0
for campaign in Campaign.objects.all():
if not campaign.edition:
campaign.edition = campaign.work.editions.all()[0]
campaign.save()
fixed +=1
print "{} campaign editions set".format(fixed)