X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=m_librarian%2Fsearch.py;h=5c1f47d65403df60bc645a144728a5c24b2076c5;hb=d9a85ec552d93e4a009f729d09f4452e6b34dac5;hp=2fbde60906edef7554ebf123f1425ad222317d88;hpb=2bde7fb4a4f400aee640a185b45c2ade7952adc6;p=m_librarian.git diff --git a/m_librarian/search.py b/m_librarian/search.py index 2fbde60..5c1f47d 100644 --- a/m_librarian/search.py +++ b/m_librarian/search.py @@ -1,21 +1,27 @@ +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 _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(): + if column == 'id': + continue _expressions.append( getattr(getattr(table.q, column), comparison_op)(value)) for expr, value in expressions: @@ -23,16 +29,19 @@ def _mk_search_conditions_with_operator(table, case_sensitive, comparison_op, getattr(expr, comparison_op)(value)) else: for column, value in values.items(): + if column == 'id': + continue _expressions.append( getattr(func.lower( - getattr(table.q, column)), comparison_op)(value.lower())) + 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 -_search_conditions_dict = { +_comparison_operators = { 'start': 'startswith', 'substring': 'contains', 'full': '__eq__', @@ -44,7 +53,7 @@ def mk_search_conditions(table, search_type, case_sensitive, values, if join_expressions is None: join_expressions = [] return _mk_search_conditions_with_operator( - table, case_sensitive, _search_conditions_dict[search_type], + table, case_sensitive, _comparison_operators[search_type], values, expressions) + join_expressions