47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from src.application.abstractions import IUnitOfWork
|
|
from src.application.contracts import ILogger
|
|
from src.application.domain.entities.organization import LegalEntityEntity
|
|
from src.infrastructure.database.decorators import transactional
|
|
|
|
|
|
class ListOrganizationsCommand:
|
|
def __init__(self, unit_of_work: IUnitOfWork, logger: ILogger):
|
|
self._unit_of_work = unit_of_work
|
|
self._logger = logger
|
|
|
|
@transactional
|
|
async def __call__(self, *, limit: int = 50, offset: int = 0) -> tuple[list[LegalEntityEntity], int]:
|
|
items = await self._unit_of_work.legal_entity_repository.list_all(limit=limit, offset=offset)
|
|
total = await self._unit_of_work.legal_entity_repository.count_all()
|
|
return items, total
|
|
|
|
|
|
class GetOrganizationCommand:
|
|
def __init__(self, unit_of_work: IUnitOfWork, logger: ILogger):
|
|
self._unit_of_work = unit_of_work
|
|
self._logger = logger
|
|
|
|
@transactional
|
|
async def __call__(self, organization_id: str) -> LegalEntityEntity:
|
|
return await self._unit_of_work.legal_entity_repository.get_by_id(organization_id)
|
|
|
|
|
|
class UpdateOrganizationCommand:
|
|
ALLOWED_FIELDS = frozenset({
|
|
'name', 'short_name', 'ogrn', 'kpp', 'legal_address', 'actual_address',
|
|
'bank_details', 'contact_person', 'contact_phone', 'status',
|
|
})
|
|
|
|
def __init__(self, unit_of_work: IUnitOfWork, logger: ILogger):
|
|
self._unit_of_work = unit_of_work
|
|
self._logger = logger
|
|
|
|
@transactional
|
|
async def __call__(self, organization_id: str, *, values: dict[str, Any]) -> LegalEntityEntity:
|
|
filtered = {k: v for k, v in values.items() if k in self.ALLOWED_FIELDS and v is not None}
|
|
return await self._unit_of_work.legal_entity_repository.update(organization_id, values=filtered)
|