117 lines
3.1 KiB
Python
117 lines
3.1 KiB
Python
from cinderclient import client as cc
|
|
from heatclient import client as hc
|
|
from keystoneclient.v2_0 import client as kc
|
|
from novaclient.v1_1 import client as nc
|
|
import os
|
|
import re
|
|
import sys
|
|
import time
|
|
|
|
CONF = dict()
|
|
keys = ["os_username", "os_password", "os_auth_url", "os_tenant_name",
|
|
"os_image_endpoint"]
|
|
|
|
|
|
def load_conf():
|
|
for item in keys:
|
|
CONF[item] = os.environ[item]
|
|
|
|
|
|
def get_nova_client():
|
|
return nc.Client(username=CONF["os_username"],
|
|
api_key=CONF["os_password"],
|
|
auth_url=CONF["os_auth_url"],
|
|
project_id=CONF["os_tenant_name"])
|
|
|
|
|
|
def get_auth_token():
|
|
keystone = kc.Client(username=CONF["os_username"],
|
|
password=CONF["os_password"],
|
|
tenant_name=CONF["os_tenant_name"],
|
|
auth_url=CONF["os_auth_url"])
|
|
return keystone.auth_token
|
|
|
|
|
|
def get_heat_client():
|
|
return hc.Client('1',
|
|
endpoint=CONF["os_image_endpoint"],
|
|
token=get_auth_token())
|
|
|
|
|
|
def get_cinder_client():
|
|
return cc.Client('1',
|
|
CONF["os_username"],
|
|
CONF["os_password"],
|
|
CONF["os_tenant_name"],
|
|
CONF["os_auth_url"])
|
|
|
|
|
|
def cleanup_heat():
|
|
current_name = sys.argv[2]
|
|
client = get_heat_client()
|
|
stacks = client.stacks.list()
|
|
name_regex = re.compile(current_name)
|
|
deleted_stacks = []
|
|
|
|
for stack in stacks:
|
|
if name_regex.match(stack.stack_name):
|
|
deleted_stacks.append(stack.stack_name)
|
|
print(stack.stack_name)
|
|
client.stacks.delete(stack.stack_name)
|
|
if not deleted_stacks:
|
|
return
|
|
else:
|
|
# Let Heat delete stacks
|
|
time.sleep(60)
|
|
|
|
stacks = client.stacks.list()
|
|
for stack in stacks:
|
|
if stack.stack_name in deleted_stacks:
|
|
# Resource cleanup is required
|
|
print("At least one stack wasn't deleted!")
|
|
print("Performing resources cleanup...")
|
|
cleanup()
|
|
|
|
|
|
def cleanup():
|
|
client = get_nova_client()
|
|
cinder_client = get_cinder_client()
|
|
servers = client.servers.list()
|
|
volumes = cinder_client.volumes.list()
|
|
secgroups = client.security_groups.list()
|
|
current_name = sys.argv[2]
|
|
name_regex = re.compile(current_name)
|
|
|
|
for server in servers:
|
|
if name_regex.match(server.name):
|
|
print(server.name)
|
|
fl_ips = client.floating_ips.findall(instance_id=server.id)
|
|
for fl_ip in fl_ips:
|
|
client.floating_ips.delete(fl_ip.id)
|
|
client.servers.delete(server.id)
|
|
|
|
time.sleep(20)
|
|
for volume in volumes:
|
|
if name_regex.match(volume.display_name):
|
|
print(volume.display_name)
|
|
volume.delete()
|
|
|
|
for group in secgroups:
|
|
if name_regex.match(group.name):
|
|
print(group.name)
|
|
group.delete()
|
|
|
|
|
|
def main(argv):
|
|
load_conf()
|
|
|
|
if "cleanup" in argv:
|
|
cleanup()
|
|
|
|
if "cleanup-heat" in argv:
|
|
cleanup_heat()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1:])
|