From 872a4f4650d40bb8189723f1fd785d2fa8c57c78 Mon Sep 17 00:00:00 2001
From: dumaresq <dumaresq@gmail.com>
Date: Sun, 7 Sep 2014 17:59:16 -0400
Subject: [PATCH] Worked on geting Oauth and adding download status

---
 couchpotato/core/downloaders/putio/main.py | 73 ++++++++++++++++------
 1 file changed, 54 insertions(+), 19 deletions(-)

diff --git a/couchpotato/core/downloaders/putio/main.py b/couchpotato/core/downloaders/putio/main.py
index 10e6aa11..5e56b103 100644
--- a/couchpotato/core/downloaders/putio/main.py
+++ b/couchpotato/core/downloaders/putio/main.py
@@ -1,8 +1,10 @@
 import shutil
 
 from couchpotato.api import addApiView
-from couchpotato.core._base.downloader.main import DownloaderBase
+from couchpotato.core._base.downloader.main import DownloaderBase, ReleaseDownloadList
+from couchpotato.core.helpers.variable import cleanHost
 from couchpotato.core.logger import CPLog
+from couchpotato.environment import Env
 import api as pio
 
 log = CPLog(__name__)
@@ -12,7 +14,9 @@ autoload = 'Putiodownload'
 
 class PutIO(DownloaderBase):
     protocol = ['torrent', 'torrent_magnet']
-    status_support = False
+    status_support = True
+    client_id = '1575'
+    client_secret = '132qbpseq1ymwn83wus4'
 
     def __init__(self):
         addApiView('downloader.putio.getfrom', self.getFromPutio, docs = {
@@ -20,6 +24,7 @@ class PutIO(DownloaderBase):
         })
 
         addApiView('downloader.putio.auth_url', self.getAuthorizationUrl)
+        addApiView('downloader.putio.credentials', self.getCredentials)
 
         return super(PutIO, self).__init__()
 
@@ -35,12 +40,10 @@ class PutIO(DownloaderBase):
         # Need to constuct a the API url a better way.
         callbackurl = None
         if self.conf('download'):
-            callbackurl = 'http://' + self.conf('callback_host') + '/' + self.conf('url_base',
-                                                                                   section = 'core') + '/api/' + self.conf(
-                'api_key', section = 'core') + '/downloader.putiodownload.getfrom/'
-        client.Transfer.add_url(url, callback_url = callbackurl)
-
-        return True
+            callbackurl = 'http://' + self.conf('callback_host') + '/' + '%sdownloader.putio.getfrom/' %Env.get('api_base'.strip('/')) 
+        resp = client.Transfer.add_url(url, callback_url = callbackurl)
+        log.debug('resp is %s', resp.id);
+        return self.downloadReturnId(resp.id) 
 
     def test(self):
         try:
@@ -51,21 +54,53 @@ class PutIO(DownloaderBase):
             log.info('Failed to get file listing, check OAUTH_TOKEN')
             return False
 
-    def getAuthorizationUrl(self):
-        # See notification/twitter
-        pass
+    def getAuthorizationUrl(self, host = None, **kwargs):
+        callback_url = cleanHost(host) + '%sdownloader.putio.credentials/' % (Env.get('api_base').lstrip('/'))
+        log.info('callback_url is %s', callback_url)
+        #I can't figure out how to pass this into getCredentials so I'm saving it here
+        self.apicallhost = host;
+        oauth = pio.AuthHelper(client_id=self.client_id,client_secret=self.client_secret,redirect_uri=callback_url)
+        resp = oauth.authentication_url
+        log.info ('reps is %s,', resp)
+        return { 
+            'success': True,
+            'url': resp,
+        }
 
-    def getCredentials(self):
-        # Save oauth_token here to settings
-        pass
+
+    def getCredentials(self, **kwargs):
+        code = kwargs.get('code')
+        log.info('getCredentials Called with code: %s', code)
+        callback_url = cleanHost(self.apicallhost)  + '%sdownloader.putio.credentials/' % (Env.get('api_base').lstrip('/'))
+        log.info('callback is %s',callback_url)
+        oauth = pio.AuthHelper(client_id=self.client_id,client_secret=self.client_secret,redirect_uri=callback_url)
+        oauth_token = oauth.get_access_token(code)
+        log.info('oauth_token is: %s', oauth_token)
+        self.conf('oauth_token', value = oauth_token);
+        return 'redirect', Env.get('web_base') + 'settings/downloaders/'
 
     def getAllDownloadStatus(self, ids):
-        # See other downloaders for examples
-
-        # Check putio for status
-
+	log.debug('Checking putio download status.')
+        client = pio.Client(self.conf('oauth_token'))
+        transfers = client.Transfer.list()
+        log.debug(transfers);
+        release_downloads = ReleaseDownloadList(self)
+        for t in transfers:
+           if t.status == "COMPLETED" and self.conf('download') == False :
+              status = 'completed'
+           else:
+              #status = t.status.lower()
+              status = 'busy'
+           release_downloads.append({
+                 'id' : t.id,
+                 'name': t.name,
+                 'status': status,
+                 'timeleft': t.estimated_time,
+           })
+            
+        log.info(release_downloads)
         # Check "getFromPutio" progress
-        pass
+        return release_downloads
 
     def getFromPutio(self, **kwargs):