adt.py

@dataclass
class InstanceGroups:
    master_nodes: Ec2InstancesGroup
    core_nodes: Ec2InstancesGroup


@dataclass
class InstanceFleets:
    config: list[dict[str, any]]
    subnets: set[KnownEc2Subnets]


Instances = InstanceGroups | InstanceFleets

match instances:
    case InstanceGroups(master_nodes, core_nodes):
        instances_config = common_instances_config | {
            'InstanceGroups': [
                {
                    'Name': "Master nodes",
                    'InstanceRole': 'MASTER',
                } | master_nodes,
                {
                    'Name': "Slave nodes",
                    'InstanceRole': 'CORE',
                } | core_nodes,
            ],
            'Ec2SubnetId': KnownEc2Subnets.InternalPrivateSubnet1a.value,
        }
    case InstanceFleets(config, subnets):
        instances_config = common_instances_config | {
            'InstanceFleets': config,
            'Ec2SubnetIds': [subnet.value for subnet in subnets],
        }