from aws.base import Base


class Task(Base):
    def __init__(self, client, cluster_arn, task_arn):
        super().__init__(client)
        self.cluster_arn = cluster_arn
        self.task_arn = task_arn

    def list_containers(self) -> list:
        client = self.get_client().describe_tasks(cluster=self.cluster_arn,
                                                  tasks=[self.task_arn])
        tasks = client.get('tasks', [])

        if len(tasks) == 0:
            return []

        return tasks[0]['containers']