def form_valid(self, form): try: cleaned_data = form.cleaned_data org = self.request.user.get_org() for key in cleaned_data: if key.startswith('field_'): idx = key[6:] label = cleaned_data["label_%s" % idx] field = cleaned_data[key] show_in_table = cleaned_data["show_%s" % idx] value_type = cleaned_data['type_%s' % idx] if field == '__new_field': if label: analytics.track(self.request.user.username, 'temba.contactfield_created') key = ContactField.make_key(label) ContactField.get_or_create( org, key, label, show_in_table=show_in_table, value_type=value_type) else: if label: ContactField.get_or_create( org, field.key, label, show_in_table=show_in_table, value_type=value_type) else: ContactField.hide_field(org, field.key) if 'HTTP_X_PJAX' not in self.request.META: return HttpResponseRedirect(self.get_success_url()) else: # pragma: no cover return self.render_to_response( self.get_context_data( form=form, success_url=self.get_success_url(), success_script=getattr(self, 'success_script', None))) except IntegrityError as e: # pragma: no cover message = str(e).capitalize() errors = self.form._errors.setdefault( forms.forms.NON_FIELD_ERRORS, forms.util.ErrorList()) errors.append(message) return self.render_to_response( self.get_context_data(form=form))
def form_valid(self, form): try: cleaned_data = form.cleaned_data org = self.request.user.get_org() for key in cleaned_data: if key.startswith('field_'): idx = key[6:] label = cleaned_data["label_%s" % idx] field = cleaned_data[key] show_in_table = cleaned_data["show_%s" % idx] value_type = cleaned_data['type_%s' % idx] if field == '__new_field': if label: analytics.track(self.request.user.username, 'temba.contactfield_created') key = ContactField.make_key(label) ContactField.get_or_create(org, key, label, show_in_table=show_in_table, value_type=value_type) else: if label: ContactField.get_or_create(org, field.key, label, show_in_table=show_in_table, value_type=value_type) else: ContactField.hide_field(org, field.key) if 'HTTP_X_PJAX' not in self.request.META: return HttpResponseRedirect(self.get_success_url()) else: # pragma: no cover return self.render_to_response(self.get_context_data(form=form, success_url=self.get_success_url(), success_script=getattr(self, 'success_script', None))) except IntegrityError as e: # pragma: no cover message = str(e).capitalize() errors = self.form._errors.setdefault(forms.forms.NON_FIELD_ERRORS, forms.utils.ErrorList()) errors.append(message) return self.render_to_response(self.get_context_data(form=form))
def test_scheduling(self): campaign = Campaign.objects.create(name="Planting Reminders", group=self.farmers, org=self.org, created_by=self.admin, modified_by=self.admin) self.assertEquals("Planting Reminders", unicode(campaign)) # create a reminder for our first planting event planting_reminder = CampaignEvent.objects.create(campaign=campaign, relative_to=self.planting_date, offset=0, flow=self.reminder_flow, delivery_hour=17, created_by=self.admin, modified_by=self.admin) self.assertEquals("Planting Date == 0 -> Color Flow", unicode(planting_reminder)) # schedule our reminders EventFire.update_campaign_events(campaign) # we should haven't any event fires created, since neither of our farmers have a planting date self.assertEquals(0, EventFire.objects.all().count()) # ok, set a planting date on one of our contacts self.farmer1.set_field('planting_date', "05-10-2020 12:30:10") # update our campaign events EventFire.update_campaign_events(campaign) # should have one event now fire = EventFire.objects.get() self.assertEquals(5, fire.scheduled.day) self.assertEquals(10, fire.scheduled.month) self.assertEquals(2020, fire.scheduled.year) # account for timezone difference, our org is in UTC+2 self.assertEquals(17 - 2, fire.scheduled.hour) self.assertEquals(self.farmer1, fire.contact) self.assertEquals(planting_reminder, fire.event) self.assertIsNone(fire.fired) # change the date of our date self.farmer1.set_field('planting_date', "06-10-2020 12:30:10") EventFire.update_campaign_events_for_contact(campaign, self.farmer1) fire = EventFire.objects.get() self.assertEquals(6, fire.scheduled.day) self.assertEquals(10, fire.scheduled.month) self.assertEquals(2020, fire.scheduled.year) self.assertEquals(self.farmer1, fire.contact) self.assertEquals(planting_reminder, fire.event) # set it to something invalid self.farmer1.set_field('planting_date', "what?") EventFire.update_campaign_events_for_contact(campaign, self.farmer1) self.assertFalse(EventFire.objects.all()) # now something valid again self.farmer1.set_field('planting_date', "07-10-2020 12:30:10") EventFire.update_campaign_events_for_contact(campaign, self.farmer1) fire = EventFire.objects.get() self.assertEquals(7, fire.scheduled.day) self.assertEquals(10, fire.scheduled.month) self.assertEquals(2020, fire.scheduled.year) self.assertEquals(self.farmer1, fire.contact) self.assertEquals(planting_reminder, fire.event) # create another reminder planting_reminder2 = CampaignEvent.objects.create(campaign=campaign, relative_to=self.planting_date, offset=1, flow=self.reminder2_flow, created_by=self.admin, modified_by=self.admin) self.assertEquals(1, planting_reminder2.abs_offset()) # update the campaign EventFire.update_campaign_events(campaign) # should have two events now, ordered by date events = EventFire.objects.all() self.assertEquals(planting_reminder, events[0].event) self.assertEquals(7, events[0].scheduled.day) self.assertEquals(planting_reminder2, events[1].event) self.assertEquals(8, events[1].scheduled.day) # mark one of the events as inactive planting_reminder2.is_active = False planting_reminder2.save() # update the campaign EventFire.update_campaign_events(campaign) # back to only one event event = EventFire.objects.get() self.assertEquals(planting_reminder, event.event) self.assertEquals(7, event.scheduled.day) # update our date self.farmer1.set_field('planting_date', '09-10-2020 12:30') # should have updated event = EventFire.objects.get() self.assertEquals(planting_reminder, event.event) self.assertEquals(9, event.scheduled.day) # let's remove our contact field ContactField.hide_field(self.org, 'planting_date') # shouldn't have anything scheduled self.assertFalse(EventFire.objects.all()) # add it back in ContactField.get_or_create(self.org, 'planting_date', "planting Date") # should be back! event = EventFire.objects.get() self.assertEquals(planting_reminder, event.event) self.assertEquals(9, event.scheduled.day) # try firing the event event.fire() # should have one flow run now run = FlowRun.objects.get() self.assertEquals(event.contact, run.contact)
def test_scheduling(self): campaign = Campaign.create(self.org, self.admin, "Planting Reminders", self.farmers) self.assertEquals("Planting Reminders", unicode(campaign)) # create a reminder for our first planting event planting_reminder = CampaignEvent.create_flow_event(self.org, self.admin, campaign, relative_to=self.planting_date, offset=0, unit='D', flow=self.reminder_flow, delivery_hour=17) self.assertEquals("Planting Date == 0 -> Color Flow", unicode(planting_reminder)) # schedule our reminders EventFire.update_campaign_events(campaign) # we should haven't any event fires created, since neither of our farmers have a planting date self.assertEquals(0, EventFire.objects.all().count()) # ok, set a planting date on one of our contacts self.farmer1.set_field('planting_date', "05-10-2020 12:30:10") # update our campaign events EventFire.update_campaign_events(campaign) # should have one event now fire = EventFire.objects.get() self.assertEquals(5, fire.scheduled.day) self.assertEquals(10, fire.scheduled.month) self.assertEquals(2020, fire.scheduled.year) # account for timezone difference, our org is in UTC+2 self.assertEquals(17 - 2, fire.scheduled.hour) self.assertEquals(self.farmer1, fire.contact) self.assertEquals(planting_reminder, fire.event) self.assertIsNone(fire.fired) # change the date of our date self.farmer1.set_field('planting_date', "06-10-2020 12:30:10") EventFire.update_campaign_events_for_contact(campaign, self.farmer1) fire = EventFire.objects.get() self.assertEquals(6, fire.scheduled.day) self.assertEquals(10, fire.scheduled.month) self.assertEquals(2020, fire.scheduled.year) self.assertEquals(self.farmer1, fire.contact) self.assertEquals(planting_reminder, fire.event) # set it to something invalid self.farmer1.set_field('planting_date', "what?") EventFire.update_campaign_events_for_contact(campaign, self.farmer1) self.assertFalse(EventFire.objects.all()) # now something valid again self.farmer1.set_field('planting_date', "07-10-2020 12:30:10") EventFire.update_campaign_events_for_contact(campaign, self.farmer1) fire = EventFire.objects.get() self.assertEquals(7, fire.scheduled.day) self.assertEquals(10, fire.scheduled.month) self.assertEquals(2020, fire.scheduled.year) self.assertEquals(self.farmer1, fire.contact) self.assertEquals(planting_reminder, fire.event) # create another reminder planting_reminder2 = CampaignEvent.create_flow_event(self.org, self.admin, campaign, relative_to=self.planting_date, offset=1, unit='D', flow=self.reminder2_flow) self.assertEquals(1, planting_reminder2.abs_offset()) # update the campaign EventFire.update_campaign_events(campaign) # should have two events now, ordered by date events = EventFire.objects.all() self.assertEquals(planting_reminder, events[0].event) self.assertEquals(7, events[0].scheduled.day) self.assertEquals(planting_reminder2, events[1].event) self.assertEquals(8, events[1].scheduled.day) # mark one of the events as inactive planting_reminder2.is_active = False planting_reminder2.save() # update the campaign EventFire.update_campaign_events(campaign) # back to only one event event = EventFire.objects.get() self.assertEquals(planting_reminder, event.event) self.assertEquals(7, event.scheduled.day) # update our date self.farmer1.set_field('planting_date', '09-10-2020 12:30') # should have updated event = EventFire.objects.get() self.assertEquals(planting_reminder, event.event) self.assertEquals(9, event.scheduled.day) # let's remove our contact field ContactField.hide_field(self.org, 'planting_date') # shouldn't have anything scheduled self.assertFalse(EventFire.objects.all()) # add it back in ContactField.get_or_create(self.org, 'planting_date', "planting Date") # should be back! event = EventFire.objects.get() self.assertEquals(planting_reminder, event.event) self.assertEquals(9, event.scheduled.day) # change our fire date to sometimein the past so it gets triggered event.scheduled = timezone.now() - timedelta(hours=1) event.save() # schedule our events to fire check_campaigns_task() # should have one flow run now run = FlowRun.objects.get() self.assertEquals(event.contact, run.contact)