def _test_prescription_voucher_alert_with_language(self, voucher,
                                                    language_code,
                                                    expected_message):
     self.assertEqual(
         prescription_voucher_alert(CaseReminder(case_id=voucher.case_id),
                                    CaseReminderHandler(default_lang='en'),
                                    FakeRecipient(language_code)),
         expected_message)
    def test_calculate_start_date_with_today_option(self):
        now = datetime.utcnow()

        with create_test_case(self.domain, 'contact', 'test-case') as case:

            reminder = CaseReminderHandler(
                domain=self.domain, use_today_if_start_date_is_blank=True)

            self.assertEqual(
                reminder.get_case_criteria_reminder_start_date_info(case, now),
                (now, True, True))

            reminder.start_date = 'start_date_case_property'
            self.assertEqual(
                reminder.get_case_criteria_reminder_start_date_info(case, now),
                (now, True, True))

            update_case(self.domain, case.case_id,
                        {'start_date_case_property': ''})
            case = CaseAccessors(self.domain).get_case(case.case_id)
            self.assertEqual(
                reminder.get_case_criteria_reminder_start_date_info(case, now),
                (now, True, True))

            update_case(self.domain, case.case_id,
                        {'start_date_case_property': '   '})
            case = CaseAccessors(self.domain).get_case(case.case_id)
            self.assertEqual(
                reminder.get_case_criteria_reminder_start_date_info(case, now),
                (now, True, True))

            update_case(self.domain, case.case_id,
                        {'start_date_case_property': 'abcdefg'})
            case = CaseAccessors(self.domain).get_case(case.case_id)
            self.assertEqual(
                reminder.get_case_criteria_reminder_start_date_info(case, now),
                (now, True, True))

            update_case(self.domain, case.case_id,
                        {'start_date_case_property': '2016-01-32'})
            case = CaseAccessors(self.domain).get_case(case.case_id)
            self.assertEqual(
                reminder.get_case_criteria_reminder_start_date_info(case, now),
                (now, True, True))

            update_case(self.domain, case.case_id,
                        {'start_date_case_property': '2016-01-10'})
            case = CaseAccessors(self.domain).get_case(case.case_id)
            self.assertEqual(
                reminder.get_case_criteria_reminder_start_date_info(case, now),
                (datetime(2016, 1, 10), True, False))

            update_case(self.domain, case.case_id,
                        {'start_date_case_property': '2016-01-12T00:00:00Z'})
            case = CaseAccessors(self.domain).get_case(case.case_id)
            self.assertEqual(
                reminder.get_case_criteria_reminder_start_date_info(case, now),
                (datetime(2016, 1, 12), True, False))
Example #3
0
    def test_recipient_case_owner_location_parent(self):
        parent_location = SQLLocation.objects.create(
            domain=self.domain,
            name='parent test',
            site_code='parent',
            location_type=self.parent_location_type)

        child_location = SQLLocation.objects.create(
            domain=self.domain,
            name='child test',
            site_code='child',
            location_type=self.child_location_type,
            parent=parent_location)

        self.user.set_location(child_location.couch_location)

        with create_test_case(self.domain,
                              'test-case',
                              'test-name',
                              owner_id=self.user.get_id) as case:
            self.assertEqual(case.owner_id, self.user.get_id)
            handler = CaseReminderHandler(
                domain=self.domain,
                recipient=RECIPIENT_CASE_OWNER_LOCATION_PARENT)
            reminder = CaseReminder(domain=self.domain, case_id=case.case_id)

            # Test the recipient is returned correctly
            with patch('corehq.apps.reminders.models.CaseReminder.handler',
                       new=handler):
                self.assertEqual(reminder.recipient, [parent_location])

            # Remove parent location
            parent_location.delete()
            child_location.parent = None
            child_location.save()
            with patch('corehq.apps.reminders.models.CaseReminder.handler',
                       new=handler):
                self.assertIsNone(reminder.recipient)

            # Remove child location
            self.user.unset_location()
            child_location.delete()
            with patch('corehq.apps.reminders.models.CaseReminder.handler',
                       new=handler):
                self.assertIsNone(reminder.recipient)

            # Remove case
            reminder.case_id = None
            with patch('corehq.apps.reminders.models.CaseReminder.handler',
                       new=handler):
                self.assertIsNone(reminder.recipient)
