From 7680faa48f2a8ede3f31c0a0f0d91c5dff452406 Mon Sep 17 00:00:00 2001 From: John Schember Date: Tue, 19 Apr 2011 20:40:11 -0400 Subject: [PATCH] Store: Diesel drm status. --- src/calibre/gui2/store/diesel_ebooks_plugin.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/src/calibre/gui2/store/diesel_ebooks_plugin.py b/src/calibre/gui2/store/diesel_ebooks_plugin.py index 66c22f847f..e44896e1f8 100644 --- a/src/calibre/gui2/store/diesel_ebooks_plugin.py +++ b/src/calibre/gui2/store/diesel_ebooks_plugin.py @@ -74,6 +74,13 @@ class DieselEbooksStore(BasicStoreConfig, StorePlugin): price_elem = data.xpath('//td[@class="price"]/text()') if price_elem: price = price_elem[0] + + with closing(br.open('http://www.diesel-ebooks.com/item/' + id.strip(), timeout=timeout/4)) as nf: + idata = html.fromstring(nf.read()) + if idata.xpath('boolean(//table[@class="format-info"]//tr[contains(th, "DRM") and contains(td, "No")])'): + drm = False + else: + drm = True counter -= 1 @@ -83,5 +90,6 @@ class DieselEbooksStore(BasicStoreConfig, StorePlugin): s.author = author.strip() s.price = price.strip() s.detail_item = '/item/' + id.strip() + s.drm = drm yield s