Added tweetburner service. Implemented shrink and expand.

joshthecoder · joshthecoder · commit 7c6c9a550f2c · 2009-07-24T18:42:57.000-05:00
diff --git a/README b/README
@@ -16,6 +16,7 @@ Supported URL shortening services:
     bit.ly
     is.gd
     cli.gs
+    twurl.nl (tweetburner)
 
 Language: Python
 Website: http://gitorious.org/shorty
diff --git a/shorty.py b/shorty.py
@@ -56,7 +56,7 @@ def __str__(self):
         return repr(self.reason)
 
 """Do a http request"""
-def request(url, parameters=None, username_pass=None):
+def request(url, parameters=None, username_pass=None, post_data=None):
 
     # build url + parameters
     if parameters:
@@ -72,6 +72,8 @@ def request(url, parameters=None, username_pass=None):
     # send request
     try:
         req = Request(url_params, headers=headers)
+        if post_data:
+            req.add_data(post_data)
         return urlopen(req)
     except URLError, e:
         raise ShortyError(e)
@@ -96,13 +98,13 @@ def http_error_301(self, req, fp, code, smg, headers):
 """Base interface that all services implement."""
 class Service(object):
 
-    def shrink(bigurl):
+    def shrink(self, bigurl):
         """Take a big url and make it smaller"""
 
         return None
 
-    def expand(tinyurl):
-        return None
+    def expand(self, tinyurl):
+        return get_redirect(tinyurl)
 
 """ Services """
 
@@ -147,9 +149,6 @@ def shrink(self, bigurl):
         resp = request('http://tinyurl.com/api-create.php', {'url': bigurl})
         return resp.read()
 
-    def expand(self, tinyurl):
-        return get_redirect(tinyurl)
-
 tinyurl = Tinyurl()
 
 # tr.im
@@ -286,9 +285,6 @@ def shrink(self, bigurl):
             raise ShortyError(turl)
         return turl
 
-    def expand(self, tinyurl):
-        return get_redirect(tinyurl)
-
 isgd = Isgd()
 
 # cli.gs
@@ -317,6 +313,15 @@ def expand(self, tinyurl):
 
 cligs = Cligs()
 
+# tweetburner.com
+class Tweetburner(Service):
+
+    def shrink(self, bigurl):
+        resp = request('http://tweetburner.com/links', post_data='link[url]=%s' % bigurl)
+        return resp.read()
+
+tweetburner = Tweetburner()
+
 """Mapping of domain to service class"""
 services = {
     'sandbox': sandbox,
@@ -325,7 +330,8 @@ def expand(self, tinyurl):
     'urlborg.com': urlborg, 'ub0.cc': urlborg,
     'bit.ly': bitly,
     'is.gd': isgd,
-    'cli.gs': cligs
+    'cli.gs': cligs,
+    'tweetburner': tweetburner, 'twurl.nl': tweetburner
 }
 
 """
