Added digg.com service.

joshthecoder · joshthecoder · commit 6bb66072f260 · 2009-07-24T23:00:19.000-05:00
diff --git a/README b/README
@@ -17,6 +17,7 @@ Supported URL shortening services for shrinking long urls:
     is.gd
     cli.gs
     twurl.nl (tweetburner)
+    digg.com
 
 Shorty can expand any url service by getting the redirect
 location by requesting the tiny url.
diff --git a/TODO b/TODO
@@ -2,6 +2,5 @@
 + buk.me
 + burnurl.com
 + chilp.it
-+ digg
 + fon.gs
 + fwd4.me
diff --git a/shorty.py b/shorty.py
@@ -322,6 +322,43 @@ def shrink(self, bigurl):
 
 tweetburner = Tweetburner()
 
+# 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)
+
+digg = Digg()
+
 """Mapping of domain to service class"""
 services = {
     'sandbox': sandbox,
@@ -331,7 +368,8 @@ def shrink(self, bigurl):
     'bit.ly': bitly,
     'is.gd': isgd,
     'cli.gs': cligs,
-    'tweetburner': tweetburner, 'twurl.nl': tweetburner
+    'tweetburner': tweetburner, 'twurl.nl': tweetburner,
+    'digg.com': digg
 }
 
 """
