# models.py
from pydantic import BaseModel, validator
from typing import Optional
from .utils.logger import SupersetLogger

class SupersetConfig(BaseModel):
    base_url: str
    auth: dict
    verify_ssl: bool = True
    timeout: int = 30
    logger: Optional[SupersetLogger] = None

    class Config:
        arbitrary_types_allowed = True  # Разрешаем произвольные типы

class DatabaseConfig(BaseModel):
    database_config: dict
    logger: Optional[SupersetLogger] = None

    class Config:
        arbitrary_types_allowed = True