237 lines
9.0 KiB
Python
237 lines
9.0 KiB
Python
# shared/clients/__init__.py
|
|
"""
|
|
Service Client Factory and Convenient Imports
|
|
Provides easy access to all service clients
|
|
"""
|
|
|
|
from .base_service_client import BaseServiceClient, ServiceAuthenticator
|
|
from .auth_client import AuthServiceClient
|
|
from .training_client import TrainingServiceClient
|
|
from .sales_client import SalesServiceClient
|
|
from .external_client import ExternalServiceClient
|
|
from .forecast_client import ForecastServiceClient
|
|
from .inventory_client import InventoryServiceClient
|
|
from .orders_client import OrdersServiceClient
|
|
from .production_client import ProductionServiceClient
|
|
from .recipes_client import RecipesServiceClient
|
|
from .suppliers_client import SuppliersServiceClient
|
|
from .tenant_client import TenantServiceClient
|
|
|
|
# Import config
|
|
from shared.config.base import BaseServiceSettings
|
|
|
|
# Cache clients to avoid recreating them
|
|
_client_cache = {}
|
|
|
|
def get_training_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> TrainingServiceClient:
|
|
"""Get or create a training service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"training_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = TrainingServiceClient(config, service_name)
|
|
return _client_cache[cache_key]
|
|
|
|
def get_sales_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> SalesServiceClient:
|
|
"""Get or create a sales service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"sales_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = SalesServiceClient(config, service_name)
|
|
return _client_cache[cache_key]
|
|
|
|
def get_external_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> ExternalServiceClient:
|
|
"""Get or create an external service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"external_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = ExternalServiceClient(config, service_name)
|
|
return _client_cache[cache_key]
|
|
|
|
def get_forecast_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> ForecastServiceClient:
|
|
"""Get or create a forecast service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"forecast_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = ForecastServiceClient(config, service_name)
|
|
return _client_cache[cache_key]
|
|
|
|
def get_inventory_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> InventoryServiceClient:
|
|
"""Get or create an inventory service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"inventory_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = InventoryServiceClient(config)
|
|
return _client_cache[cache_key]
|
|
|
|
def get_orders_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> OrdersServiceClient:
|
|
"""Get or create an orders service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"orders_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = OrdersServiceClient(config)
|
|
return _client_cache[cache_key]
|
|
|
|
def get_production_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> ProductionServiceClient:
|
|
"""Get or create a production service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"production_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = ProductionServiceClient(config)
|
|
return _client_cache[cache_key]
|
|
|
|
def get_recipes_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> RecipesServiceClient:
|
|
"""Get or create a recipes service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"recipes_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = RecipesServiceClient(config)
|
|
return _client_cache[cache_key]
|
|
|
|
def get_suppliers_client(config: BaseServiceSettings = None, service_name: str = "unknown") -> SuppliersServiceClient:
|
|
"""Get or create a suppliers service client"""
|
|
if config is None:
|
|
from app.core.config import settings as config
|
|
|
|
cache_key = f"suppliers_{service_name}"
|
|
if cache_key not in _client_cache:
|
|
_client_cache[cache_key] = SuppliersServiceClient(config)
|
|
return _client_cache[cache_key]
|
|
|
|
|
|
class ServiceClients:
|
|
"""Convenient wrapper for all service clients"""
|
|
|
|
def __init__(self, config: BaseServiceSettings = None, service_name: str = "unknown"):
|
|
self.service_name = service_name
|
|
self.config = config or self._get_default_config()
|
|
|
|
# Initialize clients lazily
|
|
self._training_client = None
|
|
self._sales_client = None
|
|
self._external_client = None
|
|
self._forecast_client = None
|
|
self._inventory_client = None
|
|
self._orders_client = None
|
|
self._production_client = None
|
|
self._recipes_client = None
|
|
self._suppliers_client = None
|
|
|
|
def _get_default_config(self):
|
|
"""Get default config from app settings"""
|
|
try:
|
|
from app.core.config import settings
|
|
return settings
|
|
except ImportError:
|
|
raise ImportError("Could not import app config. Please provide config explicitly.")
|
|
|
|
@property
|
|
def training(self) -> TrainingServiceClient:
|
|
"""Get training service client"""
|
|
if self._training_client is None:
|
|
self._training_client = get_training_client(self.config, self.service_name)
|
|
return self._training_client
|
|
|
|
@property
|
|
def sales(self) -> SalesServiceClient:
|
|
"""Get sales service client"""
|
|
if self._sales_client is None:
|
|
self._sales_client = get_sales_client(self.config, self.service_name)
|
|
return self._sales_client
|
|
|
|
@property
|
|
def external(self) -> ExternalServiceClient:
|
|
"""Get external service client"""
|
|
if self._external_client is None:
|
|
self._external_client = get_external_client(self.config, self.service_name)
|
|
return self._external_client
|
|
|
|
@property
|
|
def forecast(self) -> ForecastServiceClient:
|
|
"""Get forecast service client"""
|
|
if self._forecast_client is None:
|
|
self._forecast_client = get_forecast_client(self.config, self.service_name)
|
|
return self._forecast_client
|
|
|
|
@property
|
|
def inventory(self) -> InventoryServiceClient:
|
|
"""Get inventory service client"""
|
|
if self._inventory_client is None:
|
|
self._inventory_client = get_inventory_client(self.config, self.service_name)
|
|
return self._inventory_client
|
|
|
|
@property
|
|
def orders(self) -> OrdersServiceClient:
|
|
"""Get orders service client"""
|
|
if self._orders_client is None:
|
|
self._orders_client = get_orders_client(self.config, self.service_name)
|
|
return self._orders_client
|
|
|
|
@property
|
|
def production(self) -> ProductionServiceClient:
|
|
"""Get production service client"""
|
|
if self._production_client is None:
|
|
self._production_client = get_production_client(self.config, self.service_name)
|
|
return self._production_client
|
|
|
|
@property
|
|
def recipes(self) -> RecipesServiceClient:
|
|
"""Get recipes service client"""
|
|
if self._recipes_client is None:
|
|
self._recipes_client = get_recipes_client(self.config, self.service_name)
|
|
return self._recipes_client
|
|
|
|
@property
|
|
def suppliers(self) -> SuppliersServiceClient:
|
|
"""Get suppliers service client"""
|
|
if self._suppliers_client is None:
|
|
self._suppliers_client = get_suppliers_client(self.config, self.service_name)
|
|
return self._suppliers_client
|
|
|
|
# Convenience function to get all clients
|
|
def get_service_clients(config: BaseServiceSettings = None, service_name: str = "unknown") -> ServiceClients:
|
|
"""Get a wrapper with all service clients"""
|
|
return ServiceClients(config, service_name)
|
|
|
|
# Export all classes for direct import
|
|
__all__ = [
|
|
'BaseServiceClient',
|
|
'ServiceAuthenticator',
|
|
'AuthServiceClient',
|
|
'TrainingServiceClient',
|
|
'SalesServiceClient',
|
|
'ExternalServiceClient',
|
|
'ForecastServiceClient',
|
|
'InventoryServiceClient',
|
|
'OrdersServiceClient',
|
|
'ProductionServiceClient',
|
|
'RecipesServiceClient',
|
|
'SuppliersServiceClient',
|
|
'TenantServiceClient',
|
|
'ServiceClients',
|
|
'get_training_client',
|
|
'get_sales_client',
|
|
'get_external_client',
|
|
'get_forecast_client',
|
|
'get_inventory_client',
|
|
'get_orders_client',
|
|
'get_production_client',
|
|
'get_recipes_client',
|
|
'get_suppliers_client',
|
|
'get_service_clients'
|
|
] |