Added bit.ly service. Implemented shrink and expand methods.

joshthecoder · joshthecoder · commit db4b9c422da6 · 2009-07-23T23:56:57.000-05:00
diff --git a/README b/README
@@ -13,6 +13,7 @@ Supported URL shortening services:
     tinyurl.com  (shrink only)
     tr.im
     urlborg.com
+    bit.ly
 
 Language: Python
 Website: http://gitorious.org/shorty
diff --git a/shorty.py b/shorty.py
@@ -221,12 +221,58 @@ def expand(self, tinyurl, apikey=None):
 
 urlborg = Urlborg()
 
+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:
+            raise ShortyError('Must set a login')
+        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'])
+
+bitly = Bitly()
+
 """Mapping of domain to service class"""
 services = {
     'sandbox': sandbox,
     'tinyurl.com': tinyurl,
     'tr.im': trim,
     'urlborg.com': urlborg, 'ub0.cc': urlborg,
+    'bit.ly': bitly
 }
 
 """
