tonutils 0.5.5__tar.gz → 0.5.6__tar.gz

{tonutils-0.5.5/tonutils.egg-info → tonutils-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tonutils
-Version: 0.5.5
+Version: 0.5.6
 Summary: Tonutils is a high-level, object-oriented Python library designed to facilitate seamless interactions with the TON blockchain.
 Author: nessshon
 License: MIT

{tonutils-0.5.5 → tonutils-0.5.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tonutils"
-version = "0.5.5"
+version = "0.5.6"
 authors = [
     { name = "nessshon" },
 ]

{tonutils-0.5.5 → tonutils-0.5.6}/tonutils/client/_base.py

@@ -3,7 +3,8 @@ from __future__ import annotations
 import asyncio
 import json
 import random
-from typing import Any, Optional, List, Dict, Union
+from typing import Any, Optional, List, Dict, Union, AsyncGenerator
+from contextlib import asynccontextmanager
 
 import aiohttp
 from aiolimiter import AsyncLimiter
@@ -27,6 +28,7 @@ class Client:
         self.rps = kwargs.get("rps", 1)
         self.max_retries = kwargs.get("max_retries", 0)
 
+        self._session: aiohttp.ClientSession = kwargs.get("session", None)
         self._limiter = AsyncLimiter(
             max_rate=self.rps,
             time_period=1,
@@ -86,6 +88,24 @@ class Client:
         seconds = retry_after + random.uniform(0.2, 0.5)
         await asyncio.sleep(seconds)
 
+    @asynccontextmanager
+    async def _get_session(self) -> AsyncGenerator[aiohttp.ClientSession, None]:
+        if self._session:
+            yield self._session
+        else:
+            async with aiohttp.ClientSession(
+                    timeout=aiohttp.ClientTimeout(total=self.timeout)
+            ) as session:
+                yield session
+
+    async def close_session(self) -> None:
+        """
+        Close the aiohttp session if it was created by the client.
+        """
+        if self._session:
+            await self._session.close()
+            self._session = None
+
     async def _request(
             self,
             method: str,
@@ -114,14 +134,12 @@ class Client:
                 await self._limiter.acquire()
 
             try:
-                async with aiohttp.ClientSession(
-                        headers=self.headers,
-                        timeout=aiohttp.ClientTimeout(total=self.timeout)
-                ) as session:
+                async with self._get_session() as session:
                     async with session.request(
                             method=method,
                             url=url,
                             params=params,
+                            headers=self.headers,
                             json=body,
                     ) as response:
                         content = await self._parse_response(response)

{tonutils-0.5.5 → tonutils-0.5.6}/tonutils/client/quicknode.py

@@ -1,5 +1,7 @@
 from typing import Optional
 
+from aiohttp import ClientSession
+
 from .toncenter import ToncenterV2Client
 
 
@@ -17,6 +19,8 @@ class QuicknodeClient(ToncenterV2Client):
             http_provider_url: str,
             rps: Optional[int] = None,
             max_retries: int = 1,
+            timeout: Optional[int] = 10,
+            session: Optional[ClientSession] = None,
     ) -> None:
         """
         Initialize the QuicknodeClient.
@@ -25,9 +29,17 @@ class QuicknodeClient(ToncenterV2Client):
             You can obtain one at: https://quicknode.com
         :param rps: Optional requests per second (RPS) limit.
         :param max_retries: Number of retries for rate-limited requests. Defaults to 1.
+        :param timeout: Response timeout in seconds. Not used if session is specified. Defaults to 10.
+        :param session: Aiohttp session to avoid creating new ones every time. By default, a new one is created for each request. If specified, remember to call session_close when finished.
         """
         if not http_provider_url:
             raise ValueError("`http_provider_url` is required to initialize QuicknodeClient")
 
         base_url = http_provider_url.rstrip("/") + self.API_VERSION_PATH
-        super().__init__(base_url=base_url, rps=rps, max_retries=max_retries)
+        super().__init__(
+            base_url=base_url,
+            rps=rps,
+            max_retries=max_retries,
+            timeout=timeout,
+            session=session,
+        )

{tonutils-0.5.5 → tonutils-0.5.6}/tonutils/client/tatum.py

@@ -1,5 +1,7 @@
 from typing import Optional
 
+from aiohttp import ClientSession
+
 from .toncenter import ToncenterV2Client
 
 
@@ -20,6 +22,8 @@ class TatumClient(ToncenterV2Client):
             base_url: Optional[str] = None,
             rps: Optional[int] = None,
             max_retries: int = 1,
+            timeout: Optional[int] = 10,
+            session: Optional[ClientSession] = None,
     ) -> None:
         """
         Initialize the TatumClient.
@@ -30,6 +34,8 @@ class TatumClient(ToncenterV2Client):
         :param base_url: Optional custom base URL. Defaults to Tatum's public gateway.
         :param rps: Optional requests per second (RPS) limit.
         :param max_retries: Number of retries for rate-limited requests. Defaults to 1.
+        :param timeout: Response timeout in seconds. Not used if session is specified. Defaults to 10.
+        :param session: Aiohttp session to avoid creating new ones every time. By default, a new one is created for each request. If specified, remember to call session_close when finished.
         """
         if not api_key:
             raise ValueError("`api_key` is required to initialize TatumClient.")
@@ -47,4 +53,6 @@ class TatumClient(ToncenterV2Client):
             is_testnet=is_testnet,
             rps=rps,
             max_retries=max_retries,
+            timeout=timeout,
+            session=session,
         )

{tonutils-0.5.5 → tonutils-0.5.6}/tonutils/client/tonapi.py

@@ -1,6 +1,7 @@
 from typing import Any, Dict, List, Optional, Union
 
 from pytoniq_core import Address, Builder, Cell, HashMap, Transaction, Slice
+from aiohttp import ClientSession
 
 from ._base import Client
 from .utils import RunGetMethodStack, RunGetMethodResult, unpack_config
@@ -24,6 +25,8 @@ class TonapiClient(Client):
             base_url: Optional[str] = None,
             rps: Optional[int] = None,
             max_retries: int = 1,
+            timeout: Optional[int] = 10,
+            session: Optional[ClientSession] = None,
     ) -> None:
         """
         Initialize the TonapiClient.
@@ -33,6 +36,8 @@ class TonapiClient(Client):
         :param base_url: Optional custom base URL. If not provided, uses the official endpoint.
         :param rps: Optional requests per second (RPS) limit.
         :param max_retries: Number of retries for rate-limited requests. Defaults to 1.
+        :param timeout: Response timeout in seconds. Not used if session is specified. Defaults to 10.
+        :param session: Aiohttp session to avoid creating new ones every time. By default, a new one is created for each request. If specified, remember to call session_close when finished.
         """
         if not api_key:
             raise ValueError("`api_key` is required to initialize TonapiClient.")
