EbookFoundation
/
regluit
mirror of https://github.com/EbookFoundation/regluit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #648 from Gluejar/doab-covers 

Doab covers
pull/1/head
Raymond Yee 2016-11-21 15:55:29 -08:00 committed by GitHub
parent 405af3b2e7 b7f2fb196c
commit dba88ba6b3
2 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
core/loaders/doab.py
View File

@ -40,16 +40,14 @@ def store_doab_cover(doab_id, redo=False):
return (default_storage.url(cover_file_name), True)
except Exception, e:
# if there is a problem, return None for cover URL
logger.warning('Failed to make cover image for doab_id={}: {}'.format(doab_id, e))
return (None, False)
def update_cover_doab(doab_id, store_cover=True):
def update_cover_doab(doab_id, edition, store_cover=True):
"""
update the cover url for work with doab_id
if store_cover is True, use the cover from our own storage
"""
work = models.Identifier.objects.get(type='doab', value=doab_id).work
edition = work.preferred_edition
if store_cover:
(cover_url, new_cover) = store_doab_cover(doab_id)
else:
@ -147,7 +145,7 @@ def load_doab_edition(title, doab_id, url, format, rights,
doab_identifer = models.Identifier.get_or_add(type='doab',value=doab_id,
work=ebook.edition.work)
# update the cover id
cover_url = update_cover_doab(doab_id)
cover_url = update_cover_doab(doab_id, ebook.edition)
# attach more metadata
attach_more_doab_metadata(ebook.edition,
@ -230,7 +228,7 @@ def load_doab_edition(title, doab_id, url, format, rights,
ebook.save()
# update the cover id (could be done separately)
cover_url = update_cover_doab(doab_id)
cover_url = update_cover_doab(doab_id, edition)
# attach more metadata
attach_more_doab_metadata(edition,

22
core/management/commands/add_missing_doab_covers.py Normal file
View File

@ -0,0 +1,22 @@
from __future__ import print_function
from django.core.management.base import BaseCommand
from regluit.core.models import Work
from regluit.core.loaders.doab import update_cover_doab
class Command(BaseCommand):
help = "make covers for doab editions"
def handle(self, **options):
works = Work.objects.filter(selected_edition__isnull=False, selected_edition__cover_image__isnull=True)
#.filter(selected_edition__isnull=False, selected_edition__cover_image__isnull=True)
#.exclude(selected_edition__identifiers__type='goog')
added = 0
for (i, work) in enumerate(works):
if work.doab and work.selected_edition.googlebooks_id == '':
update_cover_doab(work.doab, work.selected_edition)
added += 1
print ('\r {}:{}'.format(i, added), end='')
print('added {} covers'.format(added))