Example #4
0
def add_reminder(request,
                 domain,
                 handler_id=None,
                 template="reminders/partial/add_reminder.html"):

    if handler_id:
        handler = CaseReminderHandler.get(handler_id)
        if handler.doc_type != 'CaseReminderHandler' or handler.domain != domain:
            raise Http404
    else:
        handler = None

    if request.method == "POST":
        reminder_form = CaseReminderForm(request.POST)
        if reminder_form.is_valid():
            if not handler:
                handler = CaseReminderHandler(domain=domain)
                handler.ui_type = UI_SIMPLE_FIXED
            for key, value in reminder_form.cleaned_data.items():
                if (key != "frequency") and (key != "message"):
                    handler[key] = value
            handler.max_iteration_count = REPEAT_SCHEDULE_INDEFINITELY
            handler.schedule_length = reminder_form.cleaned_data["frequency"]
            handler.event_interpretation = EVENT_AS_OFFSET
            handler.events = [
                CaseReminderEvent(
                    day_num=0,
                    fire_time=time(hour=0, minute=0, second=0),
                    message=reminder_form.cleaned_data["message"],
                    callback_timeout_intervals=[])
            ]
            handler.save()
            return HttpResponseRedirect(
                reverse('list_reminders', args=[domain]))
    elif handler:
        initial = {}
        for key in handler.to_json():
            if (key !=
                    "max_iteration_count") and (key != "schedule_length") and (
                        key != "events") and (key != "event_interpretation"):
                initial[key] = handler[key]
        initial["message"] = json.dumps(handler.events[0].message)
        initial["frequency"] = handler.schedule_length
        reminder_form = CaseReminderForm(initial=initial)
    else:
        reminder_form = CaseReminderForm()

    return render(request, template, {
        'reminder_form': reminder_form,
        'domain': domain
    })
Example #5
0
 def create_new_broadcast(self):
     return CaseReminderHandler(
         domain=self.domain,
         nickname='One-time Reminder',
         reminder_type=REMINDER_TYPE_ONE_TIME,
     )
Example #6
0
 def process_schedule_form(self):
     new_handler = CaseReminderHandler(
         use_today_if_start_date_is_blank=False)
     self.schedule_form.save(new_handler)
Example #7
0
 handler = CaseReminderHandler(
     domain=contact.domain,
     reminder_type=reminder_type,
     nickname="One-time Reminder",
     default_lang="xx",
     method=content_type,
     recipient=recipient,
     start_condition_type=ON_DATETIME,
     start_datetime=datetime.utcnow(),
     start_offset=0,
     events=[
         CaseReminderEvent(
             day_num=0,
             fire_time=time(0, 0),
             form_unique_id=form_unique_id
             if content_type == METHOD_SMS_SURVEY else None,
             message={'xx': message} if content_type == METHOD_SMS else {},
             callback_timeout_intervals=[],
         )
     ],
     schedule_length=1,
     event_interpretation=EVENT_AS_OFFSET,
     max_iteration_count=1,
     case_id=case_id,
     user_id=contact.get_id if recipient == RECIPIENT_USER else None,
     sample_id=contact.get_id
     if recipient == RECIPIENT_SURVEY_SAMPLE else None,
     user_group_id=contact.get_id
     if recipient == RECIPIENT_USER_GROUP else None,
     messaging_event_id=logged_event.pk if logged_event else None,
 )
