Python 3.4-3.7 support;
[librarian.git] / librarian / text.py
index c0591e0..7ba6d29 100644 (file)
@@ -3,9 +3,13 @@
 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
-from librarian import dcparser, parser, functions
+from __future__ import unicode_literals
+
+import copy
+from librarian import functions, OutputFile
 from lxml import etree
 import os
+import six
 
 
 functions.reg_substitute_entities()
@@ -23,55 +27,92 @@ Wersja lektury w opracowaniu merytorycznym i krytycznym (przypisy i motywy) dost
 
 Utwór opracowany został w ramach projektu Wolne Lektury przez fundację Nowoczesna Polska.
 
-%(license_description)s.%(source)s
+%(license_description)s.%(source)s%(publisher)s
 
-%(description)s%(contributors)s
+%(description)s%(contributors)s%(funders)s%(isbn)s
 """
 
-def transform(input_file, output_file, parse_dublincore=True, **options):
-    """Transforms input_file in XML to output_file in TXT."""
+
+def transform(wldoc, flags=None, **options):
+    """
+    Transforms input_file in XML to output_file in TXT.
+    possible flags: raw-text,
+    """
     # Parse XSLT
     style_filename = os.path.join(os.path.dirname(__file__), 'xslt/book2txt.xslt')
     style = etree.parse(style_filename)
 
-    document = parser.WLDocument.from_file(input_file, True, parse_dublincore=parse_dublincore)
-    result = document.transform(style, **options)
+    document = copy.deepcopy(wldoc)
+    del wldoc
+    document.swap_endlines()
 
-    if parse_dublincore:
-        parsed_dc = dcparser.BookInfo.from_element(document.edoc)
-        description = parsed_dc.description
-        url = parsed_dc.url
+    if flags:
+        for flag in flags:
+            document.edoc.getroot().set(flag, 'yes')
+    if 'wrapping' in options:
+        options['wrapping'] = str(options['wrapping'])
 
-        license_description = parsed_dc.license_description
-        license = parsed_dc.license
-        if license:
-            license_description = u"Ten utwór jest udostepniony na licencji %s: \n%s" % (license_description, license)
-        else:
-            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/)"
+    result = document.transform(style, **options)
 
-        source = parsed_dc.source_name
-        if source:
-            source = "\n\nTekst opracowany na podstawie: " + source
-        else:
-            source = ''
+    if not flags or 'raw-text' not in flags:
+        if document.book_info:
+            parsed_dc = document.book_info
+            description = parsed_dc.description
+            url = document.book_info.url
+    
+            license_description = parsed_dc.license_description
+            license = parsed_dc.license
+            if license:
+                license_description = u"Ten utwór jest udostępniony na licencji %s: \n%s" % (
+                    license_description, license)
+            else:
+                license_description = u"Ten utwór nie jest objęty majątkowym prawem autorskim i znajduje się " \
+                                      u"w domenie publicznej, co oznacza że możesz go swobodnie wykorzystywać, " \
+                                      u"publikować i rozpowszechniać. Jeśli utwór opatrzony jest dodatkowymi " \
+                                      u"materiałami (przypisy, motywy literackie etc.), które podlegają prawu " \
+                                      u"autorskiemu, to te dodatkowe materiały udostępnione są na licencji " \
+                                      u"Creative Commons Uznanie Autorstwa – Na Tych Samych Warunkach 3.0 PL " \
+                                      u"(http://creativecommons.org/licenses/by-sa/3.0/)"
 
-        contributors = ', '.join(person.readable() for person in
-                                 sorted(set(p for p in (parsed_dc.technical_editors + parsed_dc.editors) if p)))
-        if contributors:
-            contributors = "\n\nOpracowanie redakcyjne i przypisy: %s" % contributors
+            source = parsed_dc.source_name
+            if source:
+                source = "\n\nTekst opracowany na podstawie: " + source
+            else:
+                source = ''
+    
+            contributors = ', '.join(person.readable() for person in 
+                                     sorted(set(p for p in (parsed_dc.technical_editors + parsed_dc.editors) if p)))
+            if contributors:
+                contributors = "\n\nOpracowanie redakcyjne i przypisy: %s." % contributors
+            funders = ', '.join(parsed_dc.funders)
+            if funders:
+                funders = u"\n\nPublikację wsparli i wsparły: %s." % funders
+            publisher = '\n\nWydawca: ' + ', '.join(parsed_dc.publisher)
+            isbn = getattr(parsed_dc, 'isbn_txt', None)
+            if isbn:
+                isbn = '\n\n' + isbn
+            else:
+                isbn = ''
+        else:
+            description = 'Publikacja zrealizowana w ramach projektu Wolne Lektury (http://wolnelektury.pl).'
+            url = '*' * 10
+            license_description = ""
+            source = ""
+            contributors = ""
+            funders = ""
+            publisher = ""
+            isbn = ""
+        result = (TEMPLATE % {
+            'description': description,
+            'url': url,
+            'license_description': license_description,
+            'text': six.text_type(result),
+            'source': source,
+            'contributors': contributors,
+            'funders': funders,
+            'publisher': publisher,
+            'isbn': isbn,
+        }).encode('utf-8')
     else:
-        description = 'Publikacja zrealizowana w ramach projektu Wolne Lektury (http://wolnelektury.pl).'
-        url = '*' * 10
-        license = ""
-        license_description = ""
-        source = ""
-        contributors = ""
-    output_file.write((TEMPLATE % {
-        'description': description,
-        'url': url,
-        'license_description': license_description,
-        'text': unicode(result),
-        'source': source,
-        'contributors': contributors,
-    }).encode('utf-8'))
-
+        result = six.text_type(result).encode('utf-8')
+    return OutputFile.from_bytes(b"\r\n".join(result.splitlines()) + b"\r\n")