Added stats for bit.ly

astagi · astagi · commit 2aa187e36cbf · 2011-07-23T18:30:34.000+02:00
diff --git a/services/bitly.py b/services/bitly.py
@@ -59,6 +59,17 @@ def expand(self, tinyurl):
             raise ShortyError(jdata['errorMessage'])
         return str(jdata['results'].values()[0]['longUrl'])
 
+    def stats(self, tinyurl):
+        if not self.login:
+            return get_redirect(tinyurl)
+        parameters, username_pass = self._setup()
+        parameters['shortUrl'] = tinyurl
+        resp = request('http://api.bit.ly/v3/clicks', parameters, username_pass)
+        jdata = json.loads(resp.read())
+        if jdata['status_code'] != 200:
+            raise ShortyError(jdata['errorMessage'])
+        return str(jdata['data']['clicks'][0]['global_clicks'])
+
     def qrcode(self, tinyurl):
         qrdata = request(tinyurl + '.qrcode').read()
         return qrdata
