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     if case_sensitive:
  14         for column, value in values.items():
  15             expressions.append(
  16                 getattr(getattr(table.q, column), comparison_op)(value))
  17     else:
  18         for column, value in values.items():
  19             expressions.append(
  20                 getattr(func.lower(
  21                     getattr(table.q, column)), comparison_op)(value.lower()))
  22     return AND(*expressions)
  23
  24
  25 def _search_exact(table, case_sensitive, values):
  26     return _search_with_operator(table, case_sensitive, '__eq__', values)
  27
  28
  29 def _search_start(table, case_sensitive, values):
  30     return _search_with_operator(table, case_sensitive, 'startswith', values)
  31
  32
  33 def _search_substring(table, case_sensitive, values):
  34     return _search_with_operator(table, case_sensitive, 'contains', values)
  35
  36
  37 def _search(table, search_type, case_sensitive, values):
  38     _search_f = globals()['_search_%s' % search_type]
  39     conditions = _search_f(table, case_sensitive, values)
  40     return table.select(conditions)
  41
  42
  43 def search_authors(search_type, case_sensitive, values):
  44     return _search(Author, search_type, case_sensitive, values)
  45
  46
  47 def search_books(search_type, case_sensitive, values):
  48     return _search(Book, search_type, case_sensitive, values)
  49
  50
  51 def search_extensions(search_type, case_sensitive, values):
  52     return _search(Extension, search_type, case_sensitive, values)
  53
  54
  55 def search_genres(search_type, case_sensitive, values):
  56     return _search(Genre, search_type, case_sensitive, values)
  57
  58
  59 def search_languages(search_type, case_sensitive, values):
  60     return _search(Language, search_type, case_sensitive, values)