service/src/piracyshield_service/dda/get_all_by_account.py
2024-01-19 15:32:27 +01:00

57 lines
1.4 KiB
Python

from __future__ import annotations
from piracyshield_service.base import BaseService
from piracyshield_component.exception import ApplicationException
from piracyshield_data_storage.dda.storage import DDAStorage, DDAStorageGetException
from piracyshield_service.dda.errors import DDAErrorCode, DDAErrorMessage
class DDAGetAllByAccountService(BaseService):
"""
Get all the DDA identifiers assigned to an account.
"""
data_storage = None
def __init__(self):
"""
Inizialize logger and required modules.
"""
super().__init__()
self._prepare_modules()
def execute(self, account_id: str) -> list | Exception:
try:
response = self.data_storage.get_all_by_account(
account_id = account_id
)
batch = response.batch()
if not len(batch):
self.logger.debug(f'No DDA identifier found')
return list(batch)
except DDAStorageGetException as e:
self.logger.error(f'Cannot get all the DDA identifiers')
raise ApplicationException(DDAErrorCode.GENERIC, DDAErrorMessage.GENERIC, e)
def _schedule_task(self):
pass
def _validate_parameters(self):
pass
def _prepare_configs(self):
pass
def _prepare_modules(self):
self.data_storage = DDAStorage()