tonutils 0.5.5__py3-none-any.whl → 0.5.6__py3-none-any.whl

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/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/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/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/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.dist-info → tonutils-0.5.6.dist-info}/METADATA

@@ -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.dist-info → tonutils-0.5.6.dist-info}/RECORD

@@ -6,12 +6,12 @@ tonutils/exceptions.py,sha256=FCgD7mr4K5KDv7ZLkQmy2CseMSqi1S4UzOQuLybNkSk,1351
 tonutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tonutils/utils.py,sha256=ZsIDGIDCzoH34qv79uSE3KH_Gqa2nYxD1csalVJX90I,8679
 tonutils/client/__init__.py,sha256=-MmgreKvoHTywzzBKFNbuX4wOJrwiFfKz2kE38_FeK8,404
-tonutils/client/_base.py,sha256=h997m-E5ziG-j1xsYF7szT4jCQ7s5ScuXk5loQ8CSF8,9924
+tonutils/client/_base.py,sha256=W4IQof61jWyRBL-_Fim9TWACHqhr0LdJ3Uw_y-ImRGs,10552
 tonutils/client/lite.py,sha256=iLgVSUWwDNrMwRFOP1VUq8iMItvV4g-PI9n0wfVQQK4,5853
-tonutils/client/quicknode.py,sha256=X2TajtKr-52ksb-ZuLvHjQnh9beefZUkpnQ7MaKJp2g,1147
-tonutils/client/tatum.py,sha256=aYUZPm4YKIxlD8OYRFmKZHPrK6RXcSTjxozp-MdPVWs,1676
-tonutils/client/tonapi.py,sha256=PsKFX6Q2ShMs44QH-2NKx19Z-Ls9vgIGOf1-3JqWGfA,5235
-tonutils/client/toncenter.py,sha256=yn3ctOusYLauAfkPBpRf7vsBH95E4tu-iGNlGRdwJ6Y,10406
+tonutils/client/quicknode.py,sha256=HfeZwhdjYx2ob6T2EJT8dgfLR2bm67q2BZSS4ZSh5Sg,1674
+tonutils/client/tatum.py,sha256=cLCj_DUaL8butSWfuCGT47f4rD6B4stYQVX_3vBJNOE,2156
+tonutils/client/tonapi.py,sha256=g8eabxEY3cYoG4Sr1TqvTmOxwtx0prXggAK9NiPkYOg,5714
+tonutils/client/toncenter.py,sha256=LPuBmtqGpvzRKmIeC6vBNXh4-2-X_8sa4VUcZ073y9Q,11330
 tonutils/client/utils.py,sha256=ouzzS_r0YYHplmzoE9-2seLhtjU7O32FLGwSIgIoo3k,7339
 tonutils/dns/__init__.py,sha256=zEOmml_INEoBfr3h5S1PHvr1Wksr_9Owf8S3gLWDHWI,52
 tonutils/dns/categories.py,sha256=1rWN1LmuHt0rmPOf0EJDNMeYkmnKR4nYCPNwLJf2dvc,281
@@ -125,8 +125,8 @@ tonutils/wallet/contract/v2.py,sha256=SWik7CtQzNjcyjNc2G_hu8xPYun28GC_Ky0Sky0jX5
 tonutils/wallet/contract/v3.py,sha256=0mTWsZqTom1dWsKtjcUjyjt7Z7f0pExdDxeQi0lbk_Y,3107
 tonutils/wallet/contract/v4.py,sha256=LfCujQUNy_ScBfTqgSWsr6cRs_IGXkq2pxAML0Qvsz4,6347
 tonutils/wallet/contract/v5.py,sha256=uiLVI_tAI-69__MUeZZFKzNo3GRuoNqKdQRUbZ1NL20,8355
-tonutils-0.5.5.dist-info/licenses/LICENSE,sha256=fr0RzMsygzM2gNKLyH8NdrwYGFC4RCMhqMrtICms5mM,1060
-tonutils-0.5.5.dist-info/METADATA,sha256=URkhRJTv8P9GHlXswdQdlYYLTsDCAfnq4NJoZlax63E,4552
-tonutils-0.5.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tonutils-0.5.5.dist-info/top_level.txt,sha256=-7H_mGl8S9HKQrkUiTLmEbtMM-knzRzd_a0cZZnuZIU,9
-tonutils-0.5.5.dist-info/RECORD,,
+tonutils-0.5.6.dist-info/licenses/LICENSE,sha256=fr0RzMsygzM2gNKLyH8NdrwYGFC4RCMhqMrtICms5mM,1060
+tonutils-0.5.6.dist-info/METADATA,sha256=-RyJGfXBs9lLMN_QLhmGIKfCY8vDkqDLdr27xiSfd-0,4552
+tonutils-0.5.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tonutils-0.5.6.dist-info/top_level.txt,sha256=-7H_mGl8S9HKQrkUiTLmEbtMM-knzRzd_a0cZZnuZIU,9
+tonutils-0.5.6.dist-info/RECORD,,