67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
import datetime
|
|
from pathlib import Path
|
|
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.responses import HTMLResponse, RedirectResponse
|
|
from fastapi.staticfiles import StaticFiles
|
|
from fastapi.templating import Jinja2Templates
|
|
|
|
from gismeteo import datehelp
|
|
from gismeteo.location import LOCATION_BUNDLE
|
|
from gismeteo.mock import MOCK_DATA
|
|
from weather.model import WeatherResponse
|
|
|
|
from .filters import cloudness_icon, wind_direction_icon
|
|
|
|
|
|
def mount(app: FastAPI):
|
|
base_dir = Path(__file__).parent
|
|
app.mount("/static", StaticFiles(directory=base_dir / "static"), name="static")
|
|
templates = Jinja2Templates(directory=base_dir / "templates")
|
|
templates.env.filters["wind_direction_icon"] = wind_direction_icon
|
|
templates.env.filters["cloudness_icon"] = cloudness_icon
|
|
|
|
def build_weather_response(request: Request, response: WeatherResponse):
|
|
return templates.TemplateResponse(
|
|
request=request,
|
|
name="weather.html",
|
|
context={
|
|
"datetime": datetime,
|
|
"response": response,
|
|
},
|
|
)
|
|
|
|
@app.get("/", response_class=RedirectResponse)
|
|
async def get_weather_root():
|
|
return RedirectResponse("weather")
|
|
|
|
@app.get("/weather", response_class=HTMLResponse)
|
|
async def get_weather_list(request: Request):
|
|
return templates.TemplateResponse(
|
|
request=request,
|
|
name="index.html",
|
|
context={
|
|
"locations": LOCATION_BUNDLE._values,
|
|
},
|
|
)
|
|
|
|
@app.get("/weather/{location}", response_class=RedirectResponse)
|
|
async def get_weather_default(location: str):
|
|
return RedirectResponse(f"{location}/{datetime.date.today()}")
|
|
|
|
@app.get("/weather/{location}/mock", response_class=HTMLResponse)
|
|
async def get_weather_mock(request: Request):
|
|
response = MOCK_DATA.response
|
|
return build_weather_response(request, response)
|
|
|
|
# @app.get("/weather/{location}/{day}", response_class=HTMLResponse)
|
|
async def get_weather_day(request: Request, location: str, day: datehelp.Day):
|
|
date = datehelp.parse(day)
|
|
response = await request.app.state.weather_api.get_day(location, date)
|
|
return build_weather_response(request, response)
|
|
|
|
@app.get("/weather/{location}/{date}", response_class=HTMLResponse)
|
|
async def get_weather(request: Request, location: str, date: datetime.date):
|
|
response = await request.app.state.weather_api.get_day(location, date)
|
|
return build_weather_response(request, response)
|