src/depot/publishers/legimi.py

   1 from datetime import date
   2 import re
   3 from django.conf import settings
   4 from django.utils.html import escape
   5 from django.utils.safestring import mark_safe
   6 from librarian.functions import lang_code_3to2
   7 from librarian.builders import EpubBuilder, MobiBuilder
   8 from librarian.covers.marquise import MarquiseCover, LabelMarquiseCover
   9 from catalogue.models import Audience
  10 from .base import BasePublisher
  11
  12
  13 class Legimi(BasePublisher):
  14     BASE_URL = 'https://panel.legimi.pl'
  15     LOGIN_URL = BASE_URL + '/publishers/membership'
  16     UPLOAD_URL = BASE_URL + '/administration/upload/start'
  17     CREATE_URL = BASE_URL + '/publishers/publications/create'
  18     EDIT_URL = BASE_URL + '/publishers/publications/edit/%s'
  19     EDIT_FILES_URL = BASE_URL + '/publishers/publications/editfiles/%s'
  20     EDIT_SALE_URL = BASE_URL + '/publishers/publications/editsale/%s'
  21
  22     CATEGORIES = {
  23         'Dla dzieci i młodzieży': 94,
  24         'Książki dla dzieci': 15,
  25         'Literatura młodzieżowa': 24,
  26         'Kryminał': 29,
  27         'Kryminał klasyczny': 31,
  28         'Kryminał współczesny': 32,
  29         'Kryminał historyczny': 30,
  30         'default': 8886,
  31         'Edukacja': 10,
  32         'Słowniki i leksykony': 14,
  33         'Encyklopedie': 13,
  34         'Lektury': 11,
  35         'Starożytność': 80,
  36         'Barok': 83,
  37         'Oświecenie': 84,
  38         'Dwudziestolecie międzywojenne': 88,
  39         'Średniowiecze': 81,
  40         'Współczesność': 90,
  41         'Modernizm': 87,
  42         'Pozytywizm': 86,
  43         'Renesans': 82,
  44         'Romantyzm': 85,
  45         'Młoda Polska': 89,
  46         'Podręczniki': 52,
  47         'Fantastyka i sci-fi': 25,
  48         'Fantastyka': 26,
  49         'Science fiction': 27,
  50         'Języki obce': 59,
  51         'Antyki i kolekcjonerstwo': 53,
  52         'Astrologia i wróżbiarstwo': 54,
  53         'Zdrowie i rodzina': 57,
  54         'Hobby': 55,
  55         'Medycyna i zdrowie': 58,
  56         'Psychologiczne': 78,
  57         'Styl': 56,
  58         'Humanistyka': 97,
  59         'Kultura i sztuka': 64,
  60         'Film': 66,
  61         'Muzyka': 65,
  62         'Eseje literackie': 49,
  63         'Historia': 60,
  64         'Styl życia': 73,
  65         'Wakacje i podróże': 69,
  66         'Dla mężczyzn': 79,
  67         'Sport': 76,
  68         'Obyczajowe i romanse': 93,
  69         'Humor': 68,
  70         'Obyczajowe': 35,
  71         'Powieść': 41,
  72         'Powieść przygodowa': 42,
  73         'Współczesna powieść przygodowa': 44,
  74         'Historyczna powieść przygodowa': 43,
  75         'Powieść historyczna': 46,
  76         'Powieść psychologiczna': 47,
  77         'Powieść religijna': 45,
  78         'Romans': 36,
  79         'Romans klasyczny': 38,
  80         'Romans współczesny': 39,
  81         'Literatura erotyczna': 40,
  82         'Romans historyczny': 37,
  83         'Dla kobiet': 77,
  84         'Sensacja, thriller, horror': 91,
  85         'Horror': 28,
  86         'Sensacja': 33,
  87         'Thriller': 34,
  88         'Aktualności': 70,
  89         'Czasopisma': 71,
  90         'Literatura faktu, reportaże, biografie': 92,
  91         'Literatura faktu': 16,
  92         'Biografie': 17,
  93         'Publicystyka': 20,
  94         'Dzienniki': 19,
  95         'Dokument, esej': 18,
  96         'Historia literatury i krytyka literacka': 23,
  97         'Literatura popularnonaukowa': 22,
  98         'Reportaż': 21,
  99         'Społeczno-polityczne': 72,
 100         'Poezja i dramat': 95,
 101         'Dramat': 48,
 102         'Poezja': 50,
 103         'Religia i duchowość': 51,
 104         'Nauka i nowe technologie': 98,
 105         'Nauka i technika': 61,
 106         'Nauki ścisłe': 62,
 107         'Nauki humanistyczne': 63,
 108         'Technologia i Internet': 75,
 109         'Specjalistyczne': 99,
 110         'Biznes i finanse': 1,
 111         'Ekonomia': 5,
 112         'Finanse': 6,
 113         'Zarządzanie': 3,
 114         'Marketing': 2,
 115         'Rozwój osobisty': 7,
 116         'Kariera i sukces zawodowy': 8,
 117         'Psychologia, motywacja': 9,
 118         'PR': 4,
 119         'Prawo': 67,
 120         'Branżowe': 74,
 121     }
 122
 123     def login(self):
 124         self._session.post(
 125             self.LOGIN_URL,
 126             data={
 127                 'ValidationTrue': 'true',
 128                 'UserName': self.username,
 129                 'Password': self.password,
 130             })
 131
 132     def can_publish(self, site, book):
 133         meta = book.wldocument(librarian2=True).meta
 134         d = {
 135             'errors': [],
 136             'warnings': [],
 137             'info': []
 138         }
 139         thema = self.get_thema(meta)
 140         if thema:
 141             d['info'].append(mark_safe(
 142                 "w kategorii " + ", ".join(
 143                     "<b><tt>{code}</tt></b>".format(code=escape(t))
 144                     for t in thema
 145                 )
 146             ))
 147             if not meta.thema_main:
 148                 d['warnings'].append('Brak głównej kategorii Thema')
 149         else:
 150             d['errors'].append('Brak kategorii Thema.')
 151         return d
 152
 153     def list(self):
 154         return self.session.get('https://wydawca.legimi.com/publishers/publications')
 155
 156     def upload(self, content):
 157         response = self.session.post(
 158             self.UPLOAD_URL,
 159             files={
 160                 "files": content,
 161             })
 162         model = response.json()['model']
 163         return {
 164             "name": model['Name'],
 165             "token": model['Token'],
 166             "url": model['Url'],
 167         }
 168
 169     def get_thema(self, meta):
 170         thema = []
 171         if meta.thema_main:
 172             thema.append(meta.thema_main)
 173         thema.extend(meta.thema)
 174
 175         thema.extend(
 176             Audience.objects.filter(code__in=meta.audiences).exclude(
 177                 thema=None).values_list('thema', flat=True)
 178         )
 179         return thema
 180
 181     def send_book(self, site_book_publish, changes=None):
 182         site_book = site_book_publish.site_book
 183         site = site_book.site
 184         book = site_book.book
 185         wlbook = book.wldocument(librarian2=True, changes=changes)
 186         meta = wlbook.meta
 187
 188         cover = LabelMarquiseCover(meta, width=1200).output_file()
 189         texts = site.get_texts()
 190         epub_file = EpubBuilder(
 191             cover=MarquiseCover,
 192             fundraising=texts,
 193             base_url='file://' + book.gallery_path() + '/'
 194         ).build(wlbook).get_file()
 195         mobi_file = MobiBuilder(
 196             cover=MarquiseCover,
 197             fundraising=texts,
 198             base_url='file://' + book.gallery_path() + '/'
 199         ).build(wlbook).get_file()
 200
 201         book_data = {
 202             "Title": meta.title,
 203             "Author": ", ".join(p.readable() for p in meta.authors),
 204             "Year": str(date.today().year),
 205
 206             'GenreId': str(self.get_genre(wlbook)),
 207             'themaCategories': ';'.join(self.get_thema(meta)),
 208             'thema-search': '',
 209             'Isbn': '',
 210             'LanguageLocale': lang_code_3to2(meta.language),
 211
 212             'Description': self.get_description(wlbook, site.description_add),
 213         }
 214         if meta.isbn_html:
 215             isbn = meta.isbn_html
 216             if isbn.upper().startswith(('ISBN ', 'ISBN-')):
 217                 isbn = isbn[5:]
 218             isbn = isbn.strip()
 219             book_data['Isbn'] = isbn
 220
 221         files_data = {}
 222
 223         cover_data = self.upload(
 224             (meta.url.slug + '.jpg', cover.get_file(), 'image/jpeg')
 225         )
 226         book_data.update({
 227             "Cover.Name": cover_data['name'],
 228             "Cover.Token": cover_data['token'],
 229             "Cover.Url": cover_data['url'],
 230         })
 231
 232         epub_data = self.upload(
 233             (meta.url.slug + '.epub', epub_file, 'application/epub+zip')
 234         )
 235         files_data.update({
 236             'BookEpub.Token': epub_data['token'],
 237             'BookEpub.Name': epub_data['name'],
 238             'SampleEpubType': 'Generation',
 239         })
 240
 241         mobi_data = self.upload(
 242             (meta.url.slug + '.mobi', mobi_file, 'application/x-mobipocket-ebook')
 243         )
 244         files_data.update({
 245             'BookMobi.Token': mobi_data['token'],
 246             'BookMobi.Name': mobi_data['name'],
 247         })
 248
 249         if site_book.external_id:
 250             self.edit(
 251                 site_book.external_id,
 252                 book_data
 253             )
 254             self.edit_files(
 255                 site_book.external_id,
 256                 files_data
 257             )
 258         else:
 259             legimi_id = self.create_book(book_data, files_data)
 260             if legimi_id:
 261                 site_book.external_id = legimi_id
 262                 site_book.save(update_fields=['external_id'])
 263
 264         self.edit_sale(site_book)
 265
 266     def get_genre(self, wlbook):
 267         if wlbook.meta.legimi and wlbook.meta.legimi in self.CATEGORIES:
 268             return self.CATEGORIES[wlbook.meta.legimi]
 269         for epoch in wlbook.meta.epochs:
 270             if epoch in self.CATEGORIES:
 271                 return self.CATEGORIES[epoch]
 272         return self.CATEGORIES['Lektury']
 273
 274     def create_book(self, book_data, files_data):
 275         data = {
 276             'createValidationTrue': 'true',
 277             'PublisherId': self.publisher_handle,
 278             'IsLibraryPass': 'False',
 279
 280             'SamplesGenerationType': 'Quantity',
 281             'SamplesGenerationPercent': '10',
 282
 283             'EnterToTheMarketType': 'No',
 284             'EnterToTheMarketDate': '',
 285             'HidingDate': '',
 286             'SalesNoLimitOption': 'false',
 287             'SalesNoLimitKindle': 'false',
 288             'SalesInStoreEbookGrossValue': '0,00',
 289             'SalesPromotion': 'False',
 290             'SalesPromotionGrossValue': '0,00',
 291             'SalesPromotionDatesRange.DateStart': '',
 292             'SalesPromotionDatesRange.DateEnd': '',
 293         }
 294
 295         for form in 'Epub', 'Mobi', 'Pdf':
 296             data.update({
 297                 f'Book{form}.Token': '',
 298                 f'Book{form}.Name': '',
 299                 f'Book{form}.StorageName': '',
 300                 f'Book{form}.Order': '',
 301
 302                 f'Sample{form}Type': 'Files',
 303                 f'Sample{form}.Token': '',
 304                 f'Sample{form}.Name': '',
 305                 f'Sample{form}.StorageName': '',
 306                 f'Sample{form}.Order': '',
 307             })
 308
 309         data.update(book_data)
 310         data.update(files_data)
 311
 312         response = self.session.post(self.CREATE_URL, data=data)
 313         m = re.search(r'/(\d+)$', response.url)
 314         if m is not None:
 315             return m.group(1)
 316
 317     def edit(self, legimi_id, data):
 318         current = {
 319             'ValidationTrue': 'true',
 320             'Id': legimi_id
 321         }
 322
 323         current.update(data)
 324
 325         self.session.post(
 326             self.EDIT_URL % legimi_id,
 327             data=current
 328         )
 329
 330     def edit_files(self, legimi_id, files_data):
 331         current = {
 332             'ValidationTrue': 'true',
 333             'Id': legimi_id,
 334             'SamplesGenerationType': 'Quantity',
 335             'SamplesGenerationPercent': '10',
 336         }
 337
 338         for form in 'Epub', 'Mobi', 'Pdf':
 339             current.update({
 340                 f'Book{form}.Token': '',
 341                 f'Book{form}.Name': '',
 342                 f'Book{form}.StorageName': '',
 343                 f'Book{form}.Order': '',
 344
 345                 f'Sample{form}.Type': 'Files',
 346                 f'Sample{form}.Token': '',
 347                 f'Sample{form}.Name': '',
 348                 f'Sample{form}.StorageName': '',
 349                 f'Sample{form}.Order': '',
 350             })
 351
 352         current.update(files_data)
 353
 354         response = self.session.post(
 355             self.EDIT_FILES_URL % legimi_id,
 356             data=current
 357         )
 358
 359     def edit_sale(self, site_book):
 360         book = site_book.book
 361         assert site_book.external_id
 362
 363         words = book.wldocument(librarian2=True).get_statistics()['total']['words_with_fn']
 364
 365         price = settings.LEGIMI_SMALL_PRICE
 366         if words > settings.LEGIMI_SMALL_WORDS:
 367             price = settings.LEGIMI_BIG_PRICE
 368
 369         abo = 'true' if words > settings.LEGIMI_BIG_WORDS else 'false'
 370
 371         data = {
 372             'ValidationTrue': 'true',
 373             'Id': site_book.external_id,
 374             'SalesPromotionId': "0",
 375             'IsLibraryPass': "False",
 376             'OriginalEnterToTheMarketType': "No",
 377             'OriginalHidingDate': "",
 378             'OriginalEnterToTheMarketDate': "",
 379             'EnterToTheMarketType': "Yes",
 380             'EnterToTheMarketDate': "",
 381             'HidingDate': "",
 382             'SalesNoLimitOption': abo,
 383             'SalesNoLimitKindle': abo,
 384             'SalesInStoreEbookGrossValue': f'{price},00',
 385             'SalesPromotion': "False",
 386             'SalesPromotionGrossValue': "0,00",
 387             'SalesPromotionDatesRange.DateStart': "",
 388             'SalesPromotionDatesRange.DateEnd': "",
 389         }
 390
 391         self.session.post(
 392             self.EDIT_SALE_URL % site_book.external_id,
 393             data=data
 394         )