Example #8
0
    def test_cache(self, check_state_mock):
        domain = 'reminder-cache-test'

        # Nothing expected at first
        self.assertEqual(CaseReminderHandler.get_handler_ids(domain), [])

        # Create two reminder definitions of different types
        handler1 = CaseReminderHandler(
            domain=domain,
            reminder_type=REMINDER_TYPE_DEFAULT,
        )
        handler1.save()
        self.addCleanup(handler1.delete)

        handler2 = CaseReminderHandler(
            domain=domain,
            reminder_type=REMINDER_TYPE_ONE_TIME,
        )
        handler2.save()
        self.addCleanup(handler2.delete)

        self.assertEqual(
            CaseReminderHandler.get_handler_ids(
                domain, reminder_type_filter=REMINDER_TYPE_DEFAULT),
            [handler1._id])

        self.assertEqual(
            CaseReminderHandler.get_handler_ids(
                domain, reminder_type_filter=REMINDER_TYPE_ONE_TIME),
            [handler2._id])

        self.assertEqual(sorted(CaseReminderHandler.get_handler_ids(domain)),
                         sorted([handler1._id, handler2._id]))

        # Add another with default type
        handler3 = CaseReminderHandler(
            domain=domain,
            reminder_type=REMINDER_TYPE_DEFAULT,
        )
        handler3.save()
        self.addCleanup(handler3.delete)

        self.assertEqual(
            sorted(
                CaseReminderHandler.get_handler_ids(
                    domain, reminder_type_filter=REMINDER_TYPE_DEFAULT)),
            sorted([handler1._id, handler3._id]))

        self.assertEqual(
            CaseReminderHandler.get_handler_ids(
                domain, reminder_type_filter=REMINDER_TYPE_ONE_TIME),
            [handler2._id])

        self.assertEqual(sorted(CaseReminderHandler.get_handler_ids(domain)),
                         sorted([handler1._id, handler2._id, handler3._id]))

        # Retire the one-time reminder
        handler2.retire()

        self.assertEqual(
            sorted(
                CaseReminderHandler.get_handler_ids(
                    domain, reminder_type_filter=REMINDER_TYPE_DEFAULT)),
            sorted([handler1._id, handler3._id]))

        self.assertEqual(
            CaseReminderHandler.get_handler_ids(
                domain, reminder_type_filter=REMINDER_TYPE_ONE_TIME), [])

        self.assertEqual(sorted(CaseReminderHandler.get_handler_ids(domain)),
                         sorted([handler1._id, handler3._id]))
    def handle(self, *args, **options):
        if len(args) == 0 or len(args) > 1:
            raise CommandError(
                "Usage: manage.py create_definitions_2012_07 <domain>")

        domain = args[0]
        case_type = "participant"
        times_of_day = {
            "0600": time(6, 0, 0),
            "0900": time(9, 0, 0),
            "1700": time(17, 0, 0),
            "2100": time(21, 0, 0)
        }
        days = [
            "Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday",
            "Sunday"
        ]

        for time_code, actual_time in times_of_day.items():
            c = CaseReminderHandler(domain=domain,
                                    case_type=case_type,
                                    nickname="Group 1, @ " + time_code,
                                    default_lang="en",
                                    method="survey",
                                    ui_type="COMPLEX",
                                    recipient="CASE",
                                    start_property="send_normal_message",
                                    start_value=time_code,
                                    start_date="normal_result_message_date",
                                    start_offset=0,
                                    start_match_type="EXACT",
                                    events=[
                                        CaseReminderEvent(
                                            day_num=0,
                                            fire_time=actual_time,
                                            message={},
                                            callback_timeout_intervals=[],
                                            form_unique_id="")
                                    ],
                                    schedule_length=1,
                                    event_interpretation="SCHEDULE",
                                    max_iteration_count=1,
                                    until=None)
            c.save()
            print "Created " + c.nickname

            for day in days:
                c = CaseReminderHandler(
                    domain=domain,
                    case_type=case_type,
                    nickname="Group 2 - 4, " + day + ", @ " + time_code,
                    default_lang="en",
                    method="survey",
                    ui_type="COMPLEX",
                    recipient="CASE",
                    start_property="send_" + day.lower(),
                    start_value=time_code,
                    start_date=day.lower() + "_date",
                    start_offset=0,
                    start_match_type="EXACT",
                    events=[
                        CaseReminderEvent(day_num=0,
                                          fire_time=actual_time,
                                          message={},
                                          callback_timeout_intervals=[],
                                          form_unique_id="")
                    ],
                    schedule_length=1,
                    event_interpretation="SCHEDULE",
                    max_iteration_count=1,
                    until=None)
                c.save()
                print "Created " + c.nickname
Example #10
0
    def setUp(self):
        super(KooKooTestCase, self).setUp()
        self.ivr_backend = MobileBackend(
            _id="MOBILE_BACKEND_KOOKOO",
            outbound_module="corehq.apps.kookoo.api",
            outbound_params={
                "is_test": True,
                "api_key": "xyz"
            },
        )
        self.ivr_backend.save()

        self.user1 = self.create_mobile_worker("user1",
                                               "123",
                                               "91001",
                                               save_vn=False)
        self.user2 = self.create_mobile_worker("user2",
                                               "123",
                                               "91002",
                                               save_vn=False)
        self.create_group("group1", [self.user1, self.user2])

        dirname = os.path.dirname(os.path.abspath(__file__))
        self.load_app("app1.json", dirname)
        self.load_app("app2.json", dirname)

        self.reminder1 = CaseReminderHandler(
            domain=self.domain,
            active=True,
            case_type="participant",
            method=METHOD_IVR_SURVEY,
            recipient=RECIPIENT_CASE,
            sample_id=None,
            user_group_id=None,
            user_id=None,
            case_id=None,
            reminder_type=REMINDER_TYPE_DEFAULT,
            submit_partial_forms=True,
            include_case_side_effects=False,
            max_question_retries=5,
            start_condition_type=CASE_CRITERIA,
            start_property="name",
            start_value="case1",
            start_date=None,
            start_offset=0,
            start_match_type=MATCH_EXACT,
            events=[
                CaseReminderEvent(
                    day_num=0,
                    fire_time=time(12, 0),
                    fire_time_type=FIRE_TIME_DEFAULT,
                    callback_timeout_intervals=[30],
                    form_unique_id=self.apps[0].modules[0].forms[0].unique_id,
                ),
                CaseReminderEvent(
                    day_num=0,
                    fire_time=time(13, 0),
                    fire_time_type=FIRE_TIME_DEFAULT,
                    callback_timeout_intervals=[30],
                    form_unique_id=self.apps[0].modules[0].forms[1].unique_id,
                ),
            ],
            schedule_length=1,
            event_interpretation=EVENT_AS_SCHEDULE,
            max_iteration_count=7,
            until=None,
            force_surveys_to_use_triggered_case=False,
        )
        self.reminder1.save()

        self.reminder2 = CaseReminderHandler(
            domain=self.domain,
            active=True,
            case_type="participant",
            method=METHOD_IVR_SURVEY,
            recipient=RECIPIENT_OWNER,
            sample_id=None,
            user_group_id=None,
            user_id=None,
            case_id=None,
            reminder_type=REMINDER_TYPE_DEFAULT,
            submit_partial_forms=True,
            include_case_side_effects=True,
            max_question_retries=5,
            start_condition_type=CASE_CRITERIA,
            start_property="name",
            start_value="case2",
            start_date=None,
            start_offset=0,
            start_match_type=MATCH_EXACT,
            events=[
                CaseReminderEvent(
                    day_num=0,
                    fire_time=time(12, 0),
                    fire_time_type=FIRE_TIME_DEFAULT,
                    callback_timeout_intervals=[30, 30],
                    form_unique_id=self.apps[1].modules[0].forms[0].unique_id,
                ),
            ],
            schedule_length=1,
            event_interpretation=EVENT_AS_SCHEDULE,
            max_iteration_count=7,
            until=None,
            force_surveys_to_use_triggered_case=False,
        )
        self.reminder2.save()
