Richard
GitHub
commit
6daa2bdf3c
15 changed files with 649 additions and 766 deletions
86
HISTORY.md
Normal file
86
HISTORY.md
Normal file
|
|
@ -0,0 +1,86 @@
|
|||
Release History
|
||||
===============
|
||||
|
||||
1.0.8 (unreleased)
|
||||
------------------
|
||||
|
||||
- Fixes find method returning the wrong file when more than one file
|
||||
exists with that name.
|
||||
- Handle new shared file URLS.
|
||||
|
||||
1.0.7 (2020-03-25)
|
||||
------------------
|
||||
|
||||
- Fix login by calculating public RSA exponent instead of hardcoding.
|
||||
|
||||
1.0.6 (2020-02-03)
|
||||
------------------
|
||||
|
||||
- Fixes RSA public exponent issue.
|
||||
- Switches dependency pycrypto to pycryptodome.
|
||||
|
||||
1.0.5 (2019-11-18)
|
||||
------------------
|
||||
|
||||
- Increase the wait time in between failed API request retries.
|
||||
|
||||
1.0.4 (2019-11-18)
|
||||
------------------
|
||||
|
||||
- Increase the wait time in between failed API request retries.
|
||||
|
||||
1.0.3 (2019-11-12)
|
||||
------------------
|
||||
|
||||
- Fixes broken `download` method.
|
||||
- Changes `download` and `download_url` methods to return the path to
|
||||
the downloaded file, previously returned `None`.
|
||||
- Added LICENSE.
|
||||
|
||||
1.0.2 (2019-11-07)
|
||||
------------------
|
||||
|
||||
- Reverts, "Replace pycrypto dependency with pycryptodome" as breaks
|
||||
login process.
|
||||
|
||||
1.0.1 (2019-11-06)
|
||||
------------------
|
||||
|
||||
- When a request fails due to EAGAIN response, retry with exp backoff
|
||||
up to 20 seconds.
|
||||
- Adds logging, removes print statements.
|
||||
- Replace pycrypto dependency with pycryptodome.
|
||||
- Removes Python 2 specific code.
|
||||
|
||||
1.0.0 (2019-10-31)
|
||||
------------------
|
||||
|
||||
- Removes broken method `get_contacts()`.
|
||||
- Adds support for login with a v2 Mega user account.
|
||||
- Adds `export()` method to share a file or folder, returning public
|
||||
share URL with key.
|
||||
- Adds code, message attrs to RequestError exception, makes message in
|
||||
raised exceptions include more details.
|
||||
- Alters `create_folder()` to accept a path including multiple sub
|
||||
directories, adds support to create them all (similar to 'mkdir -p'
|
||||
on unix systems).
|
||||
- Adds `exclude_deleted=True` optional arg to `find()` method, to
|
||||
exclude deleted nodes from results.
|
||||
|
||||
0.9.20 (2019-10-17)
|
||||
-------------------
|
||||
|
||||
- Python 3 bugfix to `upload` method.
|
||||
|
||||
0.9.19 (2019-10-16)
|
||||
-------------------
|
||||
|
||||
- Python 3 support and bugfixes.
|
||||
- Update packaging code.
|
||||
- Added changelog.
|
||||
|
||||
0.9.18 (2013-07-04)
|
||||
-------------------
|
||||
|
||||
- Unknown
|
||||
|
||||
79
HISTORY.rst
79
HISTORY.rst
|
|
@ -1,79 +0,0 @@
|
|||
.. :changelog:
|
||||
|
||||
Release History
|
||||
===============
|
||||
|
||||
1.0.7 (2020-03-25)
|
||||
------------------
|
||||
|
||||
- Fix login by calculating public RSA exponent instead of hardcoding.
|
||||
|
||||
|
||||
1.0.6 (2020-02-03)
|
||||
------------------
|
||||
|
||||
- Fixes RSA public exponent issue.
|
||||
- Switches dependency pycrypto to pycryptodome.
|
||||
|
||||
|
||||
1.0.5 (2019-11-18)
|
||||
------------------
|
||||
|
||||
- Increase the wait time in between failed API request retries.
|
||||
|
||||
|
||||
1.0.4 (2019-11-18)
|
||||
------------------
|
||||
|
||||
- Increase the wait time in between failed API request retries.
|
||||
|
||||
|
||||
1.0.3 (2019-11-12)
|
||||
------------------
|
||||
|
||||
- Fixes broken ``download`` method.
|
||||
- Changes ``download`` and ``download_url`` methods to return the path to the downloaded file, previously returned ``None``.
|
||||
- Added LICENSE.
|
||||
|
||||
|
||||
1.0.2 (2019-11-07)
|
||||
------------------
|
||||
|
||||
- Reverts, "Replace pycrypto dependency with pycryptodome" as breaks login process.
|
||||
|
||||
|
||||
1.0.1 (2019-11-06)
|
||||
------------------
|
||||
|
||||
- When a request fails due to EAGAIN response, retry with exp backoff up to 20 seconds.
|
||||
- Adds logging, removes print statements.
|
||||
- Replace pycrypto dependency with pycryptodome.
|
||||
- Removes Python 2 specific code.
|
||||
|
||||
|
||||
1.0.0 (2019-10-31)
|
||||
------------------
|
||||
|
||||
- Removes broken method ``get_contacts()``.
|
||||
- Adds support for login with a v2 Mega user account.
|
||||
- Adds ``export()`` method to share a file or folder, returning public share URL with key.
|
||||
- Adds code, message attrs to RequestError exception, makes message in raised exceptions include more details.
|
||||
- Alters ``create_folder()`` to accept a path including multiple sub directories, adds support to create them all (similar to 'mkdir -p' on unix systems).
|
||||
- Adds ``exclude_deleted=True`` optional arg to ``find()`` method, to exclude deleted nodes from results.
|
||||
|
||||
0.9.20 (2019-10-17)
|
||||
-------------------
|
||||
|
||||
- Python 3 bugfix to ``upload`` method.
|
||||
|
||||
0.9.19 (2019-10-16)
|
||||
-------------------
|
||||
|
||||
- Python 3 support and bugfixes.
|
||||
- Update packaging code.
|
||||
- Added changelog.
|
||||
|
||||
0.9.18 (2013-07-04)
|
||||
-------------------
|
||||
|
||||
- Unknown
|
||||
|
|
@ -1,9 +1,9 @@
|
|||
include HISTORY.rst
|
||||
include README.rst
|
||||
include HISTORY.md
|
||||
include README.md
|
||||
include requirements.txt
|
||||
|
||||
recursive-include tests *
|
||||
recursive-exclude * __pycache__
|
||||
recursive-exclude * *.py[co]
|
||||
|
||||
recursive-include docs *.rst conf.py Makefile make.bat
|
||||
recursive-include docs *.md conf.py Makefile make.bat
|
||||
|
|
|
|||
169
README.md
Normal file
169
README.md
Normal file
|
|
@ -0,0 +1,169 @@
|
|||
Mega.py
|
||||
=======
|
||||
|
||||
[](https://travis-ci.org/odwyersoftware/mega.py)
|
||||
[](https://crate.io/packages/mega.py/) [](https://pypi.org/project/mega.py/)
|
||||
|
||||
Python library for the [Mega.co.nz](https://mega.nz/aff=Zo6IxNaHw14)
|
||||
API, currently supporting:
|
||||
|
||||
- login
|
||||
- uploading
|
||||
- downloading
|
||||
- deleting
|
||||
- searching
|
||||
- sharing
|
||||
- renaming
|
||||
- moving files
|
||||
|
||||
This is a work in progress, further functionality coming shortly.
|
||||
|
||||
For more detailed information see API\_INFO.md
|
||||
|
||||
How To Use
|
||||
----------
|
||||
|
||||
### Create a Mega account
|
||||
|
||||
First, [create an account with Mega](https://mega.nz/aff=Zo6IxNaHw14) .
|
||||
|
||||
### Install mega.py package
|
||||
|
||||
Run the following command, or run setup from the latest github source.
|
||||
|
||||
```python
|
||||
pip install mega.py
|
||||
```
|
||||
|
||||
### Import mega.py
|
||||
|
||||
```python
|
||||
from mega import Mega
|
||||
```
|
||||
|
||||
### Create an instance of Mega.py
|
||||
|
||||
```python
|
||||
mega = Mega()
|
||||
```
|
||||
|
||||
### Login to Mega
|
||||
|
||||
```python
|
||||
m = mega.login(email, password)
|
||||
# login using a temporary anonymous account
|
||||
m = mega.login()
|
||||
```
|
||||
|
||||
### Get user details
|
||||
|
||||
```python
|
||||
details = m.get_user()
|
||||
```
|
||||
|
||||
### Get account balance (Pro accounts only)
|
||||
|
||||
```python
|
||||
balance = m.get_balance()
|
||||
```
|
||||
|
||||
### Get account disk quota
|
||||
|
||||
```python
|
||||
quota = m.get_quota()
|
||||
```
|
||||
|
||||
### Get account storage space
|
||||
|
||||
```python
|
||||
# specify unit output kilo, mega, gig, else bytes will output
|
||||
space = m.get_storage_space(kilo=True)
|
||||
```
|
||||
|
||||
### Get account files
|
||||
|
||||
```python
|
||||
files = m.get_files()
|
||||
```
|
||||
|
||||
### Upload a file, and get its public link
|
||||
|
||||
```python
|
||||
file = m.upload('myfile.doc')
|
||||
m.get_upload_link(file)
|
||||
# see mega.py for destination and filename options
|
||||
```
|
||||
|
||||
### Export a file or folder
|
||||
|
||||
```python
|
||||
public_exported_web_link = m.export('myfile.doc')
|
||||
public_exported_web_link = m.export('my_mega_folder/my_sub_folder_to_share')
|
||||
# e.g. https://mega.nz/#F!WlVl1CbZ!M3wmhwZDENMNUJoBsdzFng
|
||||
```
|
||||
|
||||
### Find a file or folder
|
||||
|
||||
```python
|
||||
folder = m.find('my_mega_folder')
|
||||
# Excludes results which are in the Trash folder (i.e. deleted)
|
||||
folder = m.find('my_mega_folder', exclude_deleted=True)
|
||||
```
|
||||
|
||||
### Upload a file to a destination folder
|
||||
|
||||
```python
|
||||
folder = m.find('my_mega_folder')
|
||||
m.upload('myfile.doc', folder[0])
|
||||
```
|
||||
|
||||
### Download a file from URL or file obj, optionally specify destination folder
|
||||
|
||||
```python
|
||||
file = m.find('myfile.doc')
|
||||
m.download(file)
|
||||
m.download_url('https://mega.co.nz/#!utYjgSTQ!OM4U3V5v_W4N5edSo0wolg1D5H0fwSrLD3oLnLuS9pc')
|
||||
m.download(file, '/home/john-smith/Desktop')
|
||||
# specify optional download filename (download_url() supports this also)
|
||||
m.download(file, '/home/john-smith/Desktop', 'myfile.zip')
|
||||
```
|
||||
|
||||
### Import a file from URL, optionally specify destination folder
|
||||
|
||||
```python
|
||||
m.import_public_url('https://mega.co.nz/#!utYjgSTQ!OM4U3V5v_W4N5edSo0wolg1D5H0fwSrLD3oLnLuS9pc')
|
||||
folder_node = m.find('Documents')[1]
|
||||
m.import_public_url('https://mega.co.nz/#!utYjgSTQ!OM4U3V5v_W4N5edSo0wolg1D5H0fwSrLD3oLnLuS9pc', dest_node=folder_node)
|
||||
```
|
||||
|
||||
### Create a folder
|
||||
|
||||
```python
|
||||
m.create_folder('new_folder')
|
||||
m.create_folder('new_folder/sub_folder/subsub_folder')
|
||||
```
|
||||
|
||||
Returns a dict of folder node name and node\_id, e.g.
|
||||
|
||||
```python
|
||||
{
|
||||
'new_folder': 'qpFhAYwA',
|
||||
'sub_folder': '2pdlmY4Z',
|
||||
'subsub_folder': 'GgMFCKLZ'
|
||||
}
|
||||
```
|
||||
|
||||
### Rename a file or a folder
|
||||
|
||||
```python
|
||||
file = m.find('myfile.doc')
|
||||
m.rename(file, 'my_file.doc')
|
||||
```
|
||||
|
||||
## Contact Support
|
||||
|
||||
For paid priority support contact [mega@odwyer.software](mailto:mega@odwyer.software).
|
||||
**[UK Python Development Agency](https://odwyer.software/)**
|
||||
|
||||
|
||||
204
README.rst
204
README.rst
|
|
@ -1,204 +0,0 @@
|
|||
**NOTICE**: If you're reading this on GitHub.com please be aware this is
|
||||
a mirror of the primary remote located at https://code.richard.do/richardARPANET/mega.py_. Please direct issues and
|
||||
pull requests there.
|
||||
|
||||
--------------
|
||||
|
||||
.. _megapy:
|
||||
|
||||
Mega.py
|
||||
=======
|
||||
|
||||
|Build Status| |Downloads|
|
||||
|
||||
Python library for the `Mega.co.nz <https://mega.nz/aff=Zo6IxNaHw14>`_ API, currently supporting:
|
||||
|
||||
- login
|
||||
- uploading
|
||||
- downloading
|
||||
- deleting
|
||||
- searching
|
||||
- sharing
|
||||
- renaming
|
||||
- moving files
|
||||
|
||||
This is a work in progress, further functionality coming shortly.
|
||||
|
||||
For more detailed information see API_INFO.md
|
||||
|
||||
How To Use
|
||||
----------
|
||||
|
||||
.. _create-mega-account:
|
||||
|
||||
Create a Mega account
|
||||
~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
First, `create an account with Mega <https://mega.nz/aff=Zo6IxNaHw14>`_
|
||||
.
|
||||
|
||||
.. _install-megapy-package:
|
||||
|
||||
Install mega.py package
|
||||
~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
Run the following command, or run setup from the latest github source
|
||||
pip install mega.py
|
||||
|
||||
.. _import-megapy:
|
||||
|
||||
Import mega.py
|
||||
~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
from mega import Mega
|
||||
|
||||
.. _create-an-instance-of-megapy:
|
||||
|
||||
Create an instance of Mega.py
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
mega = Mega()
|
||||
|
||||
Login to Mega
|
||||
~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
m = mega.login(email, password)
|
||||
# login using a temporary anonymous account
|
||||
m = mega.login()
|
||||
|
||||
Get user details
|
||||
~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
details = m.get_user()
|
||||
|
||||
Get account balance (Pro accounts only)
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
balance = m.get_balance()
|
||||
|
||||
Get account disk quota
|
||||
~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
quota = m.get_quota()
|
||||
|
||||
Get account storage space
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
# specify unit output kilo, mega, gig, else bytes will output
|
||||
space = m.get_storage_space(kilo=True)
|
||||
|
||||
Get account files
|
||||
~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
files = m.get_files()
|
||||
|
||||
Upload a file, and get its public link
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
file = m.upload('myfile.doc')
|
||||
m.get_upload_link(file)
|
||||
# see mega.py for destination and filename options
|
||||
|
||||
Export a file or folder
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
public_exported_web_link = m.export('myfile.doc')
|
||||
public_exported_web_link = m.export('my_mega_folder/my_sub_folder_to_share')
|
||||
# e.g. https://mega.nz/#F!WlVl1CbZ!M3wmhwZDENMNUJoBsdzFng
|
||||
|
||||
Find a file or folder
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
folder = m.find('my_mega_folder')
|
||||
# Excludes results which are in the Trash folder (i.e. deleted)
|
||||
folder = m.find('my_mega_folder', exclude_deleted=True)
|
||||
|
||||
Upload a file to a destination folder
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
folder = m.find('my_mega_folder')
|
||||
m.upload('myfile.doc', folder[0])
|
||||
|
||||
Download a file from URL or file obj, optionally specify destination folder
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
file = m.find('myfile.doc')
|
||||
m.download(file)
|
||||
m.download_url('https://mega.co.nz/#!utYjgSTQ!OM4U3V5v_W4N5edSo0wolg1D5H0fwSrLD3oLnLuS9pc')
|
||||
m.download(file, '/home/john-smith/Desktop')
|
||||
# specify optional download filename (download_url() supports this also)
|
||||
m.download(file, '/home/john-smith/Desktop', 'myfile.zip')
|
||||
|
||||
Import a file from URL, optionally specify destination folder
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
m.import_public_url('https://mega.co.nz/#!utYjgSTQ!OM4U3V5v_W4N5edSo0wolg1D5H0fwSrLD3oLnLuS9pc')
|
||||
folder_node = m.find('Documents')[1]
|
||||
m.import_public_url('https://mega.co.nz/#!utYjgSTQ!OM4U3V5v_W4N5edSo0wolg1D5H0fwSrLD3oLnLuS9pc', dest_node=folder_node)
|
||||
|
||||
Create a folder
|
||||
~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
m.create_folder('new_folder')
|
||||
m.create_folder('new_folder/sub_folder/subsub_folder')
|
||||
|
||||
Returns a dict of folder node name and node_id, e.g.
|
||||
|
||||
.. code:: python
|
||||
|
||||
{
|
||||
'new_folder': 'qpFhAYwA',
|
||||
'sub_folder': '2pdlmY4Z',
|
||||
'subsub_folder': 'GgMFCKLZ'
|
||||
}
|
||||
|
||||
Rename a file or a folder
|
||||
~~~~~~~~~~~~~~~~~~~~~~~~~
|
||||
|
||||
.. code:: python
|
||||
|
||||
file = m.find('myfile.doc')
|
||||
m.rename(file, 'my_file.doc')
|
||||
|
||||
|
||||
~
|
||||
|
||||
.. _`https://code.richard.do/richardARPANET/mega.py`: https://code.richard.do/richardARPANET/mega.py
|
||||
.. _`https://github.com/meganz/sdk`: https://github.com/meganz/sdk
|
||||
|
||||
.. |Build Status| image:: https://travis-ci.org/richardARPANET/mega.py.png?branch=master
|
||||
:target: https://travis-ci.org/richardARPANET/mega.py
|
||||
.. |Downloads| image:: https://pypip.in/d/mega.py/badge.png
|
||||
:target: https://crate.io/packages/mega.py/
|
||||
|
|
@ -1,4 +1,5 @@
|
|||
import os
|
||||
import uuid
|
||||
from mega import Mega
|
||||
|
||||
|
||||
|
|
@ -8,7 +9,7 @@ def test():
|
|||
comment/uncomment lines to test various parts of the API
|
||||
see readme.md for more information
|
||||
"""
|
||||
|
||||
unique = str(uuid.uuid4())
|
||||
# user details
|
||||
email = os.environ['EMAIL']
|
||||
password = os.environ['PASS']
|
||||
|
|
@ -36,10 +37,11 @@ def test():
|
|||
print((files[file]))
|
||||
|
||||
# upload file
|
||||
print((m.upload('examples.py')))
|
||||
print((m.upload(filename='examples.py',
|
||||
dest_filename=f'examples_{unique}.py')))
|
||||
|
||||
# search for a file in account
|
||||
file = m.find('examples.py')
|
||||
file = m.find(f'examples_{unique}.py')
|
||||
|
||||
if file:
|
||||
# get public link
|
||||
|
|
|
|||
|
|
@ -1,17 +1,20 @@
|
|||
-r requirements.txt
|
||||
pytest
|
||||
ipdb
|
||||
flake8
|
||||
pep8-naming
|
||||
autoflake
|
||||
mccabe
|
||||
yapf
|
||||
tox
|
||||
coverage
|
||||
pytest-cov
|
||||
zest.releaser
|
||||
setuptools
|
||||
twine
|
||||
wheel
|
||||
rope
|
||||
pytest-mock
|
||||
pytest==5.4.3
|
||||
ipdb==0.13.3
|
||||
flake8==3.8.3
|
||||
pep8-naming==0.11.1
|
||||
autoflake==1.3.1
|
||||
mccabe==0.6.1
|
||||
brunette==0.1.5
|
||||
tox==3.15.2
|
||||
coverage==5.1
|
||||
pytest-cov==2.10.0
|
||||
zest.releaser==6.20.1
|
||||
setuptools==47.3.1
|
||||
twine==3.2.0
|
||||
wheel==0.34.2
|
||||
rope==0.17.0
|
||||
pytest-mock==3.1.1
|
||||
brunette==0.1.5
|
||||
lock-requirements==0.1.1
|
||||
requests-mock==1.8.0
|
||||
|
|
|
|||
16
setup.cfg
16
setup.cfg
|
|
@ -13,18 +13,10 @@ exclude = .git,__pycache__,legacy,build,dist,.tox
|
|||
max-complexity = 15
|
||||
ignore = E741,W504,W503
|
||||
|
||||
[yapf]
|
||||
based_on_style = pep8
|
||||
spaces_before_comment = 2
|
||||
split_before_logical_operator = true
|
||||
indent_width = 4
|
||||
split_complex_comprehension = true
|
||||
column_limit = 79
|
||||
dedent_closing_brackets = true
|
||||
spaces_around_power_operator = true
|
||||
no_spaces_around_selected_binary_operators = false
|
||||
split_penalty_import_names = 500
|
||||
join_multiple_lines = true
|
||||
[tool:brunette]
|
||||
line-length = 79
|
||||
verbose = true
|
||||
single-quotes = true
|
||||
|
||||
[coverage:run]
|
||||
omit =
|
||||
|
|
|
|||
49
setup.py
49
setup.py
|
|
@ -14,33 +14,30 @@ os.chdir(os.path.normpath(os.path.join(os.path.abspath(__file__), os.pardir)))
|
|||
with open('requirements.txt') as f:
|
||||
install_requires = f.read().splitlines()
|
||||
|
||||
with open('README.rst', 'r', encoding='utf-8') as rm_file:
|
||||
with open('README.md', 'r', encoding='utf-8') as rm_file:
|
||||
readme = rm_file.read()
|
||||
|
||||
with open('HISTORY.rst', 'r', encoding='utf-8') as hist_file:
|
||||
with open('HISTORY.md', 'r', encoding='utf-8') as hist_file:
|
||||
history = hist_file.read()
|
||||
|
||||
setup(
|
||||
name='mega.py',
|
||||
version='1.0.7',
|
||||
packages=find_packages('src', exclude=('tests', )),
|
||||
package_dir={'': 'src'},
|
||||
include_package_data=True,
|
||||
zip_safe=False,
|
||||
description='Python lib for the Mega.co.nz API',
|
||||
long_description=readme + '\n\n' + history,
|
||||
author='Richard O\'Dwyer',
|
||||
author_email='richard@richard.do',
|
||||
license='Creative Commons Attribution-Noncommercial-Share Alike license',
|
||||
install_requires=install_requires,
|
||||
classifiers=[
|
||||
'Intended Audience :: Developers',
|
||||
'Operating System :: OS Independent',
|
||||
'Programming Language :: Python',
|
||||
'Programming Language :: Python :: 3',
|
||||
'Programming Language :: Python :: 3.6',
|
||||
'Programming Language :: Python :: 3.7',
|
||||
'Programming Language :: Python :: 3.8',
|
||||
'Topic :: Internet :: WWW/HTTP',
|
||||
]
|
||||
)
|
||||
setup(name='mega.py',
|
||||
version='1.0.8.dev0',
|
||||
packages=find_packages('src', exclude=('tests', )),
|
||||
package_dir={'': 'src'},
|
||||
include_package_data=True,
|
||||
zip_safe=False,
|
||||
url='https://github.com/odwyersoftware/mega.py',
|
||||
description='Python lib for the Mega.co.nz API',
|
||||
long_description=readme + '\n\n' + history,
|
||||
long_description_content_type='text/markdown',
|
||||
author='O\'Dwyer Software',
|
||||
author_email='hello@odwyer.software',
|
||||
license='Creative Commons Attribution-Noncommercial-Share Alike license',
|
||||
install_requires=install_requires,
|
||||
classifiers=[
|
||||
'Intended Audience :: Developers',
|
||||
'Operating System :: OS Independent',
|
||||
'Programming Language :: Python',
|
||||
'Programming Language :: Python :: 3',
|
||||
'Topic :: Internet :: WWW/HTTP',
|
||||
])
|
||||
|
|
|
|||
|
|
@ -65,15 +65,13 @@ def prepare_key(arr):
|
|||
|
||||
|
||||
def encrypt_key(a, key):
|
||||
return sum(
|
||||
(aes_cbc_encrypt_a32(a[i:i + 4], key) for i in range(0, len(a), 4)), ()
|
||||
)
|
||||
return sum((aes_cbc_encrypt_a32(a[i:i + 4], key)
|
||||
for i in range(0, len(a), 4)), ())
|
||||
|
||||
|
||||
def decrypt_key(a, key):
|
||||
return sum(
|
||||
(aes_cbc_decrypt_a32(a[i:i + 4], key) for i in range(0, len(a), 4)), ()
|
||||
)
|
||||
return sum((aes_cbc_decrypt_a32(a[i:i + 4], key)
|
||||
for i in range(0, len(a), 4)), ())
|
||||
|
||||
|
||||
def encrypt_attr(attr, key):
|
||||
|
|
|
|||
|
|
@ -6,66 +6,41 @@ class ValidationError(Exception):
|
|||
|
||||
|
||||
_CODE_TO_DESCRIPTIONS = {
|
||||
-1: (
|
||||
'EINTERNAL',
|
||||
(
|
||||
'An internal error has occurred. Please submit a bug report, '
|
||||
'detailing the exact circumstances in which this error occurred'
|
||||
)
|
||||
),
|
||||
-1: ('EINTERNAL',
|
||||
('An internal error has occurred. Please submit a bug report, '
|
||||
'detailing the exact circumstances in which this error occurred')),
|
||||
-2: ('EARGS', 'You have passed invalid arguments to this command'),
|
||||
-3: (
|
||||
'EAGAIN',
|
||||
(
|
||||
'(always at the request level) A temporary congestion or server '
|
||||
'malfunction prevented your request from being processed. '
|
||||
'No data was altered. Retry. Retries must be spaced with '
|
||||
'exponential backoff'
|
||||
)
|
||||
),
|
||||
-4: (
|
||||
'ERATELIMIT',
|
||||
(
|
||||
'You have exceeded your command weight per time quota. Please '
|
||||
'wait a few seconds, then try again (this should never happen '
|
||||
'in sane real-life applications)'
|
||||
)
|
||||
),
|
||||
-3: ('EAGAIN',
|
||||
('(always at the request level) A temporary congestion or server '
|
||||
'malfunction prevented your request from being processed. '
|
||||
'No data was altered. Retry. Retries must be spaced with '
|
||||
'exponential backoff')),
|
||||
-4: ('ERATELIMIT',
|
||||
('You have exceeded your command weight per time quota. Please '
|
||||
'wait a few seconds, then try again (this should never happen '
|
||||
'in sane real-life applications)')),
|
||||
-5: ('EFAILED', 'The upload failed. Please restart it from scratch'),
|
||||
-6: (
|
||||
'ETOOMANY',
|
||||
'Too many concurrent IP addresses are accessing this upload target URL'
|
||||
),
|
||||
-7: (
|
||||
'ERANGE',
|
||||
(
|
||||
'The upload file packet is out of range or not starting and '
|
||||
'ending on a chunk boundary'
|
||||
)
|
||||
),
|
||||
-8: (
|
||||
'EEXPIRED',
|
||||
(
|
||||
'The upload target URL you are trying to access has expired. '
|
||||
'Please request a fresh one'
|
||||
)
|
||||
),
|
||||
-6:
|
||||
('ETOOMANY',
|
||||
'Too many concurrent IP addresses are accessing this upload target URL'),
|
||||
-7:
|
||||
('ERANGE', ('The upload file packet is out of range or not starting and '
|
||||
'ending on a chunk boundary')),
|
||||
-8: ('EEXPIRED',
|
||||
('The upload target URL you are trying to access has expired. '
|
||||
'Please request a fresh one')),
|
||||
-9: ('ENOENT', 'Object (typically, node or user) not found'),
|
||||
-10: ('ECIRCULAR', 'Circular linkage attempted'),
|
||||
-11: (
|
||||
'EACCESS',
|
||||
'Access violation (e.g., trying to write to a read-only share)'
|
||||
),
|
||||
-11: ('EACCESS',
|
||||
'Access violation (e.g., trying to write to a read-only share)'),
|
||||
-12: ('EEXIST', 'Trying to create an object that already exists'),
|
||||
-13: ('EINCOMPLETE', 'Trying to access an incomplete resource'),
|
||||
-14: ('EKEY', 'A decryption operation failed (never returned by the API)'),
|
||||
-15: ('ESID', 'Invalid or expired user session, please relogin'),
|
||||
-16: ('EBLOCKED', 'User blocked'),
|
||||
-17: ('EOVERQUOTA', 'Request over quota'),
|
||||
-18: (
|
||||
'ETEMPUNAVAIL',
|
||||
'Resource temporarily not available, please try again later'
|
||||
),
|
||||
-18: ('ETEMPUNAVAIL',
|
||||
'Resource temporarily not available, please try again later'),
|
||||
-19: ('ETOOMANYCONNECTIONS', 'many connections on this resource'),
|
||||
-20: ('EWRITE', 'Write failed'),
|
||||
-21: ('EREAD', 'Read failed'),
|
||||
|
|
|
|||
560
src/mega/mega.py
560
src/mega/mega.py
|
|
@ -18,12 +18,11 @@ import requests
|
|||
from tenacity import retry, wait_exponential, retry_if_exception_type
|
||||
|
||||
from .errors import ValidationError, RequestError
|
||||
from .crypto import (
|
||||
a32_to_base64, encrypt_key, base64_url_encode, encrypt_attr, base64_to_a32,
|
||||
base64_url_decode, decrypt_attr, a32_to_str, get_chunks, str_to_a32,
|
||||
decrypt_key, mpi_to_int, stringhash, prepare_key, make_id, makebyte,
|
||||
modular_inverse
|
||||
)
|
||||
from .crypto import (a32_to_base64, encrypt_key, base64_url_encode,
|
||||
encrypt_attr, base64_to_a32, base64_url_decode,
|
||||
decrypt_attr, a32_to_str, get_chunks, str_to_a32,
|
||||
decrypt_key, mpi_to_int, stringhash, prepare_key, make_id,
|
||||
makebyte, modular_inverse)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
|
@ -64,13 +63,11 @@ class Mega:
|
|||
user_hash = stringhash(email, password_aes)
|
||||
else:
|
||||
# v2 user account
|
||||
pbkdf2_key = hashlib.pbkdf2_hmac(
|
||||
hash_name='sha512',
|
||||
password=password.encode(),
|
||||
salt=a32_to_str(user_salt),
|
||||
iterations=100000,
|
||||
dklen=32
|
||||
)
|
||||
pbkdf2_key = hashlib.pbkdf2_hmac(hash_name='sha512',
|
||||
password=password.encode(),
|
||||
salt=a32_to_str(user_salt),
|
||||
iterations=100000,
|
||||
dklen=32)
|
||||
password_aes = str_to_a32(pbkdf2_key[:16])
|
||||
user_hash = base64_url_encode(pbkdf2_key[-16:])
|
||||
resp = self._api_request({'a': 'us', 'user': email, 'uh': user_hash})
|
||||
|
|
@ -84,20 +81,16 @@ class Mega:
|
|||
password_key = [random.randint(0, 0xFFFFFFFF)] * 4
|
||||
session_self_challenge = [random.randint(0, 0xFFFFFFFF)] * 4
|
||||
|
||||
user = self._api_request(
|
||||
{
|
||||
'a':
|
||||
'up',
|
||||
'k':
|
||||
a32_to_base64(encrypt_key(master_key, password_key)),
|
||||
'ts':
|
||||
base64_url_encode(
|
||||
a32_to_str(session_self_challenge) + a32_to_str(
|
||||
encrypt_key(session_self_challenge, master_key)
|
||||
)
|
||||
)
|
||||
}
|
||||
)
|
||||
user = self._api_request({
|
||||
'a':
|
||||
'up',
|
||||
'k':
|
||||
a32_to_base64(encrypt_key(master_key, password_key)),
|
||||
'ts':
|
||||
base64_url_encode(
|
||||
a32_to_str(session_self_challenge) +
|
||||
a32_to_str(encrypt_key(session_self_challenge, master_key)))
|
||||
})
|
||||
|
||||
resp = self._api_request({'a': 'us', 'user': user})
|
||||
if isinstance(resp, int):
|
||||
|
|
@ -110,15 +103,13 @@ class Mega:
|
|||
if 'tsid' in resp:
|
||||
tsid = base64_url_decode(resp['tsid'])
|
||||
key_encrypted = a32_to_str(
|
||||
encrypt_key(str_to_a32(tsid[:16]), self.master_key)
|
||||
)
|
||||
encrypt_key(str_to_a32(tsid[:16]), self.master_key))
|
||||
if key_encrypted == tsid[-16:]:
|
||||
self.sid = resp['tsid']
|
||||
elif 'csid' in resp:
|
||||
encrypted_rsa_private_key = base64_to_a32(resp['privk'])
|
||||
rsa_private_key = decrypt_key(
|
||||
encrypted_rsa_private_key, self.master_key
|
||||
)
|
||||
rsa_private_key = decrypt_key(encrypted_rsa_private_key,
|
||||
self.master_key)
|
||||
|
||||
private_key = a32_to_str(rsa_private_key)
|
||||
# The private_key contains 4 MPI integers concatenated together.
|
||||
|
|
@ -158,10 +149,8 @@ class Mega:
|
|||
sid = binascii.unhexlify('0' + sid if len(sid) % 2 else sid)
|
||||
self.sid = base64_url_encode(sid[:43])
|
||||
|
||||
@retry(
|
||||
retry=retry_if_exception_type(RuntimeError),
|
||||
wait=wait_exponential(multiplier=2, min=2, max=60)
|
||||
)
|
||||
@retry(retry=retry_if_exception_type(RuntimeError),
|
||||
wait=wait_exponential(multiplier=2, min=2, max=60))
|
||||
def _api_request(self, data):
|
||||
params = {'id': self.sequence_num}
|
||||
self.sequence_num += 1
|
||||
|
|
@ -174,24 +163,42 @@ class Mega:
|
|||
data = [data]
|
||||
|
||||
url = f'{self.schema}://g.api.{self.domain}/cs'
|
||||
req = requests.post(
|
||||
response = requests.post(
|
||||
url,
|
||||
params=params,
|
||||
data=json.dumps(data),
|
||||
timeout=self.timeout,
|
||||
)
|
||||
json_resp = json.loads(req.text)
|
||||
if isinstance(json_resp, int):
|
||||
if json_resp == -3:
|
||||
json_resp = json.loads(response.text)
|
||||
try:
|
||||
if isinstance(json_resp, list):
|
||||
int_resp = json_resp[0] if isinstance(json_resp[0],
|
||||
int) else None
|
||||
elif isinstance(json_resp, int):
|
||||
int_resp = json_resp
|
||||
except IndexError:
|
||||
int_resp = None
|
||||
if int_resp is not None:
|
||||
if int_resp == 0:
|
||||
return int_resp
|
||||
if int_resp == -3:
|
||||
msg = 'Request failed, retrying'
|
||||
logger.info(msg)
|
||||
raise RuntimeError(msg)
|
||||
raise RequestError(json_resp)
|
||||
raise RequestError(int_resp)
|
||||
return json_resp[0]
|
||||
|
||||
def _parse_url(self, url):
|
||||
# parse file id and key from url
|
||||
if '!' in url:
|
||||
"""Parse file id and key from url."""
|
||||
if '/file/' in url:
|
||||
# V2 URL structure
|
||||
url = url.replace(' ', '')
|
||||
file_id = re.findall(r'\W\w\w\w\w\w\w\w\w\W', url)[0][1:-1]
|
||||
id_index = re.search(file_id, url).end()
|
||||
key = url[id_index + 1:]
|
||||
return f'{file_id}!{key}'
|
||||
elif '!' in url:
|
||||
# V1 URL structure
|
||||
match = re.findall(r'/#!(.*)', url)
|
||||
path = match[0]
|
||||
return path
|
||||
|
|
@ -201,10 +208,8 @@ class Mega:
|
|||
def _process_file(self, file, shared_keys):
|
||||
if file['t'] == 0 or file['t'] == 1:
|
||||
keys = dict(
|
||||
keypart.split(':', 1)
|
||||
for keypart in file['k'].split('/')
|
||||
if ':' in keypart
|
||||
)
|
||||
keypart.split(':', 1) for keypart in file['k'].split('/')
|
||||
if ':' in keypart)
|
||||
uid = file['u']
|
||||
key = None
|
||||
# my objects
|
||||
|
|
@ -212,9 +217,8 @@ class Mega:
|
|||
key = decrypt_key(base64_to_a32(keys[uid]), self.master_key)
|
||||
# shared folders
|
||||
elif 'su' in file and 'sk' in file and ':' in file['k']:
|
||||
shared_key = decrypt_key(
|
||||
base64_to_a32(file['sk']), self.master_key
|
||||
)
|
||||
shared_key = decrypt_key(base64_to_a32(file['sk']),
|
||||
self.master_key)
|
||||
key = decrypt_key(base64_to_a32(keys[file['h']]), shared_key)
|
||||
if file['su'] not in shared_keys:
|
||||
shared_keys[file['su']] = {}
|
||||
|
|
@ -230,17 +234,14 @@ class Mega:
|
|||
if file['h'] and file['h'] in shared_keys.get('EXP', ()):
|
||||
shared_key = shared_keys['EXP'][file['h']]
|
||||
encrypted_key = str_to_a32(
|
||||
base64_url_decode(file['k'].split(':')[-1])
|
||||
)
|
||||
base64_url_decode(file['k'].split(':')[-1]))
|
||||
key = decrypt_key(encrypted_key, shared_key)
|
||||
file['shared_folder_key'] = shared_key
|
||||
if key is not None:
|
||||
# file
|
||||
if file['t'] == 0:
|
||||
k = (
|
||||
key[0] ^ key[4], key[1] ^ key[5], key[2] ^ key[6],
|
||||
key[3] ^ key[7]
|
||||
)
|
||||
k = (key[0] ^ key[4], key[1] ^ key[5], key[2] ^ key[6],
|
||||
key[3] ^ key[7])
|
||||
file['iv'] = key[4:6] + (0, 0)
|
||||
file['meta_mac'] = key[6:8]
|
||||
# folder
|
||||
|
|
@ -275,9 +276,8 @@ class Mega:
|
|||
"""
|
||||
ok_dict = {}
|
||||
for ok_item in files['ok']:
|
||||
shared_key = decrypt_key(
|
||||
base64_to_a32(ok_item['k']), self.master_key
|
||||
)
|
||||
shared_key = decrypt_key(base64_to_a32(ok_item['k']),
|
||||
self.master_key)
|
||||
ok_dict[ok_item['h']] = shared_key
|
||||
for s_item in files['s']:
|
||||
if s_item['u'] not in shared_keys:
|
||||
|
|
@ -302,10 +302,8 @@ class Mega:
|
|||
for foldername in paths:
|
||||
if foldername != '':
|
||||
for file in files.items():
|
||||
if (
|
||||
file[1]['a'] and file[1]['t']
|
||||
and file[1]['a']['n'] == foldername
|
||||
):
|
||||
if (file[1]['a'] and file[1]['t']
|
||||
and file[1]['a']['n'] == foldername):
|
||||
if parent_desc == file[1]['p']:
|
||||
parent_desc = file[0]
|
||||
found = True
|
||||
|
|
@ -327,28 +325,25 @@ class Mega:
|
|||
parent_dir_name = path.parent.name
|
||||
for file in list(files.items()):
|
||||
parent_node_id = None
|
||||
if parent_dir_name:
|
||||
parent_node_id = self.find_path_descriptor(
|
||||
parent_dir_name, files=files
|
||||
)
|
||||
if (
|
||||
filename and parent_node_id and file[1]['a']
|
||||
and file[1]['a']['n'] == filename
|
||||
and parent_node_id == file[1]['p']
|
||||
):
|
||||
if (
|
||||
exclude_deleted
|
||||
and self._trash_folder_node_id == file[1]['p']
|
||||
):
|
||||
try:
|
||||
if parent_dir_name:
|
||||
parent_node_id = self.find_path_descriptor(parent_dir_name,
|
||||
files=files)
|
||||
if (filename and parent_node_id and file[1]['a']
|
||||
and file[1]['a']['n'] == filename
|
||||
and parent_node_id == file[1]['p']):
|
||||
if (exclude_deleted and self._trash_folder_node_id
|
||||
== file[1]['p']):
|
||||
continue
|
||||
return file
|
||||
elif (filename and file[1]['a']
|
||||
and file[1]['a']['n'] == filename):
|
||||
if (exclude_deleted
|
||||
and self._trash_folder_node_id == file[1]['p']):
|
||||
continue
|
||||
return file
|
||||
if (filename and file[1]['a'] and file[1]['a']['n'] == filename):
|
||||
if (
|
||||
exclude_deleted
|
||||
and self._trash_folder_node_id == file[1]['p']
|
||||
):
|
||||
continue
|
||||
return file
|
||||
except TypeError:
|
||||
continue
|
||||
|
||||
def get_files(self):
|
||||
logger.info('Getting all files...')
|
||||
|
|
@ -373,17 +368,12 @@ class Mega:
|
|||
public_handle = self._api_request({'a': 'l', 'n': file['h']})
|
||||
file_key = file['k'][file['k'].index(':') + 1:]
|
||||
decrypted_key = a32_to_base64(
|
||||
decrypt_key(base64_to_a32(file_key), self.master_key)
|
||||
)
|
||||
return (
|
||||
f'{self.schema}://{self.domain}'
|
||||
f'/#!{public_handle}!{decrypted_key}'
|
||||
)
|
||||
decrypt_key(base64_to_a32(file_key), self.master_key))
|
||||
return (f'{self.schema}://{self.domain}'
|
||||
f'/#!{public_handle}!{decrypted_key}')
|
||||
else:
|
||||
raise ValueError(
|
||||
'''Upload() response required as input,
|
||||
use get_link() for regular file input'''
|
||||
)
|
||||
raise ValueError('''Upload() response required as input,
|
||||
use get_link() for regular file input''')
|
||||
|
||||
def get_link(self, file):
|
||||
"""
|
||||
|
|
@ -393,15 +383,11 @@ class Mega:
|
|||
if 'h' in file and 'k' in file:
|
||||
public_handle = self._api_request({'a': 'l', 'n': file['h']})
|
||||
if public_handle == -11:
|
||||
raise RequestError(
|
||||
"Can't get a public link from that file "
|
||||
"(is this a shared file?)"
|
||||
)
|
||||
raise RequestError("Can't get a public link from that file "
|
||||
"(is this a shared file?)")
|
||||
decrypted_key = a32_to_base64(file['key'])
|
||||
return (
|
||||
f'{self.schema}://{self.domain}'
|
||||
f'/#!{public_handle}!{decrypted_key}'
|
||||
)
|
||||
return (f'{self.schema}://{self.domain}'
|
||||
f'/#!{public_handle}!{decrypted_key}')
|
||||
else:
|
||||
raise ValidationError('File id and key must be present')
|
||||
|
||||
|
|
@ -419,15 +405,11 @@ class Mega:
|
|||
if 'h' in file and 'k' in file:
|
||||
public_handle = self._api_request({'a': 'l', 'n': file['h']})
|
||||
if public_handle == -11:
|
||||
raise RequestError(
|
||||
"Can't get a public link from that file "
|
||||
"(is this a shared file?)"
|
||||
)
|
||||
raise RequestError("Can't get a public link from that file "
|
||||
"(is this a shared file?)")
|
||||
decrypted_key = a32_to_base64(file['shared_folder_key'])
|
||||
return (
|
||||
f'{self.schema}://{self.domain}'
|
||||
f'/#F!{public_handle}!{decrypted_key}'
|
||||
)
|
||||
return (f'{self.schema}://{self.domain}'
|
||||
f'/#F!{public_handle}!{decrypted_key}')
|
||||
else:
|
||||
raise ValidationError('File id and key must be present')
|
||||
|
||||
|
|
@ -490,14 +472,12 @@ class Mega:
|
|||
"""
|
||||
Get current remaining disk quota in MegaBytes
|
||||
"""
|
||||
json_resp = self._api_request(
|
||||
{
|
||||
'a': 'uq',
|
||||
'xfer': 1,
|
||||
'strg': 1,
|
||||
'v': 1
|
||||
}
|
||||
)
|
||||
json_resp = self._api_request({
|
||||
'a': 'uq',
|
||||
'xfer': 1,
|
||||
'strg': 1,
|
||||
'v': 1
|
||||
})
|
||||
# convert bytes to megabyes
|
||||
return json_resp['mstrg'] / 1048576
|
||||
|
||||
|
|
@ -551,13 +531,11 @@ class Mega:
|
|||
"""
|
||||
Destroy a file by its private id
|
||||
"""
|
||||
return self._api_request(
|
||||
{
|
||||
'a': 'd',
|
||||
'n': file_id,
|
||||
'i': self.request_id
|
||||
}
|
||||
)
|
||||
return self._api_request({
|
||||
'a': 'd',
|
||||
'n': file_id,
|
||||
'i': self.request_id
|
||||
})
|
||||
|
||||
def destroy_url(self, url):
|
||||
"""
|
||||
|
|
@ -583,24 +561,20 @@ class Mega:
|
|||
"""
|
||||
Download a file by it's file object
|
||||
"""
|
||||
return self._download_file(
|
||||
file_handle=None,
|
||||
file_key=None,
|
||||
file=file[1],
|
||||
dest_path=dest_path,
|
||||
dest_filename=dest_filename,
|
||||
is_public=False
|
||||
)
|
||||
return self._download_file(file_handle=None,
|
||||
file_key=None,
|
||||
file=file[1],
|
||||
dest_path=dest_path,
|
||||
dest_filename=dest_filename,
|
||||
is_public=False)
|
||||
|
||||
def _export_file(self, node):
|
||||
node_data = self._node_data(node)
|
||||
self._api_request(
|
||||
[{
|
||||
'a': 'l',
|
||||
'n': node_data['h'],
|
||||
'i': self.request_id
|
||||
}]
|
||||
)
|
||||
self._api_request([{
|
||||
'a': 'l',
|
||||
'n': node_data['h'],
|
||||
'i': self.request_id
|
||||
}])
|
||||
return self.get_link(node)
|
||||
|
||||
def export(self, path=None, node_id=None):
|
||||
|
|
@ -623,10 +597,8 @@ class Mega:
|
|||
|
||||
master_key_cipher = AES.new(a32_to_str(self.master_key), AES.MODE_ECB)
|
||||
ha = base64_url_encode(
|
||||
master_key_cipher.encrypt(
|
||||
node_data['h'].encode("utf8") + node_data['h'].encode("utf8")
|
||||
)
|
||||
)
|
||||
master_key_cipher.encrypt(node_data['h'].encode("utf8") +
|
||||
node_data['h'].encode("utf8")))
|
||||
|
||||
share_key = secrets.token_bytes(16)
|
||||
ok = base64_url_encode(master_key_cipher.encrypt(share_key))
|
||||
|
|
@ -634,24 +606,26 @@ class Mega:
|
|||
share_key_cipher = AES.new(share_key, AES.MODE_ECB)
|
||||
node_key = node_data['k']
|
||||
encrypted_node_key = base64_url_encode(
|
||||
share_key_cipher.encrypt(a32_to_str(node_key))
|
||||
)
|
||||
share_key_cipher.encrypt(a32_to_str(node_key)))
|
||||
|
||||
node_id = node_data['h']
|
||||
request_body = [
|
||||
{
|
||||
'a': 's2',
|
||||
'n': node_id,
|
||||
's': [{
|
||||
'u': 'EXP',
|
||||
'r': 0
|
||||
}],
|
||||
'i': self.request_id,
|
||||
'ok': ok,
|
||||
'ha': ha,
|
||||
'cr': [[node_id], [node_id], [0, 0, encrypted_node_key]]
|
||||
}
|
||||
]
|
||||
request_body = [{
|
||||
'a':
|
||||
's2',
|
||||
'n':
|
||||
node_id,
|
||||
's': [{
|
||||
'u': 'EXP',
|
||||
'r': 0
|
||||
}],
|
||||
'i':
|
||||
self.request_id,
|
||||
'ok':
|
||||
ok,
|
||||
'ha':
|
||||
ha,
|
||||
'cr': [[node_id], [node_id], [0, 0, encrypted_node_key]]
|
||||
}]
|
||||
self._api_request(request_body)
|
||||
nodes = self.get_files()
|
||||
return self.get_folder_link(nodes[node_id])
|
||||
|
|
@ -671,38 +645,30 @@ class Mega:
|
|||
is_public=True,
|
||||
)
|
||||
|
||||
def _download_file(
|
||||
self,
|
||||
file_handle,
|
||||
file_key,
|
||||
dest_path=None,
|
||||
dest_filename=None,
|
||||
is_public=False,
|
||||
file=None
|
||||
):
|
||||
def _download_file(self,
|
||||
file_handle,
|
||||
file_key,
|
||||
dest_path=None,
|
||||
dest_filename=None,
|
||||
is_public=False,
|
||||
file=None):
|
||||
if file is None:
|
||||
if is_public:
|
||||
file_key = base64_to_a32(file_key)
|
||||
file_data = self._api_request(
|
||||
{
|
||||
'a': 'g',
|
||||
'g': 1,
|
||||
'p': file_handle
|
||||
}
|
||||
)
|
||||
file_data = self._api_request({
|
||||
'a': 'g',
|
||||
'g': 1,
|
||||
'p': file_handle
|
||||
})
|
||||
else:
|
||||
file_data = self._api_request(
|
||||
{
|
||||
'a': 'g',
|
||||
'g': 1,
|
||||
'n': file_handle
|
||||
}
|
||||
)
|
||||
file_data = self._api_request({
|
||||
'a': 'g',
|
||||
'g': 1,
|
||||
'n': file_handle
|
||||
})
|
||||
|
||||
k = (
|
||||
file_key[0] ^ file_key[4], file_key[1] ^ file_key[5],
|
||||
file_key[2] ^ file_key[6], file_key[3] ^ file_key[7]
|
||||
)
|
||||
k = (file_key[0] ^ file_key[4], file_key[1] ^ file_key[5],
|
||||
file_key[2] ^ file_key[6], file_key[3] ^ file_key[7])
|
||||
iv = file_key[4:6] + (0, 0)
|
||||
meta_mac = file_key[6:8]
|
||||
else:
|
||||
|
|
@ -733,19 +699,17 @@ class Mega:
|
|||
else:
|
||||
dest_path += '/'
|
||||
|
||||
with tempfile.NamedTemporaryFile(
|
||||
mode='w+b', prefix='megapy_', delete=False
|
||||
) as temp_output_file:
|
||||
with tempfile.NamedTemporaryFile(mode='w+b',
|
||||
prefix='megapy_',
|
||||
delete=False) as temp_output_file:
|
||||
k_str = a32_to_str(k)
|
||||
counter = Counter.new(
|
||||
128, initial_value=((iv[0] << 32) + iv[1]) << 64
|
||||
)
|
||||
counter = Counter.new(128,
|
||||
initial_value=((iv[0] << 32) + iv[1]) << 64)
|
||||
aes = AES.new(k_str, AES.MODE_CTR, counter=counter)
|
||||
|
||||
mac_str = '\0' * 16
|
||||
mac_encryptor = AES.new(
|
||||
k_str, AES.MODE_CBC, mac_str.encode("utf8")
|
||||
)
|
||||
mac_encryptor = AES.new(k_str, AES.MODE_CBC,
|
||||
mac_str.encode("utf8"))
|
||||
iv_str = a32_to_str([iv[0], iv[1], iv[0], iv[1]])
|
||||
|
||||
for chunk_start, chunk_size in get_chunks(file_size):
|
||||
|
|
@ -770,14 +734,12 @@ class Mega:
|
|||
mac_str = mac_encryptor.encrypt(encryptor.encrypt(block))
|
||||
|
||||
file_info = os.stat(temp_output_file.name)
|
||||
logger.info(
|
||||
'%s of %s downloaded', file_info.st_size, file_size
|
||||
)
|
||||
logger.info('%s of %s downloaded', file_info.st_size,
|
||||
file_size)
|
||||
file_mac = str_to_a32(mac_str)
|
||||
# check mac integrity
|
||||
if (
|
||||
file_mac[0] ^ file_mac[1], file_mac[2] ^ file_mac[3]
|
||||
) != meta_mac:
|
||||
if (file_mac[0] ^ file_mac[1],
|
||||
file_mac[2] ^ file_mac[3]) != meta_mac:
|
||||
raise ValueError('Mismatched mac')
|
||||
output_path = Path(dest_path + file_name)
|
||||
shutil.move(temp_output_file.name, output_path)
|
||||
|
|
@ -800,17 +762,15 @@ class Mega:
|
|||
ul_key = [random.randint(0, 0xFFFFFFFF) for _ in range(6)]
|
||||
k_str = a32_to_str(ul_key[:4])
|
||||
count = Counter.new(
|
||||
128, initial_value=((ul_key[4] << 32) + ul_key[5]) << 64
|
||||
)
|
||||
128, initial_value=((ul_key[4] << 32) + ul_key[5]) << 64)
|
||||
aes = AES.new(k_str, AES.MODE_CTR, counter=count)
|
||||
|
||||
upload_progress = 0
|
||||
completion_file_handle = None
|
||||
|
||||
mac_str = '\0' * 16
|
||||
mac_encryptor = AES.new(
|
||||
k_str, AES.MODE_CBC, mac_str.encode("utf8")
|
||||
)
|
||||
mac_encryptor = AES.new(k_str, AES.MODE_CBC,
|
||||
mac_str.encode("utf8"))
|
||||
iv_str = a32_to_str([ul_key[4], ul_key[5], ul_key[4], ul_key[5]])
|
||||
if file_size > 0:
|
||||
for chunk_start, chunk_size in get_chunks(file_size):
|
||||
|
|
@ -835,19 +795,17 @@ class Mega:
|
|||
|
||||
# encrypt file and upload
|
||||
chunk = aes.encrypt(chunk)
|
||||
output_file = requests.post(
|
||||
ul_url + "/" + str(chunk_start),
|
||||
data=chunk,
|
||||
timeout=self.timeout
|
||||
)
|
||||
output_file = requests.post(ul_url + "/" +
|
||||
str(chunk_start),
|
||||
data=chunk,
|
||||
timeout=self.timeout)
|
||||
completion_file_handle = output_file.text
|
||||
logger.info(
|
||||
'%s of %s uploaded', upload_progress, file_size
|
||||
)
|
||||
logger.info('%s of %s uploaded', upload_progress,
|
||||
file_size)
|
||||
else:
|
||||
output_file = requests.post(
|
||||
ul_url + "/0", data='', timeout=self.timeout
|
||||
)
|
||||
output_file = requests.post(ul_url + "/0",
|
||||
data='',
|
||||
timeout=self.timeout)
|
||||
completion_file_handle = output_file.text
|
||||
|
||||
logger.info('Chunks uploaded')
|
||||
|
|
@ -862,8 +820,7 @@ class Mega:
|
|||
attribs = {'n': dest_filename}
|
||||
|
||||
encrypt_attribs = base64_url_encode(
|
||||
encrypt_attr(attribs, ul_key[:4])
|
||||
)
|
||||
encrypt_attr(attribs, ul_key[:4]))
|
||||
key = [
|
||||
ul_key[0] ^ ul_key[4], ul_key[1] ^ ul_key[5],
|
||||
ul_key[2] ^ meta_mac[0], ul_key[3] ^ meta_mac[1], ul_key[4],
|
||||
|
|
@ -872,21 +829,20 @@ class Mega:
|
|||
encrypted_key = a32_to_base64(encrypt_key(key, self.master_key))
|
||||
logger.info('Sending request to update attributes')
|
||||
# update attributes
|
||||
data = self._api_request(
|
||||
{
|
||||
'a': 'p',
|
||||
't': dest,
|
||||
'i': self.request_id,
|
||||
'n': [
|
||||
{
|
||||
'h': completion_file_handle,
|
||||
't': 0,
|
||||
'a': encrypt_attribs,
|
||||
'k': encrypted_key
|
||||
}
|
||||
]
|
||||
}
|
||||
)
|
||||
data = self._api_request({
|
||||
'a':
|
||||
'p',
|
||||
't':
|
||||
dest,
|
||||
'i':
|
||||
self.request_id,
|
||||
'n': [{
|
||||
'h': completion_file_handle,
|
||||
't': 0,
|
||||
'a': encrypt_attribs,
|
||||
'k': encrypted_key
|
||||
}]
|
||||
})
|
||||
logger.info('Upload complete')
|
||||
return data
|
||||
|
||||
|
|
@ -900,21 +856,20 @@ class Mega:
|
|||
encrypted_key = a32_to_base64(encrypt_key(ul_key[:4], self.master_key))
|
||||
|
||||
# update attributes
|
||||
data = self._api_request(
|
||||
{
|
||||
'a': 'p',
|
||||
't': parent_node_id,
|
||||
'n': [
|
||||
{
|
||||
'h': 'xxxxxxxx',
|
||||
't': 1,
|
||||
'a': encrypt_attribs,
|
||||
'k': encrypted_key
|
||||
}
|
||||
],
|
||||
'i': self.request_id
|
||||
}
|
||||
)
|
||||
data = self._api_request({
|
||||
'a':
|
||||
'p',
|
||||
't':
|
||||
parent_node_id,
|
||||
'n': [{
|
||||
'h': 'xxxxxxxx',
|
||||
't': 1,
|
||||
'a': encrypt_attribs,
|
||||
'k': encrypted_key
|
||||
}],
|
||||
'i':
|
||||
self.request_id
|
||||
})
|
||||
return data
|
||||
|
||||
def _root_node_id(self):
|
||||
|
|
@ -937,9 +892,8 @@ class Mega:
|
|||
parent_node_id = dest
|
||||
else:
|
||||
parent_node_id = folder_node_ids[idx - 1]
|
||||
created_node = self._mkdir(
|
||||
name=directory_name, parent_node_id=parent_node_id
|
||||
)
|
||||
created_node = self._mkdir(name=directory_name,
|
||||
parent_node_id=parent_node_id)
|
||||
node_id = created_node['f'][0]['h']
|
||||
folder_node_ids[idx] = node_id
|
||||
return dict(zip(dirs, folder_node_ids.values()))
|
||||
|
|
@ -950,21 +904,16 @@ class Mega:
|
|||
attribs = {'n': new_name}
|
||||
# encrypt attribs
|
||||
encrypt_attribs = base64_url_encode(encrypt_attr(attribs, file['k']))
|
||||
encrypted_key = a32_to_base64(
|
||||
encrypt_key(file['key'], self.master_key)
|
||||
)
|
||||
encrypted_key = a32_to_base64(encrypt_key(file['key'],
|
||||
self.master_key))
|
||||
# update attributes
|
||||
return self._api_request(
|
||||
[
|
||||
{
|
||||
'a': 'a',
|
||||
'attr': encrypt_attribs,
|
||||
'key': encrypted_key,
|
||||
'n': file['h'],
|
||||
'i': self.request_id
|
||||
}
|
||||
]
|
||||
)
|
||||
return self._api_request([{
|
||||
'a': 'a',
|
||||
'attr': encrypt_attribs,
|
||||
'key': encrypted_key,
|
||||
'n': file['h'],
|
||||
'i': self.request_id
|
||||
}])
|
||||
|
||||
def move(self, file_id, target):
|
||||
"""
|
||||
|
|
@ -994,14 +943,12 @@ class Mega:
|
|||
else:
|
||||
file = target[1]
|
||||
target_node_id = file['h']
|
||||
return self._api_request(
|
||||
{
|
||||
'a': 'm',
|
||||
'n': file_id,
|
||||
't': target_node_id,
|
||||
'i': self.request_id
|
||||
}
|
||||
)
|
||||
return self._api_request({
|
||||
'a': 'm',
|
||||
'n': file_id,
|
||||
't': target_node_id,
|
||||
'i': self.request_id
|
||||
})
|
||||
|
||||
def add_contact(self, email):
|
||||
"""
|
||||
|
|
@ -1029,14 +976,12 @@ class Mega:
|
|||
if not re.match(r"[^@]+@[^@]+\.[^@]+", email):
|
||||
ValidationError('add_contact requires a valid email address')
|
||||
else:
|
||||
return self._api_request(
|
||||
{
|
||||
'a': 'ur',
|
||||
'u': email,
|
||||
'l': l,
|
||||
'i': self.request_id
|
||||
}
|
||||
)
|
||||
return self._api_request({
|
||||
'a': 'ur',
|
||||
'u': email,
|
||||
'l': l,
|
||||
'i': self.request_id
|
||||
})
|
||||
|
||||
def get_public_url_info(self, url):
|
||||
"""
|
||||
|
|
@ -1050,9 +995,10 @@ class Mega:
|
|||
Import the public url into user account
|
||||
"""
|
||||
file_handle, file_key = self._parse_url(url).split('!')
|
||||
return self.import_public_file(
|
||||
file_handle, file_key, dest_node=dest_node, dest_name=dest_name
|
||||
)
|
||||
return self.import_public_file(file_handle,
|
||||
file_key,
|
||||
dest_node=dest_node,
|
||||
dest_name=dest_name)
|
||||
|
||||
def get_public_file_info(self, file_handle, file_key):
|
||||
"""
|
||||
|
|
@ -1066,9 +1012,8 @@ class Mega:
|
|||
raise ValueError("Unexpected result", data)
|
||||
|
||||
key = base64_to_a32(file_key)
|
||||
k = (
|
||||
key[0] ^ key[4], key[1] ^ key[5], key[2] ^ key[6], key[3] ^ key[7]
|
||||
)
|
||||
k = (key[0] ^ key[4], key[1] ^ key[5], key[2] ^ key[6],
|
||||
key[3] ^ key[7])
|
||||
|
||||
size = data['s']
|
||||
unencrypted_attrs = decrypt_attr(base64_url_decode(data['at']), k)
|
||||
|
|
@ -1077,9 +1022,11 @@ class Mega:
|
|||
result = {'size': size, 'name': unencrypted_attrs['n']}
|
||||
return result
|
||||
|
||||
def import_public_file(
|
||||
self, file_handle, file_key, dest_node=None, dest_name=None
|
||||
):
|
||||
def import_public_file(self,
|
||||
file_handle,
|
||||
file_key,
|
||||
dest_node=None,
|
||||
dest_name=None):
|
||||
"""
|
||||
Import the public file into user account
|
||||
"""
|
||||
|
|
@ -1094,23 +1041,20 @@ class Mega:
|
|||
dest_name = pl_info['name']
|
||||
|
||||
key = base64_to_a32(file_key)
|
||||
k = (
|
||||
key[0] ^ key[4], key[1] ^ key[5], key[2] ^ key[6], key[3] ^ key[7]
|
||||
)
|
||||
k = (key[0] ^ key[4], key[1] ^ key[5], key[2] ^ key[6],
|
||||
key[3] ^ key[7])
|
||||
|
||||
encrypted_key = a32_to_base64(encrypt_key(key, self.master_key))
|
||||
encrypted_name = base64_url_encode(encrypt_attr({'n': dest_name}, k))
|
||||
return self._api_request(
|
||||
{
|
||||
'a': 'p',
|
||||
't': dest_node['h'],
|
||||
'n': [
|
||||
{
|
||||
'ph': file_handle,
|
||||
't': 0,
|
||||
'a': encrypted_name,
|
||||
'k': encrypted_key
|
||||
}
|
||||
]
|
||||
}
|
||||
)
|
||||
return self._api_request({
|
||||
'a':
|
||||
'p',
|
||||
't':
|
||||
dest_node['h'],
|
||||
'n': [{
|
||||
'ph': file_handle,
|
||||
't': 0,
|
||||
'a': encrypted_name,
|
||||
'k': encrypted_key
|
||||
}]
|
||||
})
|
||||
|
|
|
|||
|
|
@ -3,39 +3,17 @@ import pytest
|
|||
from mega.crypto import get_chunks
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
'file_size, exp_result', [
|
||||
(
|
||||
10,
|
||||
(
|
||||
(0, 10),
|
||||
)
|
||||
),
|
||||
(
|
||||
1000,
|
||||
(
|
||||
(0, 1000),
|
||||
)
|
||||
),
|
||||
(
|
||||
1000000,
|
||||
(
|
||||
(0, 131072), (131072, 262144), (393216, 393216),
|
||||
(786432, 213568)
|
||||
)
|
||||
),
|
||||
(
|
||||
10000000,
|
||||
(
|
||||
(0, 131072), (131072, 262144), (393216, 393216),
|
||||
@pytest.mark.parametrize('file_size, exp_result', [
|
||||
(10, ((0, 10), )),
|
||||
(1000, ((0, 1000), )),
|
||||
(1000000, ((0, 131072), (131072, 262144), (393216, 393216),
|
||||
(786432, 213568))),
|
||||
(10000000, ((0, 131072), (131072, 262144), (393216, 393216),
|
||||
(786432, 524288), (1310720, 655360), (1966080, 786432),
|
||||
(2752512, 917504), (3670016, 1048576), (4718592, 1048576),
|
||||
(5767168, 1048576), (6815744, 1048576), (7864320, 1048576),
|
||||
(8912896, 1048576), (9961472, 38528)
|
||||
)
|
||||
),
|
||||
]
|
||||
)
|
||||
(8912896, 1048576), (9961472, 38528))),
|
||||
])
|
||||
def test_get_chunks(file_size, exp_result):
|
||||
result = tuple(get_chunks(file_size))
|
||||
|
||||
|
|
|
|||
|
|
@ -3,12 +3,9 @@ import pytest
|
|||
from mega.errors import RequestError, _CODE_TO_DESCRIPTIONS
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
'code, exp_message', [
|
||||
(code, f'{desc[0]}, {desc[1]}')
|
||||
for code, desc in _CODE_TO_DESCRIPTIONS.items()
|
||||
]
|
||||
)
|
||||
@pytest.mark.parametrize('code, exp_message',
|
||||
[(code, f'{desc[0]}, {desc[1]}')
|
||||
for code, desc in _CODE_TO_DESCRIPTIONS.items()])
|
||||
def test_request_error(code, exp_message):
|
||||
exc = RequestError(code)
|
||||
|
||||
|
|
|
|||
|
|
@ -2,15 +2,16 @@ import random
|
|||
from pathlib import Path
|
||||
import os
|
||||
|
||||
import requests_mock
|
||||
import pytest
|
||||
|
||||
from mega import Mega
|
||||
|
||||
TEST_CONTACT = 'test@mega.co.nz'
|
||||
TEST_PUBLIC_URL = (
|
||||
'https://mega.nz/#!hYVmXKqL!r0d0-WRnFwulR_shhuEDwrY1Vo103-am1MyUy8oV6Ps'
|
||||
)
|
||||
'https://mega.nz/#!hYVmXKqL!r0d0-WRnFwulR_shhuEDwrY1Vo103-am1MyUy8oV6Ps')
|
||||
TEST_FILE = os.path.basename(__file__)
|
||||
MODULE = 'mega.mega'
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
|
@ -32,9 +33,7 @@ def mega(folder_name):
|
|||
def uploaded_file(mega, folder_name):
|
||||
folder = mega.find(folder_name)
|
||||
dest_node_id = folder[1]['h']
|
||||
mega.upload(
|
||||
__file__, dest=dest_node_id, dest_filename='test.py'
|
||||
)
|
||||
mega.upload(__file__, dest=dest_node_id, dest_filename='test.py')
|
||||
path = f'{folder_name}/test.py'
|
||||
return mega.find(path)
|
||||
|
||||
|
|
@ -72,8 +71,8 @@ def test_get_link(mega, uploaded_file):
|
|||
assert isinstance(link, str)
|
||||
|
||||
|
||||
@pytest.mark.skip
|
||||
class TestExport:
|
||||
|
||||
def test_export_folder(self, mega, folder_name):
|
||||
public_url = None
|
||||
for _ in range(2):
|
||||
|
|
@ -103,9 +102,7 @@ class TestExport:
|
|||
# Upload a single file into a folder
|
||||
folder = mega.find(folder_name)
|
||||
dest_node_id = folder[1]['h']
|
||||
mega.upload(
|
||||
__file__, dest=dest_node_id, dest_filename='test.py'
|
||||
)
|
||||
mega.upload(__file__, dest=dest_node_id, dest_filename='test.py')
|
||||
path = f'{folder_name}/test.py'
|
||||
assert mega.find(path)
|
||||
|
||||
|
|
@ -131,8 +128,7 @@ class TestCreateFolder:
|
|||
|
||||
def test_create_folder_with_sub_folders(self, mega, folder_name, mocker):
|
||||
folder_names_and_node_ids = mega.create_folder(
|
||||
name=(Path(folder_name) / 'subdir' / 'anothersubdir')
|
||||
)
|
||||
name=(Path(folder_name) / 'subdir' / 'anothersubdir'))
|
||||
|
||||
assert len(folder_names_and_node_ids) == 3
|
||||
assert folder_names_and_node_ids == {
|
||||
|
|
@ -143,16 +139,21 @@ class TestCreateFolder:
|
|||
|
||||
|
||||
class TestFind:
|
||||
|
||||
def test_find_file(self, mega, folder_name):
|
||||
folder = mega.find(folder_name)
|
||||
assert folder
|
||||
dest_node_id = folder[1]['h']
|
||||
mega.upload(
|
||||
__file__, dest=dest_node_id, dest_filename='test.py'
|
||||
)
|
||||
path = f'{folder_name}/test.py'
|
||||
mega.upload(__file__, dest=dest_node_id, dest_filename='test.py')
|
||||
file1 = mega.find(f'{folder_name}/test.py')
|
||||
assert file1
|
||||
|
||||
assert mega.find(path)
|
||||
dest_node_id2 = mega.create_folder('new_folder')['new_folder']
|
||||
mega.upload(__file__, dest=dest_node_id2, dest_filename='test.py')
|
||||
|
||||
file2 = mega.find('new_folder/test.py')
|
||||
assert file2
|
||||
# Check that the correct test.py was found
|
||||
assert file1 != file2
|
||||
|
||||
def test_path_not_found_returns_none(self, mega):
|
||||
assert mega.find('not_found') is None
|
||||
|
|
@ -194,15 +195,13 @@ def test_download(mega, tmpdir, folder_name):
|
|||
# Upload a single file into a folder
|
||||
folder = mega.find(folder_name)
|
||||
dest_node_id = folder[1]['h']
|
||||
mega.upload(
|
||||
__file__, dest=dest_node_id, dest_filename='test.py'
|
||||
)
|
||||
mega.upload(__file__, dest=dest_node_id, dest_filename='test.py')
|
||||
path = f'{folder_name}/test.py'
|
||||
file = mega.find(path)
|
||||
|
||||
output_path = mega.download(
|
||||
file=file, dest_path=tmpdir, dest_filename='test.py'
|
||||
)
|
||||
output_path = mega.download(file=file,
|
||||
dest_path=tmpdir,
|
||||
dest_filename='test.py')
|
||||
|
||||
assert output_path.exists()
|
||||
|
||||
|
|
@ -222,3 +221,29 @@ def test_add_contact(mega):
|
|||
def test_remove_contact(mega):
|
||||
resp = mega.remove_contact(TEST_CONTACT)
|
||||
assert isinstance(resp, int)
|
||||
|
||||
|
||||
@pytest.mark.parametrize('url, expected_file_id_and_key', [
|
||||
('https://mega.nz/#!Ue5VRSIQ!kC2E4a4JwfWWCWYNJovGFHlbz8F'
|
||||
'N-ISsBAGPzvTjT6k',
|
||||
'Ue5VRSIQ!kC2E4a4JwfWWCWYNJovGFHlbz8FN-ISsBAGPzvTjT6k'),
|
||||
('https://mega.nz/file/cH51DYDR#qH7QOfRcM-7N9riZWdSjsRq'
|
||||
'5VDTLfIhThx1capgVA30',
|
||||
'cH51DYDR!qH7QOfRcM-7N9riZWdSjsRq5VDTLfIhThx1capgVA30'),
|
||||
])
|
||||
def test_parse_url(url, expected_file_id_and_key, mega):
|
||||
assert mega._parse_url(url) == expected_file_id_and_key
|
||||
|
||||
|
||||
@pytest.mark.skip
|
||||
class TestAPIRequest:
|
||||
@pytest.mark.parametrize('response_text', ['-3', '-9'])
|
||||
def test_when_api_returns_int_raises_exception(
|
||||
self,
|
||||
mega,
|
||||
response_text,
|
||||
):
|
||||
with requests_mock.Mocker() as m:
|
||||
m.post(f'{mega.schema}://g.api.{mega.domain}/cs',
|
||||
text=response_text)
|
||||
mega._api_request(data={})
|
||||
|
|
|
|||
Loading…
Reference in a new issue