bw-essentials-core 0.0.6__py3-none-any.whl → 0.0.8__py3-none-any.whl

bw_essentials/services/api_client.py

@@ -158,9 +158,9 @@ class ApiClient:
         try:
             response = requests.request(method, formatted_url, headers=self.headers, **kwargs)
             elapsed_time_ms = (time.time() - start) * 1000
-            response.raise_for_status()
             json_data = response.json()
             self._log_response(method, formatted_url, response.status_code, elapsed_time_ms, json_data)
+            response.raise_for_status()
             return json_data
         except Exception as exc:
             elapsed_time_ms = (time.time() - start) * 1000

bw_essentials/services/master_data.py

@@ -44,7 +44,9 @@ class MasterData(ApiClient):
             "constant": "constants",
             "broker_config": "broker/config/keys",
             "broker_details": "securities/{}/details",
-            "isin_details": "company/details/isin"
+            "isin_details": "company/details/isin",
+            "shorten_url": "shorten-url",
+            "broker_partner_mapping_details": "partner/{}/broker/{}/"
         }
 
     def get_security_details(self, securities: List[str]) -> Optional[Dict[str, Any]]:
@@ -107,7 +109,8 @@ class MasterData(ApiClient):
         Returns:
             Optional[Dict[str, Any]]: Configuration key data or None.
         """
-        logger.info(f"Fetching broker config keys for broker={broker}, product_type={product_type}, category={category}")
+        logger.info(
+            f"Fetching broker config keys for broker={broker}, product_type={product_type}, category={category}")
         params = {k: v for k, v in {'broker': broker, 'product': product_type, 'category': category}.items() if v}
         response = self._get(url=self.base_url,
                              endpoint=self.urls["broker_config"],
@@ -255,3 +258,44 @@ class MasterData(ApiClient):
 
         logger.info("Market is closed based on hours and/or holiday check.")
         return False
+
+    def shorten_url(self, long_url: str) -> Dict[str, Any]:
+        """Shorten a long URL using the configured URL-shortening service.
+
+        Args:
+            long_url (str): The long URL to shorten.
+
+        Returns:
+            dict: A dictionary containing the shortened URL or any error data.
+
+        Logs:
+            Information about the long_url being shortened.
+
+        """
+        logger.info(f"Shortening URL: {long_url}")
+        response = self._post(url=self.base_url,
+                              endpoint=self.urls["shorten_url"],
+                              data={"long_url": long_url})
+        logger.info(f"Shortening URL {response =}")
+        return response.get("data", {})
+
+    def get_or_create_broker_partner_mapping(self, broker_partner: str, broker: str) -> Dict[str, Any]:
+        """
+        Fetches or creates a broker-partner mapping from master data constants table.
+
+        This method sends a GET request to retrieve the mapping details for a given
+        broker and broker partner. If the mapping does not exist, the API
+        is expected to create it automatically (based on implementation).
+
+        Parameters:
+        - broker_partner (str): The identifier of the broker partner.
+        - broker (str): The identifier of the broker.
+
+        Returns:
+        - Dict[str, Any]: The mapping data returned from the service, or an empty
+          dictionary if no data is available.
+        """
+        logger.info(f"In - get_or_create_broker_partner_mapping | broker_partner={broker_partner} broker={broker}")
+        response = self._get(url=self.base_url, endpoint=self.urls["broker_partner_mapping_details"].format(broker_partner, broker))
+        logger.info(f"Broker partner mapping {response =}")
+        return response.get('data', {})

{bw_essentials_core-0.0.6.dist-info → bw_essentials_core-0.0.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bw-essentials-core
-Version: 0.0.6
+Version: 0.0.8
 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.0.6.dist-info → bw_essentials_core-0.0.8.dist-info}/RECORD

@@ -11,15 +11,15 @@ bw_essentials/notifications/teams_notifications.py,sha256=1tkS3tTJQZ5ByMwK2WXpFg
 bw_essentials/s3_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bw_essentials/s3_utils/s3_utils.py,sha256=wzjVrTX22_8PMX86svKFYGMZwgjBHbOaeEsxO-lR7BY,12922
 bw_essentials/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bw_essentials/services/api_client.py,sha256=2aTysB4AfHqCILz2o4qOb5Z-GK5GtzhmFe3NFUfAsto,7793
+bw_essentials/services/api_client.py,sha256=74tKxhY_HiVCN3kgtk27a4GNznMaRphAspt9CWecuiY,7793
 bw_essentials/services/broker.py,sha256=As-VT1MxkCD9p-13dQJDwf0_aEJVA6H0wNqL-EYHO4g,8718
 bw_essentials/services/market_pricer.py,sha256=WAFcG5JfnqWxjzXRPrMdovPuKqWJIwa3ENvGAyEyjeM,8712
-bw_essentials/services/master_data.py,sha256=tJ-OyG9j94Piw2-rByhvkz6aVds4t6b5aprkT1TUPJQ,10068
+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/trade_placement.py,sha256=PrzeU2XXC9HF1IQ1dMDM_ZHxmC491sOl-JbA6GWPwII,20772
 bw_essentials/services/user_portfolio.py,sha256=_5M6yPfQt4MXedINBawEoPkb_o7IGzxPeHkvZkoQm8k,16191
 bw_essentials/services/user_portfolio_reporting.py,sha256=QaZzcw912Uos5ZafEaxLXDmCyetTFiVX3at3cTAv6MA,6003
-bw_essentials_core-0.0.6.dist-info/METADATA,sha256=5uvp58qqjWxL1k4YkYVp3awSDACqHf0b5a3dK-ZTeck,7501
-bw_essentials_core-0.0.6.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-bw_essentials_core-0.0.6.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
-bw_essentials_core-0.0.6.dist-info/RECORD,,
+bw_essentials_core-0.0.8.dist-info/METADATA,sha256=GHHKudJ8VIoZ0nOUAeAnpQy0FYXXuPMSTQYA_aa3Wt0,7501
+bw_essentials_core-0.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bw_essentials_core-0.0.8.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
+bw_essentials_core-0.0.8.dist-info/RECORD,,

{bw_essentials_core-0.0.6.dist-info → bw_essentials_core-0.0.8.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any