Example #11
0
def add_survey(request, domain, survey_id=None):
    survey = None

    if survey_id is not None:
        survey = Survey.get(survey_id)

    if request.method == "POST":
        form = SurveyForm(request.POST)
        if form.is_valid():
            name = form.cleaned_data.get("name")
            waves = form.cleaned_data.get("waves")
            followups = form.cleaned_data.get("followups")
            samples = form.cleaned_data.get("samples")
            send_automatically = form.cleaned_data.get("send_automatically")
            send_followup = form.cleaned_data.get("send_followup")

            sample_data = {}
            for sample in samples:
                sample_data[sample["sample_id"]] = sample

            if send_followup:
                timeout_intervals = [
                    int(followup["interval"]) * 1440 for followup in followups
                ]
            else:
                timeout_intervals = []

            timeout_duration = sum(timeout_intervals) / 1440
            final_timeout = lambda wave: [(
                (wave.end_date - wave.date).days - timeout_duration) * 1440]

            if survey is None:
                wave_list = []
                for wave in waves:
                    wave_list.append(
                        SurveyWave(
                            date=parse(wave["date"]).date(),
                            time=parse(wave["time"]).time(),
                            end_date=parse(wave["end_date"]).date(),
                            form_id=wave["form_id"],
                            reminder_definitions={},
                            delegation_tasks={},
                        ))

                if send_automatically:
                    for wave in wave_list:
                        for sample in samples:
                            if sample["method"] == "SMS":
                                handler = CaseReminderHandler(
                                    domain=domain,
                                    nickname="Survey '%s'" % name,
                                    default_lang="en",
                                    method="survey",
                                    recipient=RECIPIENT_SURVEY_SAMPLE,
                                    start_condition_type=ON_DATETIME,
                                    start_datetime=datetime.combine(
                                        wave.date, time(0, 0)),
                                    start_offset=0,
                                    events=[
                                        CaseReminderEvent(
                                            day_num=0,
                                            fire_time=wave.time,
                                            form_unique_id=wave.form_id,
                                            callback_timeout_intervals=
                                            timeout_intervals +
                                            final_timeout(wave),
                                        )
                                    ],
                                    schedule_length=1,
                                    event_interpretation=EVENT_AS_SCHEDULE,
                                    max_iteration_count=1,
                                    sample_id=sample["sample_id"],
                                    survey_incentive=sample["incentive"],
                                    submit_partial_forms=True,
                                )
                                handler.save()
                                wave.reminder_definitions[
                                    sample["sample_id"]] = handler._id

                survey = Survey(domain=domain,
                                name=name,
                                waves=wave_list,
                                followups=followups,
                                samples=samples,
                                send_automatically=send_automatically,
                                send_followup=send_followup)
            else:
                current_waves = survey.waves
                survey.waves = []
                unchanged_wave_json = []

                # Keep any waves that didn't change in case the surveys are in progress
                for wave in current_waves:
                    for wave_json in waves:
                        parsed_date = parse(wave_json["date"]).date()
                        parsed_time = parse(wave_json["time"]).time()
                        if parsed_date == wave.date and parsed_time == wave.time and wave_json[
                                "form_id"] == wave.form_id:
                            wave.end_date = parse(wave_json["end_date"]).date()
                            survey.waves.append(wave)
                            unchanged_wave_json.append(wave_json)
                            continue

                for wave in survey.waves:
                    current_waves.remove(wave)

                for wave_json in unchanged_wave_json:
                    waves.remove(wave_json)

                # Retire reminder definitions / close delegation tasks for old waves
                for wave in current_waves:
                    for sample_id, handler_id in wave.reminder_definitions.items(
                    ):
                        handler = CaseReminderHandler.get(handler_id)
                        handler.retire()
                    for sample_id, delegation_data in wave.delegation_tasks.items(
                    ):
                        for case_id, delegation_case_id in delegation_data.items(
                        ):
                            close_task(domain, delegation_case_id,
                                       request.couch_user.get_id)

                # Add in new waves
                for wave_json in waves:
                    survey.waves.append(
                        SurveyWave(
                            date=parse(wave_json["date"]).date(),
                            time=parse(wave_json["time"]).time(),
                            end_date=parse(wave_json["end_date"]).date(),
                            form_id=wave_json["form_id"],
                            reminder_definitions={},
                            delegation_tasks={},
                        ))

                # Retire reminder definitions that are no longer needed
                if send_automatically:
                    new_sample_ids = [
                        sample_json["sample_id"] for sample_json in samples
                        if sample_json["method"] == "SMS"
                    ]
                else:
                    new_sample_ids = []

                for wave in survey.waves:
                    for sample_id, handler_id in wave.reminder_definitions.items(
                    ):
                        if sample_id not in new_sample_ids:
                            handler = CaseReminderHandler.get(handler_id)
                            handler.retire()
                            del wave.reminder_definitions[sample_id]

                # Update existing reminder definitions
                for wave in survey.waves:
                    for sample_id, handler_id in wave.reminder_definitions.items(
                    ):
                        handler = CaseReminderHandler.get(handler_id)
                        handler.events[
                            0].callback_timeout_intervals = timeout_intervals + final_timeout(
                                wave)
                        handler.nickname = "Survey '%s'" % name
                        handler.survey_incentive = sample_data[sample_id][
                            "incentive"]
                        handler.save()

                # Create additional reminder definitions as necessary
                for wave in survey.waves:
                    for sample_id in new_sample_ids:
                        if sample_id not in wave.reminder_definitions:
                            handler = CaseReminderHandler(
                                domain=domain,
                                nickname="Survey '%s'" % name,
                                default_lang="en",
                                method="survey",
                                recipient=RECIPIENT_SURVEY_SAMPLE,
                                start_condition_type=ON_DATETIME,
                                start_datetime=datetime.combine(
                                    wave.date, time(0, 0)),
                                start_offset=0,
                                events=[
                                    CaseReminderEvent(
                                        day_num=0,
                                        fire_time=wave.time,
                                        form_unique_id=wave.form_id,
                                        callback_timeout_intervals=
                                        timeout_intervals +
                                        final_timeout(wave),
                                    )
                                ],
                                schedule_length=1,
                                event_interpretation=EVENT_AS_SCHEDULE,
                                max_iteration_count=1,
                                sample_id=sample_id,
                                survey_incentive=sample_data[sample_id]
                                ["incentive"],
                                submit_partial_forms=True,
                            )
                            handler.save()
                            wave.reminder_definitions[sample_id] = handler._id

                # Set the rest of the survey info
                survey.name = name
                survey.followups = followups
                survey.samples = samples
                survey.send_automatically = send_automatically
                survey.send_followup = send_followup

            # Sort the questionnaire waves by date and time
            survey.waves = sorted(
                survey.waves,
                key=lambda wave: datetime.combine(wave.date, wave.time))

            # Create / Close delegation tasks as necessary for samples with method "CATI"
            survey.update_delegation_tasks(request.couch_user.get_id)

            survey.save()
            return HttpResponseRedirect(reverse("survey_list", args=[domain]))
    else:
        initial = {}
        if survey is not None:
            waves = []
            samples = [
                SurveySample.get(sample["sample_id"])
                for sample in survey.samples
            ]
            utcnow = datetime.utcnow()
            for wave in survey.waves:
                wave_json = {
                    "date": str(wave.date),
                    "form_id": wave.form_id,
                    "time": str(wave.time),
                    "ignore": wave.has_started(survey),
                    "end_date": str(wave.end_date),
                }

                waves.append(wave_json)

            initial["name"] = survey.name
            initial["waves"] = waves
            initial["followups"] = survey.followups
            initial["samples"] = survey.samples
            initial["send_automatically"] = survey.send_automatically
            initial["send_followup"] = survey.send_followup

        form = SurveyForm(initial=initial)

    form_list = get_form_list(domain)
    form_list.insert(0, {"code": "--choose--", "name": "-- Choose --"})
    sample_list = get_sample_list(domain)
    sample_list.insert(0, {"code": "--choose--", "name": "-- Choose --"})

    context = {
        "domain": domain,
        "survey_id": survey_id,
        "form": form,
        "form_list": form_list,
        "sample_list": sample_list,
        "method_list": SURVEY_METHOD_LIST,
        "user_list": CommCareUser.by_domain(domain),
        "started": survey.has_started() if survey is not None else False,
    }
    return render(request, "reminders/partial/add_survey.html", context)
