-# -*- coding: utf-8 -*-
-#
-# This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
-# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
-#
-
-"""
-Classes for packaging ebooks for some old partners.
-These should be removed from librarian to separate package,
-along with custom cover images etc.
-
-New partners shouldn't be added here, but in the partners repository.
-"""
-from __future__ import print_function, unicode_literals
-
-from librarian import packagers, cover
-from .util import makedirs
-
-
-class GandalfEpub(packagers.EpubPackager):
- cover = cover.GandalfCover
-
-
-class GandalfPdf(packagers.PdfPackager):
- cover = cover.GandalfCover
-
-
-class BookotekaEpub(packagers.EpubPackager):
- cover = cover.BookotekaCover
-
-
-class PrestigioEpub(packagers.EpubPackager):
- cover = cover.PrestigioCover
- flags = ('less-advertising',)
-
-
-class PrestigioPdf(packagers.PdfPackager):
- cover = cover.PrestigioCover
- flags = ('less-advertising',)
-
-
-class Virtualo(packagers.Packager):
- @staticmethod
- def utf_trunc(text, limit):
- """ truncates text to at most `limit' bytes in utf-8 """
- if text is None:
- return text
- if len(text.encode('utf-8')) > limit:
- newlimit = limit - 3
- while len(text.encode('utf-8')) > newlimit:
- text = text[:(newlimit - len(text.encode('utf-8'))) / 4]
- text += '...'
- return text
-
- @classmethod
- def prepare(cls, input_filenames, output_dir='', verbose=False):
- from lxml import etree
- from librarian import DirDocProvider, ParseError
- from librarian.parser import WLDocument
- from copy import deepcopy
- import os.path
-
- xml = etree.fromstring("""<?xml version="1.0" encoding="utf-8"?>
- <products xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"></products>""")
- product = etree.fromstring("""<product>
- <publisherProductId></publisherProductId>
- <title></title>
- <info></info>
- <description></description>
- <authors>
- <author>
- <names>Jan</names>
- <lastName>Kowalski</lastName>
- </author>
- </authors>
- <price>0.0</price>
- <language>PL</language>
- </product>""")
-
- try:
- for main_input in input_filenames:
- if verbose:
- print(main_input)
- path, fname = os.path.realpath(main_input).rsplit('/', 1)
- provider = DirDocProvider(path)
- slug, ext = os.path.splitext(fname)
-
- outfile_dir = os.path.join(output_dir, slug)
- makedirs(os.path.join(output_dir, slug))
-
- doc = WLDocument.from_file(main_input, provider=provider)
- info = doc.book_info
-
- product_elem = deepcopy(product)
- product_elem[0].text = cls.utf_trunc(slug, 100)
- product_elem[1].text = cls.utf_trunc(info.title, 255)
- product_elem[2].text = cls.utf_trunc(info.description, 255)
- product_elem[3].text = cls.utf_trunc(info.source_name, 3000)
- product_elem[4][0][0].text = cls.utf_trunc(u' '.join(info.author.first_names), 100)
- product_elem[4][0][1].text = cls.utf_trunc(info.author.last_name, 100)
- xml.append(product_elem)
-
- cover.VirtualoCover(info).save(os.path.join(outfile_dir, slug+'.jpg'))
- outfile = os.path.join(outfile_dir, '1.epub')
- outfile_sample = os.path.join(outfile_dir, '1.sample.epub')
- doc.save_output_file(doc.as_epub(), output_path=outfile)
- doc.save_output_file(doc.as_epub(doc, sample=25), output_path=outfile_sample)
- outfile = os.path.join(outfile_dir, '1.mobi')
- outfile_sample = os.path.join(outfile_dir, '1.sample.mobi')
- doc.save_output_file(doc.as_mobi(cover=cover.VirtualoCover), output_path=outfile)
- doc.save_output_file(
- doc.as_mobi(doc, cover=cover.VirtualoCover, sample=25),
- output_path=outfile_sample)
- except ParseError as e:
- print('%(file)s:%(name)s:%(message)s' % {
- 'file': main_input,
- 'name': e.__class__.__name__,
- 'message': e.message
- })
-
- xml_file = open(os.path.join(output_dir, 'import_products.xml'), 'w')
- xml_file.write(etree.tostring(xml, pretty_print=True, encoding='unicode').encode('utf-8'))
- xml_file.close()