Implement Python helpers for creating fio resources
Related-PROD: PROD-37187
Change-Id: Iee1d85955fe10876a85da99025211946c2135178
diff --git a/fio/fio_cleanup.py b/fio/fio_cleanup.py
new file mode 100644
index 0000000..ccbe38e
--- /dev/null
+++ b/fio/fio_cleanup.py
@@ -0,0 +1,81 @@
+import connection as conn
+from openstack.exceptions import ResourceFailure
+from typing import Final
+
+
+compute = conn.cloud.compute
+network = conn.cloud.network
+volume = conn.cloud.volume
+
+CLIENT_NAME_MASK: Final[str] = conn.FIO_CLIENT_NAME_MASK
+FLAVOR_NAME: Final[str] = conn.FIO_FLAVOR_NAME
+KEYPAIR_NAME: Final[str] = conn.FIO_KEYPAIR_NAME
+SG_NAME: Final[str] = conn.FIO_SG_NAME
+
+ROUTER_NAME: Final[str] = conn.FIO_ROUTER_NAME
+NET_NAME: Final[str] = conn.FIO_NET_NAME
+
+
+if __name__ == "__main__":
+ # Find fio clients and server
+ vms = compute.servers(name=CLIENT_NAME_MASK)
+ for vm in vms:
+ attachments = compute.volume_attachments(vm)
+ # Delete fio volume attachment (and any other attachments
+ # that the VM could have)
+ # Delete the volume and the server
+ for att in attachments:
+ vol_id = att.volume_id
+ vol = volume.get_volume(vol_id)
+ try:
+ conn.detach_volume(vm, vol)
+ print(
+ f"'{vol.id}' volume has been detached from fio '{vm.name}'"
+ " server.")
+ conn.delete_volume(vol)
+ print(f"'{vol.id}' volume has been deleted.")
+ conn.delete_server(vm)
+ print(f"'{vm.name}' server has been deleted.")
+ except ResourceFailure as e:
+ print(
+ f"Cleanup of '{vm.id}' with volume '{vol.id}' attached "
+ f"failed with '{e.message}' error.")
+ conn.delete_volume(vol)
+ continue
+
+ # Remove ports from fio router (including external GW)
+ router = network.find_router(ROUTER_NAME)
+ if router:
+ network.update_router(router.id, external_gateway_info={})
+ print("Externa GW port has been deleted from fio router.")
+ router_ports = network.ports(device_id=router.id)
+ for p in router_ports:
+ network.remove_interface_from_router(router.id, port_id=p.id)
+ print(f"'{p.id}' port has been deleted from fio router.")
+
+ # Delete fio network topology
+ net = network.find_network(NET_NAME)
+ if net:
+ network.delete_network(net.id)
+ print(f"fio '{net.id}' network has been deleted.")
+ if router:
+ network.delete_router(router.id)
+ print(f"fio '{router.id}' router has been deleted.")
+
+ # Delete fio flavor
+ flavor = compute.find_flavor(FLAVOR_NAME)
+ if flavor:
+ compute.delete_flavor(flavor.id)
+ print(f"fio '{flavor.id}' flavor has been deleted.")
+
+ # # Delete fio keypair
+ kp = compute.find_keypair(KEYPAIR_NAME)
+ if kp:
+ compute.delete_keypair(kp)
+ print(f"fio '{kp.id}' keypair has been deleted.")
+
+ # Delete fio security group
+ sg = network.find_security_group(SG_NAME)
+ if sg:
+ network.delete_security_group(sg)
+ print(f"fio '{sg.id}' security group has been deleted.")