Random Generator Ansible Module

stacksofplates

Just practicing with some custom modules. I can't really think of anything useful to write because there are so many already, so I just made a random string generator that can be passed into a variable for other tasks.

Here's the output:

Here's the module:

#!/usr/bin/python

DOCUMENTATION='''
---
module: random_generator
short_description: Generates random string
description:
    - Generates random string
    - from a given size
options:
  size:
    description:
      - Size of the string to generate
    required: true
'''
RETURN = '''
msg:
    description: Returns random string
    type: string
'''
EXAMPLES = '''
- random_generator:
    size: 25
'''

import string
import random 
from ansible.module_utils.basic import AnsibleModule


def random_generator(size, chars=string.ascii_uppercase + string.digits):
    return ''.join(random.choice(chars) for _ in range(size))

module = AnsibleModule(
    argument_spec = dict(
    size = dict(required=True, type='int')
    )
)
size = module.params.get('size')

try:
    random_generator(size)
    success = True
    ret_msg = str(random_generator(size))
except KeyError:
   success = False
   ret_msg = "Error"

if success:
    module.exit_json(msg=ret_msg)
else:
    module.fail_json(msg=ret_msg)


if __name__ == "__main__":
  main()

Here's how the documentation looks: