bw-essentials-core 0.1.10__tar.gz → 0.1.12__tar.gz

{bw_essentials_core-0.1.10 → bw_essentials_core-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bw-essentials-core
-Version: 0.1.10
+Version: 0.1.12
 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.10 → bw_essentials_core-0.1.12}/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_core-0.1.10 → bw_essentials_core-0.1.12}/bw_essentials/services/api_client.py

@@ -80,6 +80,19 @@ class ApiClient:
         env_key = f"{service_name.upper()}_BASE_URL"
         return self._get_env_var(env_key)
 
+    def get_api_key(self, service_name: str) -> str:
+        """
+        Resolve the service API Key for a given service name using environment variables.
+
+        Args:
+            service_name (str): The logical name of the service.
+
+        Returns:
+            str: The resolved api key from the environment.
+        """
+        env_key = f"{service_name.upper()}_API_KEY"
+        return self._get_env_var(env_key)
+
     def set_tenant_id(self, tenant_id: str) -> None:
         """
         Set the tenant ID in the request headers.

{bw_essentials_core-0.1.10 → bw_essentials_core-0.1.12}/bw_essentials/services/broker.py

@@ -62,7 +62,8 @@ class Broker(ApiClient):
             "authorised_holdings": "brokers/{}/holdings/authorise",
             "user_instructions": "brokers/{}/user/instructions",
             "details": "brokers/{}/trade/details",
-            "surveillance_orders": "brokers/{}/surveillance/orders"
+            "surveillance_orders": "brokers/{}/surveillance/orders",
+            "profile": "brokers/{}/profile"
         }
 
     def authenticate(self, broker_name: str, user_id: str, entity_id: str,
@@ -118,6 +119,26 @@ class Broker(ApiClient):
         logger.info(f"{response =}")
         return response["data"]["balance"]
 
+    def get_profile(self, broker_name: str, user_id: str) -> dict:
+        """
+        Fetch user profile from the broker service.
+
+        Args:
+            broker_name (str): Broker name.
+            user_id (str): User ID.
+
+        Returns:
+            Dict: User Profile.
+        """
+        logger.info(f"In - get_profile {user_id =}")
+        response = self._get(
+            url=self.base_url,
+            endpoint=self.urls["profile"].format(broker_name),
+            params={"user_id": user_id}
+        )
+        logger.info(f"{response =}")
+        return response["data"]
+
     def get_holdings(self, broker_name: str, user_id: str) -> list:
         """
         Fetch user holdings from the broker service.

bw_essentials_core-0.1.12/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_core-0.1.10 → bw_essentials_core-0.1.12}/bw_essentials/services/user_app.py

@@ -34,16 +34,20 @@ class PrometheusUserApp(ApiClient):
         urls (Dict[str, str]): A dictionary of endpoint path templates
     """
 
-    def __init__(self, service_user: str):
+    def __init__(self, service_user: str, tenant_id: str = None):
         """
-        Initialize the PrometheusUserApp client with the given service user.
+        Initialize the PrometheusUserApp client with the given service user and tenant id.
 
         Args:
             service_user (str): Username or service identifier for authentication.
+            tenant_id (str, optional): Tenant identifier for multi-tenant usage.
+                                       Defaults to None if not applicable.
         """
         super().__init__(user=service_user)
         self.base_url = self.get_base_url(Services.PROMETHEUS_USER_APP.value)
+        self.api_key = self.get_api_key(Services.PROMETHEUS_USER_APP.value)
         self.name = Services.PROMETHEUS_USER_APP.value
+        self.tenant_id = tenant_id
         self.urls = {
             "user_details": "user/details",
             "profile": "user/profile",
@@ -60,6 +64,20 @@ class PrometheusUserApp(ApiClient):
             "sso": "user/{}/sso"
         }
 
+
+    def _headers(self):
+        """
+        Prepares headers for API calls.
+
+        Returns:
+            dict: Headers
+        """
+        return {
+            'Content-Type': 'application/json',
+            **({'X-Tenant-ID': self.tenant_id} if self.tenant_id else {}),
+            'x-api-key': self.api_key
+        }
+
     def get_user_details(self, user_id: str) -> Optional[Dict[str, Any]]:
         """
         Fetch detailed user information by user ID.
@@ -71,10 +89,11 @@ class PrometheusUserApp(ApiClient):
             Optional[Dict[str, Any]]: Parsed user data if found, else None.
         """
         logger.info("Fetching user details for user_id=%s", user_id)
+        self.headers = self._headers()
         response = self._get(
             endpoint=self.urls["user_details"],
             url=self.base_url,
-            params={'user_id': user_id}
+            params={'user_id': user_id},
         )
         return response.get('data')
 

{bw_essentials_core-0.1.10 → bw_essentials_core-0.1.12}/bw_essentials_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bw-essentials-core
-Version: 0.1.10
+Version: 0.1.12
 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.10 → bw_essentials_core-0.1.12}/bw_essentials_core.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ bw_essentials/s3_utils/s3_utils.py
 bw_essentials/services/__init__.py
 bw_essentials/services/api_client.py
 bw_essentials/services/broker.py
+bw_essentials/services/job_scheduler.py
 bw_essentials/services/market_pricer.py
 bw_essentials/services/master_data.py
 bw_essentials/services/model_portfolio_reporting.py

{bw_essentials_core-0.1.10 → bw_essentials_core-0.1.12}/setup.py

@@ -10,7 +10,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="bw-essentials-core",
-    version="0.1.10",
+    version="0.1.12",
     author="InvestorAI",
     author_email="support+tech@investorai.in",
     description="Reusable utilities for S3, email, Data Loch, Microsoft Teams Notifications and more.",