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

bw_essentials/services/notification.py

@@ -0,0 +1,77 @@
+"""
+Module to make API calls to Notification Service
+"""
+import logging
+
+from bw_essentials.constants.services import Services
+from bw_essentials.services.api_client import ApiClient
+
+logger = logging.getLogger(__name__)
+
+
+class NotificationService(ApiClient):
+    """
+    Handles notifications through various channels.
+    """
+    PLATFORM = "PLATFORM"
+    NOTIFICATION_API_KEY = 'NOTIFICATION_API_KEY'
+    NOTIFICATION_SENDER_NUMBER = "NOTIFICATION_SENDER_NUMBER"
+
+    def __init__(self):
+        """
+        Initializes the Notification object.
+
+        Args:
+        - user: The user associated with the notification.
+        """
+        super().__init__("notification")
+        self.name = Services.NOTIFICATION.value
+        self.base_url = self.get_base_url(self.name)
+        self.urls = {
+            "whatsapp": "whatsapp"
+        }
+
+    def _whatsapp(self, title, template, platform, params, to, user_id):
+        """
+        Sends a WhatsApp notification.
+
+        Args:
+        - user_id (str): The ID of the user receiving the notification.
+        - number (str): The phone number for the recipient.
+        - title (str): Title of the notification.
+        - template (str): Template for the WhatsApp message.
+        - platform (str): The platform used for sending
+            (WhatsApp, in this case).
+        - params: Parameters for the notification message.
+
+        Returns:
+        None
+        """
+        logger.info(f"In - whatsapp {user_id =}, {to =}, "
+                    f"{title =}, {template =}, {platform =}, {params =}")
+        payload = {
+            "from": self._get_env_var(NotificationService.NOTIFICATION_SENDER_NUMBER),
+            "to": to,
+            "userId": user_id,
+            "platform": platform,
+            "title": title,
+            "template": template,
+            "params": params
+        }
+        logger.info(f"whatsapp {payload =}")
+        headers = {
+            'api-key': self._get_env_var(NotificationService.NOTIFICATION_API_KEY)
+        }
+        self.set_headers(headers)
+        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 = }")
+
+        self._whatsapp(title=title,
+                       template=template,
+                       platform=self._get_env_var(NotificationService.PLATFORM),
+                       params=params,
+                       to=to,
+                       user_id=user_id)

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

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

@@ -16,11 +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/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.0.dist-info/METADATA,sha256=jara8u2pGS_rtKqecDPjm1TkRMhy8GdZwFlmC7YYz5s,7501
-bw_essentials_core-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bw_essentials_core-0.1.0.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
-bw_essentials_core-0.1.0.dist-info/RECORD,,
+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,,