async-mega-py 2.2.0__tar.gz → 2.2.2__tar.gz

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: async-mega.py
-Version: 2.2.0
+Version: 2.2.2
 Summary: Python library for the Mega.nz and Transfer.it API
 Keywords: api,downloader,mega,mega.nz,transfer.it
 Author: NTFSvolume
@@ -79,7 +79,7 @@ Please check [`src/mega/data_structures.py`](src/mega/data_structures.py) for de
 > [!TIP]
 > You can run the commands bellow in an interactive python (the asyncio REPL). Try them in real time as is, using `async/await` keywords!
 >
-> ```uv run -p3.12 --with async-mega-py python -m asyncio```
+> `uv run -p3.12 --with async-mega-py python -m asyncio`
 
 ```python
 from mega.client import MegaNzClient
@@ -99,7 +99,7 @@ from mega.client import MegaNzClient
 # Also works without using it as a context manager, but you have the responsability to close the session at the end
 mega = MegaNzClient()
 await mega.login() # login using a temporary anonymous account
-# await mega.close()  
+# await mega.close()
 
 ```
 
@@ -163,7 +163,7 @@ await mega.download(uploaded_file, output_dir)
 
 # Download a public file
 file_url = "https://mega.nz/file/vJ9EBJBC#vA1XpbngXcnN7lqXEWGQFPDc5ZlG6yltCHwGN-0O2LQ"
-public_handle, public_key = mega.parse_file_url(file_url)  
+public_handle, public_key = mega.parse_file_url(file_url)
 await mega.download_public_file(public_handle, public_key, output_dir)
 
 # Download a public folder
@@ -186,7 +186,7 @@ print (result.is_folder)
 > You can show a progress bar on the terminal for each download/upload by calling them within the `progress_bar` context manager (needs optional dependency `rich` to be installed):
 
 ```python
-with mega.progress_bar:  
+with mega.progress_bar:
     results = await mega.download_public_folder(public_handle, public_key, output_dir / "with_bar")
 ```
 
@@ -264,16 +264,14 @@ The output will be:
 
 ```json
 {
-    "oImwb6nN": "/tests/script.js",
-    "FIXitv4F": "/tests/scripts",
-    "0fPFklV3": "/tests/scripts/notes.txt",
-    "l9zkz1GU": "/tests/scripts/script.js",
-    "E4LqT4EF": "/tests/scripts/styles.css",
+  "oImwb6nN": "/tests/script.js",
+  "FIXitv4F": "/tests/scripts",
+  "0fPFklV3": "/tests/scripts/notes.txt",
+  "l9zkz1GU": "/tests/scripts/script.js",
+  "E4LqT4EF": "/tests/scripts/styles.css"
 }
 ```
 
-
-
 ```python
 # Get the path to a node
 fs.absolute_path("0fPFklV3") # /tests/scripts/notes.txt
@@ -294,24 +292,25 @@ for node in fs.iterdir(folder.id, recursive=True):
     print(fs.absolute_path(node.id))
 
 ```
+
 Output will be:
 
 ```json
 [
-    "/tests/logo.png",
-    "/tests/logo.png",
-    "/tests/script.js",
-    "/tests/scripts",
-    "/tests/scripts/notes.txt",
-    "/tests/scripts/script.js",
-    "/tests/scripts/styles.css",
-    "/tests/setup.sh",
-    "/tests/utils.py",
+  "/tests/logo.png",
+  "/tests/logo.png",
+  "/tests/script.js",
+  "/tests/scripts",
+  "/tests/scripts/notes.txt",
+  "/tests/scripts/script.js",
+  "/tests/scripts/styles.css",
+  "/tests/setup.sh",
+  "/tests/utils.py"
 ]
 ```
 
 > [!IMPORTANT]  
-> Mega's filesystem is *not* POSIX-compliant: multiple nodes may have the same path.
+> Mega's filesystem is _not_ POSIX-compliant: multiple nodes may have the same path.
 >
 > If 2 nodes have the same path, `find` will throw an error.
 
@@ -352,10 +351,10 @@ mega-py --help
 ```
 
 ```powershell
-Usage: mega-py [OPTIONS] COMMAND [ARGS]...  
+Usage: mega-py [OPTIONS] COMMAND [ARGS]...
 
