settings.py

import sys
from enum import Enum

from loguru import logger
from minio import Minio
from pydantic import Field
from pydantic_settings import BaseSettings

logger.add(sys.stdout, format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {message}")


class StorageType(str, Enum):
    MINIO = "minio"
    LOCAL = "local"


class Settings(BaseSettings):
    IMAGE_DIR: str = Field(default="images", description="Local image dir")
    MINIO_ACCESS_KEY: str
    MINIO_BUCKET: str
    MINIO_SECRET_KEY: str
    MINIO_ENDPOINT: str
    MINIO_SECURE: bool = Field(default=False, description="Set to True if your minio endpoint supports https")
    SERVER_URL: str = Field(description="The url that point to this server, it will be used for images' url prefix")
    STORAGE_TYPE: str = Field(default="local")
    XOSCAR_ACTORS: int = Field(default=4, description="How many processes should xoscar spawn")
    XOSCAR_ADDRESS: str = Field(default="localhost:7777")

    class Config:
        env_file = ".env"
        env_file_encoding = "utf-8"

    @property
    def minio_endpoint_url(self):
        url_prefix = "http://"
        if self.MINIO_SECURE is True:
            url_prefix = "https://"
        return f"{url_prefix}{self.MINIO_ENDPOINT}"


settings = Settings()

minio_client = Minio(
    endpoint=settings.MINIO_ENDPOINT,
    access_key=settings.MINIO_ACCESS_KEY,
    secret_key=settings.MINIO_SECRET_KEY,
    secure=settings.MINIO_SECURE,
)