X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=m_librarian%2Fsearch.py;h=5c1f47d65403df60bc645a144728a5c24b2076c5;hb=35190cea6dda6d65537b110f82a5d3f5277bd94c;hp=a310757fba4069d659d0eb402db1e5d8222a3b32;hpb=408ee16ffe9da1c0b5bd1404da19edc7582b517a;p=m_librarian.git

diff --git a/m_librarian/search.py b/m_librarian/search.py
index a310757..5c1f47d 100644
--- a/m_librarian/search.py
+++ b/m_librarian/search.py
@@ -1,60 +1,92 @@
 
+from sqlobject.sqlbuilder import AND, func
+from .db import Author, Book, Extension, Genre, Language
+
 __all__ = [
+    'mk_search_conditions',
     'search_authors', 'search_books', 'search_extensions',
     'search_genres', 'search_languages',
 ]
 
-from sqlobject.sqlbuilder import AND, func
-from .db import Author, Book, Extension, Genre, Language
-
 
-def _search_with_operator(table, case_sensitive, comparison_op, values):
-    expressions = []
+def _mk_search_conditions_with_operator(table, case_sensitive, comparison_op,
+                                        values, expressions):
+    if expressions is None:
+        expressions = []
+    _expressions = []
+    for column, value in values.items():
+        if column == 'id':
+            _expressions.append(table.q.id == value)
+            break
     if case_sensitive:
         for column, value in values.items():
-            expressions.append(
+            if column == 'id':
+                continue
+            _expressions.append(
                 getattr(getattr(table.q, column), comparison_op)(value))
+        for expr, value in expressions:
+            _expressions.append(
+                getattr(expr, comparison_op)(value))
     else:
         for column, value in values.items():
-            expressions.append(
+            if column == 'id':
+                continue
+            _expressions.append(
                 getattr(func.lower(
-                    getattr(table.q, column)), comparison_op)(value.lower()))
-    return AND(*expressions)
-
-
-def _search_exact(table, case_sensitive, values):
-    return _search_with_operator(table, case_sensitive, '__eq__', values)
+                    getattr(table.q, column)),
+                    comparison_op)(value.lower()))
+        for expr, value in expressions:
+            _expressions.append(
+                getattr(func.lower(expr), comparison_op)(value.lower()))
+    return _expressions
 
 
-def _search_start(table, case_sensitive, values):
-    return _search_with_operator(table, case_sensitive, 'startswith', values)
+_comparison_operators = {
+    'start': 'startswith',
+    'substring': 'contains',
+    'full': '__eq__',
+}
 
 
-def _search_substring(table, case_sensitive, values):
-    return _search_with_operator(table, case_sensitive, 'contains', values)
+def mk_search_conditions(table, search_type, case_sensitive, values,
+                         expressions=None, join_expressions=None):
+    if join_expressions is None:
+        join_expressions = []
+    return _mk_search_conditions_with_operator(
+        table, case_sensitive, _comparison_operators[search_type],
+        values, expressions) + join_expressions
 
 
-def _search(table, search_type, case_sensitive, values):
-    _search_f = globals()['_search_%s' % search_type]
-    conditions = _search_f(table, case_sensitive, values)
-    return table.select(conditions)
+def _search(table, search_type, case_sensitive, values,
+            expressions=None, join_expressions=None, orderBy=None):
+    conditions = mk_search_conditions(
+        table, search_type, case_sensitive, values, expressions=expressions,
+        join_expressions=join_expressions)
+    return table.select(AND(*conditions), orderBy=orderBy)
 
 
-def search_authors(search_type, case_sensitive, values):
-    return _search(Author, search_type, case_sensitive, values)
+def search_authors(search_type, case_sensitive, values,
+                   expressions=None, orderBy=None):
+    return _search(Author, search_type, case_sensitive, values,
+                   expressions=expressions, orderBy=orderBy)
 
 
-def search_books(search_type, case_sensitive, values):
-    return _search(Book, search_type, case_sensitive, values)
+def search_books(search_type, case_sensitive, values, join_expressions=None,
+                 orderBy=None):
+    return _search(Book, search_type, case_sensitive, values,
+                   join_expressions=join_expressions, orderBy=orderBy)
 
 
-def search_extensions(search_type, case_sensitive, values):
-    return _search(Extension, search_type, case_sensitive, values)
+def search_extensions(search_type, case_sensitive, values, orderBy=None):
+    return _search(Extension, search_type, case_sensitive, values,
+                   orderBy=orderBy)
 
 
-def search_genres(search_type, case_sensitive, values):
-    return _search(Genre, search_type, case_sensitive, values)
+def search_genres(search_type, case_sensitive, values, orderBy=None):
+    return _search(Genre, search_type, case_sensitive, values,
+                   orderBy=orderBy)
 
 
-def search_languages(search_type, case_sensitive, values):
-    return _search(Language, search_type, case_sensitive, values)
+def search_languages(search_type, case_sensitive, values, orderBy=None):
+    return _search(Language, search_type, case_sensitive, values,
+                   orderBy=orderBy)