def test_custom_tag_1(self): # make a new tagset TagSet.objects_by_account(self.account).get_or_create(account=self.account,name="new test tagset") populate_rule_components_for_an_account(self.account) left_side = LeftSide.objects_by_account(self.account).get(display_name="have a new test tagset tag that") self.assertEqual(left_side.query_string_partial, "taggeditem__tag__in=Tag.objects_by_account(self.account).filter(tagset__name='new test tagset',name") self.assertEqualQuerySets(left_side.operators, self._tag_operators ) self.assertEqualQuerySets(left_side.right_side_types, self._text_right_side_types) self.assertEqual(left_side.add_closing_paren, True) return left_side
def _tab_or_manage_tags_redirect(context): request = context["request"] context["all_tagsets"] = TagSet.objects_by_account(request.account).using('default').all() if request.is_ajax(): fragment_html = {"tagset_details" : render_to_string("generic_tags/_manage_tags_tagset_details.html", RequestContext(request,context)),} c = { "fragments":fragment_html, "success": context["success"], } return HttpResponse(simplejson.dumps(c)) else: return HttpResponseRedirect(reverse("generic_tags:manage"))
def test_populate_cleans_up_unused_rules(self): # create a custom tag from generic_tags.models import TagSet ts1 = TagSet.objects_by_account(self.account).get_or_create(account=self.account,name="new test tagset")[0] # repopulate populate_rule_components_for_an_account(self.account) # make sure it's there left_side = LeftSide.objects_by_account(self.account).get(display_name="have a new test tagset tag that") self.assertEqual(left_side.query_string_partial, "taggeditem__tag__in=Tag.objects_by_account(self.account).filter(tagset__name='new test tagset',name") self.assertEqualQuerySets(left_side.operators, self._tag_operators ) self.assertEqualQuerySets(left_side.right_side_types, self._text_right_side_types) # delete it ts1.delete() # repopulate populate_rule_components_for_an_account(self.account) # make sure it's gone self.assertEqual(LeftSide.objects_by_account(self.account).filter(display_name="have a new test tagset tag that").count(), 0)
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)
def tagsets(self): return TagSet.objects_by_account(self.account).all()
def create_default_tagsets(cls, instance, created=None, *args, **kwargs): from generic_tags.models import TagSet if instance and created: TagSet.create_default_tagsets_for_an_account(instance)
def manage(request): section = "admin" all_tagsets = TagSet.objects_by_account(request.account).all() return locals()
def save_tags_and_tagsets(request): success = False data = json.loads(request.POST["data"]) created_tagsets = [] created_tags = [] deleted_tagsets = [] deleted_tags = [] for ts in data["tag_sets"]: tagset = None if ts["db_pk"]: try: tagset = TagSet.objects_by_account(request.account).using('default').get(pk=ts["db_pk"]) tagset.name = ts["name"] except TagSet.DoesNotExist: pass else: tagset, created = TagSet.objects_by_account(request.account).using('default').get_or_create(account=request.account, name=ts["name"]) created_tagsets.append(tagset) if tagset: tagset.page_pk = ts["page_pk"] if ts["is_deleted"]: tagset.delete() deleted_tagsets.append(tagset) else: tagset.order = ts["order"] tagset.save() for t in ts["tags"]: tag = None if t["db_pk"]: try: tag = Tag.objects_by_account(request.account).using('default').get(pk=t["db_pk"]) tag.name = t["name"] tag.tagset = tagset except Tag.DoesNotExist: pass else: tag, created = Tag.objects_by_account(request.account).using('default').get_or_create(account=request.account, name=t["name"], tagset=tagset) try: transaction.commit() except: pass created_tags.append(tag) if tag: tag.page_pk = t["page_pk"] if t["is_deleted"]: tag.delete() deleted_tags.append(tag) else: tag.order = t["order"] try: transaction.commit() except: pass tag.save() ret_dict = {} ret_dict["created_tagsets"] = [{"name":ts.name, "order":ts.order, "db_pk":ts.pk, "page_pk": ts.page_pk} for ts in created_tagsets] ret_dict["created_tags"] = [{"name":t.name, "order":t.order, "db_pk":t.pk, "page_pk": t.page_pk} for t in created_tags] ret_dict["deleted_tagsets"] = [{"name":ts.name, "order":ts.order, "db_pk":ts.pk, "page_pk": ts.page_pk} for ts in deleted_tagsets] ret_dict["deleted_tags"] = [{"name":t.name, "order":t.order, "db_pk":t.pk, "page_pk": t.page_pk} for t in deleted_tags] ret_dict["success"] = True return ret_dict