from .service import Service
from aws.base import Base


class Cluster(Base):
    def __init__(self, client, cluster_arn, launch_type='FARGATE'):
        super().__init__(client)
        self.cluster_arn = cluster_arn
        self.launch_type = launch_type

    def get_cluster_name(self) -> str:
        return self.cluster_arn.split('/')[-1]

    def get_services(self) -> list:
        client = self.get_client()
        paginator = client.get_paginator('list_services')
        page_iterator = paginator.paginate(cluster=self.cluster_arn,
                                           launchType=self.launch_type)
        return [Service(client, self.cluster_arn, item) for sublist in
                page_iterator for item in
                sublist.get('serviceArns', [])]