1 files changed, 81 insertions, 0 deletions
diff --git a/97suifangqa/apps/recommend/tools.py b/97suifangqa/apps/recommend/tools.py
new file mode 100644
index 0000000..3b98c1f
--- /dev/null
+++ b/97suifangqa/apps/recommend/tools.py
@@ -0,0 +1,81 @@
+# -*- coding: utf-8 -*-
+#
+# tools for apps/recommend
+#
+
+from django.db.models import Count
+
+from recommend import models as rm
+
+import re
+
+
+def make_tag(ids=[], tag='tag', sep='_'):                       # {{{
+    """
+    make tag by using given list of ids
+
+    if 'ids' is a list of integers, then sort them by magnitude
+    """
+    if isinstance(ids, tuple):
+        ids = list(ids)
+    # check list element type
+    all_ints = all(isinstance(item, int) for item in ids)
+    if all_ints:
+        # sort ints by magnitude
+        ids.sort()
+    #
+    tag_name = tag
+    for id in ids:
+        tag_name = '%s%s%s' % (tag_name, sep, id)
+    return tag_name
+# }}}
+
+
+def get_research_config(atoms=[]):                              # {{{
+    """
+    return the found ResearchConfig object
+    by filtering on the given atoms list
+    """
+    if not atoms:
+        return False
+    # convert id list to obj list
+    if isinstance(atoms[0], int):
+        atoms = [rm.ResearchAtom.objects.get(id=id) for id in atoms]
+    qs = rm.ResearchConfig.objects.annotate(c=Count('researchAtoms'))\
+            .filter(c=len(atoms))
+    for atom in atoms:
+        qs = qs.filter(researchAtoms=atom)
+    if not qs:
+        return None
+    elif len(qs) == 1:
+        return qs[0]
+    else:
+        return False
+# }}}
+
+
+def make_config_display(atoms=[]):                              # {{{
+    """
+    make a display string for the given config
+    """
+    disp_str = u''
+    if not atoms:
+        return disp_str
+    # convert to list
+    if isinstance(atoms, tuple):
+        atoms = list(atoms)
+    # check list element type
+    all_ints = all(isinstance(item, int) for item in atoms)
+    if all_ints:
+        # sort ints by magnitude and convert to objects list
+        atoms.sort()
+        atoms = [rm.ResearchAtom.objects.get(id=id) for id in atoms]
+    #
+    for atom in atoms:
+        disp_str = '%s%s | ' % (disp_str, atom.display())
+    disp_str = re.sub(r'(^\s*\|\s*)|(\s*\|\s*$)', '', disp_str)
+    #
+    return disp_str
+# }}}
+
+