Move services into their own modules.

joshthecoder · joshthecoder · commit 7cf2944f11e6 · 2009-08-02T16:12:50.000-05:00
diff --git a/services/agd.py b/services/agd.py
@@ -0,0 +1,23 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# a.gd
+class Agd(Service):
+
+    def shrink(self, bigurl, tag=None, password=None, expires=None):
+        post_param = {'url': bigurl}
+        if tag:
+            post_param['tag'] = tag
+        if password:
+            post_param['pass'] = password
+        if expires:
+            post_param['validTill'] = expires
+        resp = request('http://a.gd/?module=ShortURL&file=Add&mode=API',
+                        post_data = urlencode(post_param))
+        url = resp.read()
+        if url.startswith('http://'):
+            return url
+        else:
+            raise ShortyError(url[url.find('>')+1:url.rfind('<')])
+
diff --git a/services/bitly.py b/services/bitly.py
@@ -0,0 +1,48 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# bit.ly
+class Bitly(Service):
+
+    version = '2.0.1'
+
+    def __init__(self, login=None, apikey=None, password=None):
+        self.login = login
+        self.apikey = apikey
+        self.password = password
+
+    def _setup(self):
+        parameters = {'version': self.version}
+        if self.apikey:
+            parameters['apiKey'] = self.apikey
+            parameters['login'] = self.login
+            username_pass = None
+        elif self.password:
+            username_pass = (self.login, self.password)
+        else:
+            raise ShortyError('Must set an apikey or password')
+        return parameters, username_pass
+
+    def shrink(self, bigurl):
+        if not self.login:
+            raise ShortyError('Must set a login')
+        parameters, username_pass = self._setup()
+        parameters['longUrl'] = bigurl
+        resp = request('http://api.bit.ly/shorten', parameters, username_pass)
+        jdata = json.loads(resp.read())
+        if jdata['errorCode'] != 0:
+            raise ShortyError(jdata['errorMessage'])
+        return str(jdata['results'][bigurl]['shortUrl'])
+
+    def expand(self, tinyurl):
+        if not self.login:
+            return get_redirect(tinyurl)
+        parameters, username_pass = self._setup()
+        parameters['shortUrl'] = tinyurl
+        resp = request('http://api.bit.ly/expand', parameters, username_pass)
+        jdata = json.loads(resp.read())
+        if jdata['errorCode'] != 0:
+            raise ShortyError(jdata['errorMessage'])
+        return str(jdata['results'].values()[0]['longUrl'])
+
diff --git a/services/bukme.py b/services/bukme.py
@@ -0,0 +1,26 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# buk.me
+class Bukme(Service):
+
+    def _process(self, resp):
+        # bukme returns some markup after the url, so chop it off
+        url = resp[:resp.find('<')]
+        if url.startswith('http://'):
+            return url
+        else:
+            raise ShortyError(url)
+
+    def shrink(self, bigurl, tag=None):
+        parameters = {'url': bigurl}
+        if tag:
+            parameters['buk'] = tag
+        resp = request('http://buk.me/api.php', parameters)
+        return self._process(resp.read())
+
+    def expand(self, tinyurl):
+        resp = request('http://buk.me/api.php', {'rev': tinyurl})
+        return self._process(resp.read())
+
diff --git a/services/chilpit.py b/services/chilpit.py
@@ -0,0 +1,38 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# chilp.it
+class Chilpit(Service):
+
+    def shrink(self, bigurl):
+        resp = request('http://chilp.it/api.php', {'url': bigurl})
+        url = resp.read()
+        if url.startswith('http://'):
+            return url
+        else:
+            raise ShortyError(url)
+
+    def expand(self, tinyurl):
+        turl = urlparse(tinyurl)
+        if turl.netloc.lstrip('www.') != 'chilp.it':
+            raise ShortyError('Not a chilp.it url')
+        resp = request('http://p.chilp.it/api.php?' + turl.path.strip('/'))
+        url = resp.read()
+        if url.startswith('http://'):
+            return url
+        else:
+            raise ShortyError(url)
+
+    # get click stats of the tinyurl
+    def stats(self, tinyurl):
+        turl = urlparse(tinyurl)
+        if turl.netloc.lstrip('www.') != 'chilp.it':
+            raise ShortyError('Not a chilp.it url')
+        resp = request('http://s.chilp.it/api.php?' + turl.query)
+        hit_count = resp.read()
+        try:
+            return int(hit_count)
+        except:
+            raise ShortyError('Url not found or invalid')
+
diff --git a/services/cligs.py b/services/cligs.py
@@ -0,0 +1,27 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+
+# cli.gs
+class Cligs(Service):
+
+    def __init__(self, apikey=None, appid=None):
+        self.apikey = apikey
+        self.appid = appid
+
+    def shrink(self, bigurl, title=None):
+        parameters = {'url': bigurl}
+        if title:
+            parameters['title'] = title
+        if self.apikey:
+            parameters['key'] = self.apikey
+        if self.appid:
+            parameters['appid'] = self.appid
+        resp = request('http://cli.gs/api/v1/cligs/create', parameters)
+        return resp.read()
+
+    def expand(self, tinyurl):
+        # TODO: debug this some more, not working properly
+        '''resp = request('http://cli.gs/api/v1/cligs/expand', {'clig': tinyurl})
+        return resp.read()'''
+        return get_redirect(tinyurl)
+
diff --git a/services/digg.py b/services/digg.py
@@ -0,0 +1,39 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# digg
+class Digg(Service):
+
+    def __init__(self, appkey=None):
+        self.appkey = appkey
+        self.itemid = None
+        self.view_count = None
+
+    def shrink(self, bigurl):
+        if not self.appkey:
+            raise ShortyError('Must set an appkey')
+        resp = request('http://services.digg.com/url/short/create',
+            {'url': bigurl, 'appkey': self.appkey, 'type': 'json'})
+        jdata = json.loads(resp.read())['shorturls'][0]
+        self.itemid = jdata['itemid']
+        self.view_count = jdata['view_count']
+        return str(jdata['short_url'])
+
+    def expand(self, tinyurl):
+        if self.appkey:
+            turl = urlparse(tinyurl)
+            if turl.netloc != 'digg.com' and turl.netloc != 'www.digg.com':
+                raise ShortyError('Not a valid digg url')
+            resp = request('http://services.digg.com/url/short/%s' % quote(
+                            turl.path.strip('/')),
+                            {'appkey': self.appkey, 'type': 'json'})
+            jdata = json.loads(resp.read())['shorturls'][0]
+            self.itemid = jdata['itemid']
+            self.view_count = jdata['view_count']
+            return str(jdata['link'])
+        else:
+            self.itemid = None
+            self.view_count = None
+            return get_redirect(tinyurl)
+
diff --git a/services/fongs.py b/services/fongs.py
@@ -0,0 +1,39 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# fon.gs
+class Fongs(Service):
+
+    def shrink(self, bigurl, tag=None):
+        parameters = {'url': bigurl}
+        if tag:
+            parameters['linkname'] = tag
+        resp = request('http://fon.gs/create.php', parameters)
+        data = resp.read()
+        if data.startswith('OK:'):
+            return data.lstrip('OK: ')
+        elif data.startswith('MODIFIED:'):
+            return data.lstrip('MODIFIED: ')
+        else:
+            raise ShortyError(data)
+
+    # check if the given tag is taken
+    # returns true if available false if taken
+    def check(self, tag):
+        resp = request('http://fon.gs/check.php', {'linkname': tag})
+        data = resp.read()
+        if data.startswith('AVAILABLE'):
+            return True
+        elif data.startswith('TAKEN'):
+            return False
+        else:
+            raise ShortyError(data)
+
+    def expand(self, tinyurl):
+        if tinyurl[-1] != '/':
+            turl = tinyurl + '/'
+        else:
+            turl = tinyurl
+        return Service.expand(self, turl)
+
diff --git a/services/fwd4me.py b/services/fwd4me.py
@@ -0,0 +1,24 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# fwd4.me
+class Fwd4me(Service):
+
+    ecodes = {
+        '1': 'Invalid long url',
+        '2': 'Invalid api key',
+        '3': 'Account suspended or revoked',
+        '4': 'Long url is black listed',
+        '5': 'Internal system error'
+    }
+
+    def shrink(self, bigurl):
+        resp = request('http://api.fwd4.me/', {'url': bigurl})
+        data = resp.read()
+        if data.startswith('ERROR:'):
+            ecode = data.lstrip('ERROR:')
+            raise ShortyError(Fwd4me.ecodes.get(ecode, 'Unkown error'))
+        else:
+            return data
+
diff --git a/services/isgd.py b/services/isgd.py
@@ -0,0 +1,14 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# is.gd
+class Isgd(Service):
+
+    def shrink(self, bigurl):
+        resp = request('http://is.gd/api.php', {'longurl': bigurl})
+        turl = resp.read()
+        if turl.startswith('Error:'):
+            raise ShortyError(turl)
+        return turl
+
diff --git a/services/sandbox.py b/services/sandbox.py
@@ -0,0 +1,35 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+class Sandbox(Service):
+
+    def __init__(self, length=4, letters='abcdefghijklmnopqrstuvwxyz'):
+        self.urls = {}
+        self.letters = letters
+        self.length = length
+        self.base = len(letters) - 1
+
+    def shrink(self, bigurl):
+        # generate the tiny path
+        tpath = ''
+        while True:
+            for i in range(self.length):
+                tpath += self.letters[randint(0, self.base)]
+            if self.urls.has_key(tpath):
+                # tpath already in use, regen another
+                tpath = ''
+            else:
+                break
+
+        # store url and return tiny link
+        self.urls[tpath] = bigurl
+        return 'http://sandbox/' + tpath
+
+    def expand(self, tinyurl):
+        # lookup big url and return
+        turl = urlparse(tinyurl)
+        if turl.netloc != 'sandbox':
+            raise ShortyError('Not a sandbox url')
+        return self.urls.get(turl.path.strip('/'))
+
diff --git a/services/tinyurl.py b/services/tinyurl.py
@@ -0,0 +1,11 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# tinyurl.com
+class Tinyurl(Service):
+
+    def shrink(self, bigurl):
+        resp = request('http://tinyurl.com/api-create.php', {'url': bigurl})
+        return resp.read()
+
diff --git a/services/trim.py b/services/trim.py
@@ -0,0 +1,52 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# tr.im
+class Trim(Service):
+
+    def __init__(self, apikey=None, username_pass=None):
+        self.apikey = apikey
+        self.username_pass = username_pass
+
+    def shrink(self, bigurl, custom=None, searchtags=None, privacycode=None,
+                newtrim=False, sandbox=False):
+        parameters = {}
+        parameters['url'] = bigurl
+        if custom:
+            parameters['custom'] = custom
+        if searchtags:
+            parameters['searchtags'] = searchtags
+        if privacycode:
+            parameters['privacycode'] = privacycode
+        if newtrim:
+            parameters['newtrim'] = '1'
+        if sandbox:
+            parameters['sandbox'] = '1'
+        if self.apikey:
+            parameters['api_key'] = self.apikey
+        resp = request('http://api.tr.im/api/trim_url.json', parameters, self.username_pass)
+        jdata = json.loads(resp.read())
+        self.status = (int(jdata['status']['code']), str(jdata['status']['message']))
+        if not 200 <= self.status[0] < 300:
+            raise ShortyError(self.status[1])
+        self.trimpath = str(jdata['trimpath'])
+        self.reference = str(jdata['reference'])
+        self.destination = str(jdata['destination'])
+        self.domain = str(jdata['domain'])
+        return str(jdata['url'])
+
+    def expand(self, tinyurl):
+        turl = urlparse(tinyurl)
+        if turl.netloc != 'tr.im' and turl.netloc != 'www.tr.im':
+            raise ShortyError('Not a valid tr.im url')
+        parameters = {'trimpath': turl.path.strip('/')}
+        if self.apikey:
+            parameters['api_key'] = self.apikey
+        resp = request('http://api.tr.im/api/trim_destination.json', parameters)
+        jdata = json.loads(resp.read())
+        self.status = (int(jdata['status']['code']), str(jdata['status']['message']))
+        if not 200 <= self.status[0] < 300:
+            raise ShortyError(self.status[1])
+        return str(jdata['destination'])
+
diff --git a/services/tweetburner.py b/services/tweetburner.py
@@ -0,0 +1,11 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+# tweetburner.com
+class Tweetburner(Service):
+
+    def shrink(self, bigurl):
+        resp = request('http://tweetburner.com/links', post_data='link[url]=%s' % bigurl)
+        return resp.read()
+
diff --git a/services/urlborg.py b/services/urlborg.py
