quantplay 2.0.135__tar.gz → 2.0.136__tar.gz

{quantplay-2.0.135 → quantplay-2.0.136}/PKG-INFO

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

{quantplay-2.0.135 → quantplay-2.0.136}/quantplay/broker/icici_direct.py

@@ -238,6 +238,12 @@ class ICICI(Broker):
             .when(pl.col("right") == "Put")
             .then(pl.lit("PE"))
             .alias("instrument_type"),
+            (
+                    pl.col("unrealized_profit").cast(pl.Float32)
+                    + pl.col("realized_profit").cast(pl.Float32)
+            ).alias("pnl"),
+            pl.col("quantity").cast(pl.Int32).alias("quantity"),
+            pl.col("average_price").cast(pl.Float32).alias("average_price")
         )
 
         positions_df = positions_df.with_columns(
@@ -264,17 +270,17 @@ class ICICI(Broker):
         )
 
         positions_df = positions_df.with_columns(
-            pl.lit(0).alias("sell_value"),
-            pl.lit(0).alias("buy_value"),
+            pl.when(pl.col("quantity") < 0)
+            .then(pl.col("quantity").abs()*pl.col("average_price")).otherwise(pl.lit(0)).alias("sell_value"),
+            pl.when(pl.col("quantity") > 0)
+            .then(pl.col("quantity").abs() * pl.col("average_price")).otherwise(pl.lit(0)).alias("buy_value"),
             pl.lit(0).alias("ltp"),
-            pl.lit(0).alias("sell_quantity"),
+            pl.when(pl.col("quantity") < 0)
+            .then(pl.col("quantity").abs()).otherwise(pl.lit(0)).alias("sell_quantity"),
+            pl.when(pl.col("quantity") > 0)
+            .then(pl.col("quantity").abs()).otherwise(pl.lit(0)).alias("buy_quantity"),
             pl.lit("NRML").alias("product"),
-            pl.lit(0).alias("buy_quantity"),
             pl.col("instrument_type").alias("option_type"),
-            (
-                pl.col("unrealized_profit").cast(pl.Float32)
-                + pl.col("realized_profit").cast(pl.Float32)
-            ).alias("pnl"),
         )
         return positions_df[list(self.positions_schema.keys())].cast(
             self.positions_schema
@@ -382,7 +388,7 @@ class ICICI(Broker):
         }
 
     def get_exchange(self, exchange: ExchangeType) -> ...:
-        return
+        return exchange
 
     def get_product(self, product: ProductType) -> ...:
         return

{quantplay-2.0.135 → quantplay-2.0.136}/quantplay.egg-info/PKG-INFO

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

{quantplay-2.0.135 → quantplay-2.0.136}/setup.py

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