From: Eevee <git@veekun.com>
Date: Sun, 6 Dec 2009 10:02:12 +0000 (-0800)
Subject: Added support for querying by user.
X-Git-Url: http://git.veekun.com/zzz-floof.git/commitdiff_plain/e0b6c733c6bea7ff31b9361650fb8534b9433597

Added support for querying by user.
---

diff --git a/floof/controllers/search.py b/floof/controllers/search.py
index f87ddae..720ca67 100644
--- a/floof/controllers/search.py
+++ b/floof/controllers/search.py
@@ -5,7 +5,7 @@ from pylons.controllers.util import abort, redirect
 from pylons import url
 
 from floof.lib.base import BaseController, render
-from floof.lib.search import do_search
+from floof.lib.search import parse
 
 log = logging.getLogger(__name__)
 
@@ -20,7 +20,7 @@ class SearchController(BaseController):
             return self.save()
 
         c.query = request.params.get('query', '')
-        c.artwork = do_search(c.query)
+        c.artwork = parse(c.query).all()
         return render('/index.mako')
 
     # TODO: login required
diff --git a/floof/lib/search.py b/floof/lib/search.py
index 37d762d..8874824 100644
--- a/floof/lib/search.py
+++ b/floof/lib/search.py
@@ -1,43 +1,51 @@
-from floof.model import Art, Tag, TagText
+from floof.model import Art, ArtUser, ArtUserType, Tag, TagText, User
 
-def do_search(query):
-    tags = query.split()
+def parse(search_string):
+    """Parses a search query, and returns a query object on Art.
 
-    tagtexts = TagText.query.filter(TagText.text.in_(tags))
-    tagtext_ids = [_.id for _ in tagtexts]
+    Queries can contain:
+    - Regular tags: foo
+    - User relations: by:kalu, of:eevee, for:ootachi
 
-    # Fetch art that has all the tags
-    artwork = Art.query.join(Tag) \
-                   .filter(Tag.tagtext_id.in_(tagtext_ids)) \
-                   .all()
-    return artwork
+    Later:
+    - Negative versions of anything above: -by:eevee, -dongs
+    """
 
+    # XXX doesn't do negative querying yet.
+    # XXX could use some sane limits.
 
+    # We'll be building this as we go.
+    q = Art.query
 
+    terms = search_string.split()
+    for tag in terms:
+        if ':' in tag:
+            # This is a special tag; at the moment, by/for/of to indicate
+            # related users
+            prefix, tag = tag.split(':', 1)
 
+            # XXX what to do if this fails?  abort?  return empty query?
+            target_user = User.get_by(name=tag)
 
-
-# unfinished stuff
-def parse(query):
-    words = query.split()
-
-    tags = []
-    for word in words:
-        components = word.split(':')
-        if len(components) == 1:
-            # tags are plain.
-            tags.append(word)
-        elif components[0] == "rating":
-            if components[1].isnumeric():
-                score = int(components[1])
+            if prefix == 'by':
+                rel = ArtUserType.BY
+            elif prefix == 'for':
+                rel = ArtUserType.FOR
+            elif prefix == 'of':
+                rel = ArtUserType.OF
             else:
-                score = Rating.reverse_options.get(components[1])
+                # Bogus tag.  Not sure what to do here, so for the moment,
+                # ignore it
+                continue
 
-            if -1 <= score <= 3:
-                pass
-                # TODO: Find stuff that has this rating
-                # Rating.query.filter(Rating.s)
+            # Inner join to the ArtUser table
+            q = q.join(ArtUser, aliased=True) \
+                 .filter(ArtUser.user == target_user) \
+                 .filter(ArtUser.type == rel)
 
-    tagtexts = TagText.query.filter(TagText.text.in_(tags))
-    tagtext_ids = map(lambda x:x.id, tagtexts)
+        else:
+            # Regular ol' tag
+            q = q.join(Tag, TagText, aliased=True) \
+                 .filter(TagText.text == tag)
 
+    return q
diff --git a/floof/model/search.py b/floof/model/search.py
index fb86f35..37f2d5f 100644
--- a/floof/model/search.py
+++ b/floof/model/search.py
@@ -10,13 +10,6 @@ class SavedSearch(Entity):
     def __unicode__(self):
         return self.string
 
-    @property
-    def results(self):
-        # This caused some cyclic dependencies when I tried importing it
-        # at the module level.  I wonder why that is...
-        from floof.lib.search import do_search
-        return do_search(self.string)
-
 
 
 class GalleryWidget(Entity):
diff --git a/floof/templates/users/view.mako b/floof/templates/users/view.mako
index 5b7e12d..9d34279 100644
--- a/floof/templates/users/view.mako
+++ b/floof/templates/users/view.mako
@@ -21,7 +21,8 @@ ${h.form(url(controller='user_settings', action='rel_toggle', name=c.user.name.l
     % endif
 % endif
 
+<%! from floof.lib.search import parse %>
 % for gallery in c.this_user.primary_page.galleries:
 <h2>${gallery.string}</h2>
-${macros.thumbs(gallery.search.results)}
+${macros.thumbs(parse(gallery.search.string))}
 % endfor