43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from dataclasses import dataclass
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from app.core.config import Settings
|
|
|
|
DEFAULT_LIBLIB_TEMPLATE_UUID = "5d7e67009b344550bc1aa6ccbfa1d7f4"
|
|
DEFAULT_IMAGE_PROVIDER = "liblib"
|
|
DEFAULT_IMAGE_STYLE = "watercolor"
|
|
DEFAULT_IMAGE_SIZE = "1280x720"
|
|
DEFAULT_POLL_INTERVAL_SECONDS = 5
|
|
DEFAULT_MAX_ATTEMPTS = 60
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class MemoirImageSettings:
|
|
enabled: bool = False
|
|
provider: str = DEFAULT_IMAGE_PROVIDER
|
|
default_style: str = DEFAULT_IMAGE_STYLE
|
|
default_size: str = DEFAULT_IMAGE_SIZE
|
|
poll_interval_seconds: int = DEFAULT_POLL_INTERVAL_SECONDS
|
|
max_attempts: int = DEFAULT_MAX_ATTEMPTS
|
|
liblib_template_uuid: str = DEFAULT_LIBLIB_TEMPLATE_UUID
|
|
|
|
@classmethod
|
|
def from_settings(cls, settings: "Settings") -> "MemoirImageSettings":
|
|
s = settings
|
|
return cls(
|
|
enabled=bool(s.memoir_image_enabled),
|
|
provider=s.memoir_image_provider or DEFAULT_IMAGE_PROVIDER,
|
|
default_style=s.memoir_image_style_default or DEFAULT_IMAGE_STYLE,
|
|
default_size=s.memoir_image_size_default or DEFAULT_IMAGE_SIZE,
|
|
poll_interval_seconds=s.memoir_image_poll_interval,
|
|
max_attempts=s.memoir_image_max_attempts,
|
|
liblib_template_uuid=s.liblib_template_uuid or DEFAULT_LIBLIB_TEMPLATE_UUID,
|
|
)
|
|
|
|
@classmethod
|
|
def from_env(cls) -> "MemoirImageSettings":
|
|
from app.core.config import settings as _s
|
|
|
|
return cls.from_settings(_s)
|