93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
"""
|
|
Utilities for calling mobigen for management. do not use in application
|
|
|
|
"""
|
|
|
|
from itertools import islice
|
|
from StringIO import StringIO
|
|
import uuid
|
|
|
|
from django.core.files.storage import default_storage
|
|
from django.core.files.base import ContentFile, File
|
|
|
|
from regluit.core.models import (Campaign, Ebook)
|
|
from regluit.core import parameters
|
|
from regluit.core.mobi import convert_to_mobi
|
|
|
|
|
|
# compute whether we can apply mobigen to a given edition to produce a mobi file
|
|
# need to have an ebook in epub or pdf format
|
|
# possible return values: already has a mobi file / can generate a mobi file / not possible
|
|
|
|
def edition_mobi_status(edition):
|
|
"""
|
|
for a given edition, return:
|
|
* 1 if there is already a mobi ebook
|
|
* 0 if there is none but we have an epub or html to convert from
|
|
* -1 for no epub/html to convert from
|
|
"""
|
|
formats = set([ebook.format for ebook in edition.work.ebooks()])
|
|
if 'mobi' in formats:
|
|
return 1
|
|
elif ('epub' in formats) or ('html' in formats):
|
|
return 0
|
|
else:
|
|
return -1
|
|
|
|
|
|
def write_file_to_storage(file_object, content_type, path):
|
|
"""
|
|
write file_object to the default_storage at given path
|
|
"""
|
|
file_s3 = ContentFile(file_object)
|
|
file_s3.content_type = content_type
|
|
|
|
default_storage.save(path, file_s3)
|
|
return file_s3
|
|
|
|
|
|
# generator for editions to add mobi to
|
|
# campaigns that can have mobi files but don't yet.
|
|
|
|
def editions_to_convert():
|
|
for campaign in Campaign.objects.filter(edition__ebooks__isnull=False).distinct():
|
|
# need to make sure campaign type is not B2U because kindlegen is for books we give awy free of charge
|
|
if (edition_mobi_status(campaign.edition) == 0) and (campaign.type != parameters.BUY2UNGLUE): # possible to generate mobi
|
|
yield campaign.edition
|
|
|
|
|
|
def generate_mobi_ebook_for_edition(edition):
|
|
|
|
# pull out the sister edition to convert from
|
|
sister_ebook = edition.ebooks.filter(format__in=['epub', 'html'])[0]
|
|
|
|
# run the conversion process
|
|
|
|
output = convert_to_mobi(sister_ebook.url)
|
|
#output = open("/Users/raymondyee/Downloads/hello.mobi").read()
|
|
|
|
file_ = write_file_to_storage(output,
|
|
"application/x-mobipocket-ebook",
|
|
"/ebf/{0}.mobi".format(uuid.uuid4().get_hex()))
|
|
|
|
# create a path for the ebookfile: IS THIS NECESSARY?
|
|
# https://github.com/Gluejar/regluit/blob/25dcb06f464dc11b5e589ab6859dfcc487f8f3ef/core/models.py#L1771
|
|
|
|
#ebfile = EbookFile(edition=edition, file=file_, format='mobi')
|
|
#ebfile.save()
|
|
|
|
# maybe need to create an ebook pointing to ebookFile ?
|
|
# copy metadata from sister ebook
|
|
|
|
ebfile_url = default_storage.url(file_.name)
|
|
#print (ebfile_url)
|
|
|
|
ebook = Ebook(url=ebfile_url,
|
|
format="mobi",
|
|
provider="Unglue.it",
|
|
rights=sister_ebook.rights,
|
|
edition=edition)
|
|
ebook.save()
|
|
|
|
return ebook
|