from typing import Optional

from pydantic_settings import BaseSettings


class Settings(BaseSettings):
    # SFTP Settings (Optional)
    SFTP_HOST: Optional[str] = None
    SFTP_USER: Optional[str] = None
    SFTP_KEY_PATH: Optional[str] = "~/.ssh/id_rsa"
    SFTP_PATH: Optional[str] = None
    SFTP_PORT: int = 22
    TRAINING_LOG_REMOTE_PATH: Optional[str] = None
    TRAINING_CONFIG_LOCAL_PATH: Optional[str] = None

    # Local Settings (Optional)
    LOCAL_PATH: Optional[str] = None
    TRAINING_LOG_LOCAL_PATH: Optional[str] = None
    TRAINING_CONFIG_REMOTE_PATH: Optional[str] = None

    # API Settings
    PROJECT_NAME: str = "Training Monitor"
    API_VER_STR: str = "/api/v1"

    class Config:
        env_file = ".env"
        case_sensitive = True


settings = Settings()