"""Reit Service""" from typing import Optional, List from ..common import SitarAPIBaseService from .schemas import ReitResponse class ReitService(SitarAPIBaseService): _instance: Optional["ReitService"] = None @classmethod def get_instance(cls) -> "ReitService": if cls._instance is None: cls._instance = cls() return cls._instance async def search( self, fraccion: Optional[str] = None, nico: Optional[str] = None, skip: int = 0, limit: int = 100, ) -> List[ReitResponse]: params = {"skip": skip, "limit": min(limit, 1000)} if fraccion: params["fraccion"] = fraccion if nico: params["nico"] = nico data = await self._make_request("GET", "/api/v1/reit/", params=params) return [ReitResponse(**item) for item in data] async def get_by_id(self, sysid: int) -> ReitResponse: data = await self._make_request("GET", f"/api/v1/reit/{sysid}") return ReitResponse(**data)