bw-essentials-core 0.0.6__tar.gz → 0.0.8__tar.gz

{bw_essentials_core-0.0.6 → bw_essentials_core-0.0.8}/PKG-INFO

@@ -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 → bw_essentials_core-0.0.8}/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 → bw_essentials_core-0.0.8}/bw_essentials_core.egg-info/PKG-INFO

@@ -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 → bw_essentials_core-0.0.8}/setup.py

@@ -10,7 +10,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="bw-essentials-core",
-    version="0.0.6",
+    version="0.0.8",
     author="InvestorAI",
     author_email="support+tech@investorai.in",
     description="Reusable utilities for S3, email, Data Loch, Microsoft Teams Notifications and more.",

{bw_essentials_core-0.0.6 → bw_essentials_core-0.0.8}/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