1 from sqlobject.sqlbuilder import AND, OR, func, CONCAT
3 from .config import get_config
4 from .db import Author, Book, Extension, Genre, Language
7 'mk_search_conditions',
8 'search_authors', 'search_books', 'search_extensions',
9 'search_genres', 'search_languages',
13 def _mk_search_conditions_with_operator(table, case_sensitive, comparison_op,
15 if expressions is None:
18 for column, value in values.items():
20 _expressions.append(table.q.id == value)
23 for column, value in values.items():
27 getattr(getattr(table.q, column), comparison_op)(value))
28 for expr, value in expressions:
30 getattr(expr, comparison_op)(value))
32 for column, value in values.items():
37 getattr(table.q, column)),
38 comparison_op)(value.lower()))
39 for expr, value in expressions:
41 getattr(func.lower(expr), comparison_op)(value.lower()))
45 _comparison_operators = {
46 'start': 'startswith',
47 'substring': 'contains',
52 def mk_search_conditions(table, search_type, case_sensitive, values,
53 expressions=None, join_expressions=None):
54 if join_expressions is None:
56 return _mk_search_conditions_with_operator(
57 table, case_sensitive, _comparison_operators[search_type],
58 values, expressions) + join_expressions
61 def _search(table, search_type, case_sensitive, values,
62 expressions=None, join_expressions=None, orderBy=None):
63 conditions = mk_search_conditions(
64 table, search_type, case_sensitive, values, expressions=expressions,
65 join_expressions=join_expressions)
66 return table.select(AND(*conditions), orderBy=orderBy)
69 def search_authors(search_type, case_sensitive, values,
70 expressions=None, orderBy=None):
71 return _search(Author, search_type, case_sensitive, values,
72 expressions=expressions, orderBy=orderBy)
75 def search_books(search_type, case_sensitive, values, join_expressions=None,
76 orderBy=None, use_filters=False):
79 lang_filter = config.getlist('filters', 'lang')
80 deleted_filter = config.getint('filters', 'deleted')
82 if join_expressions is None:
85 for lang in lang_filter:
86 lvalues = {'name': lang}
87 conditions = mk_search_conditions(
88 Language, search_type, case_sensitive, lvalues)
89 lang_conditions.append(conditions)
90 join_expressions.append(Book.j.language)
91 join_expressions.append(OR(*lang_conditions))
92 conditions = mk_search_conditions(
93 Book, search_type, case_sensitive, values,
94 join_expressions=join_expressions)
95 if use_filters and not deleted_filter:
96 conditions.append(Book.q.deleted == False) # noqa: E712
97 return Book.select(AND(*conditions), orderBy=orderBy)
100 def search_extensions(search_type, case_sensitive, values, orderBy=None):
101 return _search(Extension, search_type, case_sensitive, values,
105 def search_genres(search_type, case_sensitive, values, orderBy=None):
106 return _search(Genre, search_type, case_sensitive, values,
110 def search_languages(search_type, case_sensitive, values, orderBy=None):
111 return _search(Language, search_type, case_sensitive, values,
116 if isinstance(value, bytes):
117 return value.decode('utf-8')
121 def _guess_case_sensitivity(value):
122 return not value.islower()
125 def search_authors_raw(value, search_type, case_sensitive):
126 value = decode(value)
128 search_type = 'start'
129 if case_sensitive is None:
130 case_sensitive = _guess_case_sensitivity(value)
132 CONCAT(Author.q.surname, ' ', Author.q.name, ' ', Author.q.misc_name),
135 authors = search_authors(search_type, case_sensitive, {}, expressions,
136 orderBy=('surname', 'name', 'misc_name'))
137 columns = get_config().getlist('columns', 'author', ['fullname'])
139 'authors': list(authors),
140 'search_authors': value,
141 'search_type': search_type,
142 'case_sensitive': case_sensitive,
147 def books_by_author(aid):
148 use_filters = get_config().getint('filters', 'use_in_books_list', 1)
149 columns = get_config().getlist('columns', 'book', ['title'])
150 author = Author.get(aid)
152 join_expressions = []
153 join_expressions.append(Book.j.authors)
154 join_expressions.append(Author.q.id == aid)
155 books = search_books('full', None, {}, join_expressions,
156 orderBy=('series', 'ser_no', 'title', '-date'),
157 use_filters=use_filters)
160 Book.j.authors & (Author.q.id == aid),
161 orderBy=['series', 'ser_no', 'title'],
164 'books_by_author': {author.fullname: list(books)},
169 def search_books_raw(value, search_type, case_sensitive, use_filters):
170 value = decode(value)
172 search_type = 'start'
173 if case_sensitive is None:
174 case_sensitive = _guess_case_sensitivity(value)
175 books = search_books(search_type, case_sensitive, {'title': value}, None,
176 orderBy=('title',), use_filters=use_filters)
177 books_by_authors = {}
179 author = book.author1
180 if author in books_by_authors:
181 books_by_author = books_by_authors[author]
183 books_by_author = books_by_authors[author] = []
184 books_by_author.append(book)
185 columns = get_config().getlist('columns', 'book', ['title'])
187 'books_by_author': books_by_authors,
188 'search_books': value,
189 'search_type': search_type,
190 'case_sensitive': case_sensitive,