diff --git a/scone/default/recipes/docker.py b/scone/default/recipes/docker.py index be0f4c2..3d7b8b2 100644 --- a/scone/default/recipes/docker.py +++ b/scone/default/recipes/docker.py @@ -28,7 +28,7 @@ class DockerContainer(Recipe): async def cook(self, kitchen: Kitchen) -> None: kitchen.get_dependency_tracker() - current_state = await kitchen.ut1(DockerContainerState(self.name)) + current_state = ContainerState(await kitchen.ut1(DockerContainerState(self.name))) if current_state == ContainerState.NOTFOUND: await kitchen.ut1areq( diff --git a/scone/default/utensils/docker_utensils.py b/scone/default/utensils/docker_utensils.py index 95ce197..e6384df 100644 --- a/scone/default/utensils/docker_utensils.py +++ b/scone/default/utensils/docker_utensils.py @@ -51,16 +51,16 @@ class DockerContainerState(Utensil): container: Container if self.name == container.name: if container.status == "running": - await channel.send(ContainerState.RUNNING) + await channel.send(ContainerState.RUNNING.value) elif container.status == "exited": - await channel.send(ContainerState.EXITED) + await channel.send(ContainerState.EXITED.value) elif container.status == "restarting": - await channel.send(ContainerState.RESTARTING) + await channel.send(ContainerState.RESTARTING.value) else: raise ValueError(f"Unknown container status: {container.status}") break else: - await channel.send(ContainerState.NOTFOUND) + await channel.send(ContainerState.NOTFOUND.value) @attr.s(auto_attribs=True)