m_librarian/search.py

   1
   2 __all__ = [
   3     'search_authors', 'search_books', 'search_extensions',
   4     'search_genres', 'search_languages',
   5 ]
   6
   7 from sqlobject.sqlbuilder import AND, func
   8 from .db import Author, Book, Extension, Genre, Language
   9
  10
  11 def _search_with_operator(table, case_sensitive, comparison_op, values,
  12                           expressions):
  13     _expressions = []
  14     if case_sensitive:
  15         for column, value in values.items():
  16             _expressions.append(
  17                 getattr(getattr(table.q, column), comparison_op)(value))
  18         for expr, value in expressions:
  19             _expressions.append(
  20                 getattr(expr, comparison_op)(value))
  21     else:
  22         for column, value in values.items():
  23             _expressions.append(
  24                 getattr(func.lower(
  25                     getattr(table.q, column)), comparison_op)(value.lower()))
  26         for expr, value in expressions:
  27             _expressions.append(
  28                 getattr(func.lower(expr), comparison_op)(value.lower()))
  29     return AND(*_expressions)
  30
  31
  32 def _search_exact(table, case_sensitive, values, expressions):
  33     return _search_with_operator(table, case_sensitive, '__eq__', values,
  34                                  expressions)
  35
  36
  37 def _search_start(table, case_sensitive, values, expressions):
  38     return _search_with_operator(table, case_sensitive, 'startswith', values,
  39                                  expressions)
  40
  41
  42 def _search_substring(table, case_sensitive, values, expressions):
  43     return _search_with_operator(table, case_sensitive, 'contains', values,
  44                                  expressions)
  45
  46
  47 def _search(table, search_type, case_sensitive, values,
  48             expressions, orderBy=None):
  49     _search_f = globals()['_search_%s' % search_type]
  50     conditions = _search_f(table, case_sensitive, values, expressions)
  51     return table.select(conditions, orderBy=orderBy)
  52
  53
  54 def search_authors(search_type, case_sensitive, values,
  55                    expressions, orderBy=None):
  56     return _search(Author, search_type, case_sensitive, values,
  57                    expressions, orderBy)
  58
  59
  60 def search_books(search_type, case_sensitive, values, orderBy=None):
  61     return _search(Book, search_type, case_sensitive, values,
  62                    [], orderBy)
  63
  64
  65 def search_extensions(search_type, case_sensitive, values, orderBy=None):
  66     return _search(Extension, search_type, case_sensitive, values,
  67                    [], orderBy)
  68
  69
  70 def search_genres(search_type, case_sensitive, values, orderBy=None):
  71     return _search(Genre, search_type, case_sensitive, values,
  72                    [], orderBy)
  73
  74
  75 def search_languages(search_type, case_sensitive, values, orderBy=None):
  76     return _search(Language, search_type, case_sensitive, values,
  77                    [], orderBy)