dinislage/CouchPotatoServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Re-use resursion code

Browse Source
This commit is contained in:
Ruud
2015-02-10 13:15:08 +01:00
parent 578effc538
commit c1266a36e4
1 changed files with 9 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
couchpotato/core/downloaders/putio/main.py
View File

@@ -29,15 +29,16 @@ class PutIO(DownloaderBase):
return super(PutIO, self).__init__()
# This is a recusive function to check for the folders
def recursionFolder(self, client, folder, tfolder):
def recursionFolder(self, client, folder = 0, tfolder = ''):
files = client.File.list(folder)
for f in files:
if f.name == tfolder and f.content_type == "application/x-directory":
return f.id
elif f.content_type == "application/x-directory":
result = self.recursionFolder(client, f.id, tfolder)
if result != 0:
return result
if f.content_type == 'application/x-directory':
if f.name == tfolder:
return f.id
else:
result = self.recursionFolder(client, f.id, tfolder)
if result != 0:
return result
return 0
# This will check the root for the folder, and kick of recusively checking sub folder
@@ -45,16 +46,7 @@ class PutIO(DownloaderBase):
if folder == 0:
return 0
else:
files = client.File.list()
for f in files:
if f.name == folder and f.content_type == "application/x-directory":
return f.id
elif f.content_type == "application/x-directory":
result = self.recursionFolder(client, f.id, folder)
if result != 0:
return result
#If we get through the whole list and don't get a match we will use the root
return 0
return self.recursionFolder(client, 0, folder)
def download(self, data = None, media = None, filedata = None):
if not media: media = {}