quantplay 2.0.23__tar.gz → 2.0.24__tar.gz

{quantplay-2.0.23 → quantplay-2.0.24}/PKG-INFO

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

{quantplay-2.0.23 → quantplay-2.0.24}/quantplay/broker/noren.py

@@ -505,7 +505,6 @@ class Noren(Broker):
                 "norenordno": "order_id",
                 "uid": "user_id",
                 "exch": "exchange",
-                "avgprc": "average_price",
                 "prd": "product",
                 "trantype": "transaction_type",
                 "qty": "quantity",
@@ -542,6 +541,12 @@ class Noren(Broker):
             orders_df = orders_df.with_columns(
                 pl.col("trgprc").cast(pl.Float64).alias("trigger_price")
             )
+        if "avgprc" not in orders_df.columns:
+            orders_df = orders_df.with_columns(pl.lit(None).alias("average_price"))
+        else:
+            orders_df = orders_df.with_columns(
+                pl.col("avgprc").cast(pl.Float64).alias("average_price")
+            )
         if "remarks" not in orders_df.columns:
             orders_df = orders_df.with_columns(pl.lit(None).alias("tag"))
         else:

{quantplay-2.0.23 → quantplay-2.0.24}/quantplay/broker/upstox.py

@@ -354,7 +354,8 @@ class Upstox(Broker):
             # Get Positions
             api_response = api_instance.get_positions(self.api_version)
             positions = [
-                position.to_dict() for position in api_response.data  # type:ignore
+                position.to_dict()
+                for position in api_response.data  # type:ignore
             ]
             positions_df = pl.DataFrame(positions)
         except ApiException as e:
@@ -484,13 +485,6 @@ class Upstox(Broker):
         if tag:
             orders_df = orders_df.filter(pl.col("tag") == tag)
 
-        orders_df = orders_df.with_columns(
-            pl.col("instrument_token")
-            .str.split_exact("|", 1)
-            .struct.rename_fields(["upstox_exchange", "token"])
-            .alias("fields")
-        ).unnest("fields")
-
         orders_df = orders_df.with_columns(
             pl.when(pl.col("exchange") == "NSE")
             .then(pl.col("tradingsymbol").str.replace("-EQ", ""))
@@ -498,6 +492,17 @@ class Upstox(Broker):
             .alias("tradingsymbol")
         )
 
+        orders_df = orders_df.with_columns(
+            pl.struct(["exchange", "tradingsymbol"])
+            .map_elements(
+                lambda x: int(
+                    self.symbol_attribute(x["exchange"], x["tradingsymbol"], "token")
+                ),
+                return_dtype=pl.Int64,
+            )
+            .alias("token")
+        )
+
         orders_df = orders_df.with_columns(
             pl.col("order_timestamp")
             .str.strptime(pl.Datetime, "%Y-%m-%d %H:%M:%S")

{quantplay-2.0.23 → quantplay-2.0.24}/quantplay.egg-info/PKG-INFO

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

{quantplay-2.0.23 → quantplay-2.0.24}/setup.py

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