HedgeTech 0.2.5__tar.gz → 0.3.0b2__tar.gz

hedgetech-0.3.0b2/HedgeTech/DataEngine/__ime/__AsyncClient.py

@@ -0,0 +1,349 @@
+# ========================================|======================================== #
+#                                      Imports                                      #
+# ========================================|======================================== #
+
+from asyncio import (
+    Event,
+    wait_for,
+    TimeoutError,
+    sleep,
+)
+from websockets.asyncio.client import connect
+from json import loads
+from HedgeTech.Auth import AuthAsyncClient
+from typing import (
+    Literal,
+    List,
+    AsyncGenerator,
+)
+from .__io_types import (
+    WebsocketBase_contractName,
+    WebsocketBase_contractId,
+    
+    OverviewResponse,
+    Contracts,
+    
+    ContractInfoResponse,
+    AllowedPriceRangeResponse,
+    AggregateResponse,
+    BestLimitResponse,
+)
+
+# ========================================|======================================== #
+#                                 Class Definitions                                 #
+# ========================================|======================================== #
+
+class DataEngine_Ime_AsyncClient:
+    
+    def __init__(
+        self,
+        AuthAsyncClient : AuthAsyncClient,
+    ):
+        
+        self.__AuthAsyncClient = AuthAsyncClient
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def search_contracts(
+        self,
+        market : Literal[
+            "IMEDepositCertificate",
+            "IMEFutures",
+            "IMEOption",
+        ],
+        Search_char : str,
+    )-> Contracts :
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/static/data/contract/search',
+            params={
+                'market' : market,
+                'Search_char' : Search_char
+            } 
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def contracts_static_info_by_name(self,contract_names : List[str])-> Contracts:
+        
+            data = await self.__AuthAsyncClient.httpx_Client.get(
+                url='https://core.hedgetech.ir/data-engine/ime/static/data/contract/name',
+                params=[('contract_names', name) for name in contract_names]
+            )
+            
+            if data.is_success :
+                
+                return data.json()
+            
+            else :
+                
+                raise ValueError(data.json().get('detail'))
+            
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def contracts_static_info_by_id(self,contract_ids : List[str | int])-> Contracts:
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/static/data/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def live_overview_by_name(self,contract_names : List[str])-> OverviewResponse:
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/overview/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def live_overview_by_id(self,contract_ids : List[str | int])-> OverviewResponse:
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/overview/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def live_best_limit_by_name(self,contract_names : List[str])-> BestLimitResponse:
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/best-limit/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def live_best_limit_by_id(self,contract_ids : List[str | int])-> BestLimitResponse:
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/best-limit/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def live_aggregate_by_name(self,contract_names : List[str])-> AggregateResponse:
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/aggregate/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+
+    # +--------------------------------------------------------------------------------------+ #
+
+
+    async def live_aggregate_by_id(self,contract_ids : List[str | int])-> AggregateResponse :
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/aggregate/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def live_contract_info_by_name(self,contract_names : List[str])-> ContractInfoResponse:
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/contract-info/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+
+    # +--------------------------------------------------------------------------------------+ #
+
+
+    async def live_contract_info_by_id(self,contract_ids : List[str | int])-> ContractInfoResponse :
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/contract-info/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def live_allowed_price_range_by_name(self,contract_names : List[str])-> AllowedPriceRangeResponse:
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/allowed-price-range/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+
+    # +--------------------------------------------------------------------------------------+ #
+
+
+    async def live_allowed_price_range_by_id(self,contract_ids : List[str | int])-> AllowedPriceRangeResponse :
+        
+        data = await self.__AuthAsyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/allowed-price-range/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+
+    # +--------------------------------------------------------------------------------------+ #
+
+    
+    async def websocket_by_name(
+        self,
+        contract_names: List[str],
+        event: Event | None = None,
+    )-> AsyncGenerator[
+        WebsocketBase_contractName,
+        None
+    ]:
+        
+        if not isinstance(event,Event):
+            event = Event()
+            event.set()
+        
+        async with connect(
+            uri=(
+                f"wss://core.hedgetech.ir/data-engine/ime/live/data/websocket/contract/name?"
+                + "&".join(f"contract_name={s}" for s in contract_names)
+            ),
+            additional_headers=self.__AuthAsyncClient.token
+        ) as ws:
+                            
+            while event.is_set():
+                
+                try : yield loads(await wait_for(ws.recv(), timeout=2))
+                
+                except TimeoutError: 
+                    await sleep(0)
+                    continue
+                
+                except Exception: 
+                    await ws.close()
+                    break
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    async def websocket_by_id(
+        self,
+        contract_ids: List[str | int],
+        event: Event | None = None,
+    )-> AsyncGenerator[
+        WebsocketBase_contractId,
+        None
+    ]:
+        
+        if not isinstance(event,Event):
+            event = Event()
+            event.set()
+        
+        async with connect(
+            uri=(
+                f"wss://core.hedgetech.ir/data-engine/ime/live/data/websocket/contract/id?"
+                + "&".join(f"contract_id={s}" for s in contract_ids)
+            ),
+            additional_headers=self.__AuthAsyncClient.token
+        ) as ws:
+                            
+            while event.is_set():
+                
+                try : yield loads(await wait_for(ws.recv(), timeout=2))
+                
+                except TimeoutError: 
+                    await sleep(0)
+                    continue
+                
+                except Exception: 
+                    await ws.close()
+                    break

