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

unsee: Restrict URL to unsee domains

Browse source 
Otherwise it would crash any pastoob get!
This commit is contained in:
Laurent Bachelier 2014-10-10 23:30:58 +02:00
commit 31a96228db
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/unsee/module.py
View file

@ -22,6 +22,7 @@ from weboob.tools.backend import Module
from weboob.capabilities.paste import BasePaste from weboob.capabilities.paste import BasePaste
from weboob.tools.capabilities.paste import BasePasteModule from weboob.tools.capabilities.paste import BasePasteModule
from weboob.tools.capabilities.paste import image_mime from weboob.tools.capabilities.paste import image_mime
from weboob.deprecated.browser.decorators import check_url
import re import re
from .browser import UnseeBrowser from .browser import UnseeBrowser
@ -57,6 +58,7 @@ class UnseeModule(Module, BasePasteModule):
mime = image_mime(contents, ('gif', 'jpeg', 'png')) mime = image_mime(contents, ('gif', 'jpeg', 'png'))
return 20 * int(mime is not None) return 20 * int(mime is not None)
@check_url('https?://unsee.cc/.+')
def get_paste(self, id): def get_paste(self, id):
paste = UnPaste(id) paste = UnPaste(id)
paste.contents = self.browser.get_image(id).encode('base64') paste.contents = self.browser.get_image(id).encode('base64')