infix/test/infamy/container.py

"""Manage Infix containers"""

class Container:
    """Helper methods"""

    def __init__(self, target):
        self.system = target

    def _find(self, name):
        oper = self.system.get_data("/infix-containers:containers/container")
        if not oper:
            return None

        for container in oper["containers"]["container"]:
            if container["name"] == name:
                return container
        return None

    def exists(self, name):
        """Check if container {name} runs on target."""
        container = self._find(name)
        if not container:
            return False
        return True

    def running(self, name):
        """Check if container {name} exists and is running."""
        container = self._find(name)
        if container and container["running"]:
            return True
        return False

    def action(self, name, act):
        """Call NETCONF action 'type' on container 'name'"""
        return self.system.call_action_dict("infix-containers", {
            "containers": {
                "container": [
                    {
                        "name": f"{name}",
                        f"{act}": {}
                    }
                ]
            }
        })