Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Complete content server Ajax API

Browse Source
This commit is contained in:
Kovid Goyal 2011-08-29 17:41:34 -06:00
parent 3bd6dc3359
commit 30b3fdb925
1 changed files with 45 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
src/calibre/library/server/ajax.py
View File

@ -19,7 +19,7 @@ from calibre.utils.config import prefs
from calibre.ebooks.metadata.book.json_codec import JsonCodec from calibre.ebooks.metadata.book.json_codec import JsonCodec
from calibre.utils.icu import sort_key from calibre.utils.icu import sort_key
from calibre.library.server import custom_fields_to_display from calibre.library.server import custom_fields_to_display
from calibre import force_unicode from calibre import force_unicode, isbytestring
from calibre.library.field_metadata import category_icon_map from calibre.library.field_metadata import category_icon_map
class Endpoint(object): # {{{ class Endpoint(object): # {{{
@ -66,6 +66,7 @@ def category_icon(category, meta): # {{{
return icon return icon
# }}} # }}}
# URL Encoding {{{
def encode_name(name): def encode_name(name):
if isinstance(name, unicode): if isinstance(name, unicode):
name = name.encode('utf-8') name = name.encode('utf-8')
@ -82,6 +83,7 @@ def category_url(prefix, cid):
def icon_url(prefix, name): def icon_url(prefix, name):
return absurl(prefix, '/browse/icon/'+name) return absurl(prefix, '/browse/icon/'+name)
# }}}
class AjaxServer(object): class AjaxServer(object):
@ -106,6 +108,9 @@ class AjaxServer(object):
connect('ajax_books_in', base_href+'/books_in/{category}/{item}', connect('ajax_books_in', base_href+'/books_in/{category}/{item}',
self.ajax_books_in) self.ajax_books_in)
# Search
connect('ajax_search', base_href+'/search', self.ajax_search)
# Get book metadata {{{ # Get book metadata {{{
@Endpoint() @Endpoint()
@ -422,6 +427,10 @@ class AjaxServer(object):
if sort_order not in ('asc', 'desc'): if sort_order not in ('asc', 'desc'):
sort_order = 'asc' sort_order = 'asc'
sfield = self.db.data.sanitize_sort_field_name(sort)
if sfield not in self.db.field_metadata.sortable_field_keys():
raise cherrypy.HTTPError(404, '%s is not a valid sort field'%sort)
if dname in ('allbooks', 'newest'): if dname in ('allbooks', 'newest'):
ids = self.search_cache('') ids = self.search_cache('')
elif dname == 'search': elif dname == 'search':
@ -444,9 +453,6 @@ class AjaxServer(object):
ids = ids.intersection(all_ids) ids = ids.intersection(all_ids)
ids = list(ids) ids = list(ids)
sfield = self.db.data.sanitize_sort_field_name(sort)
if sfield not in self.db.field_metadata.sortable_field_keys():
raise cherrypy.HTTPError(404, '%s is not a valid sort field'%sort)
self.db.data.multisort(fields=[(sfield, sort_order == 'asc')], subsort=True, self.db.data.multisort(fields=[(sfield, sort_order == 'asc')], subsort=True,
only_ids=ids) only_ids=ids)
total_num = len(ids) total_num = len(ids)
@ -454,10 +460,44 @@ class AjaxServer(object):
return { return {
'total_num': total_num, 'sort_order':sort_order, 'total_num': total_num, 'sort_order':sort_order,
'offset':offset, 'num':len(ids), 'sort':sort, 'offset':offset, 'num':len(ids), 'sort':sort,
'base_url':'/ajax/books_in/%s/%s'%(category, item), 'base_url':absurl(self.opts.url_prefix, '/ajax/books_in/%s/%s'%(category, item)),
'book_ids':ids 'book_ids':ids
} }
# }}} # }}}
# Search {{{
@Endpoint()
def ajax_search(self, query='', sort='title', offset=0, num=25,
sort_order='asc'):
try:
num = int(num)
except:
raise cherrypy.HTTPError(404, "Invalid num: %r"%num)
try:
offset = int(offset)
except:
raise cherrypy.HTTPError(404, "Invalid offset: %r"%offset)
sfield = self.db.data.sanitize_sort_field_name(sort)
if sfield not in self.db.field_metadata.sortable_field_keys():
raise cherrypy.HTTPError(404, '%s is not a valid sort field'%sort)
if isbytestring(query):
query = query.decode('UTF-8')
ids = self.db.search_getting_ids(query.strip(), self.search_restriction)
ids = list(ids)
self.db.data.multisort(fields=[(sfield, sort_order == 'asc')], subsort=True,
only_ids=ids)
total_num = len(ids)
ids = ids[offset:offset+num]
return {
'total_num': total_num, 'sort_order':sort_order,
'offset':offset, 'num':len(ids), 'sort':sort,
'base_url':absurl(self.opts.url_prefix, '/ajax/search'),
'query': query,
'book_ids':ids
}
# }}}