bw-essentials-core 0.1.11__py3-none-any.whl → 0.1.13__py3-none-any.whl

bw_essentials/constants/services.py

@@ -20,6 +20,7 @@ class Services(Enum):
     PROMETHEUS_USER_APP = "Prometheus_User_App"
     PORTFOLIO_CATALOGUE = "Portfolio_Catalogue"
     PORTFOLIO_CONTENT = "Portfolio_Content"
+    JOB_SCHEDULER = "Job_Scheduler"
 
 class PortfolioStatus(Enum):
     """

bw_essentials/services/job_scheduler.py

@@ -0,0 +1,60 @@
+import logging
+from bw_essentials.constants.services import Services
+from bw_essentials.services.api_client import ApiClient
+
+logger = logging.getLogger(__name__)
+
+
+class JobScheduler(ApiClient):
+    """
+    Class for making API calls to the Job Scheduler Service.
+
+    Args:
+    user (str): The user for whom the API calls are being made.
+    """
+
+    def __init__(self, user):
+        """
+        Initialize the Job Scheduler object.
+
+        Args:
+        user (str): The user for whom the API calls are being made.
+        """
+        super().__init__(user)
+        self.base_url = self.get_base_url(Services.JOB_SCHEDULER.value)
+        self.api_key = self.get_api_key(Services.JOB_SCHEDULER.value)
+        self.name = Services.JOB_SCHEDULER.value
+        self.urls = {
+            "process_user_profile": "taskmanager/process/user-portfolios/"
+        }
+
+    def _headers(self):
+        """
+        Prepares headers for API calls.
+
+        Returns:
+            dict: Headers
+        """
+        return {
+            'Content-Type': 'application/json',
+            'x-api-key': self.api_key
+        }
+
+    def process_user_profile(self, data):
+        """
+        Send a request to process a user profile.
+
+        This method prepares the required headers and sends a POST request
+        to the `process_user_profile` endpoint with the provided user data.
+
+        Args:
+            data (dict): A dictionary containing user profile information
+                to be sent in the request body.
+
+        Returns:
+            None: The method does not return a value.
+        """
+        logger.info(f"In - process_user_profile {self.user =}")
+        self.headers = self._headers()
+        self._post(url=self.base_url, endpoint=self.urls.get('process_user_profile'), json=data)
+        logger.info(f"Out - process_user_profile")

bw_essentials/services/user_portfolio.py

@@ -440,7 +440,7 @@ class UserPortfolio(ApiClient):
         logger.info(f"In create_user_portfolio_threshold {payload =}")
         data = self._post(url=self.base_url,
                           endpoint=self.urls.get("user_portfolio_thresholds"),
-                          data=payload)
+                          json=payload)
         logger.info(f"{data =}")
         return data.get("data")
 
@@ -522,7 +522,7 @@ class UserPortfolio(ApiClient):
         logger.info(f"In create_holding_threshold {payload =}")
         data = self._post(url=self.base_url,
                           endpoint=self.urls.get("holding_thresholds"),
-                          data=payload)
+                          json=payload)
         logger.info(f"{data =}")
         return data.get("data")
 

{bw_essentials_core-0.1.11.dist-info → bw_essentials_core-0.1.13.dist-info}/METADATA

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

@@ -1,6 +1,6 @@
 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=jFP9QnzgYK-PUJT3oigpdaesYnnzLN-D5REe3Bdkwms,864
+bw_essentials/constants/services.py,sha256=HxeM05zJP1oCqwLRLb9wLxAOCmbdu-jWgtlmnPE4KFU,900
 bw_essentials/data_loch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bw_essentials/data_loch/data_loch.py,sha256=iJO0oPhZyQjoUZOaeBP1nsbuc5iR9ZNWoAXj2pGEpT8,11716
 bw_essentials/email_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -13,6 +13,7 @@ bw_essentials/s3_utils/s3_utils.py,sha256=wzjVrTX22_8PMX86svKFYGMZwgjBHbOaeEsxO-
 bw_essentials/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bw_essentials/services/api_client.py,sha256=5XFbdw5gfxXz7mQlLWNC8efDrvZgAztM-Oo0Dp1jDF8,8215
 bw_essentials/services/broker.py,sha256=W4y31K7dF4LHfQSvnAwVWfGoYKkdz-W2TXqTLhkUxn8,9348
+bw_essentials/services/job_scheduler.py,sha256=AmlaYJsCCjHxTg9MQ58dMuiZOUDfxmbd43jzN4L-lNU,1834
 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
@@ -22,9 +23,9 @@ bw_essentials/services/portfolio_catalogue.py,sha256=F4ne1eF6Koj0CCV1j07ebIgIlRC
 bw_essentials/services/portfolio_content.py,sha256=rdFTTsUUIt-AgFfA1SAItqg2njakVOm4dUKU0FHZQ2E,2148
 bw_essentials/services/trade_placement.py,sha256=PrzeU2XXC9HF1IQ1dMDM_ZHxmC491sOl-JbA6GWPwII,20772
 bw_essentials/services/user_app.py,sha256=jEpdT2W0Bj65xwQG2X1wnsQQEvjMHMbtjuvuYlzhJ6Q,11773
-bw_essentials/services/user_portfolio.py,sha256=G3PrMOryqEXSERodP4hwbrxV2JHiXUVwxdFri_E1v3k,22179
+bw_essentials/services/user_portfolio.py,sha256=uUQUCG8Av8eKBIw-TlNtOPpQluO9HjAkxMzzO5jfzbo,22179
 bw_essentials/services/user_portfolio_reporting.py,sha256=pwqfW95LTiGOGufcTphljFxPlOd-G4Q263UtoQURPxM,6772
-bw_essentials_core-0.1.11.dist-info/METADATA,sha256=ZQXEqpqVlQmlTiFB7kl2v1p8rif8JBTWGH3ba0njS6Y,7502
-bw_essentials_core-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bw_essentials_core-0.1.11.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
-bw_essentials_core-0.1.11.dist-info/RECORD,,
+bw_essentials_core-0.1.13.dist-info/METADATA,sha256=DTCc-XwQdanVEFfUW9HP4I30NGEEaFvlifdkmDudjDg,7502
+bw_essentials_core-0.1.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bw_essentials_core-0.1.13.dist-info/top_level.txt,sha256=gDc5T_y5snwKGXDQUusEus-FEt0RFwG644Yn_58wQOQ,14
+bw_essentials_core-0.1.13.dist-info/RECORD,,