bw-essentials-core 0.0.5__tar.gz → 0.0.7__tar.gz

{bw_essentials_core-0.0.5 → bw_essentials_core-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bw-essentials-core
-Version: 0.0.5
+Version: 0.0.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.0.5 → bw_essentials_core-0.0.7}/bw_essentials/data_loch/data_loch.py

@@ -132,15 +132,17 @@ class DataLoch:
             file_name (str): File name to use in LakeFS.
             commit (bool): Whether to commit the change after upload. Defaults to True.
         """
-        logger.info(f"Uploading file {file_name} from {local_file_path}")
-        branch_obj = self._get_branch(repository, branch)
-        self.s3.upload_file(repository, local_file_path, f"{branch}/{file_name}", "application/octet-stream")
-        if commit:
-            commit_response = branch_obj.commit(
-                message=f"{file_name=} uploaded to data lake.",
-                metadata=self._get_meta_data(repository, branch)
-            )
-            logger.info(f"Changes committed: {commit_response=}")
+        logger.info(f"Uploading {file_name=} from {local_file_path=}")
+        logger.info(f"{file_name=} uploading to LakeFS")
+        branch_obj = self._get_branch(branch=branch, repository=repository)
+        obj = branch_obj.object(path=f"{file_name}")
+        with open(local_file_path, mode='rb') as reader, obj.writer("wb") as writer:
+            writer.write(reader.read())
+        commit_response = branch_obj.commit(
+            message=f"{file_name=} uploaded to data lake.",
+            metadata=self._get_meta_data(branch=branch,repository=repository)
+        )
+        logger.info(f"Changes committed: {commit_response=}")
 
     def download_data(self, repository, branch, local_file_path, file_name):
         """

{bw_essentials_core-0.0.5 → bw_essentials_core-0.0.7}/bw_essentials/services/master_data.py

@@ -44,7 +44,8 @@ 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"
         }
 
     def get_security_details(self, securities: List[str]) -> Optional[Dict[str, Any]]:
@@ -107,7 +108,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 +257,23 @@ 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", {})

{bw_essentials_core-0.0.5 → bw_essentials_core-0.0.7}/bw_essentials_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bw-essentials-core
-Version: 0.0.5
+Version: 0.0.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.0.5 → bw_essentials_core-0.0.7}/setup.py

@@ -10,7 +10,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="bw-essentials-core",
-    version="0.0.5",
+    version="0.0.7",
     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.5 → bw_essentials_core-0.0.7}/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