brynq-sdk-brynq 2.0.4__tar.gz → 2.1.0__tar.gz

{brynq_sdk_brynq-2.0.4 → brynq_sdk_brynq-2.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq_sdk_brynq
-Version: 2.0.4
+Version: 2.1.0
 Summary: BrynQ SDK for the BrynQ.com platform
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_brynq-2.0.4 → brynq_sdk_brynq-2.1.0}/brynq_sdk_brynq/brynq.py

@@ -2,7 +2,7 @@ import os
 import requests
 import json
 import pandas as pd
-from typing import Union, Literal
+from typing import Union, Literal, Optional, List
 
 
 class BrynQ:
@@ -146,6 +146,83 @@ class BrynQ:
 
         return credentials[0]
 
+    def get_interface_credential(self, interface_id: str, system: str, system_type: Optional[str] = None,
+                                 test_environment: bool = False) -> Union[dict, List[dict]]:
+        """
+        This method retrieves authentication credentials from BrynQ for a specific interface and system.
+        :param interface_id: ID of the interface to get credentials for
+        :param system: The app name to search for in credentials (e.g., 'bob', 'profit')
+        :param system_type: Optional parameter to specify 'source' or 'target'. If not provided,
+                           searches in both lists
+        :param test_environment: boolean if the test environment is used (only for profit system)
+        :return: Credential dictionary or list of credential dictionaries for the specified system
+        """
+        # Get credentials from interface configuration
+        response = requests.get(
+            url=f'{self.url}interfaces/{interface_id}/config/auth',
+            headers=self._get_headers()
+        )
+        response.raise_for_status()
+        config = response.json()
+        matching_credentials = []
+        # If system_type is provided, only search in that list
+        if system_type:
+            if system_type not in ['source', 'target']:
+                raise ValueError("system_type must be either 'source' or 'target'")
+            credentials_list = config.get(f'{system_type}s', [])
+            for cred in credentials_list:
+                if cred.get('app') == system:
+                    # Check test environment for profit system
+                    if system == 'profit':
+                        is_test = cred.get('data', {}).get('isTestEnvironment', False)
+                        if is_test == test_environment:
+                            matching_credentials.append({'credential': cred, 'type': system_type})
+                    else:
+                        matching_credentials.append({'credential': cred, 'type': system_type})
+        # If no system_type provided, search in both lists
+        else:
+            source_credentials = []
+            target_credentials = []
+            # Check sources
+            for source in config.get('sources', []):
+                if source.get('app') == system:
+                    if system == 'profit':
+                        is_test = source.get('data', {}).get('isTestEnvironment', False)
+                        if is_test == test_environment:
+                            source_credentials.append({'credential': source, 'type': 'source'})
+                    else:
+                        source_credentials.append({'credential': source, 'type': 'source'})
+            # Check targets
+            for target in config.get('targets', []):
+                if target.get('app') == system:
+                    if system == 'profit':
+                        is_test = target.get('data', {}).get('isTestEnvironment', False)
+                        if is_test == test_environment:
+                            target_credentials.append({'credential': target, 'type': 'target'})
+                    else:
+                        target_credentials.append({'credential': target, 'type': 'target'})
+            # Combine matching credentials based on type
+            if source_credentials and target_credentials:
+                raise ValueError(
+                    f'Multiple credentials found for system {system} in both source and target. '
+                    f'Please specify system_type as "source" or "target"'
+                )
+            matching_credentials = source_credentials or target_credentials
+        # Handle results
+        if len(matching_credentials) == 0:
+            if system == 'profit':
+                raise ValueError(f'No credentials found for system {system} with test_environment={test_environment}')
+            else:
+                raise ValueError(f'No credentials found for system {system}')
+        if len(matching_credentials) == 1:
+            return matching_credentials[0]['credential']
+        if len(matching_credentials) > 1:
+            warning_msg = f'Multiple credentials found for system {system}'
+            if system_type:
+                warning_msg += f' in {system_type}'
+            warnings.warn(warning_msg)
+            return [cred['credential'] for cred in matching_credentials]
+
     def refresh_system_credential(self, system: str, system_id: int) -> json:
         """
         This method refreshes Oauth authentication credentials in BrynQ.

{brynq_sdk_brynq-2.0.4 → brynq_sdk_brynq-2.1.0}/brynq_sdk_brynq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq-sdk-brynq
-Version: 2.0.4
+Version: 2.1.0
 Summary: BrynQ SDK for the BrynQ.com platform
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_brynq-2.0.4 → brynq_sdk_brynq-2.1.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 
 setup(
     name='brynq_sdk_brynq',
-    version='2.0.4',
+    version='2.1.0',
     description='BrynQ SDK for the BrynQ.com platform',
     long_description='BrynQ SDK for the BrynQ.com platform',
     author='BrynQ',