voussoir/megapy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Dedent by raising early.

Browse source
This commit is contained in:
Ethan Dalool 2020-10-01 11:27:31 -07:00
parent 5609db3482
commit 400701f50d
1 changed files with 15 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/mega/mega.py
View file

@ -427,23 +427,23 @@ class Mega:
files = [self._node_data(file) for file in files]
if all('h' in file and 'k' in file for file in files):
request = [{'a': 'l', 'n': file['h']} for file in files]
public_handles = self._api_request(request)
if public_handles == -11:
raise errors.RequestError(
"Can't get a public link from that file "
"(is this a shared file?)"
)
urls = []
for (file, public_handle) in zip(files, public_handles):
decrypted_key = crypto.a32_to_base64(file['key'])
url = f'{self.schema}://{self.domain}/#!{public_handle}!{decrypted_key}'
urls.append(url)
return urls
else:
if not all('h' in file and 'k' in file for file in files):
raise errors.ValidationError('File id and key must be present')
request = [{'a': 'l', 'n': file['h']} for file in files]
public_handles = self._api_request(request)
if public_handles == -11:
raise errors.RequestError(
"Can't get a public link from that file "
"(is this a shared file?)"
)
urls = []
for (file, public_handle) in zip(files, public_handles):
decrypted_key = crypto.a32_to_base64(file['key'])
url = f'{self.schema}://{self.domain}/#!{public_handle}!{decrypted_key}'
urls.append(url)
return urls
def _node_data(self, node):
if isinstance(node, dict):
return node