More tags

Author: znedw

pythonbits/bb.py

@@ -25,7 +25,7 @@
 from . import musicbrainz as mb
 from . import imagehosting
 from . import goodreads
-from .googlebooks import find_cover
+from .googlebooks import find_cover, find_categories
 from .ffmpeg import FFMpeg
 from . import templating as bb
 from .submission import (Submission, form_field, finalize, cat_map,
@@ -934,8 +934,10 @@ def _render_form_description(self):
 
 
 class BookSubmission(BbSubmission):
-    default_fields = BbSubmission.default_fields + ("isbn", "format", "title", "author", "publisher", "language", "description", "year", "cover",
-                                                    "title")
+    default_fields = BbSubmission.default_fields + ("isbn", "format", "title",
+                                                    "author", "publisher",
+                                                    "language", "description",
+                                                    "year", "cover", "title")
 
     _cat_id = 'book'
     _form_type = 'E-Books'
@@ -1003,8 +1005,10 @@ def _render_summary(self):
 
     @form_field('tags')
     def _render_tags(self):
+        categories = find_categories(self['book']['isbn'])
         authors = self['book']['authors']
-        return uniq(list(format_tag(a['name']) for a in authors))
+        return uniq(list(format_tag(a['name']) for a in authors) +
+                    list(format_tag(a) for a in categories))
 
     def _render_section_information(self):
         def gr_author_link(gra):
@@ -1046,6 +1050,8 @@ def _render_description(self):
     @finalize
     @form_field('image')
     def _render_cover(self):
+        # Goodreads usually won't give you a cover image as they don't have the
+        # the right to distribute them
         if "nophoto" in self['book']['image_url']:
             return find_cover(self['book']['isbn'])
         else:

pythonbits/calibre.py

@@ -18,7 +18,8 @@ def get_version():
         return ebook_meta.communicate()[0].decode('utf8')
     except OSError:
         raise EbookMetaException(
-            "Could not find %s, please ensure it is installed (via Calibre)." % COMMAND)
+            "Could not find {}, please ensure it is installed (via Calibre)."
+            .format(COMMAND))
 
 
 def read_metadata(path):
@@ -34,6 +35,6 @@ def read_metadata(path):
             try:
                 key, value = row.split(': ')
                 result[key.strip(' .')] = value.strip()
-            except:
+            except KeyError:
                 pass
     return result

pythonbits/goodreads.py

@@ -76,13 +76,16 @@ def search(self, path):
                 print('{}: {} by {} ({})'
                       .format(i, book['best_book']['title'],
                               book['best_book']['author']['name'],
-                              book['original_publication_year'].get('#text', '')))
+                              book['original_publication_year']
+                              .get('#text', '')))
 
             while True:
                 choice = input('Select number or enter an alternate'
-                               ' search term (or an ISBN with isbn: prefix):'
-                               ' [0-{}, 0 default] '
-                               .format(len(book_results['results']['work']) - 1))
+                               ' search term'
+                               ' (or an ISBN with isbn: prefix):'
+                               ' [0-{}, 0 default]'
+                               .format(
+                                   len(book_results['results']['work']) - 1))
                 try:
                     choice = int(choice)
                 except ValueError:

pythonbits/googlebooks.py

@@ -13,8 +13,21 @@ def find_cover(isbn):
     log.debug('Fetching alt cover art from {}'.format(resp.url))
     if resp.status_code == 200:
         content = json.loads(resp.content)
-        return content['items'][0]
-        ['volumeInfo']['imageLinks']['thumbnail'] or ''
+        return (content['items'][0]['volumeInfo']
+                ['imageLinks']['thumbnail'] or '')
     else:
         log.warn('Couldn\'t find cover art for ISBN {}'.format(isbn))
         return ''
+
+
+def find_categories(isbn):
+    path = 'volumes?q=isbn:{}'.format(isbn)
+    resp = requests.get(API_URL+path)
+    log.debug('Fetching categories from {}'.format(resp.url))
+    if resp.status_code == 200:
+        content = json.loads(resp.content)
+        return (content['items'][0]['volumeInfo']
+                ['categories'] or '')
+    else:
+        log.warn('Couldn\'t find categories for ISBN {}'.format(isbn))
+        return ''