cmueller/ansible
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Update group module to use new shared module code

Browse source
This commit is contained in:
Stephen Fromm 2012-07-21 22:30:13 -07:00
parent 0bad1e7de2
commit aeed872aee
1 changed files with 60 additions and 105 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

129
group
View file

@ -17,50 +17,27 @@
# You should have received a copy of the GNU General Public License # You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>. # along with Ansible. If not, see <http://www.gnu.org/licenses/>.
try:
import json
except ImportError:
import simplejson as json
import os import os
import grp import grp
import shlex
import subprocess import subprocess
import sys
import syslog
GROUPADD = "/usr/sbin/groupadd" def get_bin_path(module, arg):
GROUPDEL = "/usr/sbin/groupdel" if os.path.exists('/usr/sbin/%s' % arg):
GROUPMOD = "/usr/sbin/groupmod" return '/usr/sbin/%s' % arg
elif os.path.exists('/sbin/%s' % arg):
def exit_json(rc=0, **kwargs): return '/sbin/%s' % arg
if 'name' in kwargs:
add_group_info(kwargs)
print json.dumps(kwargs)
sys.exit(rc)
def fail_json(**kwargs):
kwargs['failed'] = True
exit_json(rc=1, **kwargs)
def add_group_info(kwargs):
name = kwargs['name']
if group_exists(name):
kwargs['state'] = 'present'
info = group_info(name)
kwargs['gid'] = info[2]
else: else:
kwargs['state'] = 'absent' module.fail_json(msg="Cannot find %s" % arg)
return kwargs
def group_del(group): def group_del(module, group):
cmd = [GROUPDEL, group] cmd = [get_bin_path(module, 'groupdel'), group]
p = subprocess.Popen(cmd, shell=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE) p = subprocess.Popen(cmd, shell=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
(out, err) = p.communicate() (out, err) = p.communicate()
rc = p.returncode rc = p.returncode
return (rc, out, err) return (rc, out, err)
def group_add(group, **kwargs): def group_add(module, group, **kwargs):
cmd = [GROUPADD] cmd = [get_bin_path(module, 'groupadd')]
for key in kwargs: for key in kwargs:
if key == 'gid' and kwargs[key] is not None: if key == 'gid' and kwargs[key] is not None:
cmd.append('-g') cmd.append('-g')
@ -73,8 +50,8 @@ def group_add(group, **kwargs):
rc = p.returncode rc = p.returncode
return (rc, out, err) return (rc, out, err)
def group_mod(group, **kwargs): def group_mod(module, group, **kwargs):
cmd = [GROUPMOD] cmd = [get_bin_path(module, 'groupmod')]
info = group_info(group) info = group_info(group)
for key in kwargs: for key in kwargs:
if key == 'gid': if key == 'gid':
@ -107,70 +84,39 @@ def group_info(group):
# =========================================== # ===========================================
if not os.path.exists(GROUPADD): def main():
if os.path.exists("/sbin/groupadd"): module = AnsibleModule(
GROUPADD = "/sbin/groupadd" argument_spec = dict(
else: state=dict(default='present', choices=['present', 'absent']),
fail_json(msg="Cannot find groupadd") name=dict(required=True),
if not os.path.exists(GROUPDEL): gid=dict(default=None),
if os.path.exists("/sbin/groupdel"): system=dict(default='no', choices=['yes', 'no']),
GROUPDEL = "/sbin/groupdel" )
else: )
fail_json(msg="Cannot find groupdel")
if not os.path.exists(GROUPMOD):
if os.path.exists("/sbin/groupmod"):
GROUPDEL = "/sbin/groupmod"
else:
fail_json(msg="Cannot find groupmod")
if len(sys.argv) == 2 and os.path.exists(sys.argv[1]):
argfile = sys.argv[1]
args = open(argfile, 'r').read()
else:
args = ' '.join(sys.argv[1:])
items = shlex.split(args)
syslog.openlog('ansible-%s' % os.path.basename(__file__))
syslog.syslog(syslog.LOG_NOTICE, 'Invoked with %s' % args)
if not len(items):
fail_json(msg='the module requires arguments -a')
sys.exit(1)
params = {}
for x in items:
(k, v) = x.split("=")
params[k] = v
state = params.get('state','present')
name = params.get('name', None)
gid = params.get('gid', None)
system = params.get('system', 'no')
if state not in [ 'present', 'absent' ]:
fail_json(msg='invalid state')
if system not in ['yes', 'no']:
fail_json(msg='invalid system')
if name is None:
fail_json(msg='name is required')
state = module.params['state']
name = module.params['name']
gid = module.params['gid']
system = module.params['system']
rc = None rc = None
out = '' out = ''
err = '' err = ''
result = {} result = {}
result['name'] = name result['name'] = name
result['state'] = state
if state == 'absent': if state == 'absent':
if group_exists(name): if group_exists(name):
(rc, out, err) = group_del(name) (rc, out, err) = group_del(module, name)
if rc != 0: if rc != 0:
fail_json(name=name, msg=err) module.fail_json(name=name, msg=err)
elif state == 'present': elif state == 'present':
if not group_exists(name): if not group_exists(name):
(rc, out, err) = group_add(name, gid=gid, system=system) (rc, out, err) = group_add(module, name, gid=gid, system=system)
else: else:
(rc, out, err) = group_mod(name, gid=gid) (rc, out, err) = group_mod(module, name, gid=gid)
if rc is not None and rc != 0: if rc is not None and rc != 0:
fail_json(name=name, msg=err) module.fail_json(name=name, msg=err)
if rc is None: if rc is None:
result['changed'] = False result['changed'] = False
@ -180,5 +126,14 @@ if out:
result['stdout'] = out result['stdout'] = out
if err: if err:
result['stderr'] = err result['stderr'] = err
exit_json(**result)
fail_json(name=name, msg='Unexpected position reached') if group_exists(name):
info = group_info(name)
result['system'] = system
result['gid'] = info[2]
module.exit_json(**result)
# include magic from lib/ansible/module_common.py
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
main()