Added configurable maximum download size in shorturl plugin

Bytespeicher · May 9, 2014 · 583d12e · 583d12e
1 parent 67e4fc9
commit 583d12e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plugins/shorturl.py b/plugins/shorturl.py
@@ -8,7 +8,7 @@
 from plugins.plugin import Plugin
 from bs4 import BeautifulSoup
 
-from bytebot_config import BYTEBOT_HTTP_TIMEOUT
+from bytebot_config import BYTEBOT_HTTP_TIMEOUT, BYTEBOT_HTTP_MAXSIZE
 
 class shorturl(Plugin):
     def googl(self, url):
@@ -23,7 +23,7 @@ def googl(self, url):
             json.dumps(postdata),
             headers
         )
-        data = urllib2.urlopen(url=req, timeout=BYTEBOT_HTTP_TIMEOUT).read()
+        data = urllib2.urlopen(url=req, timeout=BYTEBOT_HTTP_TIMEOUT).read(BYTEBOT_HTTP_MAXSIZE)
         ret = json.loads(data)[u'id'].encode('ascii','ignore')
 
         return ret