# -*- coding: utf-8 -*-
import os
-
-from sqlobject.sqlbuilder import CONCAT
from bottle import cheetah_view, redirect, request, route, static_file
-from m_librarian.config import get_config
-from m_librarian.db import Author, Book
-from m_librarian.download import download
-from m_librarian.search import search_authors, search_books
+from ..config import get_config
+from ..db import Book
+from ..download import download
+from ..search import search_authors_raw, books_by_author, search_books_raw
@route('/')
@cheetah_view('index.tmpl')
def index():
- return {}
+ return {
+ 'get_config': get_config,
+ }
@route('/search_authors', method='GET')
return {}
-def decode(value):
- if isinstance(value, bytes):
- return value.decode('utf-8')
- return value
-
-
-def _guess_case_sensitivity(value):
- return not value.islower()
-
-
@route('/search_authors/', method='POST')
@cheetah_view('list_authors.tmpl')
def search_authors_post():
value = request.forms.get('search_authors')
if not value:
return redirect('/search_authors/')
- value = decode(value)
search_type = request.forms.get('search_type')
- if not search_type:
- search_type = 'start'
case_sensitive = request.forms.get('case_sensitive')
- if case_sensitive is None:
- case_sensitive = _guess_case_sensitivity(value)
- expressions = [(
- CONCAT(Author.q.surname, ' ', Author.q.name, ' ', Author.q.misc_name),
- decode(value)
- )]
- authors = search_authors(search_type, case_sensitive, {}, expressions,
- orderBy=('surname', 'name', 'misc_name'))
- return {
- 'authors': list(authors),
- 'search_authors': value,
- 'search_type': search_type,
- 'case_sensitive': case_sensitive,
- }
+ return search_authors_raw(value, search_type, case_sensitive)
-@route('/books-by-author/<id:int>/', method='GET')
-@cheetah_view('books_by_author.tmpl')
-def books_by_author(id):
- return {
- 'author': Author.get(id),
- 'books': Book.select(
- Book.j.authors & (Author.q.id == id),
- orderBy=['series', 'ser_no', 'title'],
- )
- }
+@route('/books-by-author/<aid:int>/', method='GET')
+@cheetah_view('list_books.tmpl')
+def _books_by_author(aid):
+ return books_by_author(aid)
@route('/static/<filename:path>')
@route('/download/', method='POST')
@cheetah_view('download.tmpl')
def download_books():
- books_ids = request.forms.getall('books')
- download_path = get_config().get('download', 'path')
+ books_ids = []
+ form = request.forms
+ for k in form:
+ if k.split('_')[-1] == 'books':
+ for bid in form.getall(k):
+ books_ids.append(bid)
if books_ids:
- for id in books_ids:
- book = Book.get(int(id))
+ download_path = get_config().getpath('download', 'path')
+ for bid in books_ids:
+ book = Book.get(int(bid))
download(book, download_path)
return {
'message': u'Книги сохранены.',
@route('/search_books/', method='GET')
@cheetah_view('search_books.tmpl')
def search_books_get():
- return {}
+ return {
+ 'get_config': get_config,
+ }
@route('/search_books/', method='POST')
value = request.forms.get('search_books')
if not value:
return redirect('/search_books/')
- value = decode(value)
search_type = request.forms.get('search_type')
- if not search_type:
- search_type = 'start'
case_sensitive = request.forms.get('case_sensitive')
- if case_sensitive is None:
- case_sensitive = _guess_case_sensitivity(value)
- books = search_books(search_type, case_sensitive, {'title': value}, None,
- orderBy=('title',))
- books_by_authors = {}
- for book in books:
- author = book.authors[0].fullname
- if author in books_by_authors:
- books_by_author = books_by_authors[author]
- else:
- books_by_author = books_by_authors[author] = []
- books_by_author.append(book)
- return {
- 'books_by_author': books_by_authors,
- 'search_books': value,
- 'search_type': search_type,
- 'case_sensitive': case_sensitive,
- }
+ use_filters = request.forms.get('use_filters')
+ return search_books_raw(value, search_type, case_sensitive, use_filters)