quantplay 2.0.57__tar.gz → 2.0.58__tar.gz

{quantplay-2.0.57 → quantplay-2.0.58}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantplay
-Version: 2.0.57
+Version: 2.0.58
 Summary: This python package will be stored in AWS CodeArtifact
 Home-page: 
 Author: 

{quantplay-2.0.57 → quantplay-2.0.58}/quantplay/broker/broker_factory.py

@@ -142,7 +142,9 @@ class BrokerFactory:
                 f"Missing Arguments for {user_broker_account['username']}:{user_broker_account['nickname']} in broker '{broker}' -> {compare_map[broker].difference(user_broker_account.keys())}"
             )
 
-    def generate_token(self, user_broker_account: Dict[str, Any]):
+    def generate_token(
+        self, user_broker_account: Dict[str, Any]
+    ) -> Dict[str, BrokerType | Any]:
         broker_client: BrokerType | None = None
 
         broker_data = user_broker_account["broker_data"]

{quantplay-2.0.57 → quantplay-2.0.58}/quantplay/broker/dhan.py

@@ -10,6 +10,7 @@ from quantplay.exception.exceptions import (
     InvalidArgumentException,
     QuantplayOrderPlacementException,
     RetryableException,
+    TokenException,
     retry_exception,
 )
 from quantplay.model.broker import (
@@ -32,10 +33,14 @@ class Dhan(Broker):
         access_token: str,
         load_instrument: bool = True,
     ):
-        self.dhan = dhanhq(
-            client_id=user_id,
-            access_token=access_token,
-        )
+        try:
+            self.dhan = dhanhq(
+                client_id=user_id,
+                access_token=access_token,
+            )
+
+        except Exception:
+            raise TokenException("Dhan Session Expired")
 
     def load_instrument(self, file_name: str | None = None) -> None:
         super().load_instrument("upstox_instruments")

{quantplay-2.0.57 → quantplay-2.0.58}/quantplay/broker/upstox.py

@@ -51,39 +51,20 @@ class Upstox(Broker):
     ):
         super().__init__()
 
-        try:
-            if access_token:
-                self.set_access_token(access_token)
-                self.user_id = user_id
-
-            # TODO: Implement Auto Login
-            # elif (
-            #     api_key
-            #     and api_secret
-            #     and totp
-            #     and mobile_number
-            #     and account_pin
-            #     and redirect_url
-            # ):
-            #     new_access_token = self.generate_token(
-            #         api_key, api_secret, totp, mobile_number, account_pin, redirect_url
-            #     )
-            #     self.set_access_token(new_access_token)
-
-            #     self.configuration = upstox_client.Configuration()
-            #     self.configuration.access_token = self.access_token
-
-            else:
-                raise InvalidArgumentException("Missing Args")
-
-        except Exception as e:
-            raise e
+        if access_token:
+            self.set_access_token(access_token)
+            self.user_id = user_id
+        else:
+            raise TokenException("Upstox Session Expired")
 
-        self.configuration = upstox_client.Configuration()
-        self.configuration.access_token = self.access_token
-        self.api_version = "2.0"
+        try:
+            self.configuration = upstox_client.Configuration()
+            self.configuration.access_token = self.access_token
+            self.api_version = "2.0"
 
-        self.api_client = upstox_client.ApiClient(self.configuration)
+            self.api_client = upstox_client.ApiClient(self.configuration)
+        except Exception:
+            raise TokenException("Upstox Session Expired")
 
         if load_instrument:
             self.load_instrument()

{quantplay-2.0.57 → quantplay-2.0.58}/quantplay.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quantplay
-Version: 2.0.57
+Version: 2.0.58
 Summary: This python package will be stored in AWS CodeArtifact
 Home-page: 
 Author: 

{quantplay-2.0.57 → quantplay-2.0.58}/setup.py

@@ -21,7 +21,7 @@ requirements = [
 setup(
     name="quantplay",
     long_description=Path("README.md").read_text(),
-    version="2.0.57",
+    version="2.0.58",
     setup_requires=["pytest-runner"],
     install_requires=requirements,
     tests_require=[],