3f19346f27c91f8bd847584633b9a60c37a15fa7
[librarian.git] / 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, **kwargs):
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'))