@@ -47,6 +52,8 @@ class TonapiClient(Client):
             is_testnet=is_testnet,
             rps=rps,
             max_retries=max_retries,
+            timeout=timeout,
+            session=session,
         )
 
     async def run_get_method(

{tonutils-0.5.5 → tonutils-0.5.6}/tonutils/client/toncenter.py

@@ -2,6 +2,7 @@ import base64
 from typing import Any, Dict, List, Optional, Union
 
 from pytoniq_core import Address, Builder, Cell, HashMap, Transaction, Slice
+from aiohttp import ClientSession
 
 from ._base import Client
 from .utils import RunGetMethodStack, RunGetMethodResult, unpack_config
@@ -26,6 +27,8 @@ class ToncenterV2Client(Client):
             base_url: Optional[str] = None,
             rps: Optional[int] = None,
             max_retries: int = 1,
+            timeout: Optional[int] = 10,
+            session: Optional[ClientSession] = None,
     ) -> None:
         """
         Initialize the ToncenterV2Client.
@@ -37,6 +40,8 @@ class ToncenterV2Client(Client):
             If not set, defaults to the official Toncenter URLs.
         :param rps: Optional requests per second (RPS) limit.
         :param max_retries: Number of retries for rate-limited requests. Defaults to 1.
+        :param timeout: Response timeout in seconds. Not used if session is specified. Defaults to 10.
+        :param session: Aiohttp session to avoid creating new ones every time. By default, a new one is created for each request. If specified, remember to call session_close when finished.
         """
         default_url = "https://testnet.toncenter.com" if is_testnet else "https://toncenter.com"
         base_url = (base_url or default_url).rstrip("/") + self.API_VERSION_PATH
@@ -48,6 +53,8 @@ class ToncenterV2Client(Client):
             is_testnet=is_testnet,
             rps=rps,
             max_retries=max_retries,
+            timeout=timeout,
+            session=session,
         )
 
     async def run_get_method(
@@ -178,6 +185,8 @@ class ToncenterV3Client(Client):
             base_url: Optional[str] = None,
             rps: Optional[int] = None,
             max_retries: int = 1,
+            timeout: Optional[int] = 10,
+            session: Optional[ClientSession] = None,
     ) -> None:
         """
         Initialize the ToncenterV3Client.
@@ -189,6 +198,8 @@ class ToncenterV3Client(Client):
             Defaults to official Toncenter endpoints.
         :param rps: Optional requests per second (RPS) limit.
         :param max_retries: Number of retries for rate-limited requests. Defaults to 1.
+        :param timeout: Response timeout in seconds. Not used if session is specified. Defaults to 10.
+        :param session: Aiohttp session to avoid creating new ones every time. By default, a new one is created for each request. If specified, remember to call session_close when finished.
         """
         default_url = (
             "https://testnet.toncenter.com"
@@ -204,6 +215,8 @@ class ToncenterV3Client(Client):
             is_testnet=is_testnet,
             rps=rps,
             max_retries=max_retries,
+            timeout=timeout,
+            session=session,
         )
 
     async def run_get_method(

{tonutils-0.5.5 → tonutils-0.5.6/tonutils.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tonutils
-Version: 0.5.5
+Version: 0.5.6
 Summary: Tonutils is a high-level, object-oriented Python library designed to facilitate seamless interactions with the TON blockchain.
 Author: nessshon
 License: MIT