Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2024-12-24 11:42:52 +01:00
Code Issues Projects Releases Wiki Activity

ShrinkUrl: Retry shrinking twice if it fails.

Browse Source
This commit is contained in:
Valentin Lorentz 2013-01-07 20:22:23 +01:00
parent 589d55ebcd
commit 018ee3456d
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
plugins/ShrinkUrl/plugin.py
View File

@ -29,9 +29,11 @@
### ###
import re import re
import time
import json import json
import urllib import urllib
import supybot.log as log
import supybot.conf as conf import supybot.conf as conf
import supybot.utils as utils import supybot.utils as utils
from supybot.commands import * from supybot.commands import *
@ -72,6 +74,17 @@ ShrunkenUrlDB = plugins.DB('ShrinkUrl', {'cdb': CdbShrunkenUrlDB})
class ShrinkError(Exception): class ShrinkError(Exception):
pass pass
def retry(f):
def newf(*args, **kwargs):
for x in xrange(0, 3):
try:
return f(*args, **kwargs)
except Exception:
log.exception('Shrinking URL failed. Trying again.')
time.sleep(1)
return f(*args, **kwargs)
return newf
class ShrinkUrl(callbacks.PluginRegexp): class ShrinkUrl(callbacks.PluginRegexp):
regexps = ['shrinkSnarfer'] regexps = ['shrinkSnarfer']
def __init__(self, irc): def __init__(self, irc):
@ -155,6 +168,7 @@ class ShrinkUrl(callbacks.PluginRegexp):
shrinkSnarfer = urlSnarfer(shrinkSnarfer) shrinkSnarfer = urlSnarfer(shrinkSnarfer)
shrinkSnarfer.__doc__ = utils.web._httpUrlRe shrinkSnarfer.__doc__ = utils.web._httpUrlRe
@retry
def _getLnUrl(self, url): def _getLnUrl(self, url):
url = utils.web.urlquote(url) url = utils.web.urlquote(url)
try: try:
@ -185,6 +199,7 @@ class ShrinkUrl(callbacks.PluginRegexp):
irc.error(str(e)) irc.error(str(e))
ln = thread(wrap(ln, ['url'])) ln = thread(wrap(ln, ['url']))
@retry
def _getTinyUrl(self, url): def _getTinyUrl(self, url):
try: try:
return self.db.get('tiny', url) return self.db.get('tiny', url)
@ -212,6 +227,7 @@ class ShrinkUrl(callbacks.PluginRegexp):
tiny = thread(wrap(tiny, ['url'])) tiny = thread(wrap(tiny, ['url']))
_xrlApi = 'http://metamark.net/api/rest/simple' _xrlApi = 'http://metamark.net/api/rest/simple'
@retry
def _getXrlUrl(self, url): def _getXrlUrl(self, url):
quotedurl = utils.web.urlquote(url) quotedurl = utils.web.urlquote(url)
try: try:
@ -240,6 +256,7 @@ class ShrinkUrl(callbacks.PluginRegexp):
xrl = thread(wrap(xrl, ['url'])) xrl = thread(wrap(xrl, ['url']))
_gooApi = 'https://www.googleapis.com/urlshortener/v1/url' _gooApi = 'https://www.googleapis.com/urlshortener/v1/url'
@retry
def _getGooUrl(self, url): def _getGooUrl(self, url):
try: try:
return self.db.get('goo', url) return self.db.get('goo', url)
@ -270,6 +287,7 @@ class ShrinkUrl(callbacks.PluginRegexp):
_ur1Api = 'http://ur1.ca/' _ur1Api = 'http://ur1.ca/'
_ur1Regexp = re.compile(r'<a href="(?P<url>[^"]+)">') _ur1Regexp = re.compile(r'<a href="(?P<url>[^"]+)">')
@retry
def _getUr1Url(self, url): def _getUr1Url(self, url):
try: try:
return self.db.get('ur1ca', utils.web.urlquote(url)) return self.db.get('ur1ca', utils.web.urlquote(url))
@ -298,6 +316,7 @@ class ShrinkUrl(callbacks.PluginRegexp):
ur1 = thread(wrap(ur1, ['url'])) ur1 = thread(wrap(ur1, ['url']))
_x0Api = 'http://api.x0.no/?%s' _x0Api = 'http://api.x0.no/?%s'
@retry
def _getX0Url(self, url): def _getX0Url(self, url):
try: try:
return self.db.get('x0', url) return self.db.get('x0', url)
@ -323,6 +342,7 @@ class ShrinkUrl(callbacks.PluginRegexp):
irc.error(str(e)) irc.error(str(e))
x0 = thread(wrap(x0, ['url'])) x0 = thread(wrap(x0, ['url']))
@retry
def _getExpandUrl(self, url): def _getExpandUrl(self, url):
url = utils.web.urlquote(url) url = utils.web.urlquote(url)
try: try: