service_man/api/application/usecase/company/get_users_company.py

from api.application.abstractions.transaction import TransactionContextManager
from api.application.contracts.company.company_request import \
    CompanyByOwnerEmail
from api.application.contracts.company.company_response import \
    CompanyBaseResponse
from api.domain.company.repository import CompanyRepository


class GetCompaniesByOwnerEmail:
    def __init__(
        self,
        transaction: TransactionContextManager,
        company_repository: CompanyRepository,
    ) -> None:
        self.company_repository = company_repository
        self.transaction = transaction

    async def execute(self, request: CompanyByOwnerEmail) -> list[CompanyBaseResponse]:
        async with self.transaction:
            companies = await self.company_repository.get_companies_by_owner_email(
                filter={"email": request.email}
            )
            return [
                CompanyBaseResponse(name=comp.name.value, email=comp.email.value)
                for comp in companies
            ]