hedgetech-0.3.0b2/HedgeTech/DataEngine/__ime/__SyncClient.py

@@ -0,0 +1,337 @@
+# ========================================|======================================== #
+#                                      Imports                                      #
+# ========================================|======================================== #
+
+from threading import Event
+from websockets.sync.client import connect
+from json import loads
+from HedgeTech.Auth import AuthSyncClient
+from typing import (
+    Literal,
+    List,
+    Generator,
+)
+from .__io_types import (
+    WebsocketBase_contractName,
+    WebsocketBase_contractId,
+    
+    OverviewResponse,
+    Contracts,
+    
+    ContractInfoResponse,
+    AllowedPriceRangeResponse,
+    AggregateResponse,
+    BestLimitResponse,
+)
+
+# ========================================|======================================== #
+#                                 Class Definitions                                 #
+# ========================================|======================================== #
+
+class DataEngine_Ime_SyncClient:
+    
+    def __init__(
+        self,
+        AuthSyncClient : AuthSyncClient,
+    ):
+        
+        self.__AuthSyncClient = AuthSyncClient
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def search_contracts(
+        self,
+        market : Literal[
+            "IMEDepositCertificate",
+            "IMEFutures",
+            "IMEOption",
+        ],
+        Search_char : str,
+    )-> Contracts :
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/static/data/contract/search',
+            params={
+                'market' : market,
+                'Search_char' : Search_char
+            } 
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def contracts_static_info_by_name(self,contract_names : List[str])-> Contracts:
+        
+            data = self.__AuthSyncClient.httpx_Client.get(
+                url='https://core.hedgetech.ir/data-engine/ime/static/data/contract/name',
+                params=[('contract_names', name) for name in contract_names]
+            )
+            
+            if data.is_success :
+                
+                return data.json()
+            
+            else :
+                
+                raise ValueError(data.json().get('detail'))
+            
+    # +--------------------------------------------------------------------------------------+ #
+
+    def contracts_static_info_by_id(self,contract_ids : List[str | int])-> Contracts:
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/static/data/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def live_overview_by_name(self,contract_names : List[str])-> OverviewResponse:
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/overview/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def live_overview_by_id(self,contract_ids : List[str | int])-> OverviewResponse:
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/overview/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def live_best_limit_by_name(self,contract_names : List[str])-> BestLimitResponse:
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/best-limit/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def live_best_limit_by_id(self,contract_ids : List[str | int])-> BestLimitResponse:
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/best-limit/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def live_aggregate_by_name(self,contract_names : List[str])-> AggregateResponse:
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/aggregate/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+
+    # +--------------------------------------------------------------------------------------+ #
+
+
+    def live_aggregate_by_id(self,contract_ids : List[str | int])-> AggregateResponse :
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/aggregate/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def live_contract_info_by_name(self,contract_names : List[str])-> ContractInfoResponse:
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/contract-info/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+
+    # +--------------------------------------------------------------------------------------+ #
+
+
+    def live_contract_info_by_id(self,contract_ids : List[str | int])-> ContractInfoResponse :
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/contract-info/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+        
+    # +--------------------------------------------------------------------------------------+ #
+
+    def live_allowed_price_range_by_name(self,contract_names : List[str])-> AllowedPriceRangeResponse:
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/allowed-price-range/contract/name',
+            params=[('contract_names', name) for name in contract_names]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+
+    # +--------------------------------------------------------------------------------------+ #
+
+
+    def live_allowed_price_range_by_id(self,contract_ids : List[str | int])-> AllowedPriceRangeResponse :
+        
+        data = self.__AuthSyncClient.httpx_Client.get(
+            url='https://core.hedgetech.ir/data-engine/ime/live/data/allowed-price-range/contract/id',
+            params=[('contract_ids', id) for id in contract_ids]
+        )
+        
+        if data.is_success :
+            
+            return data.json()
+        
+        else :
+            
+            raise ValueError(data.json().get('detail'))
+
+    # +--------------------------------------------------------------------------------------+ #
+    
+    def websocket_by_name(
+        self,
+        contract_names: List[str],
+        event: Event | None = None,
+    )-> Generator[
+        WebsocketBase_contractName,
+        None
+    ]:
+        
+        if not isinstance(event,Event):
+            event = Event()
+            event.set()
+
+        with connect(
+            uri=(
+                f"wss://core.hedgetech.ir/data-engine/ime/live/data/websocket/contract/name?"
+                + "&".join(f"contract_name={s}" for s in contract_names)
+            ),
+            additional_headers=self.__AuthSyncClient.token
+        ) as ws:
+                        
+            while event.is_set():
+                
+                try : yield loads(ws.recv(timeout=2))
+                except TimeoutError : continue
+                except Exception: 
+                    ws.close()
+                    break
+        
+    # +--------------------------------------------------------------------------------------+ #
+    
+
+    def websocket_by_id(
+        self,
+        contract_ids: List[str | int],
+        event: Event | None = None,
+    )-> Generator[
+        WebsocketBase_contractId,
+        None
+    ]:
+        
+        if not isinstance(event,Event):
+            event = Event()
+            event.set()
+            
+        with connect(
+            uri=(
+                f"wss://core.hedgetech.ir/data-engine/ime/live/data/websocket/contract/id?"
+                + "&".join(f"contract_id={s}" for s in contract_ids)
+            ),
+            additional_headers=self.__AuthSyncClient.token
+        ) as ws:
+            
+            while event.is_set():
+                
+                try : yield loads(ws.recv(timeout=2))
+                except TimeoutError : continue
+                except Exception: 
+                    ws.close()
+                    break
+    

hedgetech-0.3.0b2/HedgeTech/DataEngine/__ime/__init__.py

@@ -0,0 +1,2 @@
+from .__AsyncClient import DataEngine_Ime_AsyncClient
+from .__SyncClient import DataEngine_Ime_SyncClient

hedgetech-0.3.0b2/HedgeTech/DataEngine/__ime/__io_types/__init__.py

@@ -0,0 +1,12 @@
+from .__response import(
+    WebsocketBase_contractName,
+    WebsocketBase_contractId,
+    
+    OverviewResponse,
+    Contracts,
+    
+    ContractInfoResponse,
+    AllowedPriceRangeResponse,
+    AggregateResponse,
+    BestLimitResponse,
+)