src/librarian/builders/txt.py

   1 # coding: utf-8
   2 from __future__ import unicode_literals
   3
   4 import io
   5 from librarian import OutputFile, get_resource
   6
   7
   8 with io.open(get_resource("res/text/template.txt")) as f:
   9     TEMPLATE = f.read()
  10
  11
  12 class TxtFragment:
  13     def __init__(self):
  14         self.pieces = []
  15         self.current_margin = 0
  16         self.starting_block = True
  17
  18     def push_legacy_margin(self, margin):
  19         if margin:
  20             if self.pieces:
  21                 self.pieces[-1] = self.pieces[-1].rstrip(' ')
  22             self.pieces.append('\r\n' * margin)
  23             self.current_margin += margin
  24             self.starting_block = True
  25
  26     def push_margin(self, margin):
  27         if margin:
  28             if self.pieces:
  29                 self.pieces[-1] = self.pieces[-1].rstrip(' ')
  30             if margin > self.current_margin:
  31                 self.pieces.append('\r\n' * (margin - self.current_margin))
  32                 self.current_margin = margin
  33                 self.starting_block = True
  34
  35     def push_text(self, text, prepared=False):
  36         if text:
  37             if self.starting_block and not prepared:
  38                 text = text.lstrip()
  39             self.pieces.append(text)
  40             self.current_margin = 0
  41             if not prepared:
  42                 self.starting_block = False
  43
  44
  45 class TxtBuilder:
  46     """
  47     """
  48     file_extension = "txt"
  49     identifier = "txt"
  50
  51     default_license_description = {
  52         "pol": (
  53             "Ten utwór nie jest objęty majątkowym prawem autorskim "
  54             "i znajduje się w domenie publicznej, co oznacza że "
  55             "możesz go swobodnie wykorzystywać, publikować "
  56             "i rozpowszechniać. Jeśli utwór opatrzony jest "
  57             "dodatkowymi materiałami (przypisy, motywy literackie "
  58             "etc.), które podlegają prawu autorskiemu, to te "
  59             "dodatkowe materiały udostępnione są na licencji "
  60             "Creative Commons Uznanie Autorstwa – Na Tych Samych "
  61             "Warunkach 3.0 PL "
  62             "(http://creativecommons.org/licenses/by-sa/3.0/)"
  63         )
  64     }
  65     license_description = {
  66         "pol": "Ten utwór jest udostępniony na licencji {meta.license_description}: \n{meta.license}",
  67     }
  68
  69     def __init__(self):
  70         self.fragments = {
  71             None: TxtFragment(),
  72             'header': TxtFragment()
  73         }
  74         self.current_fragments = [self.fragments[None]]
  75
  76     def enter_fragment(self, fragment):
  77         self.current_fragments.append(self.fragments[fragment])
  78
  79     def exit_fragment(self):
  80         self.current_fragments.pop()
  81
  82     def push_text(self, text, prepared=False):
  83         self.current_fragments[-1].push_text(text, prepared=prepared)
  84
  85     def push_margin(self, margin):
  86         self.current_fragments[-1].push_margin(margin)
  87
  88     def push_legacy_margin(self, margin, where=None):
  89         self.current_fragments[-1].push_legacy_margin(margin)
  90
  91     def build(self, document, raw_text=False):
  92         document.tree.getroot().txt_build(self)
  93         meta = document.meta
  94
  95         self.enter_fragment('header')
  96         if meta.translators:
  97             self.push_text("tłum. ")
  98             self.push_text(
  99                 ", ".join(
 100                     translator.readable()
 101                     for translator in meta.translators
 102                 )
 103             )
 104             #builder.push_margin(2)
 105             self.push_legacy_margin(1)
 106
 107         if meta.isbn_txt:
 108             #builder.push_margin(2)
 109             self.push_legacy_margin(1)
 110             isbn = meta.isbn_txt
 111             if isbn.startswith(('ISBN-' , 'ISBN ')):
 112                 isbn = isbn[5:]
 113             self.push_text('ISBN {isbn}'.format(isbn=isbn))
 114             #builder.push_margin(5)
 115
 116         #builder.push_margin(4)
 117         self.push_legacy_margin(1)
 118         self.exit_fragment()
 119
 120         text = ''.join(self.fragments['header'].pieces) +  ''.join(self.fragments[None].pieces)
 121
 122         if raw_text:
 123             result = text
 124         else:
 125             if meta.license:
 126                 license_description = self.license_description['pol'].format(meta=meta)
 127             else:
 128                 license_description = self.default_license_description['pol']
 129
 130             if meta.source_name:
 131                 source = "\n\nTekst opracowany na podstawie: " + meta.source_name
 132             else:
 133                 source = ''
 134
 135             contributors = ', '.join(
 136                 person.readable()
 137                 for person in sorted(set(
 138                     p for p in (
 139                         meta.technical_editors + meta.editors
 140                     ) if p))
 141             )
 142             if contributors:
 143                 contributors = (
 144                     "\n\nOpracowanie redakcyjne i przypisy: %s."
 145                     % contributors
 146                 )
 147
 148             funders = ', '.join(meta.funders)
 149             if funders:
 150                 funders = u"\n\nPublikację wsparli i wsparły: %s." % funders
 151
 152             isbn = getattr(meta, 'isbn_txt', None)
 153             if isbn:
 154                 isbn = '\n\n' + isbn
 155             else:
 156                 isbn = ''
 157
 158             result = TEMPLATE % {
 159                 "text": text,
 160                 "description": meta.description,
 161                 "url": meta.url,
 162                 "license_description": license_description,
 163                 "source": source,
 164                 "contributors": contributors,
 165                 "funders": funders,
 166                 "publisher":  '\n\nWydawca: ' + ', '.join(meta.publisher),
 167                 "isbn": isbn,
 168             }
 169
 170         result = '\r\n'.join(result.splitlines()) + '\r\n'
 171         return OutputFile.from_bytes(result.encode('utf-8'))