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

Don't fetch trash node id until needed.

Browse source 
Fetching trash's ID performs a call to get_files which can be
somewhat slow. Some use cases do not require ever calling get_files.
This commit is contained in:
Ethan Dalool 2020-11-11 21:58:34 -08:00
parent 40644921dc
commit 572ce2d627
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/mega/mega.py
View file

@ -35,7 +35,7 @@ class Mega:
self.sid = None self.sid = None
self.sequence_num = random.randint(0, 0xFFFFFFFF) self.sequence_num = random.randint(0, 0xFFFFFFFF)
self.request_id = crypto.make_id(10) self.request_id = crypto.make_id(10)
self._trash_folder_node_id = None self._cached_trash_folder_node_id = None
self.shared_keys = {} self.shared_keys = {}
self.requests_session = requests.Session() self.requests_session = requests.Session()
@ -110,12 +110,17 @@ class Mega:
resp = self._api_request(request) resp = self._api_request(request)
return resp return resp
@property
def _trash_folder_node_id(self):
if self._cached_trash_folder_node_id is None:
self._cached_trash_folder_node_id = self.get_node_by_type(NODE_TYPE_TRASH)[0]
return self._cached_trash_folder_node_id
def login(self, email=None, password=None): def login(self, email=None, password=None):
if email: if email:
self._login_user(email, password) self._login_user(email, password)
else: else:
self.login_anonymous() self.login_anonymous()
self._trash_folder_node_id = self.get_node_by_type(NODE_TYPE_TRASH)[0]
logger.info('Login complete') logger.info('Login complete')
return self return self