mirror of https://github.com/ansible/ansible.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
232 lines
8.1 KiB
Python
232 lines
8.1 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# (c) 2013, Alexander Bulimov <lazywolf0@gmail.com>
|
|
# based on lvol module by Jeroen Hoekx <jeroen.hoekx@dsquare.be>
|
|
#
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
DOCUMENTATION = '''
|
|
---
|
|
author: Alexander Bulimov
|
|
module: lvg
|
|
short_description: Configure LVM volume groups
|
|
description:
|
|
- This module creates, removes or resizes volume groups.
|
|
version_added: "1.1"
|
|
options:
|
|
vg:
|
|
description:
|
|
- The name of the volume group.
|
|
required: true
|
|
dev:
|
|
description:
|
|
- List of comma-separated devices to use in this volume group.
|
|
required: true
|
|
pesize:
|
|
description:
|
|
- The size of the physical extent in megabytes.
|
|
default: 4
|
|
required: false
|
|
state:
|
|
choices: [ "present", "absent" ]
|
|
default: present
|
|
description:
|
|
- Control if the volume group exists.
|
|
required: false
|
|
force:
|
|
choices: [ "yes", "no" ]
|
|
default: "no"
|
|
description:
|
|
- If yes, allows to remove volume group with logical volumes.
|
|
required: false
|
|
examples:
|
|
- description: Create a volume group on top of /dev/sda1 with physical extent size = 32MB.
|
|
code: lvg vg=vg.services dev=/dev/sda1 pesize=32
|
|
- description: Create a volume group on top of /dev/sdb1 and /dev/sdc5.
|
|
code: lvg vg=vg.services dev=/dev/sdb1,/dev/sdc5
|
|
- description: Remove a volume group with name vg.services.
|
|
code: lvg vg=vg.services state=absent
|
|
notes:
|
|
- module does not modify PE size for already present volume group
|
|
'''
|
|
|
|
def parse_vgs(data):
|
|
vgs = []
|
|
for line in data.splitlines():
|
|
parts = line.strip().split(';')
|
|
vgs.append({
|
|
'name': parts[0],
|
|
'pv_count': int(parts[1]),
|
|
'lv_count': int(parts[2]),
|
|
})
|
|
return vgs
|
|
|
|
def parse_pvs(data):
|
|
pvs = []
|
|
for line in data.splitlines():
|
|
parts = line.strip().split(';')
|
|
pvs.append({
|
|
'name': parts[0],
|
|
'vg_name': parts[1],
|
|
})
|
|
return pvs
|
|
|
|
def main():
|
|
module = AnsibleModule(
|
|
argument_spec = dict(
|
|
vg=dict(required=True),
|
|
dev=dict(),
|
|
pesize=dict(),
|
|
state=dict(choices=["absent", "present"], default='present'),
|
|
force=dict(type='bool', default='no'),
|
|
),
|
|
supports_check_mode=True,
|
|
)
|
|
|
|
vg = module.params['vg']
|
|
state = module.params['state']
|
|
force = module.boolean(module.params['force'])
|
|
if module.params['pesize']:
|
|
pesize = int(module.params['pesize'])
|
|
else:
|
|
pesize = 4
|
|
|
|
if module.params['dev']:
|
|
dev = module.params['dev'].replace(',',' ')
|
|
dev_list = module.params['dev'].split(',')
|
|
elif state == 'present':
|
|
module.fail_json(msg="No dev given.")
|
|
|
|
if state=='present':
|
|
### check given devices
|
|
for test_dev in dev_list:
|
|
if not os.path.exists(test_dev):
|
|
module.fail_json(msg="No dev %s found."%test_dev)
|
|
|
|
### get pv list
|
|
rc,current_pvs,err = module.run_command("pvs --noheadings -o pv_name,vg_name --separator ';'")
|
|
if rc != 0:
|
|
module.fail_json(msg="Failed executing pvs command.",rc=rc, err=err)
|
|
|
|
### check pv for devices
|
|
pvs = parse_pvs(current_pvs)
|
|
used_pvs = filter(lambda pv: pv['name'] in dev_list and pv['vg_name'] and pv['vg_name'] != vg, pvs)
|
|
if used_pvs:
|
|
module.fail_json(msg="dev %s is already in %s volume group."%(used_pvs[0]['name'],used_pvs[0]['vg_name']))
|
|
|
|
rc,current_vgs,err = module.run_command("vgs --noheadings -o vg_name,pv_count,lv_count --separator ';'")
|
|
|
|
if rc != 0:
|
|
module.fail_json(msg="Failed executing vgs command.",rc=rc, err=err)
|
|
|
|
changed = False
|
|
|
|
vgs = parse_vgs(current_vgs)
|
|
|
|
for test_vg in vgs:
|
|
if test_vg['name'] == vg:
|
|
this_vg = test_vg
|
|
break
|
|
else:
|
|
this_vg = None
|
|
|
|
if this_vg is None:
|
|
if state == 'present':
|
|
### create VG
|
|
if module.check_mode:
|
|
changed = True
|
|
else:
|
|
### create PV
|
|
for current_dev in dev_list:
|
|
rc,_,err = module.run_command("pvcreate %s"%current_dev)
|
|
if rc == 0:
|
|
changed = True
|
|
else:
|
|
module.fail_json(msg="Creating physical volume '%s' failed"%current_dev, rc=rc, err=err)
|
|
rc,_,err = module.run_command("vgcreate -s %s %s %s"%(pesize, vg, dev))
|
|
if rc == 0:
|
|
changed = True
|
|
else:
|
|
module.fail_json(msg="Creating volume group '%s' failed"%vg, rc=rc, err=err)
|
|
else:
|
|
if state == 'absent':
|
|
if module.check_mode:
|
|
module.exit_json(changed=True)
|
|
else:
|
|
if this_vg['lv_count'] == 0 or force:
|
|
### remove VG
|
|
rc,_,err = module.run_command("vgremove --force %s"%(vg))
|
|
if rc == 0:
|
|
module.exit_json(changed=True)
|
|
else:
|
|
module.fail_json(msg="Failed to remove volume group %s"%(vg),rc=rc, err=err)
|
|
else:
|
|
module.fail_json(msg="Refuse to remove non-empty volume group %s without force=yes"%(vg))
|
|
|
|
### resize VG
|
|
action = None
|
|
current_devs = map(lambda x: x['name'], filter(lambda pv: pv['vg_name'] == vg, pvs))
|
|
devs_to_remove = list(set(current_devs) - set(dev_list))
|
|
devs_to_add = list(set(dev_list) - set(current_devs))
|
|
if devs_to_remove and devs_to_add:
|
|
devs_string = ' '.join([`dev` for dev in devs_to_add])
|
|
devs_to_remove_string = ' '.join([`dev` for dev in devs_to_remove])
|
|
action = 'modify'
|
|
elif devs_to_remove:
|
|
devs_string = ' '.join([`dev` for dev in devs_to_remove])
|
|
action = 'reduce'
|
|
elif devs_to_add:
|
|
devs_string = ' '.join([`dev` for dev in devs_to_add])
|
|
action = 'extend'
|
|
|
|
if action:
|
|
if module.check_mode:
|
|
changed = True
|
|
else:
|
|
if action == 'extend' or action == 'modify':
|
|
tool = 'vgextend'
|
|
### create PV
|
|
for current_dev in devs_to_add:
|
|
rc,_,err = module.run_command("pvcreate %s"%current_dev)
|
|
if rc == 0:
|
|
changed = True
|
|
else:
|
|
module.fail_json(msg="Creating physical volume '%s' failed"%current_dev, rc=rc, err=err)
|
|
else:
|
|
tool = 'vgreduce --force'
|
|
|
|
### first we add or remove PV
|
|
rc,_,err = module.run_command("%s %s %s"%(tool, vg, devs_string))
|
|
if rc == 0:
|
|
changed = True
|
|
else:
|
|
module.fail_json(msg="Unable to %s %s by %s."%(action, vg, devs_string),rc=rc,err=err)
|
|
### then if we need - remove some PV
|
|
if action == 'modify':
|
|
tool = 'vgreduce --force'
|
|
rc,_,err = module.run_command("%s %s %s"%(tool, vg, devs_to_remove_string))
|
|
if rc == 0:
|
|
changed = True
|
|
else:
|
|
module.fail_json(msg="Unable to reduce %s by %s."%(vg, devs_to_remove_string),rc=rc,err=err)
|
|
|
|
module.exit_json(changed=changed)
|
|
|
|
# this is magic, see lib/ansible/module_common.py
|
|
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
|
|
main()
|