compose-viz/compose_viz/service.py

from typing import List, Optional

from compose_viz.extends import Extends


class Service:
    def __init__(
        self,
        name: str,
        image: Optional[str] = None,
        ports: List[str] = [],
        networks: List[str] = [],
        volumes: List[str] = [],
        depends_on: List[str] = [],
        links: List[str] = [],
        extends: Optional[Extends] = None,
    ) -> None:
        self._name = name

        if image is None and extends is None:
            raise AttributeError(f"Both image and extends are not defined in service '{name}', aborting.")

        if image is not None and extends is not None:
            raise AttributeError(f"Only one of image and extends can be defined in service '{name}', aborting.")

        self._image = image
        self._ports = ports
        self._networks = networks
        self._volumes = volumes
        self._depends_on = depends_on
        self._links = links
        self._extends = extends

    @property
    def name(self):
        return self._name

    @property
    def image(self):
        return self._image

    @property
    def ports(self):
        return self._ports

    @property
    def networks(self):
        return self._networks

    @property
    def volumes(self):
        return self._volumes

    @property
    def depends_on(self):
        return self._depends_on

    @property
    def links(self):
        return self._links

    @property
    def extends(self):
        return self._extends