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

Don't put node_id into a list just to use [0].

Browse source
This commit is contained in:
Ethan Dalool 2020-03-23 19:51:50 -07:00
parent cafca108e9
commit 580b82c353
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/mega/mega.py
View file

@ -464,9 +464,9 @@ class Mega:
if type(target) == int: if type(target) == int:
if target in [NODE_TYPE_FILE, NODE_TYPE_DIR]: if target in [NODE_TYPE_FILE, NODE_TYPE_DIR]:
raise TypeError('Can\'t use file or dir node type.') raise TypeError('Can\'t use file or dir node type.')
node_id = self.get_node_by_type(target) node_id = self.get_node_by_type(target)[0]
else: else:
node_id = [target] node_id = target
files = self._api_request({'a': 'f', 'c': 1}) files = self._api_request({'a': 'f', 'c': 1})
# MERGE COMMON CODE WITH GET_FILES # MERGE COMMON CODE WITH GET_FILES
@ -475,7 +475,7 @@ class Mega:
self._init_shared_keys(files, shared_keys) self._init_shared_keys(files, shared_keys)
for file in files['f']: for file in files['f']:
processed_file = self._process_file(file, shared_keys) processed_file = self._process_file(file, shared_keys)
if processed_file['a'] and processed_file['p'] == node_id[0]: if processed_file['a'] and processed_file['p'] == node_id:
files_dict[file['h']] = processed_file files_dict[file['h']] = processed_file
return files_dict return files_dict