class NukeEks:
"""Abstract eks nuke in a class."""
def __init__(self, region_name=None) -> None:
"""Initialize eks nuke."""
self.eks = AwsClient().connect("eks", region_name)
try:
self.eks.list_clusters()
except EndpointConnectionError:
print("eks resource is not available in this aws region")
return
def nuke(self, older_than_seconds: float) -> None:
"""EKS cluster deleting function.
Deleting all EKS clusters with a timestamp greater than
older_than_seconds.
:param int older_than_seconds:
The timestamp in seconds used from which the aws
resource will be deleted
"""
for cluster in self.list_clusters(older_than_seconds):
try:
self.eks.delete_cluster(name=cluster)
print("Nuke EKS Cluster{0}".format(cluster))
except ClientError as exc:
nuke_exceptions("eks cluster", cluster, exc)
def list_clusters(self, time_delete: float) -> Iterator[str]:
"""EKS cluster list function.
List the names of all EKS clusters with a timestamp lower than
time_delete.
:param int time_delete:
Timestamp in seconds used for filter EKS clusters
:yield Iterator[str]:
EKS cluster names
"""
response = self.eks.list_clusters()
for kube in response["clusters"]:
k8s = self.eks.describe_cluster(name=kube)
if k8s["cluster"]["createdAt"].timestamp() < time_delete:
yield kube
class NukeKafka:
"""Abstract kafka nuke in a class."""
def __init__(self, region_name=None) -> None:
"""Initialize kafka nuke."""
self.kafka = AwsClient().connect("kafka", region_name)
def nuke(self, older_than_seconds: float) -> None:
"""Kafka deleting function.
Deleting all kafka cluster with a timestamp greater than
older_than_seconds.
:param int older_than_seconds:
The timestamp in seconds used from which the aws resource
will be deleted
"""
for cluster_arn in self.list_cluster(older_than_seconds):
try:
self.kafka.delete_cluster(ClusterArn=cluster_arn)
print("Nuke kafka cluster {0}".format(cluster_arn))
except ClientError as exc:
nuke_exceptions("kafka cluster", cluster_arn, exc)
def list_cluster(self, time_delete: float) -> Iterator[str]:
"""Kafka cluster list function.
List the IDs of all kafka clusters with a timestamp
lower than time_delete.
:param int time_delete:
Timestamp in seconds used for filter ebs volumes
:yield Iterator[str]:
Kafka cluster arm
"""
paginator = self.kafka.get_paginator("list_clusters")
for page in paginator.paginate():
for cluster in page["ClusterInfoList"]:
if cluster["CreationTime"].timestamp() < time_delete:
yield cluster["ClusterArn"]
class NukeRedshift:
"""Abstract redshift nuke in a class."""
def __init__(self, region_name=None) -> None:
"""Initialize redshift nuke."""
self.redshift = AwsClient().connect("redshift", region_name)
try:
self.redshift.describe_clusters()
except EndpointConnectionError:
print("Redshift resource is not available in this aws region")
return
def nuke(self, older_than_seconds) -> None:
"""Redshift resources deleting function.
Deleting all redshift resources with
a timestamp greater than older_than_seconds.
That include:
- clusters
- snapshots
- subnets
- param groups
:param int older_than_seconds:
The timestamp in seconds used from which the aws
resource will be deleted
"""
self.nuke_clusters(older_than_seconds)
self.nuke_snapshots(older_than_seconds)
self.nuke_subnets()
self.nuke_param_groups()
def nuke_clusters(self, time_delete) -> None:
"""Redshift cluster deleting function.
Deleting redshift clusters with a timestamp lower than
time_delete.
:param int older_than_seconds:
The timestamp in seconds used from which the aws resource
will be deleted
"""
for cluster in self.list_clusters(time_delete):
try:
self.redshift.delete_cluster(ClusterIdentifier=cluster,
SkipFinalClusterSnapshot=True)
print("Nuke redshift cluster{0}".format(cluster))
except ClientError as exc:
nuke_exceptions("redshift cluster", cluster, exc)
def nuke_snapshots(self, time_delete) -> None:
"""Redshift snapshot deleting function.
Deleting redshift snapshots with a timestamp lower than
time_delete.
:param int older_than_seconds:
The timestamp in seconds used from which the aws resource
will be deleted
"""
for snapshot in self.list_snapshots(time_delete):
try:
self.redshift.delete_cluster_snapshot(
SnapshotIdentifier=snapshot)
print("Nuke redshift snapshot {0}".format(snapshot))
except ClientError as exc:
nuke_exceptions("redshift snapshot", snapshot, exc)
def nuke_subnets(self) -> None:
"""Redshift subnet deleting function.
Deleting redshift subnets with a timestamp lower than
time_delete.
:param int older_than_seconds:
The timestamp in seconds used from which the aws resource
will be deleted
"""
for subnet in self.list_subnet():
try:
self.redshift.delete_cluster_subnet_group(
ClusterSubnetGroupName=subnet)
print("Nuke redshift subnet {0}".format(subnet))
except ClientError as exc:
nuke_exceptions("redshift subnet", subnet, exc)
def nuke_param_groups(self) -> None:
"""Redshift parameter group deleting function.
Deleting redshift parameter groups with a timestamp lower than
time_delete.
:param int older_than_seconds:
The timestamp in seconds used from which the aws resource
will be deleted
"""
for param in self.list_cluster_params():
try:
self.redshift.delete_cluster_parameter_group(
ParameterGroupName=param)
print("Nuke redshift param {0}".format(param))
except ClientError as exc:
nuke_exceptions("redshift param", param, exc)
def list_clusters(self, time_delete: float) -> Iterator[str]:
"""Redshift cluster list function.
List IDs of all redshift cluster with a timestamp lower than
time_delete.
:param int time_delete:
Timestamp in seconds used for filter redshift cluster
:yield Iterator[str]:
Redshift cluster IDs
"""
paginator = self.redshift.get_paginator("describe_clusters")
for page in paginator.paginate():
for cluster in page["Clusters"]:
if cluster["ClusterCreateTime"].timestamp() < time_delete:
yield cluster["ClusterIdentifier"]
def list_snapshots(self, time_delete: float) -> Iterator[str]:
"""Redshift snapshot list function.
List IDs of all redshift snapshots with a timestamp
lower than time_delete.
:param int time_delete:
Timestamp in seconds used for filter redshift snapshots
:yield Iterator[str]:
Redshift snapshots IDs
"""
paginator = self.redshift.get_paginator("describe_cluster_snapshots")
for page in paginator.paginate():
for snapshot in page["Snapshots"]:
if snapshot["SnapshotCreateTime"].timestamp() < time_delete:
yield snapshot["SnapshotIdentifier"]
def list_subnet(self) -> Iterator[str]:
"""Redshift subnet list function.
:yield Iterator[str]:
Redshift subnet names
"""
paginator = self.redshift.get_paginator(
"describe_cluster_subnet_groups")
for page in paginator.paginate():
for subnet in page["ClusterSubnetGroups"]:
yield subnet["ClusterSubnetGroupName"]
def list_cluster_params(self) -> Iterator[str]:
"""Redshift cluster parameter list function.
:yield Iterator[str]:
Redshift cluster parameter names
"""
paginator = self.redshift.get_paginator(
"describe_cluster_parameter_groups")
for page in paginator.paginate():
for param in page["ParameterGroups"]:
yield param["ParameterGroupName"]
