bw-essentials-core 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

bw_essentials/constants/services.py

@@ -20,3 +20,13 @@ class Services(Enum):
     PROMETHEUS_USER_APP = "Prometheus_User_App"
     PORTFOLIO_CATALOGUE = "Portfolio_Catalogue"
     PORTFOLIO_CONTENT = "Portfolio_Content"
+
+class PortfolioStatus(Enum):
+    """
+        Enum representing Status of Basket.
+    """
+    ACTIVE = "active"
+    INACTIVE = "inactive"
+    PAUSED = "paused"
+
+    ACTIVE_FOR_SUBSCRIBED_USER = (ACTIVE, PAUSED)

bw_essentials/services/portfolio_catalogue.py

@@ -48,7 +48,7 @@ Example
 import logging
 from typing import Optional, Dict, Any
 
-from bw_essentials.constants.services import Services
+from bw_essentials.constants.services import Services, PortfolioStatus
 from bw_essentials.services.api_client import ApiClient
 
 logger = logging.getLogger(__name__)
@@ -78,6 +78,7 @@ class PortfolioCatalogue(ApiClient):
         self.name = Services.PORTFOLIO_CATALOGUE.value
         self.urls = {
             "rebalance": "rebalance",
+            "get_rebalance": "rebalance"
         }
 
     def create_rebalance(self, json: Dict[str, Any]) -> Optional[Dict[str, Any]]:
@@ -102,3 +103,57 @@ class PortfolioCatalogue(ApiClient):
         )
         logger.info("%s", data)
         return data
+
+    def rebalance(self, portfolio_id, status):
+        """Fetch the latest rebalance record for a portfolio.
+
+        Parameters
+        ----------
+        portfolio_id : str
+            Portfolio identifier.
+        status : str
+            Portfolio status filter.
+
+        Returns
+        -------
+        dict
+            The first record from the service response's 'data' list.
+        """
+        logger.info("In - rebalance portfolio_id=%s status=%s", portfolio_id, status)
+        data = self._get(
+            url=self.base_url,
+            endpoint=self.urls["get_rebalance"],
+            params={'portfolioId': portfolio_id, 'status': status}
+        )
+        logger.info("rebalance response: %s", data)
+        return data.get('data')[0]
+
+    def get_rebalance_due_date(self, portfolio_id, status):
+        """Return the next rebalance date for an active portfolio, else None.
+
+        Parameters
+        ----------
+        portfolio_id : str
+            Portfolio identifier.
+        status : str
+            Portfolio status.
+
+        Returns
+        -------
+        str | None
+            The next rebalance date string if available; otherwise None.
+        """
+        logger.info(
+            "In - get_rebalance_due_date portfolio_id=%s status=%s",
+            portfolio_id,
+            status,
+        )
+        if status in PortfolioStatus.ACTIVE_FOR_SUBSCRIBED_USER.value:
+            rebalance_history = self.rebalance(portfolio_id, status)
+            next_rebalance_date = (
+                rebalance_history.get('nextRebalanceDate') if rebalance_history else None
+            )
+            logger.info("nextRebalanceDate=%s", next_rebalance_date)
+            return next_rebalance_date
+        logger.info("get_rebalance_due_date not applicable for status=%s", status)
+        return None

bw_essentials/services/user_portfolio_reporting.py

@@ -65,7 +65,8 @@ class UserReporting(ApiClient):
         self.urls = {
             "instructions": "reporting/instructions/",
             "portfolio_performance": "reporting/user/%s/userportfolio/%s/performance/",
-            "portfolio_performance_breakdown": "reporting/user/%s/userportfolio/%s/performance/breakdown/"
+            "portfolio_performance_breakdown": "reporting/user/%s/userportfolio/%s/performance/breakdown/",
+            'portfolio_stock_performance': "reporting/user/%s/userportfolio/%s/stock/performance/"
         }
 
     def _get_integer_quantity(self, qty, side):
@@ -151,3 +152,19 @@ class UserReporting(ApiClient):
                          endpoint=endpoint)
         logger.info(f"{data =}")
         return data.get("data")
