01f4ee2e8789be94f193cb739ebee72662c67238
[redakcja.git] / src / depot / legimi.py
1 from datetime import date
2 import re
3 from django.conf import settings
4 from librarian.functions import lang_code_3to2
5 from librarian.html import transform_abstrakt
6 from librarian.builders import EpubBuilder, MobiBuilder
7 from librarian.cover import LegimiCornerCover, LegimiCover
8 import requests
9 from slugify import slugify
10
11
12
13 fundraising=[
14             "Książka, którą czytasz, pochodzi z <a href=\"https://wolnelektury.pl/\">Wolnych Lektur</a>. Naszą misją jest wspieranie dzieciaków w dostępie do lektur szkolnych oraz zachęcanie ich do czytania. Miło Cię poznać!",
15             "Podoba Ci się to, co robimy? Jesteśmy organizacją pożytku publicznego. Wesprzyj Wolne Lektury drobną wpłatą: <a href=\"https://wolnelektury.pl/towarzystwo/\">wolnelektury.pl/towarzystwo/</a>",
16             "Przyjaciele Wolnych Lektur otrzymują dostęp do prapremier wcześniej niż inni. Zadeklaruj stałą wpłatę i dołącz do Towarzystwa Przyjaciół Wolnych Lektur: <a href=\"https://wolnelektury.pl/towarzystwo/\">wolnelektury.pl/towarzystwo/</a>",
17             "Informacje o nowościach w naszej bibliotece w Twojej skrzynce mailowej? Nic prostszego, zapisz się do newslettera. Kliknij, by pozostawić swój adres e-mail: <a href=\"https://wolnelektury.pl/newsletter/zapisz-sie/\">wolnelektury.pl/newsletter/zapisz-sie/</a>",
18             "Przekaż 1% podatku na Wolne Lektury.<br/>\nKRS: 0000070056<br/>\nNazwa organizacji: Fundacja Nowoczesna Polska<br/>\nKażda wpłacona kwota zostanie przeznaczona na rozwój Wolnych Lektur."
19 ]
20
21 description_add = '<p>Książkę polecają <a href="https://wolnelektury.pl">Wolne Lektury</a> — najpopularniejsza biblioteka on-line.</p>'
22
23
24 class Legimi:
25     #BASE_URL = 'https://wydawca.legimi.com'
26     BASE_URL = 'https://panel.legimi.pl'
27     LOGIN_URL = BASE_URL + '/publishers/membership'
28     UPLOAD_URL = BASE_URL + '/administration/upload/start'
29     CREATE_URL = BASE_URL + '/publishers/publications/create'
30     EDIT_URL = BASE_URL + '/publishers/publications/edit/%s'
31     EDIT_FILES_URL = BASE_URL + '/publishers/publications/editfiles/%s'
32     
33     def __init__(self, username, password, publisher_id):
34         self.username = username
35         self.password = password
36         self.publisher_id = publisher_id
37         self._session = None
38
39     @property
40     def session(self):
41         if self._session is None:
42             session = requests.Session()
43             response = session.post(
44                 self.LOGIN_URL,
45                 data={
46                     'ValidationTrue': 'true',
47                     'UserName': self.username,
48                     'Password': self.password,
49                 })
50             self._session = session
51         return self._session
52         
53     def list(self):
54         return self.session.get('https://wydawca.legimi.com/publishers/publications')
55         
56     def upload(self, content):
57         response = self.session.post(
58             self.UPLOAD_URL,
59             files={
60                 "files": content,
61             })
62         model = response.json()['model']
63         return {
64             "name": model['Name'],
65             "token": model['Token'],
66             "url": model['Url'],
67         }
68
69 #    name=files[]
70 #    filename
71 #    content-type
72 #    response: json
73 #     success: true
74 #     model.Url
75
76     def send_book(self, book):
77         wlbook = book.wldocument(librarian2=True)
78         meta = wlbook.meta
79
80         cover = LegimiCornerCover(meta, width=1200).output_file()
81         epub_file = EpubBuilder(cover=LegimiCover, fundraising=fundraising).build(wlbook).get_file()
82         mobi_file = MobiBuilder(cover=LegimiCover, fundraising=fundraising).build(wlbook).get_file()
83
84         book_data = {
85             "Title": meta.title,
86             "Author": ", ".join(p.readable() for p in meta.authors),
87             "Year": meta.created_at[:4],
88
89             'GenreId': '11', # TODO
90             'Isbn': '',
91             'LanguageLocale': lang_code_3to2(meta.language),
92
93             'Description': description_add,
94         }
95         if meta.isbn_html:
96             isbn = meta.isbn_html
97             if isbn.upper().startswith('ISBN '):
98                 isbn = isbn[5:]
99             isbn = isbn.strip()
100             book_data['Isbn'] = isbn
101
102         files_data = {}
103
104         abstract = wlbook.tree.find('.//abstrakt')
105         if abstract is not None:
106             book_data['Description'] = transform_abstrakt(abstract)
107         
108
109         cover_data = self.upload(
110             (meta.url.slug + '.jpg', cover.get_file(), 'image/jpeg')
111         )
112         book_data.update({
113             "Cover.Name": cover_data['name'],
114             "Cover.Token": cover_data['token'],
115             "Cover.Url": cover_data['url'],
116         })
117
118         epub_data = self.upload(
119             (meta.url.slug + '.epub', epub_file, 'application/epub+zip')
120         )
121         files_data.update({
122             'BookEpub.Token': epub_data['token'],
123             'BookEpub.Name': epub_data['name'],
124         })
125
126         mobi_data = self.upload(
127             (meta.url.slug + '.mobi', mobi_file, 'application/x-mobipocket-ebook')
128         )
129         files_data.update({
130             'BookMobi.Token': mobi_data['token'],
131             'BookMobi.Name': mobi_data['name'],
132         })
133         
134         if book.legimi_id:
135             self.edit(
136                 book.legimi_id,
137                 book_data
138             )
139             self.edit_files(
140                 book.legimi_id,
141                 files_data
142             )
143         else:
144             legimi_id = self.create_book(book_data, files_data)
145             if legimi_id:
146                 book.legimi_id = legimi_id
147                 book.save(update_fields=['legimi_id'])
148
149     def get_description(self, wlbook):
150         description = ''
151         abstract = wlbook.tree.find('.//abstrakt')
152         if abstract is not None:
153             description = transform_abstrakt(abstract)
154         description += description_add
155         description += '<p>'
156         description += ', '.join(
157             '<a href="https://wolnelektury.pl/katalog/autor/{}/">{}</a>'.format(
158                 slugify(p.readable()),
159                 p.readable(),
160             )
161             for p in wlbook.meta.authors
162         ) + '<br>'
163         description += '<a href="https://wolnelektury.pl/katalog/lektura/{}/">{}</a><br>'.format(
164             wlbook.meta.url.slug,
165             wlbook.meta.title
166         )
167         if wlbook.meta.translators:
168             description += 'tłum. ' + ', '.join(p.readable() for p in wlbook.meta.translators) + '<br>'
169         description += 'Epoka: ' + ', '.join(
170             '<a href="https://wolnelektury.pl/katalog/epoka/{}/">{}</a>'.format(
171                 slugify(p),
172                 p,
173             )
174             for p in wlbook.meta.epochs
175         ) + ' '
176         description += 'Rodzaj: ' + ', '.join(
177             '<a href="https://wolnelektury.pl/katalog/rodzaj/{}/">{}</a>'.format(
178                 slugify(p),
179                 p,
180             )
181             for p in wlbook.meta.kinds
182         ) + ' '
183         description += 'Gatunek: ' + ', '.join(
184             '<a href="https://wolnelektury.pl/katalog/gatunek/{}/">{}</a>'.format(
185                 slugify(p),
186                 p,
187             )
188             for p in wlbook.meta.genres
189         ) + '</p>'
190
191         if wlbook.meta.audience:
192             description += '<p><em>{}</em> to lektura szkolna.'.format(wlbook.meta.title)
193             if wlbook.tree.find('//pe'):
194                 description += '<br>Ebook <em>{title}</em> zawiera przypisy opracowane specjalnie dla uczennic i uczniów {school}.'.format(
195                     title=wlbook.meta.title,
196                     school='szkoły podstawowej' if wlbook.meta.audience == 'SP' else 'liceum i technikum'
197                 )
198             description += '</p>'
199         return description
200
201     def create_book(self, book_data, files_data):
202         data = {
203             'createValidationTrue': 'true',
204             'PublisherId': self.publisher_id,#3609954
205             'IsLibraryPass': 'False',
206
207             'SamplesGenerationType': 'Quantity',
208             'SamplesGenerationPercent': '10',
209
210             'EnterToTheMarketType': 'No',
211             'EnterToTheMarketDate': date.today().strftime('%d.%m.%Y'),
212             'HidingDate': '',
213             'SalesNoLimitOption': 'false',
214             'SalesNoLimitKindle': 'false',
215             'SalesInStoreEbookGrossValue': '0,00',
216             'SalesPromotion': 'False',
217             'SalesPromotionGrossValue': '0,00',
218             'SalesPromotionDatesRange.DateStart': '',
219             'SalesPromotionDatesRange.DateEnd': '',
220         }
221
222         for form in 'Epub', 'Mobi', 'Pdf':
223             data.update({
224                 f'Book{form}.Token': '',
225                 f'Book{form}.Name': '',
226                 f'Book{form}.StorageName': '',
227                 f'Book{form}.Order': '',
228
229                 f'Sample{form}Type': 'Files',
230                 f'Sample{form}.Token': '',
231                 f'Sample{form}.Name': '',
232                 f'Sample{form}.StorageName': '',
233                 f'Sample{form}.Order': '',
234             })
235
236         data.update(book_data)
237         data.update(files_data)
238
239         response = self.session.post(self.CREATE_URL, data=data)
240         m = re.search(r'/(\d+)$', response.url)
241         if m is not None:
242             return m.group(1)
243
244     def edit(self, legimi_id, data):
245         current = {
246             'ValidationTrue': 'true',
247             'Is': legimi_id
248         }
249
250         current.update(data)
251         
252         self.session.post(
253             self.EDIT_URL % legimi_id,
254             data=current
255         )
256
257     def edit_files(self, legimi_id, files_data):
258         current = {
259             'ValidationTrue': 'true',
260             'Id': legimi_id,
261             'SamplesGenerationType': 'Quantity',
262             'SamplesGenerationPercent': '10',
263         }
264
265         for form in 'Epub', 'Mobi', 'Pdf':
266             current.update({
267                 f'Book{form}.Token': '',
268                 f'Book{form}.Name': '',
269                 f'Book{form}.StorageName': '',
270                 f'Book{form}.Order': '',
271
272                 f'Sample{form}.Type': 'Files',
273                 f'Sample{form}.Token': '',
274                 f'Sample{form}.Name': '',
275                 f'Sample{form}.StorageName': '',
276                 f'Sample{form}.Order': '',
277             })
278
279         current.update(files_data)
280  
281         response = self.session.post(
282             self.EDIT_FILES_URL % legimi_id,
283             data=current
284         )
285
286
287 legimi = Legimi(
288     settings.LEGIMI_USERNAME,
289     settings.LEGIMI_PASSWORD,
290     settings.LEGIMI_PUBLISHER_ID,
291 )