mirror of
https://code.lenaisten.de/Lenaisten/advent22.git
synced 2024-11-23 08:13:01 +00:00
Merge branch 'feature/api-refactoring' into develop
This commit is contained in:
commit
3153f57613
18 changed files with 438 additions and 349 deletions
1
api/.vscode/launch.json
vendored
1
api/.vscode/launch.json
vendored
|
@ -14,6 +14,7 @@
|
||||||
],
|
],
|
||||||
"env": {
|
"env": {
|
||||||
"PYDEVD_DISABLE_FILE_VALIDATION": "1",
|
"PYDEVD_DISABLE_FILE_VALIDATION": "1",
|
||||||
|
"CACHE_TTL": "30",
|
||||||
},
|
},
|
||||||
"justMyCode": true,
|
"justMyCode": true,
|
||||||
}
|
}
|
||||||
|
|
|
@ -2,8 +2,8 @@ from fastapi import FastAPI
|
||||||
from fastapi.middleware.cors import CORSMiddleware
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
from fastapi.staticfiles import StaticFiles
|
from fastapi.staticfiles import StaticFiles
|
||||||
|
|
||||||
|
from .core.settings import SETTINGS
|
||||||
from .routers import router
|
from .routers import router
|
||||||
from .settings import SETTINGS
|
|
||||||
|
|
||||||
app = FastAPI(
|
app = FastAPI(
|
||||||
title="Advent22 API",
|
title="Advent22 API",
|
||||||
|
|
|
@ -1,52 +0,0 @@
|
||||||
import tomllib
|
|
||||||
from typing import TypeAlias
|
|
||||||
|
|
||||||
import tomli_w
|
|
||||||
from pydantic import BaseModel
|
|
||||||
|
|
||||||
from .config import Config, get_config
|
|
||||||
from .dav_common import dav_get_textfile_content, dav_write_textfile_content
|
|
||||||
|
|
||||||
|
|
||||||
class DoorSaved(BaseModel):
|
|
||||||
# Tag, an dem die Tür aufgeht
|
|
||||||
day: int
|
|
||||||
|
|
||||||
# Koordinaten für zwei Eckpunkte
|
|
||||||
x1: int
|
|
||||||
y1: int
|
|
||||||
x2: int
|
|
||||||
y2: int
|
|
||||||
|
|
||||||
|
|
||||||
DoorsSaved: TypeAlias = list[DoorSaved]
|
|
||||||
|
|
||||||
|
|
||||||
class CalendarConfig(BaseModel):
|
|
||||||
# Dateiname Hintergrundbild
|
|
||||||
background: str = "adventskalender.jpg"
|
|
||||||
|
|
||||||
# Türen für die UI
|
|
||||||
doors: DoorsSaved = []
|
|
||||||
|
|
||||||
|
|
||||||
async def get_calendar_config() -> CalendarConfig:
|
|
||||||
cfg = await get_config()
|
|
||||||
|
|
||||||
txt = await dav_get_textfile_content(
|
|
||||||
path=f"files/{cfg.puzzle.calendar}",
|
|
||||||
)
|
|
||||||
|
|
||||||
return CalendarConfig.model_validate(tomllib.loads(txt))
|
|
||||||
|
|
||||||
|
|
||||||
async def set_calendar_config(cfg: Config, cal_cfg: CalendarConfig) -> None:
|
|
||||||
await dav_write_textfile_content(
|
|
||||||
path=f"files/{cfg.puzzle.calendar}",
|
|
||||||
content=tomli_w.dumps(
|
|
||||||
cal_cfg.model_dump(
|
|
||||||
exclude_defaults=True,
|
|
||||||
exclude_unset=True,
|
|
||||||
)
|
|
||||||
),
|
|
||||||
)
|
|
|
@ -9,25 +9,24 @@ _RGB: TypeAlias = tuple[int, int, int]
|
||||||
_XY: TypeAlias = tuple[float, float]
|
_XY: TypeAlias = tuple[float, float]
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass(slots=True, frozen=True)
|
||||||
class AdventImage:
|
class AdventImage:
|
||||||
img: Image.Image
|
img: Image.Image
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
async def load_standard(cls, fp) -> Self:
|
async def from_img(cls, img: Image.Image) -> Self:
|
||||||
"""
|
"""
|
||||||
Bild laden und einen quadratischen Ausschnitt
|
Einen quadratischen Ausschnitt aus der Mitte des Bilds nehmen
|
||||||
aus der Mitte nehmen
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# Bild laden
|
# Farbmodell festlegen
|
||||||
img = Image.open(fp=fp)
|
img = img.convert(mode="RGB")
|
||||||
|
|
||||||
# Größen bestimmen
|
# Größen bestimmen
|
||||||
width, height = img.size
|
width, height = img.size
|
||||||
square = min(width, height)
|
square = min(width, height)
|
||||||
|
|
||||||
# Bild zuschneiden und skalieren
|
# zuschneiden
|
||||||
img = img.crop(
|
img = img.crop(
|
||||||
box=(
|
box=(
|
||||||
int((width - square) / 2),
|
int((width - square) / 2),
|
||||||
|
@ -37,14 +36,14 @@ class AdventImage:
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
img = img.resize(
|
# skalieren
|
||||||
size=(500, 500),
|
return cls(
|
||||||
resample=Image.LANCZOS,
|
img.resize(
|
||||||
|
size=(500, 500),
|
||||||
|
resample=Image.LANCZOS,
|
||||||
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
# Farbmodell festlegen
|
|
||||||
return cls(img=img.convert("RGB"))
|
|
||||||
|
|
||||||
async def get_text_box(
|
async def get_text_box(
|
||||||
self,
|
self,
|
||||||
xy: _XY,
|
xy: _XY,
|
||||||
|
@ -52,7 +51,7 @@ class AdventImage:
|
||||||
font: "ImageFont._Font",
|
font: "ImageFont._Font",
|
||||||
anchor: str | None = "mm",
|
anchor: str | None = "mm",
|
||||||
**text_kwargs,
|
**text_kwargs,
|
||||||
) -> tuple[int, int, int, int] | None:
|
) -> "Image._Box | None":
|
||||||
"""
|
"""
|
||||||
Koordinaten (links, oben, rechts, unten) des betroffenen
|
Koordinaten (links, oben, rechts, unten) des betroffenen
|
||||||
Rechtecks bestimmen, wenn das Bild mit einem Text
|
Rechtecks bestimmen, wenn das Bild mit einem Text
|
||||||
|
@ -77,8 +76,8 @@ class AdventImage:
|
||||||
|
|
||||||
async def get_average_color(
|
async def get_average_color(
|
||||||
self,
|
self,
|
||||||
box: tuple[int, int, int, int],
|
box: "Image._Box",
|
||||||
) -> _RGB:
|
) -> tuple[int, int, int]:
|
||||||
"""
|
"""
|
||||||
Durchschnittsfarbe eines rechteckigen Ausschnitts in
|
Durchschnittsfarbe eines rechteckigen Ausschnitts in
|
||||||
einem Bild berechnen
|
einem Bild berechnen
|
57
api/advent22_api/core/calendar_config.py
Normal file
57
api/advent22_api/core/calendar_config.py
Normal file
|
@ -0,0 +1,57 @@
|
||||||
|
import tomllib
|
||||||
|
from typing import TypeAlias
|
||||||
|
|
||||||
|
import tomli_w
|
||||||
|
from fastapi import Depends
|
||||||
|
from pydantic import BaseModel
|
||||||
|
|
||||||
|
from .config import Config, get_config
|
||||||
|
from .webdav import WebDAV
|
||||||
|
|
||||||
|
|
||||||
|
class DoorSaved(BaseModel):
|
||||||
|
# Tag, an dem die Tür aufgeht
|
||||||
|
day: int
|
||||||
|
|
||||||
|
# Koordinaten für zwei Eckpunkte
|
||||||
|
x1: int
|
||||||
|
y1: int
|
||||||
|
x2: int
|
||||||
|
y2: int
|
||||||
|
|
||||||
|
|
||||||
|
DoorsSaved: TypeAlias = list[DoorSaved]
|
||||||
|
|
||||||
|
|
||||||
|
class CalendarConfig(BaseModel):
|
||||||
|
# Dateiname Hintergrundbild
|
||||||
|
background: str = "adventskalender.jpg"
|
||||||
|
|
||||||
|
# Türen für die UI
|
||||||
|
doors: DoorsSaved = []
|
||||||
|
|
||||||
|
async def change(self, cfg: Config) -> None:
|
||||||
|
"""
|
||||||
|
Kalender Konfiguration ändern
|
||||||
|
"""
|
||||||
|
|
||||||
|
await WebDAV.write_str(
|
||||||
|
path=f"files/{cfg.puzzle.calendar}",
|
||||||
|
content=tomli_w.dumps(
|
||||||
|
self.model_dump(
|
||||||
|
exclude_defaults=True,
|
||||||
|
exclude_unset=True,
|
||||||
|
)
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def get_calendar_config(
|
||||||
|
cfg: Config = Depends(get_config),
|
||||||
|
) -> CalendarConfig:
|
||||||
|
"""
|
||||||
|
Kalender Konfiguration lesen
|
||||||
|
"""
|
||||||
|
|
||||||
|
txt = await WebDAV.read_str(path=f"files/{cfg.puzzle.calendar}")
|
||||||
|
return CalendarConfig.model_validate(tomllib.loads(txt))
|
|
@ -2,8 +2,8 @@ import tomllib
|
||||||
|
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
|
|
||||||
from .dav_common import dav_get_textfile_content
|
|
||||||
from .settings import SETTINGS
|
from .settings import SETTINGS
|
||||||
|
from .webdav import WebDAV
|
||||||
|
|
||||||
|
|
||||||
class User(BaseModel):
|
class User(BaseModel):
|
||||||
|
@ -45,7 +45,10 @@ class Config(BaseModel):
|
||||||
puzzle: Puzzle
|
puzzle: Puzzle
|
||||||
|
|
||||||
|
|
||||||
async def get_config() -> Config:
|
async def get_config() -> "Config":
|
||||||
txt = await dav_get_textfile_content(path=SETTINGS.config_filename)
|
"""
|
||||||
|
Globale Konfiguration lesen
|
||||||
|
"""
|
||||||
|
|
||||||
|
txt = await WebDAV.read_str(path=SETTINGS.config_filename)
|
||||||
return Config.model_validate(tomllib.loads(txt))
|
return Config.model_validate(tomllib.loads(txt))
|
106
api/advent22_api/core/depends.py
Normal file
106
api/advent22_api/core/depends.py
Normal file
|
@ -0,0 +1,106 @@
|
||||||
|
from io import BytesIO
|
||||||
|
from typing import cast
|
||||||
|
|
||||||
|
from fastapi import Depends
|
||||||
|
from PIL import Image, ImageFont
|
||||||
|
|
||||||
|
from .advent_image import _XY, AdventImage
|
||||||
|
from .config import Config, get_config
|
||||||
|
from .image_helpers import list_images_auto, load_image
|
||||||
|
from .sequence_helpers import Random, set_len, shuffle
|
||||||
|
from .webdav import WebDAV
|
||||||
|
|
||||||
|
|
||||||
|
async def shuffle_solution(
|
||||||
|
cfg: Config = Depends(get_config),
|
||||||
|
) -> str:
|
||||||
|
"""
|
||||||
|
Lösung: Reihenfolge zufällig bestimmen
|
||||||
|
"""
|
||||||
|
|
||||||
|
return "".join(await shuffle(cfg.puzzle.solution))
|
||||||
|
|
||||||
|
|
||||||
|
async def shuffle_images_auto(
|
||||||
|
images: list[str] = Depends(list_images_auto),
|
||||||
|
) -> list[str]:
|
||||||
|
"""
|
||||||
|
Bilder: Reihenfolge zufällig bestimmen
|
||||||
|
"""
|
||||||
|
|
||||||
|
ls = set_len(images, 24)
|
||||||
|
return await shuffle(ls)
|
||||||
|
|
||||||
|
|
||||||
|
async def get_part(
|
||||||
|
day: int,
|
||||||
|
shuffled_solution: str = Depends(shuffle_solution),
|
||||||
|
) -> str:
|
||||||
|
"""
|
||||||
|
Heute angezeigter Teil der Lösung
|
||||||
|
"""
|
||||||
|
|
||||||
|
return shuffled_solution[day]
|
||||||
|
|
||||||
|
|
||||||
|
async def get_random(
|
||||||
|
day: int,
|
||||||
|
) -> Random:
|
||||||
|
"""
|
||||||
|
Tagesabhängige Zufallszahlen
|
||||||
|
"""
|
||||||
|
|
||||||
|
return await Random.get(day)
|
||||||
|
|
||||||
|
|
||||||
|
async def gen_auto_image(
|
||||||
|
day: int,
|
||||||
|
auto_images: list[str] = Depends(shuffle_images_auto),
|
||||||
|
cfg: Config = Depends(get_config),
|
||||||
|
rnd: Random = Depends(get_random),
|
||||||
|
part: str = Depends(get_part),
|
||||||
|
) -> Image.Image:
|
||||||
|
"""
|
||||||
|
Automatisch generiertes Bild erstellen
|
||||||
|
"""
|
||||||
|
|
||||||
|
# Datei existiert garantiert!
|
||||||
|
img = await load_image(auto_images[day])
|
||||||
|
image = await AdventImage.from_img(img)
|
||||||
|
|
||||||
|
font = ImageFont.truetype(
|
||||||
|
font=BytesIO(await WebDAV.read_bytes(f"files/{cfg.server.font}")),
|
||||||
|
size=50,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Buchstaben verstecken
|
||||||
|
for letter in part:
|
||||||
|
await image.hide_text(
|
||||||
|
xy=cast(_XY, tuple(rnd.choices(range(30, 470), k=2))),
|
||||||
|
text=letter,
|
||||||
|
font=font,
|
||||||
|
)
|
||||||
|
|
||||||
|
return image.img
|
||||||
|
|
||||||
|
|
||||||
|
async def get_image(
|
||||||
|
day: int,
|
||||||
|
auto_images: list[str] = Depends(shuffle_images_auto),
|
||||||
|
cfg: Config = Depends(get_config),
|
||||||
|
rnd: Random = Depends(get_random),
|
||||||
|
part: str = Depends(get_part),
|
||||||
|
) -> Image.Image:
|
||||||
|
"""
|
||||||
|
Bild für einen Tag abrufen
|
||||||
|
"""
|
||||||
|
|
||||||
|
try:
|
||||||
|
# Versuche, aus "manual"-Ordner zu laden
|
||||||
|
return await load_image(f"images_manual/{day}.jpg")
|
||||||
|
|
||||||
|
except RuntimeError:
|
||||||
|
# Erstelle automatisch generiertes Bild
|
||||||
|
return await gen_auto_image(
|
||||||
|
day=day, auto_images=auto_images, cfg=cfg, rnd=rnd, part=part
|
||||||
|
)
|
46
api/advent22_api/core/image_helpers.py
Normal file
46
api/advent22_api/core/image_helpers.py
Normal file
|
@ -0,0 +1,46 @@
|
||||||
|
import re
|
||||||
|
from io import BytesIO
|
||||||
|
|
||||||
|
from fastapi.responses import StreamingResponse
|
||||||
|
from PIL import Image
|
||||||
|
|
||||||
|
from .webdav import WebDAV
|
||||||
|
|
||||||
|
|
||||||
|
async def list_images_auto() -> list[str]:
|
||||||
|
"""
|
||||||
|
Finde alle Bilddateien im "automatisch"-Verzeichnis
|
||||||
|
"""
|
||||||
|
|
||||||
|
return await WebDAV.list_files(
|
||||||
|
directory="/images_auto",
|
||||||
|
regex=re.compile(r"\.(gif|jpe?g|tiff?|png|bmp)$", flags=re.IGNORECASE),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def load_image(file_name: str) -> Image.Image:
|
||||||
|
"""
|
||||||
|
Versuche, Bild aus Datei zu laden
|
||||||
|
"""
|
||||||
|
|
||||||
|
if not await WebDAV.file_exists(file_name):
|
||||||
|
raise RuntimeError(f"DAV-File {file_name} does not exist!")
|
||||||
|
|
||||||
|
return Image.open(BytesIO(await WebDAV.read_bytes(file_name)))
|
||||||
|
|
||||||
|
|
||||||
|
async def api_return_image(img: Image.Image) -> StreamingResponse:
|
||||||
|
"""
|
||||||
|
Bild mit API zurückgeben
|
||||||
|
"""
|
||||||
|
|
||||||
|
# JPEG-Daten in Puffer speichern
|
||||||
|
img_buffer = BytesIO()
|
||||||
|
img.save(img_buffer, format="JPEG", quality=85)
|
||||||
|
img_buffer.seek(0)
|
||||||
|
|
||||||
|
# zurückgeben
|
||||||
|
return StreamingResponse(
|
||||||
|
media_type="image/jpeg",
|
||||||
|
content=img_buffer,
|
||||||
|
)
|
28
api/advent22_api/core/sequence_helpers.py
Normal file
28
api/advent22_api/core/sequence_helpers.py
Normal file
|
@ -0,0 +1,28 @@
|
||||||
|
import itertools
|
||||||
|
import random
|
||||||
|
from typing import Any, Self, Sequence
|
||||||
|
|
||||||
|
from .config import get_config
|
||||||
|
|
||||||
|
|
||||||
|
class Random(random.Random):
|
||||||
|
@classmethod
|
||||||
|
async def get(cls, bonus_salt: Any = "") -> Self:
|
||||||
|
cfg = await get_config()
|
||||||
|
return cls(f"{cfg.puzzle.solution}{cfg.puzzle.random_pepper}{bonus_salt}")
|
||||||
|
|
||||||
|
|
||||||
|
async def shuffle(seq: Sequence, rnd: random.Random | None = None) -> list:
|
||||||
|
# Zufallsgenerator
|
||||||
|
rnd = rnd or await Random.get()
|
||||||
|
|
||||||
|
# Elemente mischen
|
||||||
|
return rnd.sample(seq, len(seq))
|
||||||
|
|
||||||
|
|
||||||
|
def set_len(seq: Sequence, length: int) -> list:
|
||||||
|
# `seq` unendlich wiederholen
|
||||||
|
infinite = itertools.cycle(seq)
|
||||||
|
|
||||||
|
# Die ersten `length` einträge nehmen
|
||||||
|
return list(itertools.islice(infinite, length))
|
|
@ -58,7 +58,7 @@ class Settings(BaseSettings):
|
||||||
|
|
||||||
webdav: DavSettings = DavSettings()
|
webdav: DavSettings = DavSettings()
|
||||||
|
|
||||||
cache_ttl: int = 30
|
cache_ttl: int = 60 * 30
|
||||||
config_filename: str = "config.toml"
|
config_filename: str = "config.toml"
|
||||||
|
|
||||||
|
|
81
api/advent22_api/core/webdav.py
Normal file
81
api/advent22_api/core/webdav.py
Normal file
|
@ -0,0 +1,81 @@
|
||||||
|
import re
|
||||||
|
from io import BytesIO
|
||||||
|
|
||||||
|
from cache import AsyncTTL
|
||||||
|
from webdav3.client import Client as WebDAVclient
|
||||||
|
|
||||||
|
from .settings import SETTINGS
|
||||||
|
|
||||||
|
|
||||||
|
class WebDAV:
|
||||||
|
_webdav_client = WebDAVclient(
|
||||||
|
{
|
||||||
|
"webdav_hostname": SETTINGS.webdav.url,
|
||||||
|
"webdav_login": SETTINGS.webdav.username,
|
||||||
|
"webdav_password": SETTINGS.webdav.password,
|
||||||
|
"disable_check": SETTINGS.webdav.disable_check,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
@AsyncTTL(time_to_live=SETTINGS.cache_ttl)
|
||||||
|
async def list_files(
|
||||||
|
cls,
|
||||||
|
directory: str = "",
|
||||||
|
*,
|
||||||
|
regex: re.Pattern[str] = re.compile(""),
|
||||||
|
) -> list[str]:
|
||||||
|
"""
|
||||||
|
Liste aller Dateien im Ordner `directory`, die zur RegEx `regex` passen
|
||||||
|
"""
|
||||||
|
|
||||||
|
ls = cls._webdav_client.list(directory)
|
||||||
|
|
||||||
|
return [f"{directory}/{path}" for path in ls if regex.search(path)]
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
@AsyncTTL(time_to_live=SETTINGS.cache_ttl)
|
||||||
|
async def file_exists(cls, path: str) -> bool:
|
||||||
|
"""
|
||||||
|
`True`, wenn an Pfad `path` eine Datei existiert
|
||||||
|
"""
|
||||||
|
|
||||||
|
return cls._webdav_client.check(path)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
@AsyncTTL(time_to_live=SETTINGS.cache_ttl)
|
||||||
|
async def read_bytes(cls, path: str) -> bytes:
|
||||||
|
"""
|
||||||
|
Datei aus Pfad `path` als bytes laden
|
||||||
|
"""
|
||||||
|
|
||||||
|
buffer = BytesIO()
|
||||||
|
cls._webdav_client.resource(path).write_to(buffer)
|
||||||
|
buffer.seek(0)
|
||||||
|
|
||||||
|
return buffer.read()
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
@AsyncTTL(time_to_live=SETTINGS.cache_ttl)
|
||||||
|
async def read_str(cls, path: str, encoding="utf-8") -> str:
|
||||||
|
"""
|
||||||
|
Datei aus Pfad `path` als string laden
|
||||||
|
"""
|
||||||
|
|
||||||
|
return (await cls.read_bytes(path)).decode(encoding=encoding).strip()
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
async def write_bytes(cls, path: str, buffer: bytes) -> None:
|
||||||
|
"""
|
||||||
|
Bytes `buffer` in Datei in Pfad `path` schreiben
|
||||||
|
"""
|
||||||
|
|
||||||
|
cls._webdav_client.resource(path).read_from(buffer)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
async def write_str(cls, path: str, content: str, encoding="utf-8") -> None:
|
||||||
|
"""
|
||||||
|
String `content` in Datei in Pfad `path` schreiben
|
||||||
|
"""
|
||||||
|
|
||||||
|
await cls.write_bytes(path, content.encode(encoding=encoding))
|
|
@ -1,55 +0,0 @@
|
||||||
import re
|
|
||||||
from io import BytesIO, TextIOWrapper
|
|
||||||
|
|
||||||
from cache import AsyncTTL
|
|
||||||
from webdav3.client import Client as WebDAVclient
|
|
||||||
|
|
||||||
from .settings import SETTINGS
|
|
||||||
|
|
||||||
_WEBDAV_CLIENT = WebDAVclient(
|
|
||||||
{
|
|
||||||
"webdav_hostname": SETTINGS.webdav.url,
|
|
||||||
"webdav_login": SETTINGS.webdav.username,
|
|
||||||
"webdav_password": SETTINGS.webdav.password,
|
|
||||||
"disable_check": SETTINGS.webdav.disable_check,
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@AsyncTTL(time_to_live=SETTINGS.cache_ttl)
|
|
||||||
async def dav_list_files(regex: re.Pattern, directory: str = "") -> list[str]:
|
|
||||||
ls = _WEBDAV_CLIENT.list(directory)
|
|
||||||
return [f"{directory}/{path}" for path in ls if regex.search(path)]
|
|
||||||
|
|
||||||
|
|
||||||
@AsyncTTL(time_to_live=SETTINGS.cache_ttl)
|
|
||||||
async def dav_file_exists(path: str) -> bool:
|
|
||||||
return _WEBDAV_CLIENT.check(path)
|
|
||||||
|
|
||||||
|
|
||||||
@AsyncTTL(time_to_live=SETTINGS.cache_ttl)
|
|
||||||
async def dav_get_file(path: str) -> BytesIO:
|
|
||||||
resource = _WEBDAV_CLIENT.resource(path)
|
|
||||||
buffer = BytesIO()
|
|
||||||
resource.write_to(buffer)
|
|
||||||
|
|
||||||
return buffer
|
|
||||||
|
|
||||||
|
|
||||||
@AsyncTTL(time_to_live=SETTINGS.cache_ttl)
|
|
||||||
async def dav_get_textfile_content(path: str, encoding="utf-8") -> str:
|
|
||||||
buffer = await dav_get_file(path)
|
|
||||||
tio = TextIOWrapper(buffer, encoding=encoding)
|
|
||||||
tio.seek(0)
|
|
||||||
return tio.read().strip()
|
|
||||||
|
|
||||||
|
|
||||||
async def dav_write_file(path: str, buffer: BytesIO) -> None:
|
|
||||||
resource = _WEBDAV_CLIENT.resource(path)
|
|
||||||
resource.read_from(buffer)
|
|
||||||
|
|
||||||
|
|
||||||
async def dav_write_textfile_content(path: str, content: str, encoding="utf-8") -> None:
|
|
||||||
buffer = BytesIO(content.encode(encoding=encoding))
|
|
||||||
buffer.seek(0)
|
|
||||||
await dav_write_file(path, buffer)
|
|
|
@ -2,7 +2,7 @@
|
||||||
|
|
||||||
import uvicorn
|
import uvicorn
|
||||||
|
|
||||||
from .settings import SETTINGS
|
from .core.settings import SETTINGS
|
||||||
|
|
||||||
|
|
||||||
def main() -> None:
|
def main() -> None:
|
||||||
|
|
|
@ -1,152 +0,0 @@
|
||||||
import itertools
|
|
||||||
import random
|
|
||||||
import re
|
|
||||||
from io import BytesIO
|
|
||||||
from typing import Any, Self, Sequence, cast
|
|
||||||
|
|
||||||
from fastapi import Depends
|
|
||||||
from fastapi.responses import StreamingResponse
|
|
||||||
from PIL import Image, ImageFont
|
|
||||||
|
|
||||||
from ..config import Config, get_config
|
|
||||||
from ..dav_common import dav_file_exists, dav_get_file, dav_list_files
|
|
||||||
from ._image import _XY, AdventImage
|
|
||||||
|
|
||||||
##########
|
|
||||||
# RANDOM #
|
|
||||||
##########
|
|
||||||
|
|
||||||
|
|
||||||
class Random(random.Random):
|
|
||||||
@classmethod
|
|
||||||
async def get(cls, bonus_salt: Any = "") -> Self:
|
|
||||||
cfg = await get_config()
|
|
||||||
return cls(f"{cfg.puzzle.solution}{bonus_salt}{cfg.puzzle.random_pepper}")
|
|
||||||
|
|
||||||
|
|
||||||
async def set_length(seq: Sequence, length: int) -> list:
|
|
||||||
# `seq` unendlich wiederholen
|
|
||||||
infinite = itertools.cycle(seq)
|
|
||||||
# Die ersten `length` einträge nehmen
|
|
||||||
return list(itertools.islice(infinite, length))
|
|
||||||
|
|
||||||
|
|
||||||
async def shuffle(seq: Sequence, rnd: random.Random | None = None) -> list:
|
|
||||||
# Zufallsgenerator
|
|
||||||
rnd = rnd or await Random.get()
|
|
||||||
|
|
||||||
# Elemente mischen
|
|
||||||
return rnd.sample(seq, len(seq))
|
|
||||||
|
|
||||||
|
|
||||||
#########
|
|
||||||
# IMAGE #
|
|
||||||
#########
|
|
||||||
|
|
||||||
|
|
||||||
async def get_letter(
|
|
||||||
index: int,
|
|
||||||
cfg: Config = Depends(get_config),
|
|
||||||
) -> str:
|
|
||||||
return (await shuffle(cfg.puzzle.solution))[index]
|
|
||||||
|
|
||||||
|
|
||||||
_RE_IMAGE_FILE = re.compile(
|
|
||||||
r"\.(gif|jpe?g|tiff?|png|bmp)$",
|
|
||||||
flags=re.IGNORECASE,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def list_images_auto() -> list[str]:
|
|
||||||
"""
|
|
||||||
Finde alle Bilder im "automatisch"-Verzeichnis
|
|
||||||
"""
|
|
||||||
ls = await dav_list_files(_RE_IMAGE_FILE, "/images_auto")
|
|
||||||
ls = await set_length(ls, 24)
|
|
||||||
|
|
||||||
return await shuffle(ls)
|
|
||||||
|
|
||||||
|
|
||||||
async def load_image(
|
|
||||||
file_name: str,
|
|
||||||
) -> AdventImage:
|
|
||||||
"""
|
|
||||||
Versuche, Bild aus Datei zu laden
|
|
||||||
"""
|
|
||||||
|
|
||||||
if not await dav_file_exists(file_name):
|
|
||||||
raise RuntimeError(f"DAV-File {file_name} does not exist!")
|
|
||||||
|
|
||||||
img_buffer = await dav_get_file(file_name)
|
|
||||||
img_buffer.seek(0)
|
|
||||||
return await AdventImage.load_standard(img_buffer)
|
|
||||||
|
|
||||||
|
|
||||||
async def get_auto_image(
|
|
||||||
index: int,
|
|
||||||
letter: str,
|
|
||||||
images: list[str],
|
|
||||||
cfg: Config,
|
|
||||||
) -> AdventImage:
|
|
||||||
"""
|
|
||||||
Erstelle automatisch generiertes Bild
|
|
||||||
"""
|
|
||||||
|
|
||||||
# hier niemals RuntimeError!
|
|
||||||
image = await load_image(images[index])
|
|
||||||
rnd = await Random.get(index)
|
|
||||||
|
|
||||||
font = await dav_get_file(f"files/{cfg.server.font}")
|
|
||||||
font.seek(0)
|
|
||||||
|
|
||||||
# Buchstabe verstecken
|
|
||||||
await image.hide_text(
|
|
||||||
xy=cast(_XY, tuple(rnd.choices(range(30, 470), k=2))),
|
|
||||||
text=letter,
|
|
||||||
font=ImageFont.truetype(font, 50),
|
|
||||||
)
|
|
||||||
|
|
||||||
return image
|
|
||||||
|
|
||||||
|
|
||||||
async def get_image(
|
|
||||||
index: int,
|
|
||||||
letter: str = Depends(get_letter),
|
|
||||||
images: list[str] = Depends(list_images_auto),
|
|
||||||
cfg: Config = Depends(get_config),
|
|
||||||
) -> AdventImage:
|
|
||||||
"""
|
|
||||||
Bild für einen Tag erstellen
|
|
||||||
"""
|
|
||||||
|
|
||||||
try:
|
|
||||||
# Versuche, aus "manual"-Ordner zu laden
|
|
||||||
return await load_image(f"images_manual/{index}.jpg")
|
|
||||||
|
|
||||||
except RuntimeError:
|
|
||||||
# Erstelle automatisch generiertes Bild
|
|
||||||
return await get_auto_image(
|
|
||||||
index=index,
|
|
||||||
letter=letter,
|
|
||||||
images=images,
|
|
||||||
cfg=cfg,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def api_return_image(
|
|
||||||
img: Image.Image,
|
|
||||||
) -> StreamingResponse:
|
|
||||||
"""
|
|
||||||
Bild mit API zurückgeben
|
|
||||||
"""
|
|
||||||
|
|
||||||
# Bilddaten in Puffer laden
|
|
||||||
img_buffer = BytesIO()
|
|
||||||
img.save(img_buffer, format="JPEG", quality=85)
|
|
||||||
img_buffer.seek(0)
|
|
||||||
|
|
||||||
# zurückgeben
|
|
||||||
return StreamingResponse(
|
|
||||||
content=img_buffer,
|
|
||||||
media_type="image/jpeg",
|
|
||||||
)
|
|
60
api/advent22_api/routers/_security.py
Normal file
60
api/advent22_api/routers/_security.py
Normal file
|
@ -0,0 +1,60 @@
|
||||||
|
import secrets
|
||||||
|
from datetime import date
|
||||||
|
|
||||||
|
from fastapi import Depends, HTTPException, status
|
||||||
|
from fastapi.security import HTTPBasic, HTTPBasicCredentials
|
||||||
|
|
||||||
|
from ..core.config import Config, get_config
|
||||||
|
|
||||||
|
security = HTTPBasic()
|
||||||
|
|
||||||
|
|
||||||
|
async def user_is_admin(
|
||||||
|
credentials: HTTPBasicCredentials = Depends(security),
|
||||||
|
cfg: Config = Depends(get_config),
|
||||||
|
) -> bool:
|
||||||
|
"""
|
||||||
|
True iff der user "admin" ist
|
||||||
|
"""
|
||||||
|
|
||||||
|
username_correct = secrets.compare_digest(credentials.username, cfg.admin.name)
|
||||||
|
password_correct = secrets.compare_digest(credentials.password, cfg.admin.password)
|
||||||
|
|
||||||
|
return username_correct and password_correct
|
||||||
|
|
||||||
|
|
||||||
|
async def require_admin(
|
||||||
|
is_admin: bool = Depends(user_is_admin),
|
||||||
|
) -> None:
|
||||||
|
"""
|
||||||
|
HTTP 401 iff der user nicht "admin" ist
|
||||||
|
"""
|
||||||
|
|
||||||
|
if not is_admin:
|
||||||
|
raise HTTPException(status.HTTP_401_UNAUTHORIZED)
|
||||||
|
|
||||||
|
|
||||||
|
async def user_visible_doors() -> int:
|
||||||
|
"""
|
||||||
|
Anzahl der user-sichtbaren Türchen
|
||||||
|
"""
|
||||||
|
|
||||||
|
today = date.today()
|
||||||
|
|
||||||
|
if today.month == 12:
|
||||||
|
return today.day
|
||||||
|
|
||||||
|
if today.month in (1, 2, 3):
|
||||||
|
return 24
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
|
||||||
|
async def user_can_view_door(
|
||||||
|
day: int,
|
||||||
|
) -> bool:
|
||||||
|
"""
|
||||||
|
True iff das Türchen von Tag `day` user-sichtbar ist
|
||||||
|
"""
|
||||||
|
|
||||||
|
return day < await user_visible_doors()
|
|
@ -2,11 +2,12 @@ from datetime import date
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, HTTPException, status
|
from fastapi import APIRouter, Depends, HTTPException, status
|
||||||
from fastapi.responses import StreamingResponse
|
from fastapi.responses import StreamingResponse
|
||||||
|
from PIL import Image
|
||||||
|
|
||||||
from ..config import Config, get_config
|
from ..core.config import get_config
|
||||||
from ._image import AdventImage
|
from ..core.depends import get_image, get_part, shuffle_solution
|
||||||
from ._misc import api_return_image, get_image, shuffle
|
from ..core.image_helpers import api_return_image
|
||||||
from .user import user_is_admin
|
from ._security import user_can_view_door, user_is_admin, user_visible_doors
|
||||||
|
|
||||||
router = APIRouter(prefix="/days", tags=["days"])
|
router = APIRouter(prefix="/days", tags=["days"])
|
||||||
|
|
||||||
|
@ -15,48 +16,47 @@ router = APIRouter(prefix="/days", tags=["days"])
|
||||||
async def startup() -> None:
|
async def startup() -> None:
|
||||||
cfg = await get_config()
|
cfg = await get_config()
|
||||||
print(cfg.puzzle.solution)
|
print(cfg.puzzle.solution)
|
||||||
print("".join(await shuffle(cfg.puzzle.solution)))
|
|
||||||
|
|
||||||
|
shuffled_solution = await shuffle_solution(cfg)
|
||||||
@router.get("/letter/{index}")
|
print(shuffled_solution)
|
||||||
async def get_letter(
|
|
||||||
index: int,
|
|
||||||
cfg: Config = Depends(get_config),
|
|
||||||
) -> str:
|
|
||||||
return (await shuffle(cfg.puzzle.solution))[index]
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/date")
|
@router.get("/date")
|
||||||
async def get_date() -> str:
|
async def get_date() -> str:
|
||||||
|
"""
|
||||||
|
Aktuelles Server-Datum
|
||||||
|
"""
|
||||||
|
|
||||||
return date.today().isoformat()
|
return date.today().isoformat()
|
||||||
|
|
||||||
|
|
||||||
@router.get("/visible_days")
|
@router.get("/visible_days")
|
||||||
async def get_visible_days() -> int:
|
async def get_visible_days() -> int:
|
||||||
today = date.today()
|
"""
|
||||||
|
Sichtbare Türchen
|
||||||
|
"""
|
||||||
|
|
||||||
if today.month == 12:
|
return await user_visible_doors()
|
||||||
return today.day
|
|
||||||
|
|
||||||
if today.month in (1, 2, 3):
|
|
||||||
return 24
|
|
||||||
|
|
||||||
return 0
|
|
||||||
|
|
||||||
|
|
||||||
async def user_can_view(
|
@router.get("/part/{day}")
|
||||||
index: int,
|
async def get_part_for_day(
|
||||||
) -> bool:
|
part: str = Depends(get_part),
|
||||||
return index < await get_visible_days()
|
) -> str:
|
||||||
|
"""
|
||||||
|
Heutiger Lösungsteil
|
||||||
|
"""
|
||||||
|
|
||||||
|
return part
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
"/image/{index}",
|
"/image/{day}",
|
||||||
response_class=StreamingResponse,
|
response_class=StreamingResponse,
|
||||||
)
|
)
|
||||||
async def get_image_for_day(
|
async def get_image_for_day(
|
||||||
image: AdventImage = Depends(get_image),
|
image: Image.Image = Depends(get_image),
|
||||||
can_view: bool = Depends(user_can_view),
|
can_view: bool = Depends(user_can_view_door),
|
||||||
is_admin: bool = Depends(user_is_admin),
|
is_admin: bool = Depends(user_is_admin),
|
||||||
) -> StreamingResponse:
|
) -> StreamingResponse:
|
||||||
"""
|
"""
|
||||||
|
@ -66,4 +66,4 @@ async def get_image_for_day(
|
||||||
if not (can_view or is_admin):
|
if not (can_view or is_admin):
|
||||||
raise HTTPException(status.HTTP_401_UNAUTHORIZED, "Wie unhöflich!!!")
|
raise HTTPException(status.HTTP_401_UNAUTHORIZED, "Wie unhöflich!!!")
|
||||||
|
|
||||||
return await api_return_image(image.img)
|
return await api_return_image(image)
|
||||||
|
|
|
@ -1,16 +1,9 @@
|
||||||
from fastapi import APIRouter, Depends
|
from fastapi import APIRouter, Depends
|
||||||
from fastapi.responses import StreamingResponse
|
from fastapi.responses import StreamingResponse
|
||||||
from PIL import Image
|
|
||||||
|
|
||||||
from ..calendar_config import (
|
from ..core.calendar_config import CalendarConfig, DoorsSaved, get_calendar_config
|
||||||
CalendarConfig,
|
from ..core.config import Config, get_config
|
||||||
DoorsSaved,
|
from ..core.image_helpers import api_return_image, load_image
|
||||||
get_calendar_config,
|
|
||||||
set_calendar_config,
|
|
||||||
)
|
|
||||||
from ..config import Config, get_config
|
|
||||||
from ..dav_common import dav_get_file
|
|
||||||
from ._misc import api_return_image
|
|
||||||
|
|
||||||
router = APIRouter(prefix="/general", tags=["general"])
|
router = APIRouter(prefix="/general", tags=["general"])
|
||||||
|
|
||||||
|
@ -26,9 +19,7 @@ async def get_image_for_day(
|
||||||
Hintergrundbild laden
|
Hintergrundbild laden
|
||||||
"""
|
"""
|
||||||
|
|
||||||
return await api_return_image(
|
return await api_return_image(await load_image(f"files/{cal_cfg.background}"))
|
||||||
Image.open(await dav_get_file(f"files/{cal_cfg.background}"))
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/doors")
|
@router.get("/doors")
|
||||||
|
@ -56,4 +47,4 @@ async def put_doors(
|
||||||
doors,
|
doors,
|
||||||
key=lambda door: door.day,
|
key=lambda door: door.day,
|
||||||
)
|
)
|
||||||
await set_calendar_config(cfg, cal_cfg)
|
await cal_cfg.change(cfg)
|
||||||
|
|
|
@ -1,32 +1,8 @@
|
||||||
import secrets
|
from fastapi import APIRouter, Depends
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, HTTPException, status
|
from ._security import require_admin
|
||||||
from fastapi.security import HTTPBasic, HTTPBasicCredentials
|
|
||||||
|
|
||||||
from ..config import Config, get_config
|
|
||||||
|
|
||||||
router = APIRouter(prefix="/user", tags=["user"])
|
router = APIRouter(prefix="/user", tags=["user"])
|
||||||
security = HTTPBasic()
|
|
||||||
|
|
||||||
|
|
||||||
async def user_is_admin(
|
|
||||||
credentials: HTTPBasicCredentials = Depends(security),
|
|
||||||
config: Config = Depends(get_config),
|
|
||||||
) -> bool:
|
|
||||||
username_correct = secrets.compare_digest(credentials.username, config.admin.name)
|
|
||||||
|
|
||||||
password_correct = secrets.compare_digest(
|
|
||||||
credentials.password, config.admin.password
|
|
||||||
)
|
|
||||||
|
|
||||||
return username_correct and password_correct
|
|
||||||
|
|
||||||
|
|
||||||
async def require_admin(
|
|
||||||
is_admin: bool = Depends(user_is_admin),
|
|
||||||
) -> None:
|
|
||||||
if not is_admin:
|
|
||||||
raise HTTPException(status.HTTP_401_UNAUTHORIZED)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get("/admin")
|
@router.get("/admin")
|
||||||
|
|
Loading…
Reference in a new issue