Files
kayobe/ansible/roles/wipe-disks/library/blockdevice_info.py
Will Szumski faa5b33fef Fixes an issue with --wipe-disks
The awk expression that was in use prior to this change was fragile
and in some cases could incorrectly identify the unmounted block
devices. This change switches to parsing the json output which
should hopefully be more robust.

Change-Id: Ifa89e7307eb445b4f1708f0c6ac3409b3f96aafe
Story: 2010367
Task: 46578
2022-10-17 17:37:55 +01:00

74 lines
1.4 KiB
Python

#!/usr/bin/python3
DOCUMENTATION = '''
---
module: blockdevice_info
short_description: Returns information about block devices
version_added: "N/A"
description:
- "Returns information about block devices"
author:
- Will Szumski
'''
EXAMPLES = '''
- name: Retrieve information about block devices
blockdevice_info:
become: true
register: result
'''
RETURN = '''
umounted:
description: A list of all umounted block devices.
type: list
returned: always
'''
import json
from ansible.module_utils.basic import AnsibleModule
def _has_mounts(device):
if device["mountpoint"]:
return True
for child in device.get("children", []):
if _has_mounts(child):
return True
return False
def unmounted(module, lsblk):
result = []
for device in lsblk.get("blockdevices", []):
if not _has_mounts(device) and device["type"] == 'disk':
result.append(device["name"])
return result
def run_module():
# The module takes no argumnets.
module_args = dict()
result = dict(
changed=False,
unmounted=[]
)
module = AnsibleModule(
argument_spec=module_args,
supports_check_mode=True
)
_rc, stdout, _stderr = module.run_command("lsblk -J")
lsblk = json.loads(stdout)
result['unmounted'] = unmounted(module, lsblk)
module.exit_json(**result)
def main():
run_module()
if __name__ == '__main__':
main()