book2html generates full page

[librarian.git] / librarian / text.py

diff --git a/librarian/text.py b/librarian/text.py
index 57c2a73..e1ca26b 100644 (file)
--- a/librarian/text.py
+++ b/librarian/text.py
@@ -5,10 +5,7 @@
 #
 from librarian import dcparser, parser, functions
 from lxml import etree
-import cStringIO
-import codecs
 import os
-import re
 
 
 functions.reg_substitute_entities()
@@ -18,9 +15,9 @@ functions.reg_strip()
 TEMPLATE = u"""\
 Kodowanie znaków w dokumencie: UTF-8.
 -----
-Publikacja zrealizowana w ramach projektu Wolne Lektury (http://wolnelektury.pl/). Reprodukcja cyfrowa wykonana przez
-Bibliotekę Narodową z egzemplarza pochodzącego ze zbiorów BN. 
-\n%(license_description)s.
+%(description)s 
+
+%(license_description)s.%(source)s
 
 Wersja lektury w opracowaniu merytorycznym i krytycznym (przypisy i motywy) dostępna jest na stronie %(url)s.
 -----
@@ -30,37 +27,41 @@ Wersja lektury w opracowaniu merytorycznym i krytycznym (przypisy i motywy) dost
 %(text)s
 """
 
-def transform(input_filename, output_filename, is_file=True, parse_dublincore=True, **options):
-    """Transforms file input_filename in XML to output_filename in TXT."""
+def transform(input_file, output_file, parse_dublincore=True, **options):
+    """Transforms input_file in XML to output_file in TXT."""
     # Parse XSLT
     style_filename = os.path.join(os.path.dirname(__file__), 'xslt/book2txt.xslt')
     style = etree.parse(style_filename)
 
-    if is_file:
-        document = parser.WLDocument.from_file(input_filename, True, parse_dublincore=parse_dublincore)
-    else:
-        document = parser.WLDocument.from_string(input_filename, True, parse_dublincore=parse_dublincore)
-
+    document = parser.WLDocument.from_file(input_file, True, parse_dublincore=parse_dublincore)
     result = document.transform(style, **options)
 
-    output_file = codecs.open(output_filename, 'wb', encoding='utf-8')
-
     if parse_dublincore:
-        parsed_dc = dcparser.parse(input_filename)
+        parsed_dc = dcparser.BookInfo.from_element(document.edoc)
+        description = parsed_dc.description
         url = parsed_dc.url
         license_description = parsed_dc.license_description
-       license = parsed_dc.license
+        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ć" 
+            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ć"
+        source = parsed_dc.source_name
+        if source:
+            source = "\n\nNa podstawie: " + source
+        else:
+            source = ''
     else:
+        description = 'Publikacja zrealizowana w ramach projektu Wolne Lektury (http://wolnelektury.pl).'
         url = '*' * 10
         license = ""
         license_description = ""
-    output_file.write(TEMPLATE % {
+        source = ""
+    output_file.write((TEMPLATE % {
+        'description': description,
         'url': url,
-       'license_description': license_description,
+        'license_description': license_description,
         'text': unicode(result),
-    })
+        'source': source,
+    }).encode('utf-8'))