1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747 |
- '''
- OpenShiftCLI class that wraps the oc commands in a subprocess
- '''
- from __future__ import print_function
- import atexit
- import copy
- import fcntl
- import json
- import os
- import re
- import shutil
- import subprocess
- import tempfile
- try:
- import ruamel.yaml as yaml
- except ImportError:
- import yaml
- from ansible.module_utils.basic import AnsibleModule
- DOCUMENTATION = '''
- ---
- module: oc_project
- short_description: Module to manage openshift projects
- description:
- - Manage openshift projects programmatically.
- options:
- state:
- description:
- - If present, the project will be created if it doesn't exist or update if different. If absent, the project will be removed if present. If list, information about the project will be gathered and returned as part of the Ansible call results.
- required: false
- default: present
- choices: ["present", "absent", "list"]
- aliases: []
- kubeconfig:
- description:
- - The path for the kubeconfig file to use for authentication
- required: false
- default: /etc/origin/master/admin.kubeconfig
- aliases: []
- debug:
- description:
- - Turn on debug output.
- required: false
- default: False
- aliases: []
- name:
- description:
- - Name of the object that is being queried.
- required: false
- default: None
- aliases: []
- display_name:
- description:
- - The display name attribute for a project
- required: false
- default: None
- aliases: []
- description:
- description:
- - The description attribute for a project
- required: false
- default: None
- aliases: []
- admin:
- description:
- - The project admin username
- required: false
- default: false
- aliases: []
- admin_role:
- description:
- - The project admin username
- required: false
- default: 'admin'
- aliases: []
- node_selector:
- description:
- - The node selector for this project.
- - This allows certain pods in this project to run on certain nodes.
- required: false
- default: None
- aliases: []
- author:
- - "Kenny Woodson <kwoodson@redhat.com>"
- extends_documentation_fragment: []
- '''
- EXAMPLES = '''
- - name: create secret
- oc_project:
- state: present
- name: openshift-ops
- display_name: operations team project
- node_selector:
- - top=secret
- - noncustomer=True
- '''
- class YeditException(Exception):
- ''' Exception class for Yedit '''
- pass
- class Yedit(object):
- ''' Class to modify yaml files '''
- re_valid_key = r"(((\[-?\d+\])|([0-9a-zA-Z%s/_-]+)).?)+$"
- re_key = r"(?:\[(-?\d+)\])|([0-9a-zA-Z{}/_-]+)"
- com_sep = set(['.', '#', '|', ':'])
-
- def __init__(self,
- filename=None,
- content=None,
- content_type='yaml',
- separator='.',
- backup=False):
- self.content = content
- self._separator = separator
- self.filename = filename
- self.__yaml_dict = content
- self.content_type = content_type
- self.backup = backup
- self.load(content_type=self.content_type)
- if self.__yaml_dict is None:
- self.__yaml_dict = {}
- @property
- def separator(self):
- ''' getter method for separator '''
- return self._separator
- @separator.setter
- def separator(self, inc_sep):
- ''' setter method for separator '''
- self._separator = inc_sep
- @property
- def yaml_dict(self):
- ''' getter method for yaml_dict '''
- return self.__yaml_dict
- @yaml_dict.setter
- def yaml_dict(self, value):
- ''' setter method for yaml_dict '''
- self.__yaml_dict = value
- @staticmethod
- def parse_key(key, sep='.'):
- '''parse the key allowing the appropriate separator'''
- common_separators = list(Yedit.com_sep - set([sep]))
- return re.findall(Yedit.re_key.format(''.join(common_separators)), key)
- @staticmethod
- def valid_key(key, sep='.'):
- '''validate the incoming key'''
- common_separators = list(Yedit.com_sep - set([sep]))
- if not re.match(Yedit.re_valid_key.format(''.join(common_separators)), key):
- return False
- return True
- @staticmethod
- def remove_entry(data, key, sep='.'):
- ''' remove data at location key '''
- if key == '' and isinstance(data, dict):
- data.clear()
- return True
- elif key == '' and isinstance(data, list):
- del data[:]
- return True
- if not (key and Yedit.valid_key(key, sep)) and \
- isinstance(data, (list, dict)):
- return None
- key_indexes = Yedit.parse_key(key, sep)
- for arr_ind, dict_key in key_indexes[:-1]:
- if dict_key and isinstance(data, dict):
- data = data.get(dict_key)
- elif (arr_ind and isinstance(data, list) and
- int(arr_ind) <= len(data) - 1):
- data = data[int(arr_ind)]
- else:
- return None
-
-
- if key_indexes[-1][0]:
- if isinstance(data, list) and int(key_indexes[-1][0]) <= len(data) - 1:
- del data[int(key_indexes[-1][0])]
- return True
-
- elif key_indexes[-1][1]:
- if isinstance(data, dict):
- del data[key_indexes[-1][1]]
- return True
- @staticmethod
- def add_entry(data, key, item=None, sep='.'):
- ''' Get an item from a dictionary with key notation a.b.c
- d = {'a': {'b': 'c'}}}
- key = a#b
- return c
- '''
- if key == '':
- pass
- elif (not (key and Yedit.valid_key(key, sep)) and
- isinstance(data, (list, dict))):
- return None
- key_indexes = Yedit.parse_key(key, sep)
- for arr_ind, dict_key in key_indexes[:-1]:
- if dict_key:
- if isinstance(data, dict) and dict_key in data and data[dict_key]:
- data = data[dict_key]
- continue
- elif data and not isinstance(data, dict):
- raise YeditException("Unexpected item type found while going through key " +
- "path: {} (at key: {})".format(key, dict_key))
- data[dict_key] = {}
- data = data[dict_key]
- elif (arr_ind and isinstance(data, list) and
- int(arr_ind) <= len(data) - 1):
- data = data[int(arr_ind)]
- else:
- raise YeditException("Unexpected item type found while going through key path: {}".format(key))
- if key == '':
- data = item
-
-
- elif key_indexes[-1][0] and isinstance(data, list) and int(key_indexes[-1][0]) <= len(data) - 1:
- data[int(key_indexes[-1][0])] = item
-
- elif key_indexes[-1][1] and isinstance(data, dict):
- data[key_indexes[-1][1]] = item
-
-
-
- else:
- raise YeditException("Error adding to object at path: {}".format(key))
- return data
- @staticmethod
- def get_entry(data, key, sep='.'):
- ''' Get an item from a dictionary with key notation a.b.c
- d = {'a': {'b': 'c'}}}
- key = a.b
- return c
- '''
- if key == '':
- pass
- elif (not (key and Yedit.valid_key(key, sep)) and
- isinstance(data, (list, dict))):
- return None
- key_indexes = Yedit.parse_key(key, sep)
- for arr_ind, dict_key in key_indexes:
- if dict_key and isinstance(data, dict):
- data = data.get(dict_key)
- elif (arr_ind and isinstance(data, list) and
- int(arr_ind) <= len(data) - 1):
- data = data[int(arr_ind)]
- else:
- return None
- return data
- @staticmethod
- def _write(filename, contents):
- ''' Actually write the file contents to disk. This helps with mocking. '''
- tmp_filename = filename + '.yedit'
- with open(tmp_filename, 'w') as yfd:
- fcntl.flock(yfd, fcntl.LOCK_EX | fcntl.LOCK_NB)
- yfd.write(contents)
- fcntl.flock(yfd, fcntl.LOCK_UN)
- os.rename(tmp_filename, filename)
- def write(self):
- ''' write to file '''
- if not self.filename:
- raise YeditException('Please specify a filename.')
- if self.backup and self.file_exists():
- shutil.copy(self.filename, self.filename + '.orig')
-
- try:
- self.yaml_dict.fa.set_block_style()
- except AttributeError:
- pass
-
- if self.content_type == 'yaml':
- try:
- Yedit._write(self.filename, yaml.dump(self.yaml_dict, Dumper=yaml.RoundTripDumper))
- except AttributeError:
- Yedit._write(self.filename, yaml.safe_dump(self.yaml_dict, default_flow_style=False))
- elif self.content_type == 'json':
- Yedit._write(self.filename, json.dumps(self.yaml_dict, indent=4, sort_keys=True))
- else:
- raise YeditException('Unsupported content_type: {}.'.format(self.content_type) +
- 'Please specify a content_type of yaml or json.')
- return (True, self.yaml_dict)
- def read(self):
- ''' read from file '''
-
- if self.filename is None or not self.file_exists():
- return None
- contents = None
- with open(self.filename) as yfd:
- contents = yfd.read()
- return contents
- def file_exists(self):
- ''' return whether file exists '''
- if os.path.exists(self.filename):
- return True
- return False
- def load(self, content_type='yaml'):
- ''' return yaml file '''
- contents = self.read()
- if not contents and not self.content:
- return None
- if self.content:
- if isinstance(self.content, dict):
- self.yaml_dict = self.content
- return self.yaml_dict
- elif isinstance(self.content, str):
- contents = self.content
-
- try:
- if content_type == 'yaml' and contents:
-
- try:
- self.yaml_dict.fa.set_block_style()
- except AttributeError:
- pass
-
- try:
- self.yaml_dict = yaml.load(contents, yaml.RoundTripLoader)
- except AttributeError:
- self.yaml_dict = yaml.safe_load(contents)
-
- try:
- self.yaml_dict.fa.set_block_style()
- except AttributeError:
- pass
- elif content_type == 'json' and contents:
- self.yaml_dict = json.loads(contents)
- except yaml.YAMLError as err:
-
- raise YeditException('Problem with loading yaml file. {}'.format(err))
- return self.yaml_dict
- def get(self, key):
- ''' get a specified key'''
- try:
- entry = Yedit.get_entry(self.yaml_dict, key, self.separator)
- except KeyError:
- entry = None
- return entry
- def pop(self, path, key_or_item):
- ''' remove a key, value pair from a dict or an item for a list'''
- try:
- entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
- except KeyError:
- entry = None
- if entry is None:
- return (False, self.yaml_dict)
- if isinstance(entry, dict):
-
-
- if key_or_item in entry:
- entry.pop(key_or_item)
- return (True, self.yaml_dict)
- return (False, self.yaml_dict)
- elif isinstance(entry, list):
-
-
- ind = None
- try:
- ind = entry.index(key_or_item)
- except ValueError:
- return (False, self.yaml_dict)
- entry.pop(ind)
- return (True, self.yaml_dict)
- return (False, self.yaml_dict)
- def delete(self, path):
- ''' remove path from a dict'''
- try:
- entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
- except KeyError:
- entry = None
- if entry is None:
- return (False, self.yaml_dict)
- result = Yedit.remove_entry(self.yaml_dict, path, self.separator)
- if not result:
- return (False, self.yaml_dict)
- return (True, self.yaml_dict)
- def exists(self, path, value):
- ''' check if value exists at path'''
- try:
- entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
- except KeyError:
- entry = None
- if isinstance(entry, list):
- if value in entry:
- return True
- return False
- elif isinstance(entry, dict):
- if isinstance(value, dict):
- rval = False
- for key, val in value.items():
- if entry[key] != val:
- rval = False
- break
- else:
- rval = True
- return rval
- return value in entry
- return entry == value
- def append(self, path, value):
- '''append value to a list'''
- try:
- entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
- except KeyError:
- entry = None
- if entry is None:
- self.put(path, [])
- entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
- if not isinstance(entry, list):
- return (False, self.yaml_dict)
-
-
-
- entry.append(value)
- return (True, self.yaml_dict)
-
- def update(self, path, value, index=None, curr_value=None):
- ''' put path, value into a dict '''
- try:
- entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
- except KeyError:
- entry = None
- if isinstance(entry, dict):
-
-
- if not isinstance(value, dict):
- raise YeditException('Cannot replace key, value entry in dict with non-dict type. ' +
- 'value=[{}] type=[{}]'.format(value, type(value)))
- entry.update(value)
- return (True, self.yaml_dict)
- elif isinstance(entry, list):
-
-
- ind = None
- if curr_value:
- try:
- ind = entry.index(curr_value)
- except ValueError:
- return (False, self.yaml_dict)
- elif index is not None:
- ind = index
- if ind is not None and entry[ind] != value:
- entry[ind] = value
- return (True, self.yaml_dict)
-
- try:
- ind = entry.index(value)
- except ValueError:
-
- entry.append(value)
- return (True, self.yaml_dict)
-
- if ind is not None:
- return (False, self.yaml_dict)
- return (False, self.yaml_dict)
- def put(self, path, value):
- ''' put path, value into a dict '''
- try:
- entry = Yedit.get_entry(self.yaml_dict, path, self.separator)
- except KeyError:
- entry = None
- if entry == value:
- return (False, self.yaml_dict)
-
-
- try:
- tmp_copy = yaml.load(yaml.round_trip_dump(self.yaml_dict,
- default_flow_style=False),
- yaml.RoundTripLoader)
- except AttributeError:
- tmp_copy = copy.deepcopy(self.yaml_dict)
-
- try:
- tmp_copy.fa.set_block_style()
- except AttributeError:
- pass
- result = Yedit.add_entry(tmp_copy, path, value, self.separator)
- if result is None:
- return (False, self.yaml_dict)
-
-
-
- if path == '':
- if isinstance(result, list) or isinstance(result, dict):
- self.yaml_dict = result
- return (True, self.yaml_dict)
- return (False, self.yaml_dict)
- self.yaml_dict = tmp_copy
- return (True, self.yaml_dict)
- def create(self, path, value):
- ''' create a yaml file '''
- if not self.file_exists():
-
-
- try:
- tmp_copy = yaml.load(yaml.round_trip_dump(self.yaml_dict,
- default_flow_style=False),
- yaml.RoundTripLoader)
- except AttributeError:
- tmp_copy = copy.deepcopy(self.yaml_dict)
-
- try:
- tmp_copy.fa.set_block_style()
- except AttributeError:
- pass
- result = Yedit.add_entry(tmp_copy, path, value, self.separator)
- if result is not None:
- self.yaml_dict = tmp_copy
- return (True, self.yaml_dict)
- return (False, self.yaml_dict)
- @staticmethod
- def get_curr_value(invalue, val_type):
- '''return the current value'''
- if invalue is None:
- return None
- curr_value = invalue
- if val_type == 'yaml':
- try:
-
-
- curr_value = yaml.safe_load(invalue, Loader=yaml.RoundTripLoader)
- except AttributeError:
- curr_value = yaml.safe_load(invalue)
- elif val_type == 'json':
- curr_value = json.loads(invalue)
- return curr_value
- @staticmethod
- def parse_value(inc_value, vtype=''):
- '''determine value type passed'''
- true_bools = ['y', 'Y', 'yes', 'Yes', 'YES', 'true', 'True', 'TRUE',
- 'on', 'On', 'ON', ]
- false_bools = ['n', 'N', 'no', 'No', 'NO', 'false', 'False', 'FALSE',
- 'off', 'Off', 'OFF']
-
-
- if isinstance(inc_value, str) and 'bool' in vtype:
- if inc_value not in true_bools and inc_value not in false_bools:
- raise YeditException('Not a boolean type. str=[{}] vtype=[{}]'.format(inc_value, vtype))
- elif isinstance(inc_value, bool) and 'str' in vtype:
- inc_value = str(inc_value)
-
- if isinstance(inc_value, str) and inc_value == '':
- pass
-
- elif isinstance(inc_value, str) and 'str' not in vtype:
- try:
- inc_value = yaml.safe_load(inc_value)
- except Exception:
- raise YeditException('Could not determine type of incoming value. ' +
- 'value=[{}] vtype=[{}]'.format(type(inc_value), vtype))
- return inc_value
- @staticmethod
- def process_edits(edits, yamlfile):
- '''run through a list of edits and process them one-by-one'''
- results = []
- for edit in edits:
- value = Yedit.parse_value(edit['value'], edit.get('value_type', ''))
- if edit.get('action') == 'update':
-
- curr_value = Yedit.get_curr_value(
- Yedit.parse_value(edit.get('curr_value')),
- edit.get('curr_value_format'))
- rval = yamlfile.update(edit['key'],
- value,
- edit.get('index'),
- curr_value)
- elif edit.get('action') == 'append':
- rval = yamlfile.append(edit['key'], value)
- else:
- rval = yamlfile.put(edit['key'], value)
- if rval[0]:
- results.append({'key': edit['key'], 'edit': rval[1]})
- return {'changed': len(results) > 0, 'results': results}
-
- @staticmethod
- def run_ansible(params):
- '''perform the idempotent crud operations'''
- yamlfile = Yedit(filename=params['src'],
- backup=params['backup'],
- content_type=params['content_type'],
- separator=params['separator'])
- state = params['state']
- if params['src']:
- rval = yamlfile.load()
- if yamlfile.yaml_dict is None and state != 'present':
- return {'failed': True,
- 'msg': 'Error opening file [{}]. Verify that the '.format(params['src']) +
- 'file exists, that it is has correct permissions, and is valid yaml.'}
- if state == 'list':
- if params['content']:
- content = Yedit.parse_value(params['content'], params['content_type'])
- yamlfile.yaml_dict = content
- if params['key']:
- rval = yamlfile.get(params['key'])
- return {'changed': False, 'result': rval, 'state': state}
- elif state == 'absent':
- if params['content']:
- content = Yedit.parse_value(params['content'], params['content_type'])
- yamlfile.yaml_dict = content
- if params['update']:
- rval = yamlfile.pop(params['key'], params['value'])
- else:
- rval = yamlfile.delete(params['key'])
- if rval[0] and params['src']:
- yamlfile.write()
- return {'changed': rval[0], 'result': rval[1], 'state': state}
- elif state == 'present':
-
- if params['content']:
- content = Yedit.parse_value(params['content'], params['content_type'])
-
- if yamlfile.yaml_dict == content and \
- params['value'] is None:
- return {'changed': False, 'result': yamlfile.yaml_dict, 'state': state}
- yamlfile.yaml_dict = content
-
-
-
- edits = []
- _edit = {}
- if params['value'] is not None:
- _edit['value'] = params['value']
- _edit['value_type'] = params['value_type']
- _edit['key'] = params['key']
- if params['update']:
- _edit['action'] = 'update'
- _edit['curr_value'] = params['curr_value']
- _edit['curr_value_format'] = params['curr_value_format']
- _edit['index'] = params['index']
- elif params['append']:
- _edit['action'] = 'append'
- edits.append(_edit)
- elif params['edits'] is not None:
- edits = params['edits']
- if edits:
- results = Yedit.process_edits(edits, yamlfile)
-
- if results['changed'] and params['src']:
- yamlfile.write()
- return {'changed': results['changed'], 'result': results['results'], 'state': state}
-
- if params['src']:
-
- rval = yamlfile.write()
- return {'changed': rval[0],
- 'result': rval[1],
- 'state': state}
-
- return {'changed': False, 'result': yamlfile.yaml_dict, 'state': state}
- return {'failed': True, 'msg': 'Unkown state passed'}
- class OpenShiftCLIError(Exception):
- '''Exception class for openshiftcli'''
- pass
- ADDITIONAL_PATH_LOOKUPS = ['/usr/local/bin', os.path.expanduser('~/bin')]
- def locate_oc_binary():
- ''' Find and return oc binary file '''
-
-
-
- paths = os.environ.get("PATH", os.defpath).split(os.pathsep) + ADDITIONAL_PATH_LOOKUPS
- oc_binary = 'oc'
-
- try:
- which_result = shutil.which(oc_binary, path=os.pathsep.join(paths))
- if which_result is not None:
- oc_binary = which_result
- except AttributeError:
- for path in paths:
- if os.path.exists(os.path.join(path, oc_binary)):
- oc_binary = os.path.join(path, oc_binary)
- break
- return oc_binary
- class OpenShiftCLI(object):
- ''' Class to wrap the command line tools '''
- def __init__(self,
- namespace,
- kubeconfig='/etc/origin/master/admin.kubeconfig',
- verbose=False,
- all_namespaces=False):
- ''' Constructor for OpenshiftCLI '''
- self.namespace = namespace
- self.verbose = verbose
- self.kubeconfig = Utils.create_tmpfile_copy(kubeconfig)
- self.all_namespaces = all_namespaces
- self.oc_binary = locate_oc_binary()
-
-
- def _replace_content(self, resource, rname, content, force=False, sep='.'):
- ''' replace the current object with the content '''
- res = self._get(resource, rname)
- if not res['results']:
- return res
- fname = Utils.create_tmpfile(rname + '-')
- yed = Yedit(fname, res['results'][0], separator=sep)
- changes = []
- for key, value in content.items():
- changes.append(yed.put(key, value))
- if any([change[0] for change in changes]):
- yed.write()
- atexit.register(Utils.cleanup, [fname])
- return self._replace(fname, force)
- return {'returncode': 0, 'updated': False}
- def _replace(self, fname, force=False):
- '''replace the current object with oc replace'''
-
-
- yed = Yedit(fname)
- results = yed.delete('metadata.resourceVersion')
- if results[0]:
- yed.write()
- cmd = ['replace', '-f', fname]
- if force:
- cmd.append('--force')
- return self.openshift_cmd(cmd)
- def _create_from_content(self, rname, content):
- '''create a temporary file and then call oc create on it'''
- fname = Utils.create_tmpfile(rname + '-')
- yed = Yedit(fname, content=content)
- yed.write()
- atexit.register(Utils.cleanup, [fname])
- return self._create(fname)
- def _create(self, fname):
- '''call oc create on a filename'''
- return self.openshift_cmd(['create', '-f', fname])
- def _delete(self, resource, name=None, selector=None):
- '''call oc delete on a resource'''
- cmd = ['delete', resource]
- if selector is not None:
- cmd.append('--selector={}'.format(selector))
- elif name is not None:
- cmd.append(name)
- else:
- raise OpenShiftCLIError('Either name or selector is required when calling delete.')
- return self.openshift_cmd(cmd)
- def _process(self, template_name, create=False, params=None, template_data=None):
- '''process a template
- template_name: the name of the template to process
- create: whether to send to oc create after processing
- params: the parameters for the template
- template_data: the incoming template's data; instead of a file
- '''
- cmd = ['process']
- if template_data:
- cmd.extend(['-f', '-'])
- else:
- cmd.append(template_name)
- if params:
- param_str = ["{}={}".format(key, str(value).replace("'", r'"')) for key, value in params.items()]
- cmd.append('-v')
- cmd.extend(param_str)
- results = self.openshift_cmd(cmd, output=True, input_data=template_data)
- if results['returncode'] != 0 or not create:
- return results
- fname = Utils.create_tmpfile(template_name + '-')
- yed = Yedit(fname, results['results'])
- yed.write()
- atexit.register(Utils.cleanup, [fname])
- return self.openshift_cmd(['create', '-f', fname])
- def _get(self, resource, name=None, selector=None):
- '''return a resource by name '''
- cmd = ['get', resource]
- if selector is not None:
- cmd.append('--selector={}'.format(selector))
- elif name is not None:
- cmd.append(name)
- cmd.extend(['-o', 'json'])
- rval = self.openshift_cmd(cmd, output=True)
-
- if 'items' in rval:
- rval['results'] = rval['items']
- elif not isinstance(rval['results'], list):
- rval['results'] = [rval['results']]
- return rval
- def _schedulable(self, node=None, selector=None, schedulable=True):
- ''' perform oadm manage-node scheduable '''
- cmd = ['manage-node']
- if node:
- cmd.extend(node)
- else:
- cmd.append('--selector={}'.format(selector))
- cmd.append('--schedulable={}'.format(schedulable))
- return self.openshift_cmd(cmd, oadm=True, output=True, output_type='raw')
- def _list_pods(self, node=None, selector=None, pod_selector=None):
- ''' perform oadm list pods
- node: the node in which to list pods
- selector: the label selector filter if provided
- pod_selector: the pod selector filter if provided
- '''
- cmd = ['manage-node']
- if node:
- cmd.extend(node)
- else:
- cmd.append('--selector={}'.format(selector))
- if pod_selector:
- cmd.append('--pod-selector={}'.format(pod_selector))
- cmd.extend(['--list-pods', '-o', 'json'])
- return self.openshift_cmd(cmd, oadm=True, output=True, output_type='raw')
-
- def _evacuate(self, node=None, selector=None, pod_selector=None, dry_run=False, grace_period=None, force=False):
- ''' perform oadm manage-node evacuate '''
- cmd = ['manage-node']
- if node:
- cmd.extend(node)
- else:
- cmd.append('--selector={}'.format(selector))
- if dry_run:
- cmd.append('--dry-run')
- if pod_selector:
- cmd.append('--pod-selector={}'.format(pod_selector))
- if grace_period:
- cmd.append('--grace-period={}'.format(int(grace_period)))
- if force:
- cmd.append('--force')
- cmd.append('--evacuate')
- return self.openshift_cmd(cmd, oadm=True, output=True, output_type='raw')
- def _version(self):
- ''' return the openshift version'''
- return self.openshift_cmd(['version'], output=True, output_type='raw')
- def _import_image(self, url=None, name=None, tag=None):
- ''' perform image import '''
- cmd = ['import-image']
- image = '{0}'.format(name)
- if tag:
- image += ':{0}'.format(tag)
- cmd.append(image)
- if url:
- cmd.append('--from={0}/{1}'.format(url, image))
- cmd.append('-n{0}'.format(self.namespace))
- cmd.append('--confirm')
- return self.openshift_cmd(cmd)
- def _run(self, cmds, input_data):
- ''' Actually executes the command. This makes mocking easier. '''
- curr_env = os.environ.copy()
- curr_env.update({'KUBECONFIG': self.kubeconfig})
- proc = subprocess.Popen(cmds,
- stdin=subprocess.PIPE,
- stdout=subprocess.PIPE,
- stderr=subprocess.PIPE,
- env=curr_env)
- stdout, stderr = proc.communicate(input_data)
- return proc.returncode, stdout.decode('utf-8'), stderr.decode('utf-8')
-
- def openshift_cmd(self, cmd, oadm=False, output=False, output_type='json', input_data=None):
- '''Base command for oc '''
- cmds = [self.oc_binary]
- if oadm:
- cmds.append('adm')
- cmds.extend(cmd)
- if self.all_namespaces:
- cmds.extend(['--all-namespaces'])
- elif self.namespace is not None and self.namespace.lower() not in ['none', 'emtpy']:
- cmds.extend(['-n', self.namespace])
- if self.verbose:
- print(' '.join(cmds))
- try:
- returncode, stdout, stderr = self._run(cmds, input_data)
- except OSError as ex:
- returncode, stdout, stderr = 1, '', 'Failed to execute {}: {}'.format(subprocess.list2cmdline(cmds), ex)
- rval = {"returncode": returncode,
- "cmd": ' '.join(cmds)}
- if output_type == 'json':
- rval['results'] = {}
- if output and stdout:
- try:
- rval['results'] = json.loads(stdout)
- except ValueError as verr:
- if "No JSON object could be decoded" in verr.args:
- rval['err'] = verr.args
- elif output_type == 'raw':
- rval['results'] = stdout if output else ''
- if self.verbose:
- print("STDOUT: {0}".format(stdout))
- print("STDERR: {0}".format(stderr))
- if 'err' in rval or returncode != 0:
- rval.update({"stderr": stderr,
- "stdout": stdout})
- return rval
- class Utils(object):
- ''' utilities for openshiftcli modules '''
- @staticmethod
- def _write(filename, contents):
- ''' Actually write the file contents to disk. This helps with mocking. '''
- with open(filename, 'w') as sfd:
- sfd.write(str(contents))
- @staticmethod
- def create_tmp_file_from_contents(rname, data, ftype='yaml'):
- ''' create a file in tmp with name and contents'''
- tmp = Utils.create_tmpfile(prefix=rname)
- if ftype == 'yaml':
-
-
- if hasattr(yaml, 'RoundTripDumper'):
- Utils._write(tmp, yaml.dump(data, Dumper=yaml.RoundTripDumper))
- else:
- Utils._write(tmp, yaml.safe_dump(data, default_flow_style=False))
- elif ftype == 'json':
- Utils._write(tmp, json.dumps(data))
- else:
- Utils._write(tmp, data)
-
- atexit.register(Utils.cleanup, [tmp])
- return tmp
- @staticmethod
- def create_tmpfile_copy(inc_file):
- '''create a temporary copy of a file'''
- tmpfile = Utils.create_tmpfile('lib_openshift-')
- Utils._write(tmpfile, open(inc_file).read())
-
- atexit.register(Utils.cleanup, [tmpfile])
- return tmpfile
- @staticmethod
- def create_tmpfile(prefix='tmp'):
- ''' Generates and returns a temporary file name '''
- with tempfile.NamedTemporaryFile(prefix=prefix, delete=False) as tmp:
- return tmp.name
- @staticmethod
- def create_tmp_files_from_contents(content, content_type=None):
- '''Turn an array of dict: filename, content into a files array'''
- if not isinstance(content, list):
- content = [content]
- files = []
- for item in content:
- path = Utils.create_tmp_file_from_contents(item['path'] + '-',
- item['data'],
- ftype=content_type)
- files.append({'name': os.path.basename(item['path']),
- 'path': path})
- return files
- @staticmethod
- def cleanup(files):
- '''Clean up on exit '''
- for sfile in files:
- if os.path.exists(sfile):
- if os.path.isdir(sfile):
- shutil.rmtree(sfile)
- elif os.path.isfile(sfile):
- os.remove(sfile)
- @staticmethod
- def exists(results, _name):
- ''' Check to see if the results include the name '''
- if not results:
- return False
- if Utils.find_result(results, _name):
- return True
- return False
- @staticmethod
- def find_result(results, _name):
- ''' Find the specified result by name'''
- rval = None
- for result in results:
- if 'metadata' in result and result['metadata']['name'] == _name:
- rval = result
- break
- return rval
- @staticmethod
- def get_resource_file(sfile, sfile_type='yaml'):
- ''' return the service file '''
- contents = None
- with open(sfile) as sfd:
- contents = sfd.read()
- if sfile_type == 'yaml':
-
-
- if hasattr(yaml, 'RoundTripLoader'):
- contents = yaml.load(contents, yaml.RoundTripLoader)
- else:
- contents = yaml.safe_load(contents)
- elif sfile_type == 'json':
- contents = json.loads(contents)
- return contents
- @staticmethod
- def filter_versions(stdout):
- ''' filter the oc version output '''
- version_dict = {}
- version_search = ['oc', 'openshift', 'kubernetes']
- for line in stdout.strip().split('\n'):
- for term in version_search:
- if not line:
- continue
- if line.startswith(term):
- version_dict[term] = line.split()[-1]
-
-
- if "openshift" not in version_dict:
- version_dict["openshift"] = version_dict["oc"]
- return version_dict
- @staticmethod
- def add_custom_versions(versions):
- ''' create custom versions strings '''
- versions_dict = {}
- for tech, version in versions.items():
-
- if "-" in version:
- version = version.split("-")[0]
- if version.startswith('v'):
- versions_dict[tech + '_numeric'] = version[1:].split('+')[0]
-
- versions_dict[tech + '_short'] = version[1:4]
- return versions_dict
- @staticmethod
- def openshift_installed():
- ''' check if openshift is installed '''
- import rpm
- transaction_set = rpm.TransactionSet()
- rpmquery = transaction_set.dbMatch("name", "atomic-openshift")
- return rpmquery.count() > 0
-
-
- @staticmethod
- def check_def_equal(user_def, result_def, skip_keys=None, debug=False):
- ''' Given a user defined definition, compare it with the results given back by our query. '''
-
- skip = ['metadata', 'status']
- if skip_keys:
- skip.extend(skip_keys)
- for key, value in result_def.items():
- if key in skip:
- continue
-
- if isinstance(value, list):
- if key not in user_def:
- if debug:
- print('User data does not have key [%s]' % key)
- print('User data: %s' % user_def)
- return False
- if not isinstance(user_def[key], list):
- if debug:
- print('user_def[key] is not a list key=[%s] user_def[key]=%s' % (key, user_def[key]))
- return False
- if len(user_def[key]) != len(value):
- if debug:
- print("List lengths are not equal.")
- print("key=[%s]: user_def[%s] != value[%s]" % (key, len(user_def[key]), len(value)))
- print("user_def: %s" % user_def[key])
- print("value: %s" % value)
- return False
- for values in zip(user_def[key], value):
- if isinstance(values[0], dict) and isinstance(values[1], dict):
- if debug:
- print('sending list - list')
- print(type(values[0]))
- print(type(values[1]))
- result = Utils.check_def_equal(values[0], values[1], skip_keys=skip_keys, debug=debug)
- if not result:
- print('list compare returned false')
- return False
- elif value != user_def[key]:
- if debug:
- print('value should be identical')
- print(user_def[key])
- print(value)
- return False
-
- elif isinstance(value, dict):
- if key not in user_def:
- if debug:
- print("user_def does not have key [%s]" % key)
- return False
- if not isinstance(user_def[key], dict):
- if debug:
- print("dict returned false: not instance of dict")
- return False
-
- api_values = set(value.keys()) - set(skip)
- user_values = set(user_def[key].keys()) - set(skip)
- if api_values != user_values:
- if debug:
- print("keys are not equal in dict")
- print(user_values)
- print(api_values)
- return False
- result = Utils.check_def_equal(user_def[key], value, skip_keys=skip_keys, debug=debug)
- if not result:
- if debug:
- print("dict returned false")
- print(result)
- return False
-
- else:
- if key not in user_def or value != user_def[key]:
- if debug:
- print("value not equal; user_def does not have key")
- print(key)
- print(value)
- if key in user_def:
- print(user_def[key])
- return False
- if debug:
- print('returning true')
- return True
- class OpenShiftCLIConfig(object):
- '''Generic Config'''
- def __init__(self, rname, namespace, kubeconfig, options):
- self.kubeconfig = kubeconfig
- self.name = rname
- self.namespace = namespace
- self._options = options
- @property
- def config_options(self):
- ''' return config options '''
- return self._options
- def to_option_list(self, ascommalist=''):
- '''return all options as a string
- if ascommalist is set to the name of a key, and
- the value of that key is a dict, format the dict
- as a list of comma delimited key=value pairs'''
- return self.stringify(ascommalist)
- def stringify(self, ascommalist=''):
- ''' return the options hash as cli params in a string
- if ascommalist is set to the name of a key, and
- the value of that key is a dict, format the dict
- as a list of comma delimited key=value pairs '''
- rval = []
- for key in sorted(self.config_options.keys()):
- data = self.config_options[key]
- if data['include'] \
- and (data['value'] is not None or isinstance(data['value'], int)):
- if key == ascommalist:
- val = ','.join(['{}={}'.format(kk, vv) for kk, vv in sorted(data['value'].items())])
- else:
- val = data['value']
- rval.append('--{}={}'.format(key.replace('_', '-'), val))
- return rval
- class ProjectConfig(OpenShiftCLIConfig):
- ''' project config object '''
- def __init__(self, rname, namespace, kubeconfig, project_options):
- super(ProjectConfig, self).__init__(rname, None, kubeconfig, project_options)
- class Project(Yedit):
- ''' Class to wrap the oc command line tools '''
- annotations_path = "metadata.annotations"
- kind = 'Project'
- annotation_prefix = 'openshift.io/'
- def __init__(self, content):
- '''Project constructor'''
- super(Project, self).__init__(content=content)
- def get_annotations(self):
- ''' return the annotations'''
- return self.get(Project.annotations_path) or {}
- def add_annotations(self, inc_annos):
- ''' add an annotation to the other annotations'''
- if not isinstance(inc_annos, list):
- inc_annos = [inc_annos]
- annos = self.get_annotations()
- if not annos:
- self.put(Project.annotations_path, inc_annos)
- else:
- for anno in inc_annos:
- for key, value in anno.items():
- annos[key] = value
- return True
- def find_annotation(self, key):
- ''' find an annotation'''
- annotations = self.get_annotations()
- for anno in annotations:
- if Project.annotation_prefix + key == anno:
- return annotations[anno]
- return None
- def delete_annotation(self, inc_anno_keys):
- ''' remove an annotation from a project'''
- if not isinstance(inc_anno_keys, list):
- inc_anno_keys = [inc_anno_keys]
- annos = self.get(Project.annotations_path) or {}
- if not annos:
- return True
- removed = False
- for inc_anno in inc_anno_keys:
- anno = self.find_annotation(inc_anno)
- if anno:
- del annos[Project.annotation_prefix + anno]
- removed = True
- return removed
- def update_annotation(self, key, value):
- ''' remove an annotation for a project'''
- annos = self.get(Project.annotations_path) or {}
- if not annos:
- return True
- updated = False
- anno = self.find_annotation(key)
- if anno:
- annos[Project.annotation_prefix + key] = value
- updated = True
- else:
- self.add_annotations({Project.annotation_prefix + key: value})
- return updated
- class OCProject(OpenShiftCLI):
- ''' Project Class to manage project/namespace objects'''
- kind = 'namespace'
- def __init__(self,
- config,
- verbose=False):
- ''' Constructor for OCProject '''
- super(OCProject, self).__init__(None, config.kubeconfig)
- self.config = config
- self._project = None
- @property
- def project(self):
- ''' property for project'''
- if not self._project:
- self.get()
- return self._project
- @project.setter
- def project(self, data):
- ''' setter function for project propeorty'''
- self._project = data
- def exists(self):
- ''' return whether a project exists '''
- if self.project:
- return True
- return False
- def get(self):
- '''return project '''
- result = self._get(self.kind, self.config.name)
- if result['returncode'] == 0:
- self.project = Project(content=result['results'][0])
- result['results'] = self.project.yaml_dict
- elif 'namespaces "%s" not found' % self.config.name in result['stderr']:
- result = {'results': [], 'returncode': 0}
- return result
- def delete(self):
- '''delete the object'''
- return self._delete(self.kind, self.config.name)
- def create(self):
- '''create a project '''
- cmd = ['new-project', self.config.name]
- cmd.extend(self.config.to_option_list())
- return self.openshift_cmd(cmd, oadm=True)
- def update(self):
- '''update a project '''
- if self.config.config_options['display_name']['value'] is not None:
- self.project.update_annotation('display-name', self.config.config_options['display_name']['value'])
- if self.config.config_options['description']['value'] is not None:
- self.project.update_annotation('description', self.config.config_options['description']['value'])
-
- if self.config.config_options['node_selector']['value'] is not None:
- self.project.update_annotation('node-selector', self.config.config_options['node_selector']['value'])
- return self._replace_content(self.kind, self.config.name, self.project.yaml_dict)
- def needs_update(self):
- ''' verify an update is needed '''
- if self.config.config_options['display_name']['value'] is not None:
- result = self.project.find_annotation("display-name")
- if result != self.config.config_options['display_name']['value']:
- return True
- if self.config.config_options['description']['value'] is not None:
- result = self.project.find_annotation("description")
- if result != self.config.config_options['description']['value']:
- return True
- if self.config.config_options['node_selector']['value'] is not None:
- result = self.project.find_annotation("node-selector")
- if result != self.config.config_options['node_selector']['value']:
- return True
- return False
-
- @staticmethod
- def run_ansible(params, check_mode):
- '''run the idempotent ansible code'''
- node_selector = None
- if params['node_selector'] is not None:
- node_selector = ','.join(params['node_selector'])
- pconfig = ProjectConfig(
- params['name'],
- 'None',
- params['kubeconfig'],
- {
- 'admin': {'value': params['admin'], 'include': True},
- 'admin_role': {'value': params['admin_role'], 'include': True},
- 'description': {'value': params['description'], 'include': True},
- 'display_name': {'value': params['display_name'], 'include': True},
- 'node_selector': {'value': node_selector, 'include': True},
- },
- )
- oadm_project = OCProject(pconfig, verbose=params['debug'])
- state = params['state']
- api_rval = oadm_project.get()
-
-
-
- if state == 'list':
- return {'changed': False, 'results': api_rval['results'], 'state': state}
-
-
-
- if state == 'absent':
- if oadm_project.exists():
- if check_mode:
- return {'changed': True, 'msg': 'CHECK_MODE: Would have performed a delete.'}
- api_rval = oadm_project.delete()
- if api_rval['returncode'] != 0:
- return {'failed': True, 'msg': api_rval}
- return {'changed': True, 'results': api_rval, 'state': state}
- return {'changed': False, 'state': state}
- if state == 'present':
-
-
-
- if not oadm_project.exists():
- if check_mode:
- return {'changed': True, 'msg': 'CHECK_MODE: Would have performed a create.'}
-
- api_rval = oadm_project.create()
- if api_rval['returncode'] != 0:
- return {'failed': True, 'msg': api_rval}
-
- api_rval = oadm_project.get()
- if api_rval['returncode'] != 0:
- return {'failed': True, 'msg': api_rval}
- return {'changed': True, 'results': api_rval, 'state': state}
-
-
-
- if oadm_project.needs_update():
- if check_mode:
- return {'changed': True, 'msg': 'CHECK_MODE: Would have performed an update.'}
- api_rval = oadm_project.update()
- if api_rval['returncode'] != 0:
- return {'failed': True, 'msg': api_rval}
-
- api_rval = oadm_project.get()
- if api_rval['returncode'] != 0:
- return {'failed': True, 'msg': api_rval}
- return {'changed': True, 'results': api_rval, 'state': state}
- return {'changed': False, 'results': api_rval, 'state': state}
- return {'failed': True,
- 'changed': False,
- 'msg': 'Unknown state passed. [%s]' % state}
- def main():
- '''
- ansible oc module for project
- '''
- module = AnsibleModule(
- argument_spec=dict(
- kubeconfig=dict(default='/etc/origin/master/admin.kubeconfig', type='str'),
- state=dict(default='present', type='str',
- choices=['present', 'absent', 'list']),
- debug=dict(default=False, type='bool'),
- name=dict(default=None, require=True, type='str'),
- display_name=dict(default=None, type='str'),
- node_selector=dict(default=None, type='list'),
- description=dict(default=None, type='str'),
- admin=dict(default=None, type='str'),
- admin_role=dict(default='admin', type='str'),
- ),
- supports_check_mode=True,
- )
- rval = OCProject.run_ansible(module.params, module.check_mode)
- if 'failed' in rval:
- return module.fail_json(**rval)
- return module.exit_json(**rval)
- if __name__ == '__main__':
- main()
|