Code source de henrri_connect.companies.asynchro

"""
Sous-client pour les endpoints /v1/companies.

Constantes:
- ``COMPANIES_ENDPOINT`` URL de base des endpoints entreprises (/v1/companies).

Classes:
- ``henrri_connect.companies.asynchro.AsyncCompaniesClient``
    Accès asynchrone aux endpoints entreprises.

Notes:
- Utiliser de préférence l'objet ``henrri_connect.connect.AsyncHenrriClient`` pour acceder
aux endpoints.
"""

from __future__ import annotations

from typing import TYPE_CHECKING

from ..models import Address, Company

if TYPE_CHECKING:
    from ..connect import AsyncHenrriClient

COMPANIES_ENDPOINT = "/v1/companies"



class AsyncCompaniesClient:
    """
    Accès asynchrone aux endpoints entreprises.

    Arguments
    - ``client`` (AsyncHenrriClient) : Client Henrri Connect.

    Methodes
    - get(self, company_id: int) : Récupère une entreprise par son identifiant.
    - get_address(self, company_id: int) : Récupère l'adresse d'une entreprise.
    """

    def __init__(self, client: AsyncHenrriClient) -> None:
        self._c = client



    async def get(self, company_id: int) -> Company:
        """
        Récupère une entreprise par son identifiant.

        Arguments
        - ``company_id`` : Identifiant de l'entreprise.

        Returns
        - ``henrri_connect.models.Company`` : Entreprise.
        """
        resp = await self._c.request("GET", f"{COMPANIES_ENDPOINT}/{company_id}")
        return Company.model_validate(resp.json())




    async def get_address(self, company_id: int) -> Address:
        """
        Récupère l'adresse d'une entreprise.

        Arguments
        - ``company_id`` : Identifiant de l'entreprise.

        Returns
        - ``henrri_connect.models.Address`` : Adresse de l'entreprise.
        """
        resp = await self._c.request("GET", f"{COMPANIES_ENDPOINT}/{company_id}/address")
        return Address.model_validate(resp.json())