You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
2.7 KiB
88 lines
2.7 KiB
11 years ago
|
import shutil
|
||
|
|
|
||
|
|
from couchpotato.api import addApiView
|
||
|
|
from couchpotato.core._base.downloader.main import DownloaderBase
|
||
|
|
from couchpotato.core.logger import CPLog
|
||
|
|
import api as pio
|
||
|
|
|
||
|
|
log = CPLog(__name__)
|
||
|
|
|
||
|
|
autoload = 'Putiodownload'
|
||
|
|
|
||
|
|
|
||
|
|
class PutIO(DownloaderBase):
|
||
|
|
protocol = ['torrent', 'torrent_magnet']
|
||
|
|
status_support = False
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
addApiView('downloader.putio.getfrom', self.getFromPutio, docs = {
|
||
|
|
'desc': 'Allows you to download file from prom Put.io',
|
||
|
|
})
|
||
|
|
|
||
|
|
addApiView('downloader.putio.auth_url', self.getAuthorizationUrl)
|
||
|
|
|
||
|
|
return super(PutIO, self).__init__()
|
||
|
|
|
||
|
|
def download(self, data = None, media = None, filedata = None):
|
||
|
|
if not media: media = {}
|
||
|
|
if not data: data = {}
|
||
|
|
|
||
|
|
log.info('Sending "%s" to put.io', data.get('name'))
|
||
|
|
url = data.get('url')
|
||
|
|
|
||
|
|
client = pio.Client(self.conf('oauth_token'))
|
||
|
|
|
||
|
|
# 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
|
||
|
|
|
||
|
|
def test(self):
|
||
|
|
try:
|
||
|
|
client = pio.Client(self.conf('oauth_token'))
|
||
|
|
if client.File.list():
|
||
|
|
return True
|
||
|
|
except:
|
||
|
|
log.info('Failed to get file listing, check OAUTH_TOKEN')
|
||
|
|
return False
|
||
|
|
|
||
|
|
def getAuthorizationUrl(self):
|
||
|
|
# See notification/twitter
|
||
|
|
pass
|
||
|
|
|
||
|
|
def getCredentials(self):
|
||
|
|
# Save oauth_token here to settings
|
||
|
|
pass
|
||
|
|
|
||
|
|
def getAllDownloadStatus(self, ids):
|
||
|
|
# See other downloaders for examples
|
||
|
|
|
||
|
|
# Check putio for status
|
||
|
|
|
||
|
|
# Check "getFromPutio" progress
|
||
|
|
pass
|
||
|
|
|
||
|
|
def getFromPutio(self, **kwargs):
|
||
|
|
|
||
|
|
log.info('Put.io Download has been called')
|
||
|
|
client = pio.Client(self.conf('oauth_token'))
|
||
|
|
files = client.File.list()
|
||
|
|
|
||
|
|
tempdownloaddir = self.conf('tempdownload_dir')
|
||
|
|
downloaddir = self.conf('download_dir')
|
||
|
|
|
||
|
|
for f in files:
|
||
|
|
if str(f.id) == str(kwargs.get('file_id')):
|
||
|
|
# Need to read this in from somewhere
|
||
|
|
client.File.download(f, dest = tempdownloaddir, delete_after_download = self.conf('delete_file'))
|
||
|
|
shutil.move(tempdownloaddir + "/" + str(f.name), downloaddir)
|
||
|
|
|
||
|
|
# Mark status of file_id as "done" here for getAllDownloadStatus
|
||
|
|
|
||
|
|
return True
|