regluit/core/management/commands/random_campaigns.py

49 lines
1.7 KiB
Python

from datetime import timedelta
from decimal import Decimal as D
from random import randint, randrange
from django.conf import settings
from django.core.management.base import BaseCommand
from regluit.core.models import Work, Campaign
from regluit.utils.localdatetime import now
class Command(BaseCommand):
help = "creates random campaigns for any works that lack one for testing"
def handle(self, *args, **options):
for work in Work.objects.all():
if work.campaigns.all().count() > 0:
continue
campaign = Campaign()
campaign.name = work.title
campaign.work = work
campaign.description = "Test Campaign"
# random campaign target between $200 and $10,000
campaign.target = D(randint(200,10000))
# add a test rightsholder recipient right now
campaign.paypal_receiver = settings.PAYPAL_TEST_RH_EMAIL
# random deadline between 5 days from now and 180 days from now
_now = now()
campaign.deadline = random_date(_now + timedelta(days=5),
_now + timedelta(days=180))
# randomly activate some of the campaigns
coinflip = D(randint(0,10))
if coinflip > 5:
campaign.activate()
campaign.save()
print "campaign %s...status: %s" % (unicode(campaign).encode('ascii','replace') , campaign.status)
def random_date(start, end):
delta = end - start
int_delta = (delta.days * 24 * 60 * 60) + delta.seconds
random_second = randrange(int_delta)
return (start + timedelta(seconds=random_second))