Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Get Books: Fix DRM status for legimi

Browse Source
This commit is contained in:
Kovid Goyal 2011-11-17 08:15:16 +05:30
commit a5fa43ad5b
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/calibre/gui2/store/stores/legimi_plugin.py
View File

@ -43,6 +43,7 @@ class LegimiStore(BasicStoreConfig, StorePlugin):
url = 'http://www.legimi.com/pl/ebooks/?price=any&lang=pl&search=' + urllib.quote_plus(query) + '&sort=relevance' url = 'http://www.legimi.com/pl/ebooks/?price=any&lang=pl&search=' + urllib.quote_plus(query) + '&sort=relevance'
br = browser() br = browser()
drm_pattern = re.compile("(DRM)")
counter = max_results counter = max_results
with closing(br.open(url, timeout=timeout)) as f: with closing(br.open(url, timeout=timeout)) as f:
@ -61,6 +62,10 @@ class LegimiStore(BasicStoreConfig, StorePlugin):
author = re.sub(',','',author) author = re.sub(',','',author)
author = re.sub(';',',',author) author = re.sub(';',',',author)
price = ''.join(data.xpath('.//span[@class="ebook_price"]/text()')) price = ''.join(data.xpath('.//span[@class="ebook_price"]/text()'))
formats = ''.join(data.xpath('.//div[@class="item_entries"]/span[3]/text()'))
formats = re.sub('Format:','',formats)
drm = drm_pattern.search(formats)
formats = re.sub('\(DRM\)','',formats)
counter -= 1 counter -= 1
@ -70,7 +75,7 @@ class LegimiStore(BasicStoreConfig, StorePlugin):
s.author = author.strip() s.author = author.strip()
s.price = price s.price = price
s.detail_item = 'http://www.legimi.com/' + id.strip() s.detail_item = 'http://www.legimi.com/' + id.strip()
s.drm = SearchResult.DRM_LOCKED s.drm = SearchResult.DRM_LOCKED if drm else SearchResult.DRM_UNLOCKED
s.formats = 'EPUB' s.formats = formats.strip()
yield s yield s