61 lines
1.4 KiB
Python
61 lines
1.4 KiB
Python
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI, Depends
|
|
|
|
from app.api.v1.entities import router as entities_router
|
|
from app.ha.client import HaClient, HaClientSettings
|
|
from app.ha.reader import HaReader
|
|
from app.rules.recommender import Recommender
|
|
from app.rules.heating import HeatingRule
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
ha_url = getattr(app.state.settings, "ha_url", None)
|
|
ha_token = getattr(app.state.settings, "ha_token", None)
|
|
settings = HaClientSettings(
|
|
url=ha_url or "",
|
|
token=ha_token or "",
|
|
)
|
|
client = HaClient(settings=settings)
|
|
app.state.ha_reader = HaReader(client=client)
|
|
app.state.recommender = Recommender(rules=[HeatingRule()])
|
|
yield
|
|
|
|
|
|
app = FastAPI(
|
|
title="SillyHome Next API",
|
|
description="Lokales Smart-Home-Intelligenzsystem für Home Assistant.",
|
|
version="0.1.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
|
|
class Settings:
|
|
ha_url: str = "http://localhost:8123"
|
|
ha_token: str = ""
|
|
|
|
|
|
app.state.settings = Settings()
|
|
register_exception_handlers(app)
|
|
|
|
|
|
def get_ha_reader() -> HaReader:
|
|
return app.state.ha_reader
|
|
|
|
|
|
def get_recommender() -> Recommender:
|
|
return app.state.recommender
|
|
|
|
|
|
app.include_router(entities_router, dependencies=[Depends(get_ha_reader)])
|
|
|
|
|
|
@app.get("/health")
|
|
def health() -> dict[str, str]:
|
|
return {"status": "ok"}
|
|
|
|
|
|
@app.get("/")
|
|
def root() -> dict[str, str]:
|
|
return {"service": "sillyhome-next", "docs": "/docs"} |