e814ee48e9
- stacks are deleted using Heat - in case of deletion failure resources are cleaned up using direct mechanism Change-Id: Iac41d9859f2e195fa46c5a36cd8dd5862b730251
115 lines
3.1 KiB
Python
115 lines
3.1 KiB
Python
import os
|
|
import sys, getopt
|
|
import socket
|
|
import time
|
|
import ConfigParser
|
|
from novaclient.v1_1 import client as nc
|
|
import requests
|
|
from random import randint
|
|
from keystoneclient.v2_0 import client as kc
|
|
from heatclient import client as hc
|
|
from heatclient import exc as hc_exc
|
|
from cinderclient import client as cc
|
|
import re
|
|
|
|
CONF = dict()
|
|
CONF_FILE = '/etc/jenkins_jobs/credentials.conf'
|
|
|
|
def load_conf():
|
|
# load credentials and configs
|
|
config = ConfigParser.ConfigParser()
|
|
config.readfp(open(CONF_FILE))
|
|
for key, val in config.items("default"):
|
|
CONF[key] = val
|
|
|
|
for env_item in os.environ:
|
|
CONF[env_item] = os.environ[env_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:])
|