]> git.phdru.name Git - m_librarian.git/blobdiff - m_librarian/web/app.py
Refactor: Separate web forms processing from search
[m_librarian.git] / m_librarian / web / app.py
index 19fca6c4254076abb46c3b6c23c66f54656de2c7..283ddb715b8b819fb0039c57bf375337eff84a1e 100644 (file)
@@ -1,20 +1,20 @@
 # -*- 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')
@@ -28,53 +28,21 @@ 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,
-    }
+    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>')
@@ -90,11 +58,16 @@ def send_static(filename):
 @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'Книги сохранены.',
@@ -113,7 +86,9 @@ def _search_books():
 @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')
@@ -122,18 +97,7 @@ 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,
-    }
+    use_filters = request.forms.get('use_filters')
+    return search_books_raw(value, search_type, case_sensitive, use_filters)