Implemented budurl.com service.

joshthecoder · joshthecoder · commit 3f94da4dad4d · 2009-08-02T21:41:39.000-05:00
diff --git a/README b/README
@@ -32,6 +32,8 @@ Supported URL shortening services for shrinking long urls:
     chilp.it
     short.ie
     short.to
+    to.ly
+    budurl.com
 
 Shorty can expand any url service by getting the redirect
 location by requesting the tiny url.
diff --git a/TODO b/TODO
@@ -2,14 +2,12 @@ API available and ready to go!
 ++++++++++++++++++++++++++++++
 + burnurl.com
 + u.mavrev.com
-+ budurl.com
 + snipr.com/snipurl.com/sn.im
 + kl.am
 + adjix.com
 + metamark.net (xrl.us)
 + idek.net
 + xr.com
-+ to.ly
 + easyuri.com
 + rubyurl.com
 + hurl.ws
diff --git a/services/budurl.py b/services/budurl.py
@@ -0,0 +1,36 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+## @url budurl.com
+class Budurl(Service):
+
+    def __init__(self, apikey=None):
+        self.apikey = apikey
+
+    def _test(self):
+        #prompt for apikey
+        self.apikey = raw_input('budurl apikey: ')
+        Service._test(self)
+
+    def shrink(self, bigurl, notes=None):
+        if self.apikey is None:
+            raise ShortyError('Must set an apikey')
+        parameters = {'long_url': bigurl, 'api_key': self.apikey}
+        if notes:
+            parameters['notes'] = notes
+        resp = request('http://budurl.com/api/v1/budurls/shrink', parameters)
+        jdata = json.loads(resp.read())
+        if jdata['success'] != 1:
+            raise ShortyError(jdata['error_message'])
+        else:
+            return str(jdata['budurl'])
+
+    def expand(self, tinyurl):
+        resp = request('http://budurl.com/api/v1/budurls/expand', {'budurl': tinyurl})
+        jdata = json.loads(resp.read())
+        if jdata['success'] != 1:
+            raise ShortyError(jdata['error_message'])
+        else:
+            return str(jdata['long_url'])
+
diff --git a/shorty.py b/shorty.py
@@ -292,6 +292,7 @@ class Toly(Service):
     def shrink(self, bigurl):
         resp = request('http://to.ly/api.php', {'longurl': bigurl})
         return resp.read()
+
 toly = Toly()
 
 # fongs
@@ -399,6 +400,40 @@ def expand(self, tinyurl):
 
 bitly = Bitly()
 
+# budurl
+class Budurl(Service):
+
+    def __init__(self, apikey=None):
+        self.apikey = apikey
+
+    def _test(self):
+        #prompt for apikey
+        self.apikey = raw_input('budurl apikey: ')
+        Service._test(self)
+
+    def shrink(self, bigurl, notes=None):
+        if self.apikey is None:
+            raise ShortyError('Must set an apikey')
+        parameters = {'long_url': bigurl, 'api_key': self.apikey}
+        if notes:
+            parameters['notes'] = notes
+        resp = request('http://budurl.com/api/v1/budurls/shrink', parameters)
+        jdata = json.loads(resp.read())
+        if jdata['success'] != 1:
+            raise ShortyError(jdata['error_message'])
+        else:
+            return str(jdata['budurl'])
+
+    def expand(self, tinyurl):
+        resp = request('http://budurl.com/api/v1/budurls/expand', {'budurl': tinyurl})
+        jdata = json.loads(resp.read())
+        if jdata['success'] != 1:
+            raise ShortyError(jdata['error_message'])
+        else:
+            return str(jdata['long_url'])
+
+budurl = Budurl()
+
 # trim
 class Trim(Service):
 
@@ -634,7 +669,7 @@ def expand(self, tinyurl):
 services = {
     'a.gd': agd,
     'short.to': shortto,
-    'chilp.it': chilpit,
+    'budurl.com': budurl,
     'digg.com': digg,
     'tweetburner.com': tweetburner,
     'tr.im': trim,
@@ -644,6 +679,7 @@ def expand(self, tinyurl):
     'urlborg.com': urlborg,
     'is.gd': isgd,
     'fon.gs': fongs,
+    'chilp.it': chilpit,
     'ub0.cc': urlborg,
     'tinyurl.com': tinyurl,
     'bit.ly': bitly,
