75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
from fastapi import APIRouter, HTTPException, Path
|
|
from models.schemas import HAEntitiesResponse, HAEntity, HAAttributes
|
|
from services.config import SERVICES
|
|
|
|
router = APIRouter()
|
|
|
|
@router.get("/home-assistant/entities",
|
|
response_model=HAEntitiesResponse,
|
|
summary="Get Home Assistant Entities",
|
|
description="Retrieve all entities from Home Assistant",
|
|
responses={
|
|
200: {"description": "Successfully retrieved entities"},
|
|
503: {"description": "Home Assistant integration not configured"}
|
|
},
|
|
tags=["Home Assistant"])
|
|
async def get_ha_entities():
|
|
"""Get Home Assistant entities including sensors, switches, and other devices"""
|
|
if not SERVICES["home_assistant"]["enabled"]:
|
|
raise HTTPException(
|
|
status_code=503,
|
|
detail="Home Assistant integration not configured. Please set HOME_ASSISTANT_TOKEN environment variable."
|
|
)
|
|
|
|
# This would make actual API calls to Home Assistant
|
|
# For now, return mock data
|
|
return HAEntitiesResponse(
|
|
entities=[
|
|
HAEntity(
|
|
entity_id="sensor.cpu_usage",
|
|
state="45.2",
|
|
attributes=HAAttributes(
|
|
unit_of_measurement="%",
|
|
friendly_name="CPU Usage"
|
|
)
|
|
),
|
|
HAEntity(
|
|
entity_id="sensor.memory_usage",
|
|
state="2.1",
|
|
attributes=HAAttributes(
|
|
unit_of_measurement="GB",
|
|
friendly_name="Memory Usage"
|
|
)
|
|
)
|
|
]
|
|
)
|
|
|
|
@router.get("/home-assistant/entity/{entity_id}",
|
|
response_model=HAEntity,
|
|
summary="Get Specific HA Entity",
|
|
description="Get a specific Home Assistant entity by ID",
|
|
responses={
|
|
200: {"description": "Successfully retrieved entity"},
|
|
404: {"description": "Entity not found"},
|
|
503: {"description": "Home Assistant integration not configured"}
|
|
},
|
|
tags=["Home Assistant"])
|
|
async def get_ha_entity(entity_id: str = Path(..., description="Entity ID")):
|
|
"""Get a specific Home Assistant entity by its ID"""
|
|
if not SERVICES["home_assistant"]["enabled"]:
|
|
raise HTTPException(
|
|
status_code=503,
|
|
detail="Home Assistant integration not configured. Please set HOME_ASSISTANT_TOKEN environment variable."
|
|
)
|
|
|
|
# This would make actual API calls to Home Assistant
|
|
# For now, return mock data
|
|
return HAEntity(
|
|
entity_id=entity_id,
|
|
state="unknown",
|
|
attributes=HAAttributes(
|
|
unit_of_measurement="",
|
|
friendly_name=f"Entity {entity_id}"
|
|
)
|
|
)
|