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