def run_test(self, config): fail_error = None try: fail_error = None try: reverse_url = reverse(config.url_name, args=config.reverse_args, kwargs=config.reverse_kwargs) except: fail_error = "URL Reversing for '%s' failed." % (config.url_name) if not fail_error: if config.method=="POST": response = config.client.post(reverse_url, config.data) else: response = config.client.get(reverse_url, config.data) if config.verbose: print "%s" % (response.status_code) if not response.status_code == config.status_code: fail_error = "Status code fail. Expected %s. Got %s" %(response.status_code, config.status_code) assert_true(response.status_code == config.status_code) if config.check_title: fail_error = "Page is missing a title. %s" % (reverse_url) assert title_re.search(response.content) != None except: from qi_toolkit.helpers import exception_string exception_string() fail_error = "Unhandled Fail: \n%s" % (exception_string()) return fail_error
def populate_rule_components_for_an_account(account): try: from accounts.models import Account from rules.models import LeftSide, Operator, RightSideType from volunteers import VOLUNTEER_STATII from donors import DONATION_TYPES account = Account.objects.using("default").get(pk=account.pk) """This function performs several actions, and is idempotent. In order, it: - Sets up built-in rule options (like volunteer status) - Sets up data-driven rule options (like custom tag sets) - Cleans up unused rule options """ # print "populating for %s" % account request = Dummy() request.account = account # RightSideTypes all_right_side_types = [] right_type_text = RightSideType.raw_objects.using('default').get_or_create(account=account, name="text")[0] right_type_date = RightSideType.raw_objects.using('default').get_or_create(account=account, name="date")[0] right_type_number = RightSideType.raw_objects.using('default').get_or_create(account=account, name="number")[0] right_type_choices = RightSideType.raw_objects.using('default').get_or_create(account=account,name="choices")[0] all_right_side_types = [right_type_text, right_type_date, right_type_number, right_type_choices] # Operators all_operators = [] operator_contains = Operator.raw_objects.using('default').get_or_create(account=account, display_name="contains" , query_string_partial="__icontains=" , use_filter=True)[0] operator_contains.order = 10 operator_contains.save() operator_does_not_contain = Operator.raw_objects.using('default').get_or_create(account=account, display_name="does not contain" , query_string_partial="__icontains=" , use_filter=False)[0] operator_does_not_contain.order = 20 operator_does_not_contain.save() operator_is = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is" , query_string_partial="=" , use_filter=True)[0] operator_is.order = 30 operator_is.save() operator_is_not = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is not" , query_string_partial="=" , use_filter=False)[0] operator_is_not.order = 40 operator_is_not.save() operator_is_exactly = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is exactly" , query_string_partial="__iexact=" , use_filter=True)[0] operator_is_exactly.order = 50 operator_is_exactly.save() operator_is_not_exactly = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is not exactly" , query_string_partial="__iexact=" , use_filter=False)[0] operator_is_not_exactly.order = 60 operator_is_not_exactly.save() operator_is_on = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is on" , query_string_partial="=" , use_filter=True)[0] operator_is_on.order = 70 operator_is_on.save() operator_is_before = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is before" , query_string_partial="__lt=" , use_filter=True)[0] operator_is_before.order = 80 operator_is_before.save() operator_is_after = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is after" , query_string_partial="__gt=" , use_filter=True)[0] operator_is_after.order = 90 operator_is_after.save() operator_is_equal = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is equal to" , query_string_partial="=" , use_filter=True)[0] operator_is_equal.order = 100 operator_is_equal.save() operator_is_less_than = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is less than" , query_string_partial="__lt=" , use_filter=True)[0] operator_is_less_than.order = 110 operator_is_less_than.save() operator_is_more_than = Operator.raw_objects.using('default').get_or_create(account=account, display_name="is more than" , query_string_partial="__gt=" , use_filter=True)[0] operator_is_more_than.order = 120 operator_is_more_than.save() all_operators = [ operator_is_exactly, operator_is_not_exactly, operator_contains, operator_does_not_contain, operator_is, operator_is_not, operator_is_on, operator_is_before, operator_is_after, operator_is_equal, operator_is_less_than, operator_is_more_than, ] # Helper methods all_left_sides = [] def _add_text_operators(ls): ls.allowed_operators.clear() ls.allowed_operators.add(operator_is_exactly) ls.allowed_operators.add(operator_is_not_exactly) ls.allowed_operators.add(operator_contains) ls.allowed_operators.add(operator_does_not_contain) ls.save() def _add_right_type_text(ls): ls.allowed_right_side_types.clear() ls.allowed_right_side_types.add(right_type_text) ls.save() def _add_operators_and_right_side_text(ls): _add_text_operators(ls) _add_right_type_text(ls) def _add_date_operators(ls): ls.allowed_operators.clear() ls.allowed_operators.add(operator_is_on) ls.allowed_operators.add(operator_is_before) ls.allowed_operators.add(operator_is_after) ls.save() def _add_right_type_date(ls): ls.allowed_right_side_types.clear() ls.allowed_right_side_types.add(right_type_date) ls.save() def _add_operators_and_right_side_date(ls): _add_date_operators(ls) _add_right_type_date(ls) def _add_number_operators(ls): ls.allowed_operators.clear() ls.allowed_operators.add(operator_is_equal) ls.allowed_operators.add(operator_is_less_than) ls.allowed_operators.add(operator_is_more_than) ls.save() def _add_right_type_number(ls): ls.allowed_right_side_types.clear() ls.allowed_right_side_types.add(right_type_number) ls.save() def _add_operators_and_right_side_number(ls): _add_number_operators(ls) _add_right_type_number(ls) def _add_tag_operators(ls): ls.allowed_operators.clear() ls.allowed_operators.add(operator_is_exactly) ls.allowed_operators.add(operator_contains) ls.save() def _add_operators_and_right_side_tag(ls): _add_tag_operators(ls) _add_right_type_text(ls) def _add_choices_operators(ls): ls.allowed_operators.clear() ls.allowed_operators.add(operator_is) ls.allowed_operators.add(operator_is_not) ls.save() def _add_right_type_choices(ls): ls.allowed_right_side_types.clear() ls.allowed_right_side_types.add(right_type_choices) ls.save() def _add_operators_and_right_side_choices(ls): _add_choices_operators(ls) _add_right_type_choices(ls) def _add_to_all_left_sides(ls): all_left_sides.append(ls) def left_side_for_text(**kwargs): if not "display_name" in kwargs or not "query_string_partial" in kwargs: raise Exception, "display_name and query_string_partial not passed!" ls = LeftSide.raw_objects.using('default').get_or_create(**kwargs)[0] _add_operators_and_right_side_text(ls) _add_to_all_left_sides(ls) return ls def left_side_for_date(**kwargs): if not "display_name" in kwargs or not "query_string_partial" in kwargs: raise Exception, "display_name and query_string_partial not passed!" ls = LeftSide.raw_objects.using('default').get_or_create(**kwargs)[0] _add_operators_and_right_side_date(ls) _add_to_all_left_sides(ls) return ls def left_side_for_number(**kwargs): if not "display_name" in kwargs or not "query_string_partial" in kwargs: raise Exception, "display_name and query_string_partial not passed!" ls = LeftSide.raw_objects.using('default').get_or_create(**kwargs)[0] _add_operators_and_right_side_number(ls) _add_to_all_left_sides(ls) return ls def left_side_for_tag(**kwargs): if not "display_name" in kwargs or not "query_string_partial" in kwargs: raise Exception, "display_name and query_string_partial not passed!" ls = LeftSide.raw_objects.using('default').get_or_create(**kwargs)[0] _add_operators_and_right_side_tag(ls) _add_to_all_left_sides(ls) return ls def left_side_for_choices(**kwargs): if not "display_name" in kwargs or not "query_string_partial" in kwargs: raise Exception, "display_name and query_string_partial not passed!" choices = kwargs["choices"] del kwargs["choices"] ls = LeftSide.raw_objects.using('default').get_or_create(**kwargs)[0] if not ls.choices == choices: ls.choices = choices ls.save() _add_operators_and_right_side_choices(ls) _add_to_all_left_sides(ls) return ls # Left sides - built-ins ls = left_side_for_tag (account=account, display_name="have any tag that" ,query_string_partial="taggeditem__tag__name" ) ls.order=10 ls.save() ls = left_side_for_choices(account=account, display_name="volunteer status" ,query_string_partial="volunteer__status" , choices=VOLUNTEER_STATII) ls.order=100 ls.save() ls = left_side_for_date(account=account, display_name="last donation" ,query_string_partial="donor__donation__date" ) ls.order=110 ls.save() # left_side_for_number(account=account, display_name="total donations in the last 12 months" ,query_string_partial="donor__twelvemonth_total" ) # ls.order=120 # ls.save() ls = left_side_for_date(account=account, display_name="last volunteer shift" ,query_string_partial="volunteer__completedshift__date" ) ls.order=130 ls.save() # left_side_for_number(account=account, display_name="total volunteer hours in the last 12 months" ,query_string_partial="volunteer__twelvemonth_total" ) # ls.order=140 # ls.save() ls = left_side_for_date (account=account, display_name="last conversation" ,query_string_partial="conversation__date" ) ls.order=150 ls.save() ls = left_side_for_date (account=account, display_name="birthday" ,query_string_partial="actual_birthday" ) ls.order=170 ls.save() ls = left_side_for_number (account=account, display_name="age" ,query_string_partial="age" ) ls.order=190 ls.save() ls = left_side_for_choices (account=account, display_name="have a donation that" ,query_string_partial="donor__donation__type", choices=DONATION_TYPES ) ls.order=210 ls.save() # Left sides - generateds from generic_tags.models import TagSet i = 0 for ts in TagSet.objects_by_account(account).using('default').all(): i = i+1 ls = left_side_for_tag(account=account, display_name="have a %s tag that" % (ts.name), query_string_partial="taggeditem__tag__in=Tag.objects_by_account(self.account).filter(tagset__name='%s',name" % (ts.name), add_closing_paren=True ) ls.order=20+i ls.save() # Cleanup for rs in RightSideType.objects_by_account(account).using('default').all(): if rs not in all_right_side_types: rs.delete() for o in Operator.objects_by_account(account).using('default').all(): if o not in all_operators: o.delete() for ls in LeftSide.objects_by_account(account).using('default').all(): if ls not in all_left_sides: ls.delete() johnny_cache.invalidate(RightSideType) johnny_cache.invalidate(Operator) johnny_cache.invalidate(LeftSide) except: from django.core.mail import mail_admins from qi_toolkit.helpers import exception_string mail_admins("Exception generating group rules", exception_string(), fail_silently=True)