Implemented snipurl.com service.

joshthecoder · joshthecoder · commit 04ed0a4cc293 · 2009-08-03T01:21:40.000-05:00
diff --git a/README b/README
@@ -20,7 +20,7 @@ Supported URL shortening services for shrinking long urls:
     tinyurl.com                 burnurl.com
     tr.im                       hurl.ws
     urlborg.com                 xr.com
-    bit.ly
+    bit.ly                      snipurl.com
     is.gd
     cli.gs
     twurl.nl (tweetburner)
diff --git a/TODO b/TODO
@@ -1,7 +1,6 @@
 API available and ready to go!
 ++++++++++++++++++++++++++++++
 + u.mavrev.com
-+ snipr.com/snipurl.com/sn.im
 + adjix.com
 + metamark.net (xrl.us)
 + idek.net
diff --git a/services/snipurl.py b/services/snipurl.py
@@ -0,0 +1,49 @@
+## Shorty
+## Copyright 2009 Joshua Roesslein
+## See LICENSE
+
+## @url snipurl.com snipr.com sn.im snurl.com
+class Snipurl(Service):
+
+    def __init__(self, user=None, apikey=None):
+        self.user = user
+        self.apikey = apikey
+
+    def _test(self):
+        # prompt for username and apikey
+        self.user = raw_input('snipurl username: ')
+        self.apikey = raw_input('snipurl apikey: ')
+        Service._test(self)
+
+    def shrink(self, bigurl, custom=None, title=None, private_key=None,
+                owner=None, include_private_key=False):
+        if self.user is None or self.apikey is None:
+            raise ShortyError('Must set an user and apikey')
+        parameters = {
+            'sniplink': bigurl,
+            'snipuser': self.user,
+            'snipapi': self.apikey,
+            'snipformat': 'simple' 
+        }
+        if custom:
+            parameters['snipnick'] = custom
+        if title:
+            parameters['sniptitle'] = title
+        if private_key:
+            parameters['snippk'] = private_key
+        if owner:
+            parameters['snipowner'] = owner
+        if include_private_key:
+            parameters['snipformat_includepk'] = 'Y'
+        resp = request('http://snipurl.com/site/getsnip',
+                        post_data=urlencode(parameters))
+        return resp.read()
+
+    def expand(self, tinyurl):
+        # TODO: fetch detailed info
+        url = Service.expand(self, tinyurl)
+        if url.startswith('http'):
+            return url
+        else:
+            raise ShortyError('Invalid url')
+
diff --git a/shorty.py b/shorty.py
@@ -712,6 +712,53 @@ def expand(self, tinyurl):
 
 sandbox = Sandbox()
 
+# snipurl
+class Snipurl(Service):
+
+    def __init__(self, user=None, apikey=None):
+        self.user = user
+        self.apikey = apikey
+
+    def _test(self):
+        # prompt for username and apikey
+        self.user = raw_input('snipurl username: ')
+        self.apikey = raw_input('snipurl apikey: ')
+        Service._test(self)
+
+    def shrink(self, bigurl, custom=None, title=None, private_key=None,
+                owner=None, include_private_key=False):
+        if self.user is None or self.apikey is None:
+            raise ShortyError('Must set an user and apikey')
+        parameters = {
+            'sniplink': bigurl,
+            'snipuser': self.user,
+            'snipapi': self.apikey,
+            'snipformat': 'simple' 
+        }
+        if custom:
+            parameters['snipnick'] = custom
+        if title:
+            parameters['sniptitle'] = title
+        if private_key:
+            parameters['snippk'] = private_key
+        if owner:
+            parameters['snipowner'] = owner
+        if include_private_key:
+            parameters['snipformat_includepk'] = 'Y'
+        resp = request('http://snipurl.com/site/getsnip',
+                        post_data=urlencode(parameters))
+        return resp.read()
+
+    def expand(self, tinyurl):
+        # TODO: fetch detailed info
+        url = Service.expand(self, tinyurl)
+        if url.startswith('http'):
+            return url
+        else:
+            raise ShortyError('Invalid url')
+
+snipurl = Snipurl()
+
 # tinyurl
 class Tinyurl(Service):
 
@@ -746,6 +793,7 @@ def expand(self, tinyurl):
 bukme = Bukme()
 
 services = {
+    'snipr.com': snipurl,
     'chilp.it': chilpit,
     'bit.ly': bitly,
     'short.to': shortto,
@@ -754,6 +802,7 @@ def expand(self, tinyurl):
     'buk.me': bukme,
     'fon.gs': fongs,
     'ub0.cc': urlborg,
+    'snurl.com': snipurl,
     'fwd4.me': fwd4me,
     'xr.com': xr,
     'short.ie': shortie,
@@ -768,8 +817,10 @@ def expand(self, tinyurl):
     'cli.gs': cligs,
     'urlborg.com': urlborg,
     'is.gd': isgd,
+    'sn.im': snipurl,
     'tweetburner.com': tweetburner,
     'x.bb': xr,
     'tinyurl.com': tinyurl,
+    'snipurl.com': snipurl,
 }
 