Example #12
0
def add_complex_reminder_schedule(request, domain, handler_id=None):
    if handler_id:
        h = CaseReminderHandler.get(handler_id)
        if h.doc_type != 'CaseReminderHandler' or h.domain != domain:
            raise Http404
    else:
        h = None

    form_list = get_form_list(domain)
    sample_list = get_sample_list(domain)

    if request.method == "POST":
        form = ComplexCaseReminderForm(request.POST)
        if form.is_valid():
            if h is None:
                h = CaseReminderHandler(domain=domain)
                h.ui_type = UI_COMPLEX
            else:
                if h.start_condition_type != form.cleaned_data[
                        "start_condition_type"]:
                    for reminder in h.get_reminders():
                        reminder.retire()
            h.active = form.cleaned_data["active"]
            h.case_type = form.cleaned_data["case_type"]
            h.nickname = form.cleaned_data["nickname"]
            h.default_lang = form.cleaned_data["default_lang"]
            h.method = form.cleaned_data["method"]
            h.recipient = form.cleaned_data["recipient"]
            h.start_property = form.cleaned_data["start_property"]
            h.start_value = form.cleaned_data["start_value"]
            h.start_date = form.cleaned_data["start_date"]
            h.start_offset = form.cleaned_data["start_offset"]
            h.start_match_type = form.cleaned_data["start_match_type"]
            h.schedule_length = form.cleaned_data["schedule_length"]
            h.event_interpretation = form.cleaned_data["event_interpretation"]
            h.max_iteration_count = form.cleaned_data["max_iteration_count"]
            h.until = form.cleaned_data["until"]
            h.events = form.cleaned_data["events"]
            h.submit_partial_forms = form.cleaned_data["submit_partial_forms"]
            h.include_case_side_effects = form.cleaned_data[
                "include_case_side_effects"]
            h.ui_frequency = form.cleaned_data["frequency"]
            h.start_condition_type = form.cleaned_data["start_condition_type"]
            h.max_question_retries = form.cleaned_data["max_question_retries"]
            h.recipient_case_match_property = form.cleaned_data[
                "recipient_case_match_property"]
            h.recipient_case_match_type = form.cleaned_data[
                "recipient_case_match_type"]
            h.recipient_case_match_value = form.cleaned_data[
                "recipient_case_match_value"]
            if form.cleaned_data["start_condition_type"] == "ON_DATETIME":
                dt = parse(form.cleaned_data["start_datetime_date"]).date()
                tm = parse(form.cleaned_data["start_datetime_time"]).time()
                h.start_datetime = datetime.combine(dt, tm)
            else:
                h.start_datetime = None
            h.sample_id = form.cleaned_data["sample_id"]
            h.save()
            return HttpResponseRedirect(
                reverse('list_reminders', args=[domain]))
    else:
        if h is not None:
            initial = {
                "active":
                h.active,
                "case_type":
                h.case_type,
                "nickname":
                h.nickname,
                "default_lang":
                h.default_lang,
                "method":
                h.method,
                "recipient":
                h.recipient,
                "start_property":
                h.start_property,
                "start_value":
                h.start_value,
                "start_date":
                h.start_date,
                "start_match_type":
                h.start_match_type,
                "start_offset":
                h.start_offset,
                "schedule_length":
                h.schedule_length,
                "event_interpretation":
                h.event_interpretation,
                "max_iteration_count":
                h.max_iteration_count,
                "until":
                h.until,
                "events":
                h.events,
                "submit_partial_forms":
                h.submit_partial_forms,
                "include_case_side_effects":
                h.include_case_side_effects,
                "start_condition_type":
                h.start_condition_type,
                "start_datetime_date":
                str(h.start_datetime.date()) if isinstance(
                    h.start_datetime, datetime) else None,
                "start_datetime_time":
                str(h.start_datetime.time()) if isinstance(
                    h.start_datetime, datetime) else None,
                "frequency":
                h.ui_frequency,
                "sample_id":
                h.sample_id,
                "use_until":
                "Y" if h.until is not None else "N",
                "max_question_retries":
                h.max_question_retries,
                "recipient_case_match_property":
                h.recipient_case_match_property,
                "recipient_case_match_type":
                h.recipient_case_match_type,
                "recipient_case_match_value":
                h.recipient_case_match_value,
            }
        else:
            initial = {
                "events": [
                    CaseReminderEvent(day_num=0,
                                      fire_time=time(0, 0),
                                      message={"": ""},
                                      callback_timeout_intervals=[],
                                      form_unique_id=None)
                ],
                "use_until":
                "N",
                "max_question_retries":
                QUESTION_RETRY_CHOICES[-1],
                "active":
                True,
            }

        form = ComplexCaseReminderForm(initial=initial)

    return render(
        request, "reminders/partial/add_complex_reminder.html", {
            "domain": domain,
            "form": form,
            "form_list": form_list,
            "handler_id": handler_id,
            "sample_list": sample_list,
        })
