ok
Mini Shell
import logging
from xray import gettext as _
logger = logging.getLogger('api_client.init')
def get_client(interface: str = 'api_task') -> 'Client object':
"""
Get a client according to given interface:
'api' --> MongoDB APIClient
'api_req' --> ClickHouse APIClient
'base' --> basic APIClient
'dbm' --> DBMClient
:param interface: 'api'|'api_req'|'base'|'dbm'
:return: a Client object
"""
if interface == 'api_task':
logger.info('Initializing MongoDB API client')
from .api_client import APIClient
return APIClient
elif interface == 'api_req':
logger.info('Initializing ClickHouse API client')
from .api_client import SendClient
return SendClient
elif interface == 'dbm':
logger.info('Initializing DBM API client')
from .api_client import DBMClient
return DBMClient
elif interface == 'auto_tasks':
logger.info('Initializing MongoDB API client for auto tasks')
from .api_client import TasksClient
return TasksClient
elif interface == 'lists':
logger.info('Initializing MongoDB API client for UI')
from .api_client import UIAPIClient
return UIAPIClient
elif interface == 'adviser':
logger.info('Initializing Smart Advice API client')
from .api_client import SmartAdviceAPIClient
return SmartAdviceAPIClient
elif interface == 'awp_provision':
logger.info('Initializing AWP Provision API client')
from .api_client import AWPProvisionAPIClient
return AWPProvisionAPIClient
else:
logger.error('Unknown client to initialize',
extra={'requested_client': interface})
raise RuntimeError(_("Do not know client of '%s' interface") % str(interface))
Zerion Mini Shell 1.0