quantplay 2.0.166__tar.gz → 2.0.168__tar.gz

{quantplay-2.0.166 → quantplay-2.0.168}/PKG-INFO

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

{quantplay-2.0.166 → quantplay-2.0.168}/quantplay/broker/kotak.py

@@ -61,6 +61,7 @@ class Kotak(Broker):
                 "bearer_token": configuration["bearer_token"],
                 "view_token": configuration["view_token"],
                 "sid": configuration["sid"],
+                "kotakUserId": configuration["kotakUserId"],
                 "edit_token": configuration.get("edit_token"),
                 "edit_sid": configuration.get("edit_sid"),
                 "edit_rid": configuration.get("edit_rid"),
@@ -139,7 +140,7 @@ class Kotak(Broker):
             self.configuration["view_token"], options={"verify_signature": False}
         )
         userid = decode_jwt.get("sub")
-        self.kotakUserId = userid
+        self.configuration["kotakUserId"] = userid
 
         generate_otp_resp = requests.post(
             url=f"{PROD_BASE_URL}login/1.0/login/otp/generate",
@@ -154,6 +155,7 @@ class Kotak(Broker):
             self.configuration["bearer_token"] is None
             or self.configuration["sid"] is None
             or self.configuration["view_token"] is None
+            or self.configuration["kotakUserId"] is None
         ):
             raise InvalidArgumentException("Missing Params")
 
@@ -165,7 +167,7 @@ class Kotak(Broker):
                 "Auth": self.configuration["view_token"],
                 "Content-Type": "application/json",
             },
-            data=json.dumps({"userId": self.kotakUserId, "otp": otp}),
+            data=json.dumps({"userId": self.configuration["kotakUserId"], "otp": otp}),
         )
 
         edit_token_resp = json.loads(login_resp.text)
@@ -180,7 +182,39 @@ class Kotak(Broker):
     def order_history(self, order_id: str) -> pl.DataFrame:
         body = {"nOrdNo": order_id}
         order_history = self.request("order_history", body=body)["data"]  # type:ignore
-        return pl.DataFrame(order_history)
+        order_history_df = pl.DataFrame(order_history)[
+            [
+                "tok",
+                "flDtTm",
+                "nOrdNo",
+                "exch",
+                "prod",
+                "trdSym",
+                "trnsTp",
+                "qty",
+                "prcTp",
+            ]
+        ]
+        order_history_df = order_history_df.rename(
+            {
+                "tok": "token",
+                "flDtTm": "update_timestamp",
+                "nOrdNo": "order_id",
+                "exch": "exchange",
+                "prod": "product",
+                "trdSym": "tradingsymbol",
+                "trnsTp": "transaction_type",
+                "qty": "quantity",
+                "prcTp": "order_type",
+            }
+        )
+        order_history_df = order_history_df.with_columns(
+            pl.col("update_timestamp")
+            .str.strptime(pl.Datetime(time_unit="ms"), format="%d-%b-%Y %H:%M:%S")
+            .alias("update_timestamp")
+        ).sort("update_timestamp")
+
+        return order_history_df
 
     # **
     # ** GET Api's
@@ -455,11 +489,53 @@ class Kotak(Broker):
         self.request("cancel_order", body=body)
 
     def modify_order(self, order: ModifyOrderRequest) -> str:
-        return ""
+        order_id = order["order_id"]
+        price = order.get("price")
+        trigger_price = order.get("trigger_price", 0)
 
-    # **
-    # ** General Utils
-    # **
+        order_history = self.order_history(order_id).to_dicts()[-1]
+
+        order_type = self.get_order_type(
+            order.get("order_type", order_history["order_type"])
+        )
+        quantity = str(order.get("quantity", order_history["quantity"]))
+        modify_order_request = {
+            "tk": str(order_history["token"]),
+            "mp": "0",
+            "pc": order_history["product"],
+            "dd": "NA",
+            "dq": "0",
+            "vd": "DAY",
+            "ts": order_history["tradingsymbol"],
+            "tt": order_history["transaction_type"],
+            "pr": str(price),
+            "tp": str(trigger_price),
+            "qt": quantity,
+            "no": order_id,
+            "es": self.get_exchange(order_history["exchange"]),
+            "pt": order_type,
+        }
+
+        try:
+            Constants.logger.info(f"Modifying order [{order_id}] new price [{price}]")
+
+            modify_order_response = self.request(  # type:ignore
+                "modify_order", body=modify_order_request
+            )
+
+            return modify_order_response["nOrdNo"]
+
+        except Exception as e:
+            exception_message = (
+                f"OrderModificationFailed for {order_id} failed with exception {e}"
+            )
+            if (
+                "Order cannot be modified as it is being processed"
+                not in exception_message
+            ):
+                Constants.logger.error(f"{exception_message}")
+
+        return order_id
 
     def get_transaction_type(self, transaction_type: TransactionType) -> str:
         transaction_type_map: dict[TransactionType, str] = {

{quantplay-2.0.166 → quantplay-2.0.168}/quantplay.egg-info/PKG-INFO

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

{quantplay-2.0.166 → quantplay-2.0.168}/setup.py

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