def handle(self, *args, **options): num_applications = options["num"][0] available_partners = Partner.objects.all() # Don't fire any applications from the superuser. all_editors = Editor.objects.exclude(user__is_superuser=True) import_date = datetime.datetime(2017, 7, 17, 0, 0, 0) # We want to flag applications as SENT via a client later, so let's only # automatically give applications non-Sent statuses. valid_choices = [ Application.PENDING, Application.QUESTION, Application.APPROVED, Application.NOT_APPROVED, Application.INVALID, ] for _ in range(num_applications): random_editor = random.choice(all_editors) # Limit to partners this user hasn't already applied to. not_applied_partners = available_partners.exclude( applications__editor=random_editor) if not_applied_partners: random_partner = random.choice(not_applied_partners) app = ApplicationFactory( editor=random_editor, partner=random_partner, hidden=self.chance(True, False, 10), ) # Make sure partner-specific information is filled. if random_partner.specific_stream: app.specific_stream = random.choice( Stream.objects.filter(partner=random_partner)) if random_partner.specific_title: app.specific_title = Faker( random.choice( settings.FAKER_LOCALES)).sentence(nb_words=3) if random_partner.agreement_with_terms_of_use: app.agreement_with_terms_of_use = True if random_partner.account_email: app.account_email = Faker( random.choice(settings.FAKER_LOCALES)).email() # Imported applications have very specific information, and were # all imported on the same date. imported = self.chance(True, False, 50) if imported: app.status = Application.SENT app.date_created = import_date app.date_closed = import_date app.rationale = "Imported on 2017-07-17" app.comments = "Imported on 2017-07-17" app.imported = True else: app.status = random.choice(valid_choices) # Figure out earliest valid date for this app if random_editor.wp_registered < import_date.date(): start_date = import_date else: start_date = random_editor.wp_registered app.date_created = Faker( random.choice( settings.FAKER_LOCALES)).date_time_between( start_date=start_date, end_date="now", tzinfo=None) app.rationale = Faker(random.choice( settings.FAKER_LOCALES)).paragraph(nb_sentences=3) app.comments = Faker(random.choice( settings.FAKER_LOCALES)).paragraph(nb_sentences=2) # For closed applications, assign date_closed and date_open if app.status in Application.FINAL_STATUS_LIST: if not imported: potential_end_date = app.date_created + relativedelta( years=1) if potential_end_date > datetime.datetime.now(): end_date = "now" else: end_date = potential_end_date app.date_closed = Faker( random.choice( settings.FAKER_LOCALES)).date_time_between( start_date=app.date_created, end_date=end_date, tzinfo=None) app.days_open = (app.date_closed - app.date_created).days # Make sure we always set sent_by if app.status == Application.SENT and not app.sent_by: app.sent_by = twl_team app.save() # Let's mark all Approved applications that were approved more # than 3 weeks ago as Sent. old_approved_apps = Application.objects.filter( status=Application.APPROVED, date_created__lte=datetime.datetime.now() - relativedelta(weeks=3), ) # We need to be able to handle messages message_patcher = patch( "TWLight.applications.views.messages.add_message") message_patcher.start() for approved_app in old_approved_apps: client = Client(SERVER_NAME="twlight.vagrant.localdomain") coordinator = logged_in_example_coordinator( client, approved_app.partner.coordinator) url = reverse("applications:send_partner", kwargs={"pk": approved_app.partner.pk}) this_partner_access_codes = AccessCode.objects.filter( partner=approved_app.partner, authorization__isnull=True) if approved_app.partner.authorization_method == Partner.EMAIL: request = RequestFactory().post( url, data={"applications": [approved_app.pk]}) # If this partner has access codes, assign a code to # this sent application. elif (this_partner_access_codes and approved_app.partner.authorization_method == Partner.CODES): access_code = random.choice(this_partner_access_codes) request = RequestFactory().post( url, data={ "accesscode": [ "{app_pk}_{code}".format(app_pk=approved_app.pk, code=access_code.code) ] }, ) request.user = coordinator response = SendReadyApplicationsView.as_view()( request, pk=approved_app.partner.pk) # Renew a selection of sent apps. all_apps = Application.objects.filter(status=Application.SENT) num_to_renew = int(all_apps.count() * 0.5) for app_to_renew in random.sample(list(all_apps), num_to_renew): app_to_renew.renew() for application in Application.objects.filter( status=Application.PENDING, parent__isnull=False): parent_application = Application.objects.get( pk=application.parent.pk) app_date = parent_application.date_closed renewal_date = Faker(random.choice( settings.FAKER_LOCALES)).date_time_between(start_date=app_date, end_date="now", tzinfo=None) application.date_created = renewal_date application.save()
def handle(self, *args, **options): num_applications = options['num'][0] fake = Faker() available_partners = Partner.objects.all() # Don't fire any applications from the superuser. all_users = User.objects.exclude(is_superuser=True) import_date = datetime.datetime(2017, 7, 17, 0, 0, 0) for _ in range(num_applications): random_user = random.choice(all_users) random_partner = random.choice(available_partners) app = ApplicationFactory( editor = random_user.editor, partner = random_partner, hidden = self.chance(True, False, 10) ) # Make sure partner-specific information is filled. if random_partner.specific_stream: app.specific_stream = random.choice( Stream.objects.filter(partner=random_partner)) if random_partner.specific_title: app.specific_title = fake.sentence(nb_words= 3) if random_partner.agreement_with_terms_of_use: app.agreement_with_terms_of_use = True if random_partner.account_email: app.account_email = fake.email() # Imported applications have very specific information, and were # all imported on the same date. imported = self.chance(True, False, 50) if imported: app.status = Application.SENT app.date_created = import_date app.date_closed = import_date app.rationale = "Imported on 2017-07-17" app.comments = "Imported on 2017-07-17" app.imported = True else: app.status = random.choice(Application.STATUS_CHOICES)[0] app.date_created = fake.date_time_between( start_date = random_user.editor.wp_registered, end_date = "now", tzinfo=None) app.rationale = fake.paragraph(nb_sentences=3) app.comments = fake.paragraph(nb_sentences=2) # For closed applications, assign date_closed and date_open if app.status in Application.FINAL_STATUS_LIST: if not imported: potential_end_date = app.date_created + relativedelta(years=1) if potential_end_date > datetime.datetime.now(): end_date = "now" else: end_date = potential_end_date app.date_closed = fake.date_time_between( start_date = app.date_created, end_date = end_date, tzinfo=None) app.days_open = (app.date_closed - app.date_created).days if app.status == Application.SENT: # Assign sent_by if this is a non-imported sent applications if not imported: app.sent_by = random_partner.coordinator # If this partner has access codes, assign a code to # this sent application. if random_partner.authorization_method == Partner.CODES: this_partner_access_codes = AccessCode.objects.filter( partner=random_partner, authorization__isnull=True) app_code = random.choice(this_partner_access_codes) #app_code.application = app app_code.save() app.save() # Renew a selection of sent apps. all_apps = Application.objects.filter(status=Application.SENT) num_to_renew = int(all_apps.count()*0.5) for app_to_renew in random.sample(all_apps, num_to_renew): app_to_renew.renew()
def handle(self, *args, **options): num_applications = options['num'][0] fake = Faker() available_partners = Partner.objects.all() # Don't fire any applications from the superuser. all_users = User.objects.exclude(is_superuser=True) for _ in range(num_applications): random_user = random.choice(all_users) random_partner = random.choice(available_partners) app = ApplicationFactory( editor = random_user.editor, partner = random_partner, hidden = self.chance(True, False, 10) ) # Make sure partner-specific information is filled. if random_partner.specific_stream: app.specific_stream = random.choice( Stream.objects.filter(partner=random_partner)) if random_partner.specific_title: app.specific_title = fake.sentence(nb_words= 3) if random_partner.agreement_with_terms_of_use: app.agreement_with_terms_of_use = True if random_partner.account_email: app.account_email = fake.email() # Imported applications have very specific information, and were # all imported on the same date. imported = self.chance(True, False, 50) import_date = datetime.datetime(2017, 7, 17, 0, 0, 0) if imported: app.status = Application.SENT app.date_created = import_date app.date_closed = import_date app.rationale = "Imported on 2017-07-17" app.comments = "Imported on 2017-07-17" else: app.status = random.choice(Application.STATUS_CHOICES)[0] app.date_created = fake.date_time_between( start_date = random_user.editor.wp_registered, end_date = "now", tzinfo=None) app.rationale = fake.paragraph(nb_sentences=3) app.comments = fake.paragraph(nb_sentences=2) # Assign sent_by if this is a non-imported sent applications if app.status == Application.SENT: if not imported: app.sent_by = random_partner.coordinator app.date_closed = fake.date_time_between( start_date = app.date_created, end_date = "now", tzinfo=None) else: app.date_closed = import_date app.save() # Renew a selection of apps. all_apps = Application.objects.all() num_to_renew = int(num_applications*0.3) for app_to_renew in random.sample(all_apps, num_to_renew): app_to_renew.renew()