util: fix tracevol.py to work with volumeNamePrefix in storageclass

tarcevol.py works with a fixed volume name prefix "csi-vol-", this could be changed by the storageclass
and hence tracevol.py should work with the dynamic value rather than the static.

Signed-off-by: Mudit Agarwal <muagarwa@redhat.com>
This commit is contained in:
Mudit Agarwal 2020-07-07 14:57:56 +05:30 committed by mergify[bot]
parent 2ef5729a28
commit 953cfa508a

View File

@ -113,21 +113,22 @@ def format_and_print_tables(arg, pvcs, table_rbd, table_cephfs):
pvname = pvcs['spec']['volumeName']
pvdata = get_pv_data(arg, pvname)
if is_rbd_pv(arg, pvname, pvdata):
format_table(arg, pvcs, table_rbd, True)
format_table(arg, pvcs, pvdata, table_rbd, True)
else:
format_table(arg, pvcs, table_cephfs, False)
format_table(arg, pvcs, pvdata, table_cephfs, False)
else:
for pvc in pvcs['items']:
pvname = pvc['spec']['volumeName']
pvdata = get_pv_data(arg, pvname)
if is_rbd_pv(arg, pvname, pvdata):
format_table(arg, pvc, table_rbd, True)
format_table(arg, pvc, pvdata, table_rbd, True)
else:
format_table(arg, pvc, table_cephfs, False)
format_table(arg, pvc, pvdata, table_cephfs, False)
print(table_rbd)
print(table_cephfs)
def format_table(arg, pvc_data, table, is_rbd):
#pylint: disable=too-many-locals
def format_table(arg, pvc_data, pvdata, table, is_rbd):
"""
format tables for pvc and image information
"""
@ -153,15 +154,17 @@ def format_table(arg, pvc_data, table, is_rbd):
table.add_row([pvcname, pvname, "", False,
False, False])
return
# get volname prefix
volname_prefix = get_volname_prefix(arg, pvdata)
# check image/subvolume details present rados omap
pv_present, uuid_present = validate_volume_in_rados(arg, image_id, pvname, pool_name, is_rbd)
present_in_cluster = False
if is_rbd:
present_in_cluster = check_image_in_cluster(arg, image_id, pool_name)
present_in_cluster = check_image_in_cluster(arg, image_id, pool_name, volname_prefix)
else:
subvolname = "csi-vol-%s" % image_id
subvolname = volname_prefix + image_id
present_in_cluster = check_subvol_in_cluster(arg, subvolname)
image_name = "csi-vol-%s" % image_id
image_name = volname_prefix + image_id
table.add_row([pvcname, pvname, image_name, pv_present,
uuid_present, present_in_cluster])
@ -210,11 +213,11 @@ def check_pv_name_in_rados(arg, image_id, pvc_name, pool_name, is_rbd):
return False
return True
def check_image_in_cluster(arg, image_uuid, pool_name):
def check_image_in_cluster(arg, image_uuid, pool_name, volname_prefix):
"""
validate pvc information in ceph backend
"""
image = "csi-vol-%s" % image_uuid
image = volname_prefix + image_uuid
cmd = ['rbd', 'info', image, "--pool", pool_name]
if not arg.userkey:
cmd += ["--id", arg.userid, "--key", arg.userkey]
@ -513,6 +516,22 @@ def get_pv_data(arg, pvname):
sys.exit()
return pvdata
def get_volname_prefix(arg, pvdata):
"""
Returns volname prefix stored in storage class/pv,
defaults to "csi-vol-"
"""
volname_prefix = "csi-vol-"
if not pvdata:
if arg.debug:
print("failed to get pv data")
sys.exit()
volume_attr = pvdata['spec']['csi']['volumeAttributes']
key = 'volumeNamePrefix'
if key in volume_attr.keys():
volname_prefix = volume_attr[key]
return volname_prefix
if __name__ == "__main__":
ARGS = PARSER.parse_args()
if ARGS.command not in ["kubectl", "oc"]: