epub/mobi fixes
[librarian.git] / librarian / text.py
1 # -*- coding: utf-8 -*-
2 #
3 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 #
6 from librarian import dcparser, parser, functions
7 from lxml import etree
8 import os
9
10
11 functions.reg_substitute_entities()
12 functions.reg_wrap_words()
13 functions.reg_strip()
14 functions.reg_person_name()
15
16 TEMPLATE = u"""\
17 %(text)s
18
19
20 -----
21 Ta lektura, podobnie jak tysiące innych, dostępna jest na stronie wolnelektury.pl.
22 Wersja lektury w opracowaniu merytorycznym i krytycznym (przypisy i motywy) dostępna jest na stronie %(url)s.
23
24 Utwór opracowany został w ramach projektu Wolne Lektury przez fundację Nowoczesna Polska.
25
26 %(license_description)s.%(source)s
27
28 %(description)s%(contributors)s
29 """
30
31 def transform(input_file, output_file, parse_dublincore=True, flags=None, **options):
32     """
33     Transforms input_file in XML to output_file in TXT.
34     possible flags: raw-text,
35     """
36     # Parse XSLT
37     style_filename = os.path.join(os.path.dirname(__file__), 'xslt/book2txt.xslt')
38     style = etree.parse(style_filename)
39
40     document = parser.WLDocument.from_file(input_file, True, parse_dublincore=parse_dublincore)
41
42     if flags:
43         for flag in flags:
44             document.edoc.getroot().set(flag, 'yes')
45
46     result = document.transform(style, **options)
47
48     if not flags or 'raw-text' not in flags:
49         if parse_dublincore:
50             parsed_dc = dcparser.BookInfo.from_element(document.edoc)
51             description = parsed_dc.description
52             url = parsed_dc.url
53     
54             license_description = parsed_dc.license_description
55             license = parsed_dc.license
56             if license:
57                 license_description = u"Ten utwór jest udostepniony na licencji %s: \n%s" % (license_description, license)        
58             else:
59                 license_description = u"Ten utwór nie jest chroniony prawem autorskim i znajduje się w domenie publicznej, co oznacza że możesz go swobodnie wykorzystywać, publikować i rozpowszechniać. Jeśli utwór opatrzony jest dodatkowymi materiałami (przypisy, motywy literackie etc.), które podlegają prawu autorskiemu, to te dodatkowe materiały udostępnione są na licencji Creative Commons Uznanie Autorstwa – Na Tych Samych Warunkach 3.0 PL (http://creativecommons.org/licenses/by-sa/3.0/)"
60     
61             source = parsed_dc.source_name
62             if source:
63                 source = "\n\nTekst opracowany na podstawie: " + source
64             else:
65                 source = ''
66     
67             contributors = ', '.join(person.readable() for person in 
68                                      sorted(set(p for p in (parsed_dc.technical_editors + parsed_dc.editors) if p)))
69             if contributors:
70                 contributors = "\n\nOpracowanie redakcyjne i przypisy: %s" % contributors
71         else:
72             description = 'Publikacja zrealizowana w ramach projektu Wolne Lektury (http://wolnelektury.pl).'
73             url = '*' * 10
74             license = ""
75             license_description = ""
76             source = ""
77             contributors = ""
78         output_file.write((TEMPLATE % {
79             'description': description,
80             'url': url,
81             'license_description': license_description,
82             'text': unicode(result),
83             'source': source,
84             'contributors': contributors,
85         }).encode('utf-8'))
86     else:
87         output_file.write(unicode(result).encode('utf-8'))
88