bw-essentials-core 0.0.5__py3-none-any.whl → 0.0.7__py3-none-any.whl

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/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,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.dist-info → bw_essentials_core-0.0.7.dist-info}/METADATA

@@ -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.dist-info → bw_essentials_core-0.0.7.dist-info}/RECORD

@@ -2,7 +2,7 @@ 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=ZRz-cfumqafmAsetW48S3KMTjTaNAdoV1zu3bRuYfNc,513
 bw_essentials/data_loch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bw_essentials/data_loch/data_loch.py,sha256=MHKkBIqEqBuh1cS3lbX0m9hSeAz5h_FgQ_GWWC1F540,10888
+bw_essentials/data_loch/data_loch.py,sha256=Ae3WIywFawThKcpsnf8nC8ldI9ARrSNqz8e5b6sINwY,11006
 bw_essentials/email_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bw_essentials/email_client/email_client.py,sha256=_LzDWUJUkNTSn3C5v0-jjW-Ls537xt493lkcD4noQXY,9116
 bw_essentials/notifications/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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=JpI7HfyN2hgkr9Mdcm4DzddSwmAv6AvsUmAuyrW0LYo,10828
 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.5.dist-info/METADATA,sha256=WtgpSGCxi7tU-t0XHN_eoexCEWYwSvWTNZ5cyhuev6A,7501
-bw_essentials_core-0.0.5.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-bw_essentials_core-0.0.5.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
-bw_essentials_core-0.0.5.dist-info/RECORD,,
+bw_essentials_core-0.0.7.dist-info/METADATA,sha256=6nghqDj90cybbLd3LiSk_Ws3vVYydKFocJ0fgeZriGI,7501
+bw_essentials_core-0.0.7.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+bw_essentials_core-0.0.7.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
+bw_essentials_core-0.0.7.dist-info/RECORD,,

{bw_essentials_core-0.0.5.dist-info → bw_essentials_core-0.0.7.dist-info}/WHEEL

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