Adding TokenExpired Exceptions

noahsilas · Jun 2, 2011 · 5263a58 · 5263a58
1 parent f379420
commit 5263a58
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/pytube/client.py b/pytube/client.py
@@ -224,7 +224,15 @@ def _gdata_request(self, url, query=None, data=None, headers=None, timeout=None)
             headers['X-GData-Key'] = 'key=' + self.dev_key
 
         request = urllib2.Request(url, data, headers)
-        return urllib2.urlopen(request, timeout=timeout)
+        try:
+            return urllib2.urlopen(request, timeout=timeout)
+        except urllib2.HTTPError, e:
+            if e.getcode() == 401:
+                e.reponse = e.read()
+                if 'TokenExpired' in e.response:
+                    raise pytube.exceptions.TokenExpired()
+                raise e
+            raise
 
     def _gdata_json(self, url, query=None, data=None, headers=None):
         query = query or {}

diff --git a/pytube/exceptions.py b/pytube/exceptions.py
@@ -11,6 +11,10 @@ def __init__(self, data):
         self.solved = None
 
 
+class TokenExpired(AuthenticationError):
+    """ You are using an expired authentication token """
+
+
 class VideoException(Exception):
     """ Failed to access a video """