+
+    def get_portfolio_stock_performance(self, user_id, user_portfolio_id):
+        """
+        Retrieves a detailed breakdown of portfolio pnl
+
+        :param user_id: ID of the user
+        :param user_portfolio_id: ID of the user portfolio
+        :return: Performance breakdown data from the response
+        """
+        logger.info(f"In get_portfolio_stock_performance {user_id=}, {user_portfolio_id=}")
+        endpoint = self.urls.get('portfolio_stock_performance') % (
+            user_id, user_portfolio_id)
+        data = self._get(url=self.base_url,
+                         endpoint=endpoint)
+        logger.info(f"{data=}")
+        return data.get("data")

{bw_essentials_core-0.1.5.dist-info → bw_essentials_core-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bw-essentials-core
-Version: 0.1.5
+Version: 0.1.7
 Summary: Reusable utilities for S3, email, Data Loch, Microsoft Teams Notifications and more.
 Author: InvestorAI
 Author-email: support+tech@investorai.in

{bw_essentials_core-0.1.5.dist-info → bw_essentials_core-0.1.7.dist-info}/RECORD

@@ -1,6 +1,6 @@
 bw_essentials/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bw_essentials/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bw_essentials/constants/services.py,sha256=IcJNd2w7bfEjvEvrkKg9te_71l4rntvU_g4gOMbmrDE,653
+bw_essentials/constants/services.py,sha256=jFP9QnzgYK-PUJT3oigpdaesYnnzLN-D5REe3Bdkwms,864
 bw_essentials/data_loch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bw_essentials/data_loch/data_loch.py,sha256=iJO0oPhZyQjoUZOaeBP1nsbuc5iR9ZNWoAXj2pGEpT8,11716
 bw_essentials/email_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -17,13 +17,13 @@ bw_essentials/services/market_pricer.py,sha256=Qc9lxzAjhefAvjyEKsBPDu60bF6_61cnS
 bw_essentials/services/master_data.py,sha256=2o_r2gdhIKBeTFgn5IIY-becgCEpYBymO4BKmixdV1g,11987
 bw_essentials/services/model_portfolio_reporting.py,sha256=iOtm4gyfU8P7C0R1gp6gUJI4ZRxJD5K2GLOalI_gToM,3249
 bw_essentials/services/notification.py,sha256=q4jIZzIWIHvn-XaIIbkEjjhkYUV8ZGQWUXaY4NNRFgI,5078
-bw_essentials/services/portfolio_catalogue.py,sha256=sLS7tYW7AkCEXYt8oWdXhmzgzvHCH9nKKFl47y8-c9c,3070
+bw_essentials/services/portfolio_catalogue.py,sha256=y1sJENAsr8sSy_oeHqdZUGWfcV19FAHICMWYpJFRA4s,4944
 bw_essentials/services/portfolio_content.py,sha256=rdFTTsUUIt-AgFfA1SAItqg2njakVOm4dUKU0FHZQ2E,2148
 bw_essentials/services/trade_placement.py,sha256=PrzeU2XXC9HF1IQ1dMDM_ZHxmC491sOl-JbA6GWPwII,20772
 bw_essentials/services/user_app.py,sha256=_Y2NgDq6kEoco7ZRJ3-MMYJ-K2HGStJoFoeGOH_HotQ,11107
 bw_essentials/services/user_portfolio.py,sha256=_5M6yPfQt4MXedINBawEoPkb_o7IGzxPeHkvZkoQm8k,16191
-bw_essentials/services/user_portfolio_reporting.py,sha256=QaZzcw912Uos5ZafEaxLXDmCyetTFiVX3at3cTAv6MA,6003
-bw_essentials_core-0.1.5.dist-info/METADATA,sha256=O4CWWEYI-OtPf6TaJdXXZXCU3suvgqBA0jjKLbLcNBI,7501
-bw_essentials_core-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bw_essentials_core-0.1.5.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
-bw_essentials_core-0.1.5.dist-info/RECORD,,
+bw_essentials/services/user_portfolio_reporting.py,sha256=pwqfW95LTiGOGufcTphljFxPlOd-G4Q263UtoQURPxM,6772
+bw_essentials_core-0.1.7.dist-info/METADATA,sha256=QKkFUyEBpJkb6rbJ8jPV6iA1v6R0MORD9830OPz6k9c,7501
+bw_essentials_core-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bw_essentials_core-0.1.7.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
+bw_essentials_core-0.1.7.dist-info/RECORD,,