from dataclasses import dataclass
from typing import NotRequired

from docker_compose.compose.service_yaml_read import ServiceYamlReadData
from docker_compose.yaml import YamlWrapper


class ServiceYamlWriteData(ServiceYamlReadData):
    container_name: NotRequired[str]
    restart: NotRequired[str]


@dataclass(frozen=True, slots=True)
class ServiceYamlWrite(YamlWrapper[ServiceYamlWriteData]):
    pass