Example #13
0
    def test_host_case_owner_location(self):
        parent_location = SQLLocation.objects.create(
            domain=self.domain,
            name='parent test',
            site_code='parent',
            location_type=self.parent_location_type)
        self.addCleanup(parent_location.delete)

        child_location = SQLLocation.objects.create(
            domain=self.domain,
            name='child test',
            site_code='child',
            location_type=self.child_location_type,
            parent=parent_location)
        self.addCleanup(child_location.delete)

        with create_test_case(self.domain, 'test-extension-case',
                              'name') as extension_case:
            with create_test_case(self.domain, 'test-host-case',
                                  'name') as host_case:

                update_case(
                    self.domain,
                    host_case.case_id,
                    case_properties={'owner_id': child_location.location_id})
                set_parent_case(self.domain,
                                extension_case,
                                host_case,
                                relationship='extension')

                handler1 = CaseReminderHandler(
                    domain=self.domain, recipient='HOST_CASE_OWNER_LOCATION')
                handler2 = CaseReminderHandler(
                    domain=self.domain,
                    recipient='HOST_CASE_OWNER_LOCATION_PARENT')
                reminder = CaseReminder(domain=self.domain,
                                        case_id=extension_case.case_id)

                # Test the recipients are returned correctly
                with patch('corehq.apps.reminders.models.CaseReminder.handler',
                           new=handler1):
                    self.assertEqual(reminder.recipient, [child_location])

                with patch('corehq.apps.reminders.models.CaseReminder.handler',
                           new=handler2):
                    self.assertEqual(reminder.recipient, [parent_location])

                # Remove parent location reference
                child_location.parent = None
                child_location.save()
                with patch('corehq.apps.reminders.models.CaseReminder.handler',
                           new=handler2):
                    self.assertIsNone(reminder.recipient)

                # Test location that does not exist
                update_case(self.domain,
                            host_case.case_id,
                            case_properties={'owner_id': 'does-not-exist'})
                with patch('corehq.apps.reminders.models.CaseReminder.handler',
                           new=handler1):
                    self.assertIsNone(reminder.recipient)

                # Test on a case that is not an extension case
                reminder.case_id = host_case.case_id
                with patch('corehq.apps.reminders.models.CaseReminder.handler',
                           new=handler1):
                    self.assertIsNone(reminder.recipient)

                # Test on a blank case id
                reminder.case_id = None
                with patch('corehq.apps.reminders.models.CaseReminder.handler',
                           new=handler1):
                    self.assertIsNone(reminder.recipient)
