from dataclasses import dataclass, field
from typing import Optional, Dict

from frostfs_sdk.models.dto.container import Container
from frostfs_sdk.models.dto.session_token import SessionToken
from frostfs_sdk.client.parameters.wait import PrmWait


@dataclass(frozen=True)
class PrmContainerCreate:
    container: Container
    wait_params: Optional[PrmWait] = None
    session_token: Optional[SessionToken] = None
    x_headers: Dict[str, str] = field(default_factory=dict)

    def __post_init__(self):
        if self.wait_params is None:
            object.__setattr__(self, 'wait_params', PrmWait())