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