Example #14
0
    def handle(self, *args, **options):
        if len(args) == 0 or len(args) > 1:
            raise CommandError(
                "Usage: manage.py create_definitions_2012_04 <domain>")

        domain = args[0]
        case_type = "patient"
        times_of_day = {
            "0000": time(0, 0, 0),
            "0600": time(6, 0, 0),
            "0700": time(7, 0, 0),
            "0800": time(8, 0, 0),
            "1100": time(11, 0, 0),
            "1200": time(12, 0, 0),
            "1900": time(19, 0, 0),
            "2000": time(20, 0, 0),
            "2100": time(21, 0, 0),
        }
        message = {
            "xx": "{case.personal_message}",
            "en": "Please call back to this number."
        }
        default_language = "en"
        callback_timeouts = [60, 30]
        day_list = [0, 1, 2, 3, 4, 5, 6]

        # Create reminder definitions

        for time_code, actual_time in times_of_day.items():

            time_decription = time_code[:2] + ":" + time_code[2:]

            # Daily for 8 weeks

            CaseReminderHandler(
                domain=domain,
                case_type=case_type,
                nickname="Daily, 8 Weeks @ " + time_decription,
                default_lang=default_language,
                method="callback",
                recipient=RECIPIENT_CASE,
                start_property="daily_schedule",
                start_value=time_code,
                start_date="start_date",
                start_offset=0,
                start_match_type=MATCH_EXACT,
                events=[
                    CaseReminderEvent(
                        day_num=0,
                        fire_time=actual_time,
                        message=message,
                        callback_timeout_intervals=callback_timeouts)
                ],
                schedule_length=1,
                event_interpretation=EVENT_AS_SCHEDULE,
                max_iteration_count=56,
                until=None).save()

            # Daily, indefinitely

            CaseReminderHandler(
                domain=domain,
                case_type=case_type,
                nickname="Daily @ " + time_decription,
                default_lang=default_language,
                method="callback",
                recipient=RECIPIENT_CASE,
                start_property="daily_schedule",
                start_value="indefinite_" + time_code,
                start_date="start_date",
                start_offset=0,
                start_match_type=MATCH_EXACT,
                events=[
                    CaseReminderEvent(
                        day_num=0,
                        fire_time=actual_time,
                        message=message,
                        callback_timeout_intervals=callback_timeouts)
                ],
                schedule_length=1,
                event_interpretation=EVENT_AS_SCHEDULE,
                max_iteration_count=-1,
                until="stop_date").save()

            for day in day_list:

                # 3 per week schedules

                CaseReminderHandler(
                    domain=domain,
                    case_type=case_type,
                    nickname="Three Per Week @ " + time_decription,
                    default_lang=default_language,
                    method="callback",
                    recipient=RECIPIENT_CASE,
                    start_property="tpw_schedule",
                    start_value="^({0}\d\d|\d{0}\d|\d\d{0})_".format(day) +
                    time_code + "$",
                    start_date="start_date",
                    start_offset=56,
                    start_match_type=MATCH_REGEX,
                    events=[
                        CaseReminderEvent(
                            day_num=day,
                            fire_time=actual_time,
                            message=message,
                            callback_timeout_intervals=callback_timeouts)
                    ],
                    schedule_length=7,
                    event_interpretation=EVENT_AS_SCHEDULE,
                    max_iteration_count=8,
                    until=None).save()

                # 1 per week schedules

                CaseReminderHandler(
                    domain=domain,
                    case_type=case_type,
                    nickname="One Per Week @ " + time_decription,
                    default_lang=default_language,
                    method="callback",
                    recipient=RECIPIENT_CASE,
                    start_property="opw_schedule",
                    start_value="{0}_".format(day) + time_code,
                    start_date="start_date",
                    start_offset=112,
                    start_match_type=MATCH_EXACT,
                    events=[
                        CaseReminderEvent(
                            day_num=day,
                            fire_time=actual_time,
                            message=message,
                            callback_timeout_intervals=callback_timeouts)
                    ],
                    schedule_length=7,
                    event_interpretation=EVENT_AS_SCHEDULE,
                    max_iteration_count=32,
                    until=None).save()