nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Enables matching an URL in E-Hentai

Browse source
This commit is contained in:
Roger Philibert 2011-09-09 02:06:09 +02:00 • committed by Romain Bignon
commit 156903751a
1 changed files with 12 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
weboob/backends/ehentai/backend.py
View file

@ -62,10 +62,19 @@ class EHentaiBackend(BaseBackend, ICapGallery):
with self.browser: with self.browser:
return self.browser.iter_gallery_images(gallery) return self.browser.iter_gallery_images(gallery)
ID_REGEXP = r'/?\d+/[\dabcdef]+/?'
URL_REGEXP = r'.+/g/(%s)' % ID_REGEXP
def get_gallery(self, _id): def get_gallery(self, _id):
if not re.match(r'(?i)/?\d+/[\dabcdef]+/?', _id): match = re.match(r'^%s$' % self.URL_REGEXP, _id)
return None if match:
_id = match.group(1)
else:
match = re.match(r'^%s$' % self.ID_REGEXP, _id)
if match:
_id = match.group(0)
else:
return None
gallery = EHentaiGallery(_id) gallery = EHentaiGallery(_id)
with self.browser: with self.browser:
if self.browser.gallery_exists(gallery): if self.browser.gallery_exists(gallery):