quantplay 2.0.41__tar.gz → 2.0.42__tar.gz

{quantplay-2.0.41 → quantplay-2.0.42}/PKG-INFO

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

{quantplay-2.0.41 → quantplay-2.0.42}/quantplay/broker/five_paisa.py

@@ -170,16 +170,14 @@ class FivePaisa(Broker):
     def add_exchange(self, data: pl.DataFrame):
         return data.with_columns(
             pl.lit(None).alias("exchange"),
+        ).with_columns(
             pl.when((pl.col("Exch").eq("N")) & (pl.col("ExchType").eq("D")))
             .then(pl.lit("NFO"))
-            .alias("exchange"),
-            pl.when((pl.col("Exch").eq("N")) & (pl.col("ExchType").eq("C")))
+            .when((pl.col("Exch").eq("N")) & (pl.col("ExchType").eq("C")))
             .then(pl.lit("NFO"))
-            .alias("exchange"),
-            pl.when((pl.col("Exch").eq("B")) & (pl.col("ExchType").eq("D")))
+            .when((pl.col("Exch").eq("B")) & (pl.col("ExchType").eq("D")))
             .then(pl.lit("NFO"))
-            .alias("exchange"),
-            pl.when((pl.col("Exch").eq("B")) & (pl.col("ExchType").eq("C")))
+            .when((pl.col("Exch").eq("B")) & (pl.col("ExchType").eq("C")))
             .then(pl.lit("NFO"))
             .alias("exchange"),
         )
@@ -280,8 +278,7 @@ class FivePaisa(Broker):
             ).alias("pnl"),
         )
 
-        self.add_exchange(positions)
-
+        positions = self.add_exchange(positions)
         positions = positions.with_columns(
             pl.when(pl.col("exchange").is_in(["NFO", "BFO"]))
             .then(pl.col("tradingsymbol").str.split(" ").list.get(-2))
@@ -322,8 +319,7 @@ class FivePaisa(Broker):
             return pl.DataFrame(schema=self.orders_schema)
 
         orders = pl.from_dicts(orders)
-
-        self.add_exchange(orders)
+        orders = self.add_exchange(orders)
 
         orders = orders.rename(
             {
@@ -386,12 +382,12 @@ class FivePaisa(Broker):
             )
             .with_columns(
                 pl.when(pl.col("status").str.to_lowercase().str.contains("rejected"))
-                .then(OrderStatus.rejected)
+                .then(pl.lit(OrderStatus.rejected))
                 .otherwise(pl.col("status"))
                 .alias("status"),
                 pl.when(
-                    pl.col("product")
-                    == "CNC" & (pl.col("exchange").is_in(["NFO", "BFO"]))
+                    (pl.col("product").eq("CNC"))
+                    & (pl.col("exchange").is_in(["NFO", "BFO"]))
                 )
                 .then(pl.lit("NRML"))
                 .otherwise(pl.col("product"))
@@ -399,7 +395,7 @@ class FivePaisa(Broker):
             )
             .with_columns(
                 pl.when((pl.col("status") == "OPEN") & (pl.col("order_type") == "SL"))
-                .then(OrderStatus.trigger_pending)
+                .then(pl.lit(OrderStatus.trigger_pending))
                 .otherwise(pl.col("status"))
                 .alias("status")
             )

{quantplay-2.0.41 → quantplay-2.0.42}/quantplay.egg-info/PKG-INFO

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

{quantplay-2.0.41 → quantplay-2.0.42}/setup.py

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