calibre-tal/theanarchistlibrary_store/theanarchistlibrary_plugin.py

__license__ = 'GPL 3'
__copyright__ = '2012, Ruben Pollan <meskio@sindominio.net>; 2020, ibu radempa <ibu@radempa.de>'
__docformat__ = 'restructuredtext en'

import json
try:
    from urllib.parse import quote_plus as quote
except:
    from urllib2 import quote
try:
    from PyQt5.Qt import QUrl
except:
    from PyQt4.Qt import QUrl
from contextlib import closing
from calibre import browser
from calibre.gui2 import open_url
from calibre.gui2.store import StorePlugin
from calibre.gui2.store.basic_config import BasicStoreConfig
from calibre.gui2.store.search_result import SearchResult
from calibre.gui2.store.web_store_dialog import WebStoreDialog
from . import TheAnarchistLibraryStore


url1 = 'https://theanarchistlibrary.org/search?fmt=json&page=%s&query=%s'
url2 = 'https://usa.anarchistlibraries.net/search?fmt=json&page=%s&query=%s'
"""Search URLs. If the library has no fallback url, set url2 = None."""


max_pages = 10
"""Page limit. (amusewiki gives us 10 results per page.)"""


user_agent = 'Calibre plugin calibre-tal v' + '{}.{}.{}'.format(*TheAnarchistLibraryStore.version)


class TheAnarchistLibraryStore(BasicStoreConfig, StorePlugin):

    def open(self, parent=None, detail_item=None, external=False):
        url = 'http://theanarchistlibrary.org/'

        if external or self.config.get('open_external', False):
            open_url(QUrl(url_slash_cleaner(detail_item if detail_item else url)))
        else:
            d = WebStoreDialog(self.gui, url, parent, detail_item)
            d.setWindowTitle(self.name)
            d.set_tags(self.config.get('tags', ''))
            d.exec_()

    def search(self, query, max_results=10, timeout=10):
        br = browser(user_agent=user_agent)
        page = 0
        while page < max_pages:
            page += 1
            try:
                for result in self._iter_search_results(br, url1, page, query, timeout):
                    if result is False:
                        return
                    yield result
            except:
                if url2:
                    for result in self._iter_search_results(br, url2, page, query, timeout):
                        if result is False:
                            return
                        yield result

    def _iter_search_results(self, br, url, page, query, timeout):
        with closing(br.open(url % (page, quote(query)), timeout=timeout)) as f:
            doc = json.load(f)
            if not doc:
                yield False
            for data in doc:
                s = SearchResult()
                s.title = data['title'].strip()
                s.author = data['author'].strip()
                s.price = '$0.00'
                s.detail_item = data['url'].strip()
                s.drm = SearchResult.DRM_UNLOCKED
                s.downloads['EPUB'] = data['url'].strip() + '.epub'
                s.downloads['PDF'] = data['url'].strip() + '.pdf'
                s.downloads['A4.PDF'] = data['url'].strip() + '.a4.pdf'
                s.downloads['LT.PDF'] = data['url'].strip() + '.lt.pdf'
                s.formats = 'EPUB, PDF, A4.PDF, LT.PDF'
                yield s