Defaults module taken from ansible/ansible-modules-extras/pull/64. Fixes #10.pull/28/head
Alpha Chen
10 years ago
parent
315eeef7f8
commit
7b651ddbab
@ -0,0 +1,349 @@
|
|||||||
|
#!/usr/bin/python
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
# (c) 2014, GeekChimp - Franck Nijhof <franck@geekchimp.com>
|
||||||
|
#
|
||||||
|
# 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 = '''
|
||||||
|
---
|
||||||
|
module: osx_defaults
|
||||||
|
author: Franck Nijhof
|
||||||
|
short_description: osx_defaults allows users to read, write, and delete Mac OS X user defaults from Ansible
|
||||||
|
description:
|
||||||
|
- osx_defaults allows users to read, write, and delete Mac OS X user defaults from Ansible scripts.
|
||||||
|
Mac OS X applications and other programs use the defaults system to record user preferences and other
|
||||||
|
information that must be maintained when the applications aren't running (such as default font for new
|
||||||
|
documents, or the position of an Info panel).
|
||||||
|
version_added: 1.8
|
||||||
|
options:
|
||||||
|
domain:
|
||||||
|
description:
|
||||||
|
- The domain is a domain name of the form com.companyname.appname.
|
||||||
|
required: false
|
||||||
|
default: NSGlobalDomain
|
||||||
|
key:
|
||||||
|
description:
|
||||||
|
- The key of the user preference
|
||||||
|
required: true
|
||||||
|
type:
|
||||||
|
description:
|
||||||
|
- The type of value to write.
|
||||||
|
required: false
|
||||||
|
default: string
|
||||||
|
choices: [ "array", "bool", "boolean", "date", "float", "int", "integer", "string" ]
|
||||||
|
array_add:
|
||||||
|
description:
|
||||||
|
- Add new elements to the array for a key which has an array as its value.
|
||||||
|
required: false
|
||||||
|
default: string
|
||||||
|
choices: [ "true", "false" ]
|
||||||
|
value:
|
||||||
|
description:
|
||||||
|
- The value to write. Only required when state = present.
|
||||||
|
required: false
|
||||||
|
default: null
|
||||||
|
state:
|
||||||
|
description:
|
||||||
|
- The state of the user defaults
|
||||||
|
required: false
|
||||||
|
default: present
|
||||||
|
choices: [ "present", "absent" ]
|
||||||
|
notes:
|
||||||
|
- Apple Mac caches defaults. You may need to logout and login to apply the changes.
|
||||||
|
'''
|
||||||
|
|
||||||
|
EXAMPLES = '''
|
||||||
|
- osx_defaults: domain=com.apple.Safari key=IncludeInternalDebugMenu type=bool value=true state=present
|
||||||
|
- osx_defaults: domain=NSGlobalDomain key=AppleMeasurementUnits type=string value=Centimeters state=present
|
||||||
|
- osx_defaults: key=AppleMeasurementUnits type=string value=Centimeters
|
||||||
|
- osx_defaults:
|
||||||
|
key: AppleLanguages
|
||||||
|
type: array
|
||||||
|
value: ["en", "nl"]
|
||||||
|
- osx_defaults: domain=com.geekchimp.macable key=ExampleKeyToRemove state=absent
|
||||||
|
'''
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
# exceptions --------------------------------------------------------------- {{{
|
||||||
|
class OSXDefaultsException(Exception):
|
||||||
|
pass
|
||||||
|
# /exceptions -------------------------------------------------------------- }}}
|
||||||
|
|
||||||
|
# class MacDefaults -------------------------------------------------------- {{{
|
||||||
|
class OSXDefaults(object):
|
||||||
|
|
||||||
|
""" Class to manage Mac OS user defaults """
|
||||||
|
|
||||||
|
# init ---------------------------------------------------------------- {{{
|
||||||
|
""" Initialize this module. Finds 'defaults' executable and preps the parameters """
|
||||||
|
def __init__(self, **kwargs):
|
||||||
|
|
||||||
|
# Initial var for storing current defaults value
|
||||||
|
self.current_value = None
|
||||||
|
|
||||||
|
# Just set all given parameters
|
||||||
|
for key, val in kwargs.iteritems():
|
||||||
|
setattr(self, key, val)
|
||||||
|
|
||||||
|
# Try to find the defaults executable
|
||||||
|
self.executable = self.module.get_bin_path(
|
||||||
|
'defaults',
|
||||||
|
required=False,
|
||||||
|
opt_dirs=self.path.split(':'),
|
||||||
|
)
|
||||||
|
|
||||||
|
if not self.executable:
|
||||||
|
raise OSXDefaultsException("Unable to locate defaults executable.")
|
||||||
|
|
||||||
|
# When state is present, we require a parameter
|
||||||
|
if self.state == "present" and self.value is None:
|
||||||
|
raise OSXDefaultsException("Missing value parameter")
|
||||||
|
|
||||||
|
# Ensure the value is the correct type
|
||||||
|
self.value = self._convert_type(self.type, self.value)
|
||||||
|
|
||||||
|
# /init --------------------------------------------------------------- }}}
|
||||||
|
|
||||||
|
# tools --------------------------------------------------------------- {{{
|
||||||
|
""" Converts value to given type """
|
||||||
|
def _convert_type(self, type, value):
|
||||||
|
|
||||||
|
if type == "string":
|
||||||
|
return str(value)
|
||||||
|
elif type in ["bool", "boolean"]:
|
||||||
|
if value.lower() in [True, 1, "true", "1", "yes"]:
|
||||||
|
return True
|
||||||
|
elif value.lower() in [False, 0, "false", "0", "no"]:
|
||||||
|
return False
|
||||||
|
raise OSXDefaultsException("Invalid boolean value: {0}".format(repr(value)))
|
||||||
|
elif type == "date":
|
||||||
|
try:
|
||||||
|
return datetime.strptime(value.split("+")[0].strip(), "%Y-%m-%d %H:%M:%S")
|
||||||
|
except ValueError:
|
||||||
|
raise OSXDefaultsException(
|
||||||
|
"Invalid date value: {0}. Required format yyy-mm-dd hh:mm:ss.".format(repr(value))
|
||||||
|
)
|
||||||
|
elif type in ["int", "integer"]:
|
||||||
|
if not str(value).isdigit():
|
||||||
|
raise OSXDefaultsException("Invalid integer value: {0}".format(repr(value)))
|
||||||
|
return int(value)
|
||||||
|
elif type == "float":
|
||||||
|
try:
|
||||||
|
value = float(value)
|
||||||
|
except ValueError:
|
||||||
|
raise OSXDefaultsException("Invalid float value: {0}".format(repr(value)))
|
||||||
|
return value
|
||||||
|
elif type == "array":
|
||||||
|
if not isinstance(value, list):
|
||||||
|
raise OSXDefaultsException("Invalid value. Expected value to be an array")
|
||||||
|
return value
|
||||||
|
|
||||||
|
raise OSXDefaultsException('Type is not supported: {0}'.format(type))
|
||||||
|
|
||||||
|
""" Converts array output from defaults to an list """
|
||||||
|
@staticmethod
|
||||||
|
def _convert_defaults_str_to_list(value):
|
||||||
|
|
||||||
|
# Split output of defaults. Every line contains a value
|
||||||
|
value = value.splitlines()
|
||||||
|
|
||||||
|
# Remove first and last item, those are not actual values
|
||||||
|
value.pop(0)
|
||||||
|
value.pop(-1)
|
||||||
|
|
||||||
|
# Remove extra spaces and comma (,) at the end of values
|
||||||
|
value = [re.sub(',$', '', x.strip(' ')) for x in value]
|
||||||
|
|
||||||
|
return value
|
||||||
|
# /tools -------------------------------------------------------------- }}}
|
||||||
|
|
||||||
|
# commands ------------------------------------------------------------ {{{
|
||||||
|
""" Reads value of this domain & key from defaults """
|
||||||
|
def read(self):
|
||||||
|
# First try to find out the type
|
||||||
|
rc, out, err = self.module.run_command([self.executable, "read-type", self.domain, self.key])
|
||||||
|
|
||||||
|
# If RC is 1, the key does not exists
|
||||||
|
if rc == 1:
|
||||||
|
return None
|
||||||
|
|
||||||
|
# If the RC is not 0, then terrible happened! Ooooh nooo!
|
||||||
|
if rc != 0:
|
||||||
|
raise OSXDefaultsException("An error occurred while reading key type from defaults: " + out)
|
||||||
|
|
||||||
|
# Ok, lets parse the type from output
|
||||||
|
type = out.strip().replace('Type is ', '')
|
||||||
|
|
||||||
|
# Now get the current value
|
||||||
|
rc, out, err = self.module.run_command([self.executable, "read", self.domain, self.key])
|
||||||
|
|
||||||
|
# Strip output
|
||||||
|
out = out.strip()
|
||||||
|
|
||||||
|
# An non zero RC at this point is kinda strange...
|
||||||
|
if rc != 0:
|
||||||
|
raise OSXDefaultsException("An error occurred while reading key value from defaults: " + out)
|
||||||
|
|
||||||
|
# Convert string to list when type is array
|
||||||
|
if type == "array":
|
||||||
|
out = self._convert_defaults_str_to_list(out)
|
||||||
|
|
||||||
|
# Store the current_value
|
||||||
|
self.current_value = self._convert_type(type, out)
|
||||||
|
|
||||||
|
""" Writes value to this domain & key to defaults """
|
||||||
|
def write(self):
|
||||||
|
|
||||||
|
# We need to convert some values so the defaults commandline understands it
|
||||||
|
if type(self.value) is bool:
|
||||||
|
value = "TRUE" if self.value else "FALSE"
|
||||||
|
elif type(self.value) is int or type(self.value) is float:
|
||||||
|
value = str(self.value)
|
||||||
|
elif self.array_add and self.current_value is not None:
|
||||||
|
value = list(set(self.value) - set(self.current_value))
|
||||||
|
elif isinstance(self.value, datetime):
|
||||||
|
value = self.value.strftime('%Y-%m-%d %H:%M:%S')
|
||||||
|
else:
|
||||||
|
value = self.value
|
||||||
|
|
||||||
|
# When the type is array and array_add is enabled, morph the type :)
|
||||||
|
if self.type == "array" and self.array_add:
|
||||||
|
self.type = "array-add"
|
||||||
|
|
||||||
|
# All values should be a list, for easy passing it to the command
|
||||||
|
if not isinstance(value, list):
|
||||||
|
value = [value]
|
||||||
|
|
||||||
|
rc, out, err = self.module.run_command([self.executable, 'write', self.domain, self.key, '-' + self.type] + value)
|
||||||
|
|
||||||
|
if rc != 0:
|
||||||
|
raise OSXDefaultsException('An error occurred while writing value to defaults: ' + out)
|
||||||
|
|
||||||
|
""" Deletes defaults key from domain """
|
||||||
|
def delete(self):
|
||||||
|
rc, out, err = self.module.run_command([self.executable, 'delete', self.domain, self.key])
|
||||||
|
if rc != 0:
|
||||||
|
raise OSXDefaultsException("An error occurred while deleting key from defaults: " + out)
|
||||||
|
|
||||||
|
# /commands ----------------------------------------------------------- }}}
|
||||||
|
|
||||||
|
# run ----------------------------------------------------------------- {{{
|
||||||
|
""" Does the magic! :) """
|
||||||
|
def run(self):
|
||||||
|
|
||||||
|
# Get the current value from defaults
|
||||||
|
self.read()
|
||||||
|
|
||||||
|
# Handle absent state
|
||||||
|
if self.state == "absent":
|
||||||
|
print "Absent state detected!"
|
||||||
|
if self.current_value is None:
|
||||||
|
return False
|
||||||
|
self.delete()
|
||||||
|
return True
|
||||||
|
|
||||||
|
# There is a type mismatch! Given type does not match the type in defaults
|
||||||
|
if self.current_value is not None and type(self.current_value) is not type(self.value):
|
||||||
|
raise OSXDefaultsException("Type mismatch. Type in defaults: " + type(self.current_value).__name__)
|
||||||
|
|
||||||
|
# Current value matches the given value. Nothing need to be done. Arrays need extra care
|
||||||
|
if self.type == "array" and self.current_value is not None and not self.array_add and \
|
||||||
|
set(self.current_value) == set(self.value):
|
||||||
|
return False
|
||||||
|
elif self.type == "array" and self.current_value is not None and self.array_add and \
|
||||||
|
len(list(set(self.value) - set(self.current_value))) == 0:
|
||||||
|
return False
|
||||||
|
elif self.current_value == self.value:
|
||||||
|
return False
|
||||||
|
|
||||||
|
# Change/Create/Set given key/value for domain in defaults
|
||||||
|
self.write()
|
||||||
|
return True
|
||||||
|
|
||||||
|
# /run ---------------------------------------------------------------- }}}
|
||||||
|
|
||||||
|
# /class MacDefaults ------------------------------------------------------ }}}
|
||||||
|
|
||||||
|
|
||||||
|
# main -------------------------------------------------------------------- {{{
|
||||||
|
def main():
|
||||||
|
module = AnsibleModule(
|
||||||
|
argument_spec=dict(
|
||||||
|
domain=dict(
|
||||||
|
default="NSGlobalDomain",
|
||||||
|
required=False,
|
||||||
|
),
|
||||||
|
key=dict(
|
||||||
|
default=None,
|
||||||
|
),
|
||||||
|
type=dict(
|
||||||
|
default="string",
|
||||||
|
required=False,
|
||||||
|
choices=[
|
||||||
|
"array",
|
||||||
|
"bool",
|
||||||
|
"boolean",
|
||||||
|
"date",
|
||||||
|
"float",
|
||||||
|
"int",
|
||||||
|
"integer",
|
||||||
|
"string",
|
||||||
|
],
|
||||||
|
),
|
||||||
|
array_add=dict(
|
||||||
|
default=False,
|
||||||
|
required=False,
|
||||||
|
choices=BOOLEANS,
|
||||||
|
),
|
||||||
|
value=dict(
|
||||||
|
default=None,
|
||||||
|
required=False,
|
||||||
|
),
|
||||||
|
state=dict(
|
||||||
|
default="present",
|
||||||
|
required=False,
|
||||||
|
choices=[
|
||||||
|
"absent", "present"
|
||||||
|
],
|
||||||
|
),
|
||||||
|
path=dict(
|
||||||
|
default="/usr/bin:/usr/local/bin",
|
||||||
|
required=False,
|
||||||
|
)
|
||||||
|
),
|
||||||
|
supports_check_mode=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
domain = module.params['domain']
|
||||||
|
key = module.params['key']
|
||||||
|
type = module.params['type']
|
||||||
|
array_add = module.params['array_add']
|
||||||
|
value = module.params['value']
|
||||||
|
state = module.params['state']
|
||||||
|
path = module.params['path']
|
||||||
|
|
||||||
|
try:
|
||||||
|
defaults = OSXDefaults(module=module, domain=domain, key=key, type=type,
|
||||||
|
array_add=array_add, value=value, state=state, path=path)
|
||||||
|
changed = defaults.run()
|
||||||
|
module.exit_json(changed=changed)
|
||||||
|
except OSXDefaultsException as e:
|
||||||
|
module.fail_json(msg=e.message)
|
||||||
|
|
||||||
|
# /main ------------------------------------------------------------------- }}}
|
||||||
|
|
||||||
|
from ansible.module_utils.basic import *
|
||||||
|
main()
|
||||||
Loading…
Reference in new issue