bw-essentials-core 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

bw_essentials/data_loch/data_loch.py

@@ -279,3 +279,19 @@ class DataLoch:
         prefix = f"{branch}/{server_path}"
         self.s3.delete_files_by_prefix(repository, prefix)
         logger.info(f"Deleted all files under prefix {prefix}")
+
+    def delete_single_file(self, repository: str, branch: str, server_path: str) -> None:
+        """
+        Delete a single file at the specified path in a LakeFS branch.
+
+        Args:
+            repository (str): The name of the LakeFS repository.
+            branch (str): The name of the branch within the repository.
+            server_path (str): The file path (relative to the branch) to delete.
+        """
+        full_path = f"{branch}/{server_path}"
+        logger.info(f"Attempting to delete file at path: {full_path} in repository: {repository}")
+
+        self.s3.delete_file(repository, full_path)
+
+        logger.info(f"Successfully deleted file: {full_path} from repository: {repository}")

bw_essentials/services/notification.py

@@ -28,10 +28,11 @@ class NotificationService(ApiClient):
         self.name = Services.NOTIFICATION.value
         self.base_url = self.get_base_url(self.name)
         self.urls = {
-            "whatsapp": "whatsapp"
+            "whatsapp": "whatsapp",
+            "email": "email"
         }
 
-    def _whatsapp(self, title, template, platform, params, to, user_id):
+    def _whatsapp(self, title, template, platform, params, to, user_id) -> None:
         """
         Sends a WhatsApp notification.
 
@@ -66,12 +67,77 @@ class NotificationService(ApiClient):
         resp_data = self._post(url=self.base_url, endpoint=self.urls.get('whatsapp'), data=payload)
         logger.info(f"Whatsapp response {resp_data =}")
 
-    def send_whatsapp(self, template, title, params, to, user_id):
-        logger.info(f"In - send_whatsapp_notification {user_id =} {title = } {params = } {to = }")
+    def send_whatsapp(self, template, title, params, to, user_id) -> None:
+        """
+
+        Args:
+            template (str): Template for the WhatsApp message.
+            title (str): Title of the notification.
+            params: Parameters for the notification message.
+            to (str): Recipient's whatsapp number.
+            user_id (str): The ID of the user receiving the notification.
 
+        Returns:
+            None
+        """
+        logger.info(f"In - send_whatsapp_notification {user_id =} {title = } {params = } {to = }")
         self._whatsapp(title=title,
                        template=template,
                        platform=self._get_env_var(NotificationService.PLATFORM),
                        params=params,
                        to=to,
                        user_id=user_id)
+
+    def _email(self, title: str, content: str, platform: str, to: str, user_id: str) -> None:
+        """
+        Sends an email notification using the internal notification service.
+
+        Args:
+            title (str): The subject or title of the email.
+            content (str): The HTML or plain text body of the email.
+            platform (str): The platform identifier from which the email is sent (e.g., 'prometheus').
+            to (str): Recipient's email address.
+            user_id (str): The ID of the user for tracking or logging purposes.
+
+        Returns:
+            None
+
+        """
+        logger.info(f"In - email {user_id =}, {to =}, {title =}, {platform =}, {content =}")
+        payload = {
+            "to": to,
+            "userId": user_id,
+            "platform": platform,
+            "title": title,
+            "content": content
+        }
+        logger.info(f"email {payload =}")
+        headers = {
+            'api-key': self._get_env_var(NotificationService.NOTIFICATION_API_KEY),
+            'Content-Type': 'application/json'
+        }
+        self.set_headers(headers)
+        resp_data = self._post(url=self.base_url, endpoint=self.urls.get('email'), json=payload)
+        logger.info(f"Email response {resp_data =}")
+
+    def send_email(self, title: str, content: str, to: str, user_id: str) -> None:
+        """
+        Sends an email notification to the specified recipient.
+
+        Args:
+            title (str): The subject or title of the email.
+            content (str): The HTML or plain text content of the email.
+            to (str): The recipient's email address.
+            user_id (str): The ID of the user associated with the notification.
+
+        Returns:
+            None
+        """
+        logger.info(f"In - send_email {user_id =}, {title =}, {to =}")
+        self._email(
+            title=title,
+            content=content,
+            platform=self._get_env_var(NotificationService.PLATFORM),
+            to=to,
+            user_id=user_id
+        )

{bw_essentials_core-0.1.1.dist-info → bw_essentials_core-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bw-essentials-core
-Version: 0.1.1
+Version: 0.1.3
 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.1.dist-info → bw_essentials_core-0.1.3.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=ymvYcXjJpWQoGnnCnqURhQao61QJAMM-g9sJCx_Y8s4,561
 bw_essentials/data_loch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bw_essentials/data_loch/data_loch.py,sha256=Ae3WIywFawThKcpsnf8nC8ldI9ARrSNqz8e5b6sINwY,11006
+bw_essentials/data_loch/data_loch.py,sha256=iJO0oPhZyQjoUZOaeBP1nsbuc5iR9ZNWoAXj2pGEpT8,11716
 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
@@ -16,12 +16,12 @@ bw_essentials/services/broker.py,sha256=As-VT1MxkCD9p-13dQJDwf0_aEJVA6H0wNqL-EYH
 bw_essentials/services/market_pricer.py,sha256=Qc9lxzAjhefAvjyEKsBPDu60bF6_61cnSpZNfjGMyDg,11755
 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=OQD-rB7FaqcoyjRgFyl2pgeXWIgtYcUhRs_4uT1Em9A,2634
+bw_essentials/services/notification.py,sha256=q4jIZzIWIHvn-XaIIbkEjjhkYUV8ZGQWUXaY4NNRFgI,5078
 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.1.dist-info/METADATA,sha256=c5-YYlMERt3ZzOLmu4hp78zV1t5bNCawWyWh6BIGY7o,7501
-bw_essentials_core-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bw_essentials_core-0.1.1.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
-bw_essentials_core-0.1.1.dist-info/RECORD,,
+bw_essentials_core-0.1.3.dist-info/METADATA,sha256=r5Fy1wGySkKAZ54fmubQEK1EKc50kjoKw9MvnZIGPbw,7501
+bw_essentials_core-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bw_essentials_core-0.1.3.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
+bw_essentials_core-0.1.3.dist-info/RECORD,,