- CLI app for the Mega.nz and Transfer.it. Set MEGA_NZ_EMAIL and MEGA_NZ_PASSWORD  
- enviroment variables to use them as credentials for Mega  
+ CLI app for the Mega.nz and Transfer.it. Set MEGA_EMAIL and MEGA_PWD
+ enviroment variables to use them as credentials for Mega
 
 ╭─ Options ──────────────────────────────────────────────────────────────────────╮
 │ --verbose  -v               Increase verbosity (-v shows debug logs,  -vv      │
@@ -371,6 +370,6 @@ Usage: mega-py [OPTIONS] COMMAND [ARGS]...
 ```
 
 > [!TIP]
-> The CLI app does *not* accept login credentials, but you can still use your account by setting up the `MEGA_EMAIL` and `MEGA_PWD` enviroment variables
+> The CLI app does _not_ accept login credentials, but you can still use your account by setting up the `MEGA_EMAIL` and `MEGA_PWD` enviroment variables
 >
 > It will also read them from an `.env` file (if found)

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/README.md

@@ -46,7 +46,7 @@ Please check [`src/mega/data_structures.py`](src/mega/data_structures.py) for de
 > [!TIP]
 > You can run the commands bellow in an interactive python (the asyncio REPL). Try them in real time as is, using `async/await` keywords!
 >
-> ```uv run -p3.12 --with async-mega-py python -m asyncio```
+> `uv run -p3.12 --with async-mega-py python -m asyncio`
 
 ```python
 from mega.client import MegaNzClient
@@ -66,7 +66,7 @@ from mega.client import MegaNzClient
 # Also works without using it as a context manager, but you have the responsability to close the session at the end
 mega = MegaNzClient()
 await mega.login() # login using a temporary anonymous account
-# await mega.close()  
+# await mega.close()
 
 ```
 
@@ -130,7 +130,7 @@ await mega.download(uploaded_file, output_dir)
 
 # Download a public file
 file_url = "https://mega.nz/file/vJ9EBJBC#vA1XpbngXcnN7lqXEWGQFPDc5ZlG6yltCHwGN-0O2LQ"
-public_handle, public_key = mega.parse_file_url(file_url)  
+public_handle, public_key = mega.parse_file_url(file_url)
 await mega.download_public_file(public_handle, public_key, output_dir)
 
 # Download a public folder
@@ -153,7 +153,7 @@ print (result.is_folder)
 > You can show a progress bar on the terminal for each download/upload by calling them within the `progress_bar` context manager (needs optional dependency `rich` to be installed):
 
 ```python
-with mega.progress_bar:  
+with mega.progress_bar:
     results = await mega.download_public_folder(public_handle, public_key, output_dir / "with_bar")
 ```
 
@@ -231,16 +231,14 @@ The output will be:
 
 ```json
 {
-    "oImwb6nN": "/tests/script.js",
-    "FIXitv4F": "/tests/scripts",
-    "0fPFklV3": "/tests/scripts/notes.txt",
-    "l9zkz1GU": "/tests/scripts/script.js",
-    "E4LqT4EF": "/tests/scripts/styles.css",
+  "oImwb6nN": "/tests/script.js",
+  "FIXitv4F": "/tests/scripts",
+  "0fPFklV3": "/tests/scripts/notes.txt",
+  "l9zkz1GU": "/tests/scripts/script.js",
+  "E4LqT4EF": "/tests/scripts/styles.css"
 }
 ```
 
-
-
 ```python
 # Get the path to a node
 fs.absolute_path("0fPFklV3") # /tests/scripts/notes.txt
@@ -261,24 +259,25 @@ for node in fs.iterdir(folder.id, recursive=True):
     print(fs.absolute_path(node.id))
 
 ```
+
 Output will be:
 
 ```json
 [
-    "/tests/logo.png",
-    "/tests/logo.png",
-    "/tests/script.js",
-    "/tests/scripts",
-    "/tests/scripts/notes.txt",
-    "/tests/scripts/script.js",
-    "/tests/scripts/styles.css",
-    "/tests/setup.sh",
-    "/tests/utils.py",
+  "/tests/logo.png",
+  "/tests/logo.png",
+  "/tests/script.js",
+  "/tests/scripts",
+  "/tests/scripts/notes.txt",
+  "/tests/scripts/script.js",
+  "/tests/scripts/styles.css",
+  "/tests/setup.sh",
+  "/tests/utils.py"
 ]
 ```
 
 > [!IMPORTANT]  
-> Mega's filesystem is *not* POSIX-compliant: multiple nodes may have the same path.
+> Mega's filesystem is _not_ POSIX-compliant: multiple nodes may have the same path.
 >
 > If 2 nodes have the same path, `find` will throw an error.
 
@@ -319,10 +318,10 @@ mega-py --help
 ```
 
 ```powershell
-Usage: mega-py [OPTIONS] COMMAND [ARGS]...  
+Usage: mega-py [OPTIONS] COMMAND [ARGS]...
 
- CLI app for the Mega.nz and Transfer.it. Set MEGA_NZ_EMAIL and MEGA_NZ_PASSWORD  
- enviroment variables to use them as credentials for Mega  
+ CLI app for the Mega.nz and Transfer.it. Set MEGA_EMAIL and MEGA_PWD
+ enviroment variables to use them as credentials for Mega
 
 ╭─ Options ──────────────────────────────────────────────────────────────────────╮
 │ --verbose  -v               Increase verbosity (-v shows debug logs,  -vv      │
@@ -338,6 +337,6 @@ Usage: mega-py [OPTIONS] COMMAND [ARGS]...
 ```
 
 > [!TIP]
-> The CLI app does *not* accept login credentials, but you can still use your account by setting up the `MEGA_EMAIL` and `MEGA_PWD` enviroment variables
+> The CLI app does _not_ accept login credentials, but you can still use your account by setting up the `MEGA_EMAIL` and `MEGA_PWD` enviroment variables
 >
 > It will also read them from an `.env` file (if found)

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/pyproject.toml

@@ -34,7 +34,7 @@ license = "Apache-2.0"
 license-files = ["LICENSE"]
 readme = "README.md"
 requires-python = ">=3.11"
-version = "2.2.0"
+version = "2.2.2"
 
 [project.optional-dependencies]
 cli = [

async_mega_py-2.2.2/src/mega/__init__.py

@@ -0,0 +1,8 @@
+import importlib.metadata
+from contextvars import ContextVar
+
+_package_name_ = "async-mega-py"
+__version__ = importlib.metadata.version(_package_name_)
+
+LOG_FILE_PROGRESS: ContextVar[bool] = ContextVar("LOG_FILE_PROGRESS", default=True)
+LOG_HTTP_TRAFFIC: ContextVar[bool] = ContextVar("LOG_HTTP_TRAFFIC", default=False)

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/src/mega/__main__.py

@@ -14,8 +14,7 @@ _ = install_rich_traceback(width=None, max_frames=2)
 import yarl
 from cyclopts import App, Parameter
 
-from mega import __version__, env
-from mega.api import LOG_HTTP_TRAFFIC
+from mega import LOG_HTTP_TRAFFIC, __version__, env
 from mega.client import MegaNzClient
 from mega.transfer_it import TransferItClient
 from mega.utils import Site, setup_logger

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/src/mega/api.py

@@ -7,15 +7,14 @@ import json
 import logging
 import uuid
 from collections.abc import Mapping, Sequence
-from contextvars import ContextVar
 from functools import wraps
-from typing import TYPE_CHECKING, Any, ClassVar, Literal, ParamSpec, Self, TypeVar
+from typing import TYPE_CHECKING, Any, ClassVar, Generic, Literal, ParamSpec, Self, TypeVar
 
 import aiohttp
 import yarl
 from aiolimiter import AsyncLimiter
 
-from mega import __version__, _package_name_
+from mega import LOG_HTTP_TRAFFIC, __version__, _package_name_
 from mega.crypto import generate_hashcash
 from mega.errors import RequestError, RetryRequestError
 from mega.utils import random_id, random_u32int
@@ -27,8 +26,6 @@ if TYPE_CHECKING:
     _R = TypeVar("_R")
 
 
-LOG_HTTP_TRAFFIC: ContextVar[bool] = ContextVar("LOG_HTTP_TRAFFIC", default=False)
-
 logger = logging.getLogger(__name__)
 
 
@@ -66,27 +63,23 @@ def retry(
     return wrapper
 
 
-@dataclasses.dataclass(slots=True, weakref_slot=True, init=False)
+@dataclasses.dataclass(slots=True, weakref_slot=True)
 class MegaAPI:
-    session_id: str | None
-    _request_id: int
-    _client_id: str
-
-    __session: aiohttp.ClientSession | None
-    _auto_close_session: bool
-    _rate_limiter: AsyncLimiter
-
-    _entrypoint: ClassVar[yarl.URL] = yarl.URL("https://g.api.mega.co.nz/cs")
-    user_agent: str
-
-    def __init__(self, session: aiohttp.ClientSession | None = None, user_agent: str | None = None) -> None:
-        self.session_id = None
-        self._request_id = random_u32int()
-        self._client_id = random_id(10)
-        self.__session = session
-        self._auto_close_session = session is None
-        self._rate_limiter = AsyncLimiter(100, 60)
-        self.user_agent = user_agent or f"{_package_name_}/{__version__}"
+    _session: aiohttp.ClientSession | None = None
+
+    user_agent: str = f"{_package_name_}/{__version__}"
+
+    session_id: str | None = dataclasses.field(init=False, default=None)
+    _request_id: int = dataclasses.field(init=False, default_factory=random_u32int)
+    _client_id: str = dataclasses.field(init=False, default_factory=lambda: random_id(10))
+
+    _auto_close_session: bool = dataclasses.field(init=False)
+    _rate_limiter: AsyncLimiter = dataclasses.field(init=False, default_factory=lambda: AsyncLimiter(100, 60))
+
+    _entrypoint: ClassVar[yarl.URL] = dataclasses.field(init=False, default=yarl.URL("https://g.api.mega.co.nz/cs"))
+
+    def __post_init__(self) -> None:
+        self._auto_close_session = self._session is None
 
     @property
     def entrypoint(self) -> yarl.URL:
@@ -96,22 +89,12 @@ class MegaAPI:
     def client_id(self) -> str:
         return self._client_id
 
-    @property
-    def request_id(self) -> int:
-        return self._request_id
-
-    @property
-    def _session(self) -> aiohttp.ClientSession:
-        if self.__session is None:
-            self.__session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(sock_connect=160, sock_read=60))
-        return self.__session
-
     def __repr__(self) -> str:
         return f"<{type(self).__name__}>(session_id={self.session_id!r}, client_id={self.client_id!r}, auto_close_session={self._auto_close_session!r})"
 
     async def aclose(self) -> None:
-        if self._auto_close_session and self.__session:
-            await self.__session.close()
+        if self._auto_close_session and self._session:
+            await self._session.close()
 
     async def __enter__(self) -> Self:
         return self
@@ -175,7 +158,7 @@ class MegaAPI:
         headers: Mapping[str, str] | None = None,
         **kwargs: Any,
     ) -> AsyncGenerator[aiohttp.ClientResponse]:
-        kwargs["headers"] = {"User-Agent": self.user_agent} | (headers or {})
+        kwargs["headers"] = {"User-Agent": self.user_agent, **(headers or {})}
         request_id = str(uuid.uuid4())
         if LOG_HTTP_TRAFFIC.get():
             logger.debug(
@@ -187,6 +170,9 @@ class MegaAPI:
             )
 
         resp = None
+        if self._session is None:
+            self._session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(sock_connect=160, sock_read=60))
+
         try:
             async with self._rate_limiter, self._session.request(method, url, **kwargs) as resp:
                 yield resp
