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

Add method _api_start_session.

Browse source
This commit is contained in:
Ethan Dalool 2020-03-23 19:59:22 -07:00
parent 406531924c
commit ac442ced7f
1 changed files with 20 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
src/mega/mega.py
View file

@ -80,6 +80,24 @@ class Mega:
raise error_for_code(json_resp) raise error_for_code(json_resp)
return json_resp[0] return json_resp[0]
def _api_start_session(self, user, user_hash=None):
"""
The `us` request returns a dictionary like
{
'tsid': 'session' (if temporary session),
'csid': 'session' (if login session),
'privk': 'private key' (which must be decoded),
'k': 'master key' (which must be decoded),
'u': 'user id',
'ach': 1 (I don't know, it's always 1 for me)
}
"""
request = {'a': 'us', 'user': user}
if user_hash is not None:
request['uh'] = user_hash
resp = self._api_request(request)
return resp
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)
@ -111,7 +129,7 @@ class Mega:
) )
password_aes = str_to_a32(pbkdf2_key[:16]) password_aes = str_to_a32(pbkdf2_key[:16])
user_hash = base64_url_encode(pbkdf2_key[-16:]) user_hash = base64_url_encode(pbkdf2_key[-16:])
resp = self._api_request({'a': 'us', 'user': email, 'uh': user_hash}) resp = self._api_start_session(email, user_hash)
if isinstance(resp, int): if isinstance(resp, int):
raise RequestError(resp) raise RequestError(resp)
self._login_process(resp, password_aes) self._login_process(resp, password_aes)
@ -128,7 +146,7 @@ class Mega:
ts = base64_url_encode(ts) ts = base64_url_encode(ts)
user = self._api_request({'a': 'up', 'k': k, 'ts': ts}) user = self._api_request({'a': 'up', 'k': k, 'ts': ts})
resp = self._api_request({'a': 'us', 'user': user}) resp = self._api_start_session(user)
if isinstance(resp, int): if isinstance(resp, int):
raise RequestError(resp) raise RequestError(resp)
self._login_process(resp, password_key) self._login_process(resp, password_key)