Feat(web:books): Поиск по названиям книг

[m_librarian.git] / m_librarian / web / app.py

diff --git a/m_librarian/web/app.py b/m_librarian/web/app.py
index 01b15dd70f106fd3ec0d0cc856b582bcdec551a0..19fca6c4254076abb46c3b6c23c66f54656de2c7 100644 (file)
--- a/m_librarian/web/app.py
+++ b/m_librarian/web/app.py
@@ -1,6 +1,139 @@
-from bottle import route
+# -*- 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
 
 
 @route('/')
-def hello():
-    return "Hello World!"
+@cheetah_view('index.tmpl')
+def index():
+    return {}
+
+
+@route('/search_authors', method='GET')
+def _search_authors():
+    return redirect('/search_authors/')
+
+
+@route('/search_authors/', method='GET')
+@cheetah_view('search_authors.tmpl')
+def search_authors_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,
+    }
+
+
+@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('/static/<filename:path>')
+def send_static(filename):
+    return static_file(
+        filename, root=os.path.join(
+            os.path.dirname(__file__),
+            'static'
+        )
+    )
+
+
+@route('/download/', method='POST')
+@cheetah_view('download.tmpl')
+def download_books():
+    books_ids = request.forms.getall('books')
+    download_path = get_config().get('download', 'path')
+    if books_ids:
+        for id in books_ids:
+            book = Book.get(int(id))
+            download(book, download_path)
+        return {
+            'message': u'Книги сохранены.',
+        }
+    else:
+        return {
+            'message': u'Не выбрано книг для сохранения.',
+        }
+
+
+@route('/search_books', method='GET')
+def _search_books():
+    return redirect('/search_books/')
+
+
+@route('/search_books/', method='GET')
+@cheetah_view('search_books.tmpl')
+def search_books_get():
+    return {}
+
+
+@route('/search_books/', method='POST')
+@cheetah_view('list_books.tmpl')
+def search_books_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',))
+    return {
+        'books': list(books),
+        'search_books': value,
+        'search_type': search_type,
+        'case_sensitive': case_sensitive,
+    }