Spaces:
Runtime error
Runtime error
| from typing import Optional, Any, Dict, List, Union | |
| from phi.aws.api_client import AwsApiClient | |
| from phi.aws.resource.base import AwsResource | |
| from phi.aws.resource.reference import AwsReference | |
| from phi.aws.resource.cloudformation.stack import CloudFormationStack | |
| from phi.cli.console import print_info | |
| from phi.utils.log import logger | |
| class DbSubnetGroup(AwsResource): | |
| """ | |
| Reference: | |
| - https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/rds.html#RDS.Client.create_db_subnet_group | |
| Creates a new DB subnet group. DB subnet groups must contain at least one subnet in at least | |
| two AZs in the Amazon Web Services Region. | |
| """ | |
| resource_type: Optional[str] = "DbSubnetGroup" | |
| service_name: str = "rds" | |
| # The name for the DB subnet group. This value is stored as a lowercase string. | |
| # Constraints: | |
| # Must contain no more than 255 letters, numbers, periods, underscores, spaces, or hyphens. | |
| # Must not be default. | |
| # First character must be a letter. | |
| # Example: mydbsubnetgroup | |
| name: str | |
| # The description for the DB subnet group. | |
| description: Optional[str] = None | |
| # The EC2 Subnet IDs for the DB subnet group. | |
| subnet_ids: Optional[Union[List[str], AwsReference]] = None | |
| # Get Subnet IDs from a VPC CloudFormationStack | |
| # First gets private subnets from the vpc stack, then public subnets | |
| vpc_stack: Optional[CloudFormationStack] = None | |
| # Tags to assign to the DB subnet group. | |
| tags: Optional[List[Dict[str, str]]] = None | |
| def get_subnet_ids(self, aws_client: AwsApiClient) -> List[str]: | |
| """Returns the subnet_ids for the DbSubnetGroup | |
| Args: | |
| aws_client: The AwsApiClient for the current cluster | |
| """ | |
| subnet_ids = [] | |
| if self.subnet_ids is not None: | |
| if isinstance(self.subnet_ids, list): | |
| logger.debug("Getting subnet_ids from list") | |
| subnet_ids = self.subnet_ids | |
| elif isinstance(self.subnet_ids, AwsReference): | |
| logger.debug("Getting subnet_ids from reference") | |
| subnet_ids = self.subnet_ids.get_reference(aws_client=aws_client) | |
| if len(subnet_ids) == 0 and self.vpc_stack is not None: | |
| logger.debug("Getting private subnet_ids from vpc stack") | |
| private_subnet_ids = self.vpc_stack.get_private_subnets(aws_client=aws_client) | |
| if private_subnet_ids is not None: | |
| subnet_ids.extend(private_subnet_ids) | |
| if len(subnet_ids) == 0: | |
| logger.debug("Getting public subnet_ids from vpc stack") | |
| public_subnet_ids = self.vpc_stack.get_public_subnets(aws_client=aws_client) | |
| if public_subnet_ids is not None: | |
| subnet_ids.extend(public_subnet_ids) | |
| return subnet_ids | |
| def _create(self, aws_client: AwsApiClient) -> bool: | |
| """Creates the DbSubnetGroup | |
| Args: | |
| aws_client: The AwsApiClient for the current cluster | |
| """ | |
| print_info(f"Creating {self.get_resource_type()}: {self.get_resource_name()}") | |
| try: | |
| # Get subnet_ids | |
| subnet_ids = self.get_subnet_ids(aws_client=aws_client) | |
| # create a dict of args which are not null, otherwise aws type validation fails | |
| not_null_args: Dict[str, Any] = {} | |
| if self.tags: | |
| not_null_args["Tags"] = self.tags | |
| # Create DbSubnetGroup | |
| service_client = self.get_service_client(aws_client) | |
| create_response = service_client.create_db_subnet_group( | |
| DBSubnetGroupName=self.name, | |
| DBSubnetGroupDescription=self.description or f"Created for {self.name}", | |
| SubnetIds=subnet_ids, | |
| **not_null_args, | |
| ) | |
| logger.debug(f"create_response type: {type(create_response)}") | |
| logger.debug(f"create_response: {create_response}") | |
| self.active_resource = create_response.get("DBSubnetGroup", None) | |
| if self.active_resource is not None: | |
| print_info(f"{self.get_resource_type()}: {self.get_resource_name()} created") | |
| logger.debug(f"DbSubnetGroup: {self.active_resource}") | |
| return True | |
| except Exception as e: | |
| logger.error(f"{self.get_resource_type()} could not be created.") | |
| logger.error(e) | |
| return False | |
| def _read(self, aws_client: AwsApiClient) -> Optional[Any]: | |
| """Returns the DbSubnetGroup | |
| Args: | |
| aws_client: The AwsApiClient for the current cluster | |
| """ | |
| from botocore.exceptions import ClientError | |
| logger.debug(f"Reading {self.get_resource_type()}: {self.get_resource_name()}") | |
| try: | |
| service_client = self.get_service_client(aws_client) | |
| describe_response = service_client.describe_db_subnet_groups(DBSubnetGroupName=self.name) | |
| logger.debug(f"describe_response type: {type(describe_response)}") | |
| logger.debug(f"describe_response: {describe_response}") | |
| db_subnet_group_list = describe_response.get("DBSubnetGroups", None) | |
| if db_subnet_group_list is not None and isinstance(db_subnet_group_list, list): | |
| for _db_subnet_group in db_subnet_group_list: | |
| _db_sg_name = _db_subnet_group.get("DBSubnetGroupName", None) | |
| if _db_sg_name == self.name: | |
| self.active_resource = _db_subnet_group | |
| break | |
| if self.active_resource is None: | |
| logger.debug(f"No {self.get_resource_type()} found") | |
| return None | |
| logger.debug(f"DbSubnetGroup: {self.active_resource}") | |
| except ClientError as ce: | |
| logger.debug(f"ClientError: {ce}") | |
| except Exception as e: | |
| logger.error(f"Error reading {self.get_resource_type()}.") | |
| logger.error(e) | |
| return self.active_resource | |
| def _delete(self, aws_client: AwsApiClient) -> bool: | |
| """Deletes the DbSubnetGroup | |
| Args: | |
| aws_client: The AwsApiClient for the current cluster | |
| """ | |
| print_info(f"Deleting {self.get_resource_type()}: {self.get_resource_name()}") | |
| try: | |
| service_client = self.get_service_client(aws_client) | |
| self.active_resource = None | |
| delete_response = service_client.delete_db_subnet_group(DBSubnetGroupName=self.name) | |
| logger.debug(f"delete_response: {delete_response}") | |
| return True | |
| except Exception as e: | |
| logger.error(f"{self.get_resource_type()} could not be deleted.") | |
| logger.error("Please try again or delete resources manually.") | |
| logger.error(e) | |
| return False | |
| def _update(self, aws_client: AwsApiClient) -> bool: | |
| """Updates the DbSubnetGroup | |
| Args: | |
| aws_client: The AwsApiClient for the current cluster | |
| """ | |
| print_info(f"Updating {self.get_resource_type()}: {self.get_resource_name()}") | |
| try: | |
| # Get subnet_ids | |
| subnet_ids = self.get_subnet_ids(aws_client=aws_client) | |
| # Update DbSubnetGroup | |
| service_client = self.get_service_client(aws_client) | |
| update_response = service_client.modify_db_subnet_group( | |
| DBSubnetGroupName=self.name, | |
| DBSubnetGroupDescription=self.description or f"Created for {self.name}", | |
| SubnetIds=subnet_ids, | |
| ) | |
| logger.debug(f"update_response: {update_response}") | |
| self.active_resource = update_response.get("DBSubnetGroup", None) | |
| if self.active_resource is not None: | |
| print_info(f"{self.get_resource_type()}: {self.get_resource_name()} updated") | |
| return True | |
| except Exception as e: | |
| logger.error(f"{self.get_resource_type()} could not be updated.") | |
| logger.error("Please try again or update resources manually.") | |
| logger.error(e) | |
| return False | |