def main(): argument_spec = mso_argument_spec() argument_spec.update( schema=dict(type='str', required=True), site=dict(type='str', required=True), template=dict(type='str', required=True), vrf=dict(type='str', aliases=[ 'name' ]), # This parameter is not required for querying all objects state=dict(type='str', default='present', choices=['absent', 'present', 'query']), ) module = AnsibleModule( argument_spec=argument_spec, supports_check_mode=True, required_if=[ ['state', 'absent', ['vrf']], ['state', 'present', ['vrf']], ], ) schema = module.params.get('schema') site = module.params.get('site') template = module.params.get('template') vrf = module.params.get('vrf') state = module.params.get('state') mso = MSOModule(module) # Get schema_id schema_obj = mso.get_obj('schemas', displayName=schema) if not schema_obj: mso.fail_json( msg="Provided schema '{0}' does not exist".format(schema)) schema_path = 'schemas/{id}'.format(**schema_obj) schema_id = schema_obj.get('id') # Get site site_id = mso.lookup_site(site) # Get site_idx sites = [(s.get('siteId'), s.get('templateName')) for s in schema_obj.get('sites')] if (site_id, template) not in sites: mso.fail_json( msg= "Provided site/template '{0}-{1}' does not exist. Existing sites/templates: {2}" .format(site, template, ', '.join(sites))) # Schema-access uses indexes site_idx = sites.index((site_id, template)) # Path-based access uses site_id-template site_template = '{0}-{1}'.format(site_id, template) # Get VRF vrf_ref = mso.vrf_ref(schema_id=schema_id, template=template, vrf=vrf) vrfs = [v.get('vrfRef') for v in schema_obj.get('sites')[site_idx]['vrfs']] if vrf is not None and vrf_ref in vrfs: vrf_idx = vrfs.index(vrf_ref) vrf_path = '/sites/{0}/vrfs/{1}'.format(site_template, vrf) mso.existing = schema_obj.get('sites')[site_idx]['vrfs'][vrf_idx] if state == 'query': if vrf is None: mso.existing = schema_obj.get('sites')[site_idx]['vrfs'] elif not mso.existing: mso.fail_json(msg="VRF '{vrf}' not found".format(vrf=vrf)) mso.exit_json() vrfs_path = '/sites/{0}/vrfs'.format(site_template) ops = [] mso.previous = mso.existing if state == 'absent': if mso.existing: mso.sent = mso.existing = {} ops.append(dict(op='remove', path=vrf_path)) elif state == 'present': payload = dict(vrfRef=dict( schemaId=schema_id, templateName=template, vrfName=vrf, ), ) mso.sanitize(payload, collate=True) if mso.existing: ops.append(dict(op='replace', path=vrf_path, value=mso.sent)) else: ops.append(dict(op='add', path=vrfs_path + '/-', value=mso.sent)) mso.existing = mso.proposed if not module.check_mode: mso.request(schema_path, method='PATCH', data=ops) mso.exit_json()
def main(): argument_spec = mso_argument_spec() argument_spec.update( schema=dict(type='str', required=True), site=dict(type='str', required=True), template=dict(type='str', required=True), vrf=dict(type='str', required=True), region=dict(type='str', required=True), cidr=dict(type='str', aliases=[ 'ip' ]), # This parameter is not required for querying all objects primary=dict(type='bool'), state=dict(type='str', default='present', choices=['absent', 'present', 'query']), ) module = AnsibleModule( argument_spec=argument_spec, supports_check_mode=True, required_if=[ ['state', 'absent', ['cidr']], ['state', 'present', ['cidr']], ], ) schema = module.params.get('schema') site = module.params.get('site') template = module.params.get('template') vrf = module.params.get('vrf') region = module.params.get('region') cidr = module.params.get('cidr') primary = module.params.get('primary') state = module.params.get('state') mso = MSOModule(module) # Get schema_id schema_obj = mso.get_obj('schemas', displayName=schema) if not schema_obj: mso.fail_json( msg="Provided schema '{0}' does not exist".format(schema)) schema_path = 'schemas/{id}'.format(**schema_obj) schema_id = schema_obj.get('id') # Get site site_id = mso.lookup_site(site) # Get site_idx sites = [(s.get('siteId'), s.get('templateName')) for s in schema_obj.get('sites')] if (site_id, template) not in sites: mso.fail_json( msg= "Provided site/template '{0}-{1}' does not exist. Existing sites/templates: {2}" .format(site, template, ', '.join(sites))) # Schema-access uses indexes site_idx = sites.index((site_id, template)) # Path-based access uses site_id-template site_template = '{0}-{1}'.format(site_id, template) # Get VRF vrf_ref = mso.vrf_ref(schema_id=schema_id, template=template, vrf=vrf) vrfs = [v.get('vrfRef') for v in schema_obj.get('sites')[site_idx]['vrfs']] if vrf_ref not in vrfs: mso.fail_json( msg="Provided vrf '{0}' does not exist. Existing vrfs: {1}".format( vrf, ', '.join(vrfs))) vrf_idx = vrfs.index(vrf_ref) # Get Region regions = [ r.get('name') for r in schema_obj.get('sites')[site_idx]['vrfs'][vrf_idx]['regions'] ] if region not in regions: mso.fail_json( msg="Provided region '{0}' does not exist. Existing regions: {1}". format(region, ', '.join(regions))) region_idx = regions.index(region) # Get CIDR cidrs = [ c.get('ip') for c in schema_obj.get('sites')[site_idx]['vrfs'][vrf_idx] ['regions'][region_idx]['cidrs'] ] if cidr is not None and cidr in cidrs: cidr_idx = cidrs.index(cidr) # FIXME: Changes based on index are DANGEROUS cidr_path = '/sites/{0}/vrfs/{1}/regions/{2}/cidrs/{3}'.format( site_template, vrf, region, cidr_idx) mso.existing = schema_obj.get('sites')[site_idx]['vrfs'][vrf_idx][ 'regions'][region_idx]['cidrs'][cidr_idx] if state == 'query': if cidr is None: mso.existing = schema_obj.get('sites')[site_idx]['vrfs'][vrf_idx][ 'regions'][region_idx]['cidrs'] elif not mso.existing: mso.fail_json(msg="CIDR IP '{cidr}' not found".format(cidr=cidr)) mso.exit_json() cidrs_path = '/sites/{0}/vrfs/{1}/regions/{2}/cidrs'.format( site_template, vrf, region) ops = [] mso.previous = mso.existing if state == 'absent': if mso.existing: mso.sent = mso.existing = {} ops.append(dict(op='remove', path=cidr_path)) elif state == 'present': if not mso.existing: if primary is None: primary = False payload = dict( ip=cidr, primary=primary, ) mso.sanitize(payload, collate=True) if mso.existing: ops.append(dict(op='replace', path=cidr_path, value=mso.sent)) else: ops.append(dict(op='add', path=cidrs_path + '/-', value=mso.sent)) mso.existing = mso.proposed if not module.check_mode: mso.request(schema_path, method='PATCH', data=ops) mso.exit_json()