@@ -244,19 +230,19 @@ class _LazyResponseLog:
         return str(self.__json__())
 
 
-class APIContextManager:
-    __slots__ = ("_api",)
+_API_T = TypeVar("_API_T", bound=MegaAPI, covariant=True)
 
-    def __init__(self, session: aiohttp.ClientSession | None = None, *, user_agent: str | None = None) -> None:
-        self._api: MegaAPI = MegaAPI(session, user_agent)
+
+class APIContextManager(Generic[_API_T]):
+    __slots__ = ("_api",)
 
     async def __aenter__(self) -> Self:
         return self
 
     async def __aexit__(self, *_) -> None:
-        await self.close()
+        await self.aclose()
 
     async def aclose(self) -> None:
-        await self._api.close()
+        await self._api.aclose()
 
     close = aclose

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/src/mega/client.py

@@ -7,7 +7,7 @@ from pathlib import Path, PurePosixPath
 from typing import TYPE_CHECKING
 
 from mega import progress
-from mega.api import APIContextManager
+from mega.api import APIContextManager, MegaAPI
 from mega.core import MegaCore
 from mega.crypto import a32_to_base64, b64_to_a32, b64_url_encode, encrypt_attr, encrypt_key
 from mega.data_structures import (
@@ -38,13 +38,16 @@ logger = logging.getLogger(__name__)
 _DOMAIN = Site.MEGA.value
 
 
-class MegaNzClient(APIContextManager):
+class MegaNzClient(APIContextManager[MegaAPI]):
     """Interface with all the public methods of the API"""
 
     __slots__ = ("_core",)
 
     def __init__(self, session: aiohttp.ClientSession | None = None, *, user_agent: str | None = None) -> None:
-        super().__init__(session, user_agent=user_agent)
+        self._api = MegaAPI(session)
+        if user_agent:
+            self._api.user_agent = user_agent
+
         self._core: MegaCore = MegaCore(self._api)
         if hasattr(sys, "ps1"):
             setup_logger(logging.DEBUG)

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/src/mega/data_structures.py

@@ -13,7 +13,7 @@ import dataclasses
 import time
 from enum import IntEnum
 from types import MappingProxyType
-from typing import TYPE_CHECKING, Any, ClassVar, Final, Self, TypeAlias, TypedDict
+from typing import TYPE_CHECKING, Any, ClassVar, Final, Literal, Self, TypeAlias, TypedDict
 
 if TYPE_CHECKING:
     from collections.abc import Mapping
@@ -67,7 +67,7 @@ class NodeSerialized(TypedDict):
     h: NodeID  # ID
     p: NodeID  # Parent ID
     u: NotRequired[UserID]  # Owner (user ID), not present in transfer.it nodes
-    t: ReadOnly[NodeType]
+    t: ReadOnly[Literal[0, 1, 2, 3, 4, 5]]  # Node type
     a: str  # Serialized attributes
     ts: TimeStamp  # creation date
 
@@ -76,6 +76,7 @@ class NodeSerialized(TypedDict):
     sk: NotRequired[str]  # Share key, only present present in shared (public) files / folder
 
     s: NotRequired[int]  # size (files only)
+    fa: NotRequired[str]  # additional file attributes for multimedia files (ex: thumbail)
 
 
 class OwnerKeys(TypedDict):
@@ -267,6 +268,7 @@ class Node(_DictDumper):
             keys: dict[str, str] = {}
 
         size = node.get("s")
+
         return Node(
             id=node["h"],
             parent_id=node["p"],

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/src/mega/transfer_it.py

@@ -45,9 +45,13 @@ class TransferItAPI(MegaAPI):
         return await super().post(json, params)
 
 
-class TransferItClient(APIContextManager):
-    def __init__(self, session: aiohttp.ClientSession | None = None, *, user_agent: str | None = None) -> None:  # pyright: ignore[reportMissingSuperCall]
-        self._api = TransferItAPI(session, user_agent=user_agent)
+class TransferItClient(APIContextManager[TransferItAPI]):
+    __slots__ = ()
+
+    def __init__(self, session: aiohttp.ClientSession | None = None, *, user_agent: str | None = None) -> None:
+        self._api = TransferItAPI(session)
+        if user_agent:
+            self._api.user_agent = user_agent
 
     @property
     def progress_bar(self) -> _GeneratorContextManager[None, None, None]:

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/src/mega/utils.py

@@ -14,6 +14,7 @@ from typing import TYPE_CHECKING, Literal, TypeVar, overload
 import aiohttp
 import yarl
 
+from mega import LOG_FILE_PROGRESS
 from mega.errors import ValidationError
 
 if TYPE_CHECKING:
@@ -63,7 +64,7 @@ def setup_logger(level: int = logging.INFO) -> None:
 
 
 def progress_logger(output_path: Path, file_size: int, *, download: bool) -> Callable[[float], None]:
-    if not logger.isEnabledFor(10):
+    if not (LOG_FILE_PROGRESS.get() and logger.isEnabledFor(logging.DEBUG)):
         return lambda _: None
 
     from mega.data_structures import ByteSize

{async_mega_py-2.2.0 → async_mega_py-2.2.2}/src/mega/vault.py

@@ -50,12 +50,14 @@ class MegaVault:
 
         # files shared with me
         elif node.owner in self._shared_keys:
-            real_node_id, share_key = next(p for p in self._shared_keys[node.owner].items() if p[0] in node.keys)
+            real_node_id = next(node_id for node_id in node.keys if node_id in self._shared_keys[node.owner])
+            share_key = self._shared_keys[node.owner][real_node_id]
             full_key = decrypt_key(b64_to_a32(node.keys[real_node_id]), share_key)
 
         # public files/folders
         elif share_key := self._shared_keys[_EXPORTED].get(node.id):
-            encrypted_key = b64_to_a32(next(iter(node.keys.values())))
+            real_node_id = next(node_id for node_id in node.keys if node_id in self._shared_keys[_EXPORTED])
+            encrypted_key = b64_to_a32(node.keys[real_node_id])
             full_key = decrypt_key(encrypted_key, share_key)
 
         else:

async_mega_py-2.2.0/src/mega/__init__.py

@@ -1,4 +0,0 @@
-import importlib.metadata
-
-_package_name_ = "async-mega-py"
-__version__ = importlib.metadata.version(_package_name_)