mirror of
https://github.com/kata-containers/kata-containers.git
synced 2025-04-28 11:44:38 +00:00
2c89828749
Follow-up to #9898 and final PR of this set. This implements the actual deletion logic. Signed-off-by: Aurélien Bombo <abombo@microsoft.com>
62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
# Copyright (c) Microsoft Corporation.
|
|
# Adapted from original code by Jeremi Piotrowski <jpiotrowski@microsoft.com>.
|
|
from datetime import datetime, timedelta, timezone
|
|
import os
|
|
|
|
from azure.identity import AzureCliCredential
|
|
from azure.mgmt.resource import ResourceManagementClient
|
|
|
|
# Current date and time in UTC.
|
|
utc_now = datetime.now(timezone.utc)
|
|
# Cleanup time delta passed in the environment, i.e. how much time to
|
|
# wait before automatically cleaning up a resource.
|
|
cleanup_after = timedelta(hours=int(os.environ['CLEANUP_AFTER_HOURS']))
|
|
# Time considered as the cutoff to clean up a resource: if it was
|
|
# created before this time, it will be deleted.
|
|
cleanup_cutoff_time = utc_now - cleanup_after
|
|
|
|
print(f"Current time: {utc_now}")
|
|
print(f"Cleanup time delta: {cleanup_after}")
|
|
print(f"Will clean up resources created before {cleanup_cutoff_time}")
|
|
|
|
credential = AzureCliCredential()
|
|
subscription_id = os.environ['AZ_SUBSCRIPTION_ID']
|
|
client = ResourceManagementClient(credential, subscription_id)
|
|
resources = client.resources.list(expand='createdTime')
|
|
|
|
print("Processsing resources...")
|
|
num_deleted = 0
|
|
|
|
for resource in resources:
|
|
# Ignore resources that aren't AKS clusters.
|
|
if resource.type != 'Microsoft.ContainerService/managedClusters':
|
|
continue
|
|
|
|
# Ignore resources created after the cutoff time (i.e. less than
|
|
# `cleanup_after` time ago).
|
|
if resource.created_time > cleanup_cutoff_time:
|
|
print(f"{resource.name}: created at {resource.created_time}, after delta cutoff, ignored")
|
|
continue
|
|
|
|
print(f"{resource.name}: created at {resource.created_time}, before delta cutoff, deleting...")
|
|
|
|
# A resource ID looks like this:
|
|
# /subscriptions/(subscriptionId)/resourceGroups/(resourceGroupName)/providers/(resourceProviderNamespace)/(resourceType)/(resourceName)
|
|
rg_id, _, _ = resource.id.partition("/providers/")
|
|
_, _, rg_name = rg_id.partition("/resourceGroups/")
|
|
rg_resources = client.resources.list_by_resource_group(rg_name)
|
|
|
|
# XXX DANGER ZONE: Delete the resource. If it's the only resource
|
|
# in its resource group, the entire resource group is deleted.
|
|
|
|
if len(list(rg_resources)) == 1:
|
|
client.resource_groups.begin_delete(rg_name)
|
|
print(f"{resource.name}: deleted resource group")
|
|
else:
|
|
client.resources.begin_delete_by_id(resource.id)
|
|
print(f"{resource.name}: deleted resource")
|
|
|
|
num_deleted += 1
|
|
|
|
print(f"Deleted {num_deleted} resources")
|