diff options
Diffstat (limited to 'opensteak/tools/opensteak/foreman_objects')
19 files changed, 0 insertions, 1552 deletions
diff --git a/opensteak/tools/opensteak/foreman_objects/.gitignore b/opensteak/tools/opensteak/foreman_objects/.gitignore deleted file mode 100644 index a65d046..0000000 --- a/opensteak/tools/opensteak/foreman_objects/.gitignore +++ /dev/null @@ -1,58 +0,0 @@ -# Byte-compiled / optimized / DLL files -__pycache__/ -*.py[cod] -*$py.class - -# C extensions -*.so - -# Distribution / packaging -.Python -env/ -build/ -develop-eggs/ -dist/ -downloads/ -eggs/ -.eggs/ -lib/ -lib64/ -parts/ -sdist/ -var/ -*.egg-info/ -.installed.cfg -*.egg - -# PyInstaller -# Usually these files are written by a python script from a template -# before PyInstaller builds the exe, so as to inject date/other infos into it. -*.manifest -*.spec - -# Installer logs -pip-log.txt -pip-delete-this-directory.txt - -# Unit test / coverage reports -htmlcov/ -.tox/ -.coverage -.coverage.* -.cache -nosetests.xml -coverage.xml -*,cover - -# Translations -*.mo -*.pot - -# Django stuff: -*.log - -# Sphinx documentation -docs/_build/ - -# PyBuilder -target/ diff --git a/opensteak/tools/opensteak/foreman_objects/__init__.py b/opensteak/tools/opensteak/foreman_objects/__init__.py deleted file mode 100644 index 01f9c9a..0000000 --- a/opensteak/tools/opensteak/foreman_objects/__init__.py +++ /dev/null @@ -1,18 +0,0 @@ -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -# This directory is a Python package. diff --git a/opensteak/tools/opensteak/foreman_objects/api.py b/opensteak/tools/opensteak/foreman_objects/api.py deleted file mode 100644 index dc99734..0000000 --- a/opensteak/tools/opensteak/foreman_objects/api.py +++ /dev/null @@ -1,197 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -import json -import requests -from requests_futures.sessions import FuturesSession -from pprint import pformat - - -class Api: - """ - Api class - Class to deal with the foreman API v2 - """ - def __init__(self, password, login='admin', ip='127.0.0.1', printErrors=False): - """ Function __init__ - Init the API with the connection params - - @param password: authentication password - @param password: authentication login - default is admin - @param ip: api ip - default is localhost - @return RETURN: self - """ - self.base_url = 'http://{}/api/v2/'.format(ip) - self.headers = {'Accept': 'version=2', - 'Content-Type': 'application/json; charset=UTF-8'} - self.auth = (login, password) - self.errorMsg = '' - self.printErrors = printErrors - - def list(self, obj, filter=False, only_id=False, limit=20): - """ Function list - Get the list of an object - - @param obj: object name ('hosts', 'puppetclasses'...) - @param filter: filter for objects - @param only_id: boolean to only return dict with name/id - @return RETURN: the list of the object - """ - self.url = '{}{}/?per_page={}'.format(self.base_url, obj, limit) - if filter: - self.url += '&search={}'.format(filter) - self.resp = requests.get(url=self.url, auth=self.auth, - headers=self.headers) - if only_id: - if self.__process_resp__(obj) is False: - return False - if type(self.res['results']) is list: - return dict((x['name'], x['id']) for x in self.res['results']) - elif type(self.res['results']) is dict: - r = {} - for v in self.res['results'].values(): - for vv in v: - r[vv['name']] = vv['id'] - return r - else: - return False - else: - return self.__process_resp__(obj) - - def get(self, obj, id, sub_object=None): - """ Function get - Get an object by id - - @param obj: object name ('hosts', 'puppetclasses'...) - @param id: the id of the object (name or id) - @return RETURN: the targeted object - """ - self.url = '{}{}/{}'.format(self.base_url, obj, id) - if sub_object: - self.url += '/' + sub_object - self.resp = requests.get(url=self.url, auth=self.auth, - headers=self.headers) - if self.__process_resp__(obj): - return self.res - return False - - def get_id_by_name(self, obj, name): - """ Function get_id_by_name - Get the id of an object - - @param obj: object name ('hosts', 'puppetclasses'...) - @param id: the id of the object (name or id) - @return RETURN: the targeted object - """ - list = self.list(obj, filter='name = "{}"'.format(name), - only_id=True, limit=1) - return list[name] if name in list.keys() else False - - def set(self, obj, id, payload, action='', async=False): - """ Function set - Set an object by id - - @param obj: object name ('hosts', 'puppetclasses'...) - @param id: the id of the object (name or id) - @param action: specific action of an object ('power'...) - @param payload: the dict of the payload - @param async: should this request be async, if true use - return.result() to get the response - @return RETURN: the server response - """ - self.url = '{}{}/{}'.format(self.base_url, obj, id) - if action: - self.url += '/{}'.format(action) - self.payload = json.dumps(payload) - if async: - session = FuturesSession() - return session.put(url=self.url, auth=self.auth, - headers=self.headers, data=self.payload) - else: - self.resp = requests.put(url=self.url, auth=self.auth, - headers=self.headers, data=self.payload) - if self.__process_resp__(obj): - return self.res - return False - - def create(self, obj, payload, async=False): - """ Function create - Create an new object - - @param obj: object name ('hosts', 'puppetclasses'...) - @param payload: the dict of the payload - @param async: should this request be async, if true use - return.result() to get the response - @return RETURN: the server response - """ - self.url = self.base_url + obj - self.payload = json.dumps(payload) - if async: - session = FuturesSession() - return session.post(url=self.url, auth=self.auth, - headers=self.headers, data=self.payload) - else: - self.resp = requests.post(url=self.url, auth=self.auth, - headers=self.headers, - data=self.payload) - return self.__process_resp__(obj) - - def delete(self, obj, id): - """ Function delete - Delete an object by id - - @param obj: object name ('hosts', 'puppetclasses'...) - @param id: the id of the object (name or id) - @return RETURN: the server response - """ - self.url = '{}{}/{}'.format(self.base_url, obj, id) - self.resp = requests.delete(url=self.url, - auth=self.auth, - headers=self.headers, ) - return self.__process_resp__(obj) - - def __process_resp__(self, obj): - """ Function __process_resp__ - Process the response sent by the server and store the result - - @param obj: object name ('hosts', 'puppetclasses'...) - @return RETURN: the server response - """ - self.last_obj = obj - if self.resp.status_code > 299: - self.errorMsg = ">> Error {} for object '{}'".format(self.resp.status_code, - self.last_obj) - try: - self.ret = json.loads(self.resp.text) - self.errorMsg += pformat(self.ret[list(self.ret.keys())[0]]) - except: - self.ret = self.resp.text - self.errorMsg += self.ret - if self.printErrors: - print(self.errorMsg) - return False - self.res = json.loads(self.resp.text) - if 'results' in self.res.keys(): - return self.res['results'] - return self.res - - def __str__(self): - ret = pformat(self.base_url) + "\n" - ret += pformat(self.headers) + "\n" - ret += pformat(self.auth) + "\n" - return ret diff --git a/opensteak/tools/opensteak/foreman_objects/architectures.py b/opensteak/tools/opensteak/foreman_objects/architectures.py deleted file mode 100644 index 5e4303e..0000000 --- a/opensteak/tools/opensteak/foreman_objects/architectures.py +++ /dev/null @@ -1,49 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects - - -class Architectures(ForemanObjects): - """ - Architectures class - """ - objName = 'architectures' - payloadObj = 'architecture' - - def checkAndCreate(self, key, payload, osIds): - """ Function checkAndCreate - Check if an architectures exists and create it if not - - @param key: The targeted architectures - @param payload: The targeted architectures description - @param osIds: The list of os ids liked with this architecture - @return RETURN: The id of the object - """ - if key not in self: - self[key] = payload - oid = self[key]['id'] - if not oid: - return False - #~ To be sure the OS list is good, we ensure our os are in the list - for os in self[key]['operatingsystems']: - osIds.add(os['id']) - self[key]["operatingsystem_ids"] = list(osIds) - if (len(self[key]['operatingsystems']) is not len(osIds)): - return False - return oid diff --git a/opensteak/tools/opensteak/foreman_objects/compute_resources.py b/opensteak/tools/opensteak/foreman_objects/compute_resources.py deleted file mode 100644 index 9ada9c4..0000000 --- a/opensteak/tools/opensteak/foreman_objects/compute_resources.py +++ /dev/null @@ -1,62 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects -from opensteak.foreman_objects.item import ForemanItem - - -class ComputeResources(ForemanObjects): - """ - HostGroups class - """ - objName = 'compute_resources' - payloadObj = 'compute_resource' - - def list(self): - """ Function list - list the hostgroups - - @return RETURN: List of ForemanItemHostsGroup objects - """ - return list(map(lambda x: ForemanItem(self.api, x['id'], x), - self.api.list(self.objName))) - - def __getitem__(self, key): - """ Function __getitem__ - Get an hostgroup - - @param key: The hostgroup name or ID - @return RETURN: The ForemanItemHostsGroup object of an host - """ - # Because Hostgroup did not support get by name we need to do it by id - if type(key) is not int: - key = self.getId(key) - ret = self.api.get(self.objName, key) - return ForemanItem(self.api, key, ret) - - def __delitem__(self, key): - """ Function __delitem__ - Delete an hostgroup - - @param key: The hostgroup name or ID - @return RETURN: The API result - """ - # Because Hostgroup did not support get by name we need to do it by id - if type(key) is not int: - key = self.getId(key) - return self.api.delete(self.objName, key) diff --git a/opensteak/tools/opensteak/foreman_objects/domains.py b/opensteak/tools/opensteak/foreman_objects/domains.py deleted file mode 100644 index 753833f..0000000 --- a/opensteak/tools/opensteak/foreman_objects/domains.py +++ /dev/null @@ -1,44 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects - - -class Domains(ForemanObjects): - """ - Domain class - """ - objName = 'domains' - payloadObj = 'domain' - - def load(self, id='0', name=''): - """ Function load - To be rewriten - - @param id: The Domain ID - @return RETURN: DESCRIPTION - """ - - if name: - id = self.__getIdByName__(name) - self.data = self.foreman.get('domains', id) - if 'parameters' in self.data: - self.params = self.data['parameters'] - else: - self.params = [] - self.name = self.data['name'] diff --git a/opensteak/tools/opensteak/foreman_objects/freeip.py b/opensteak/tools/opensteak/foreman_objects/freeip.py deleted file mode 100644 index 86c003f..0000000 --- a/opensteak/tools/opensteak/foreman_objects/freeip.py +++ /dev/null @@ -1,79 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -#~ from foreman.api import Api -import requests -from bs4 import BeautifulSoup -import sys -import json - -class FreeIP: - """ FreeIP return an available IP in the targeted network """ - - def __init__ (self, login, password): - """ Init: get authenticity token """ - with requests.session() as self.session: - try: - #~ 1/ Get login token and authentify - payload = {} - log_soup = BeautifulSoup(self.session.get('https://127.0.0.1/users/login', verify=False).text) - payload['utf8'] = log_soup.findAll('input',attrs={'name':'utf8'})[0].get('value') - payload['authenticity_token'] = log_soup.findAll('input',attrs={'name':'authenticity_token'})[0].get('value') - if payload['authenticity_token'] == None: - raise requests.exceptions.RequestException("Bad catch of authenticity_token") - payload['commit']='Login' - payload['login[login]'] = login - payload['login[password]'] = password - #~ 2/ Log in - r = self.session.post('https://127.0.0.1/users/login', verify=False, data=payload) - if r.status_code != 200: - raise requests.exceptions.RequestException("Bad login or password") - #~ Get token for host creation - log_soup = BeautifulSoup(self.session.get('https://127.0.0.1/hosts/new', verify=False).text) - self.authenticity_token = log_soup.findAll('input',attrs={'name':'authenticity_token'})[0].get('value') - if payload['authenticity_token'] == None: - raise requests.exceptions.RequestException("Bad catch of authenticity_token") - except requests.exceptions.RequestException as e: - print("Error connection Foreman to get a free ip") - print(e) - sys.exit(1) - pass - - def get(self, subnet, mac = ""): - payload = {"host_mac": mac, "subnet_id": subnet} - payload['authenticity_token'] = self.authenticity_token - try: - self.last_ip = json.loads(self.session.post('https://127.0.0.1/subnets/freeip', verify=False, data=payload).text)['ip'] - if payload['authenticity_token'] == None: - raise requests.exceptions.RequestException("Error getting free IP") - except requests.exceptions.RequestException as e: - print("Error connection Foreman to get a free ip") - print(e) - sys.exit(1) - return self.last_ip - - - -if __name__ == "__main__": - import pprint - import sys - if len(sys.argv) == 4: - f = FreeIP(sys.argv[1], sys.argv[2]) - print(f.get(sys.argv[3])) - else: - print('Error: Usage\npython {} foreman_user foreman_password subnet'.format(sys.argv[0])) diff --git a/opensteak/tools/opensteak/foreman_objects/hostgroups.py b/opensteak/tools/opensteak/foreman_objects/hostgroups.py deleted file mode 100644 index 55b8ba6..0000000 --- a/opensteak/tools/opensteak/foreman_objects/hostgroups.py +++ /dev/null @@ -1,103 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects -from opensteak.foreman_objects.itemHostsGroup import ItemHostsGroup -from pprint import pprint as pp - - -class HostGroups(ForemanObjects): - """ - HostGroups class - """ - objName = 'hostgroups' - payloadObj = 'hostgroup' - - def list(self): - """ Function list - list the hostgroups - - @return RETURN: List of ItemHostsGroup objects - """ - return list(map(lambda x: ItemHostsGroup(self.api, x['id'], x), - self.api.list(self.objName))) - - def __getitem__(self, key): - """ Function __getitem__ - Get an hostgroup - - @param key: The hostgroup name or ID - @return RETURN: The ItemHostsGroup object of an host - """ - # Because Hostgroup did not support get by name we need to do it by id - if type(key) is not int: - key = self.getId(key) - ret = self.api.get(self.objName, key) - return ItemHostsGroup(self.api, key, ret) - - def __delitem__(self, key): - """ Function __delitem__ - Delete an hostgroup - - @param key: The hostgroup name or ID - @return RETURN: The API result - """ - # Because Hostgroup did not support get by name we need to do it by id - if type(key) is not int: - key = self.getId(key) - return self.api.delete(self.objName, key) - - def checkAndCreate(self, key, payload, - hostgroupConf, - hostgroupParent, - puppetClassesId): - """ Function checkAndCreate - check And Create procedure for an hostgroup - - check the hostgroup is not existing - - create the hostgroup - - Add puppet classes from puppetClassesId - - Add params from hostgroupConf - - @param key: The hostgroup name or ID - @param payload: The description of the hostgroup - @param hostgroupConf: The configuration of the host group from the - foreman.conf - @param hostgroupParent: The id of the parent hostgroup - @param puppetClassesId: The dict of puppet classes ids in foreman - @return RETURN: The ItemHostsGroup object of an host - """ - if key not in self: - self[key] = payload - oid = self[key]['id'] - if not oid: - return False - - # Create Hostgroup classes - hostgroupClassIds = self[key]['puppetclass_ids'] - if 'classes' in hostgroupConf.keys(): - if not self[key].checkAndCreateClasses(puppetClassesId.values()): - print("Failed in classes") - return False - - # Set params - if 'params' in hostgroupConf.keys(): - if not self[key].checkAndCreateParams(hostgroupConf['params']): - print("Failed in params") - return False - - return oid diff --git a/opensteak/tools/opensteak/foreman_objects/hosts.py b/opensteak/tools/opensteak/foreman_objects/hosts.py deleted file mode 100644 index 95d47af..0000000 --- a/opensteak/tools/opensteak/foreman_objects/hosts.py +++ /dev/null @@ -1,142 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects -from opensteak.foreman_objects.itemHost import ItemHost -import time - - -class Hosts(ForemanObjects): - """ - Host sclass - """ - objName = 'hosts' - payloadObj = 'host' - - def list(self): - """ Function list - list the hosts - - @return RETURN: List of ItemHost objects - """ - return list(map(lambda x: ItemHost(self.api, x['id'], x), - self.api.list(self.objName))) - - def __getitem__(self, key): - """ Function __getitem__ - Get an host - - @param key: The host name or ID - @return RETURN: The ItemHost object of an host - """ - return ItemHost(self.api, key, self.api.get(self.objName, key)) - - def __printProgression__(self, status, msg, eol): - """ Function __printProgression__ - Print the creation progression or not - It uses the foreman.printer lib - - @param status: Status of the message - @param msg: Message - @param eol: End Of Line (to get a new line or not) - @return RETURN: None - """ - if self.printHostProgress: - self.__printProgression__(status, msg, eol=eol) - - def createVM(self, key, attributes, printHostProgress=False): - """ Function createVM - Create a Virtual Machine - - The creation of a VM with libVirt is a bit complexe. - We first create the element in foreman, the ask to start before - the result of the creation. - To do so, we make async calls to the API and check the results - - @param key: The host name or ID - @param attributes:The payload of the host creation - @param printHostProgress: The link to opensteak.printerlib - to print or not the - progression of the host creation - @return RETURN: The API result - """ - - self.printHostProgress = printHostProgress - self.async = True - # Create the VM in foreman - self.__printProgression__('In progress', - key + ' creation: push in Foreman', eol='\r') - future1 = self.api.create('hosts', attributes, async=True) - - # Wait before asking to power on the VM - sleep = 5 - for i in range(0, sleep): - time.sleep(1) - self.__printProgression__('In progress', - key + ' creation: start in {0}s' - .format(sleep - i), - eol='\r') - - # Power on the VM - self.__printProgression__('In progress', - key + ' creation: starting', eol='\r') - future2 = self[key].powerOn() - - # Show Power on result - if future2.result().status_code is 200: - self.__printProgression__('In progress', - key + ' creation: wait for end of boot', - eol='\r') - else: - self.__printProgression__(False, - key + ' creation: Error', - failed=str(future2.result().status_code)) - return False - # Show creation result - if future1.result().status_code is 200: - self.__printProgression__('In progress', - key + ' creation: created', - eol='\r') - else: - self.__printProgression__(False, - key + ' creation: Error', - failed=str(future1.result().status_code)) - return False - - # Wait for puppet catalog to be applied - loop_stop = False - while not loop_stop: - status = self[key].getStatus() - if status == 'No Changes' or status == 'Active': - self.__printProgression__(True, - key + ' creation: provisioning OK') - loop_stop = True - elif status == 'Error': - self.__printProgression__(False, - key + ' creation: Error', - failed="Error during provisioning") - loop_stop = True - return False - else: - self.__printProgression__('In progress', - key + ' creation: provisioning ({})' - .format(status), - eol='\r') - time.sleep(5) - - return True diff --git a/opensteak/tools/opensteak/foreman_objects/item.py b/opensteak/tools/opensteak/foreman_objects/item.py deleted file mode 100644 index f418f8c..0000000 --- a/opensteak/tools/opensteak/foreman_objects/item.py +++ /dev/null @@ -1,135 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# David Blaisonneau <david.blaisonneau@orange.com> -# Arnaud Morin <arnaud1.morin@orange.com> - -from pprint import pprint as pp - - -class ForemanItem(dict): - """ - Item class - Represent the content of a foreman object as a dict - """ - - def __init__(self, api, key, - objName, payloadObj, - *args, **kwargs): - """ Function __init__ - Represent the content of a foreman object as a dict - - @param api: The foreman api - @param key: The object Key - @param *args, **kwargs: the dict representation - @return RETURN: Itself - """ - self.api = api - self.key = key - if objName: - self.objName = objName - if payloadObj: - self.payloadObj = payloadObj - self.store = dict() - if args[0]: - self.update(dict(*args, **kwargs)) - # We get the smart class parameters for the good items - if objName in ['hosts', 'hostgroups', - 'puppet_classes', 'environments']: - from opensteak.foreman_objects.itemSmartClassParameter\ - import ItemSmartClassParameter - scp_ids = map(lambda x: x['id'], - self.api.list('{}/{}/smart_class_parameters' - .format(self.objName, key))) - scp_items = list(map(lambda x: ItemSmartClassParameter(self.api, x, - self.api.get('smart_class_parameters', x)), - scp_ids)) - scp = {'{}::{}'.format(x['puppetclass']['name'], - x['parameter']): x - for x in scp_items} - self.update({'smart_class_parameters_dict': scp}) - - def __setitem__(self, key, attributes): - """ Function __setitem__ - Set a parameter of a foreman object as a dict - - @param key: The key to modify - @param attribute: The data - @return RETURN: The API result - """ - if key is 'puppetclass_ids': - payload = {"puppetclass_id": attributes, - self.payloadObj + "_class": - {"puppetclass_id": attributes}} - return self.api.create("{}/{}/{}" - .format(self.objName, - self.key, - "puppetclass_ids"), - payload) - elif key is 'parameters': - payload = {"parameter": attributes} - return self.api.create("{}/{}/{}" - .format(self.objName, - self.key, - "parameters"), - payload) - else: - payload = {self.payloadObj: {key: attributes}} - return self.api.set(self.objName, self.key, payload) - - def getParam(self, name=None): - """ Function getParam - Return a dict of parameters or a parameter value - - @param key: The parameter name - @return RETURN: dict of parameters or a parameter value - """ - if 'parameters' in self.keys(): - l = {x['name']: x['value'] for x in self['parameters']} - if name: - if name in l.keys(): - return l[name] - else: - return False - else: - return l - - def checkAndCreateClasses(self, classes): - """ Function checkAndCreateClasses - Check and add puppet classe - - @param key: The parameter name - @param classes: The classes ids list - @return RETURN: boolean - """ - actual_classes = self['puppetclass_ids'] - for v in classes: - if v not in actual_classes: - self['puppetclass_ids'] = v - return list(classes).sort() is list(self['puppetclass_ids']).sort() - - def checkAndCreateParams(self, params): - """ Function checkAndCreateParams - Check and add global parameters - - @param key: The parameter name - @param params: The params dict - @return RETURN: boolean - """ - actual_params = self['param_ids'] - for k, v in params.items(): - if k not in actual_params: - self['parameters'] = {"name": k, "value": v} - return self['param_ids'].sort() == list(params.values()).sort() diff --git a/opensteak/tools/opensteak/foreman_objects/itemHost.py b/opensteak/tools/opensteak/foreman_objects/itemHost.py deleted file mode 100644 index c531e5c..0000000 --- a/opensteak/tools/opensteak/foreman_objects/itemHost.py +++ /dev/null @@ -1,141 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -import base64 -from string import Template -from opensteak.foreman_objects.item import ForemanItem - - -class ItemHost(ForemanItem): - """ - ItemHostsGroup class - Represent the content of a foreman hostgroup as a dict - """ - - objName = 'hosts' - payloadObj = 'host' - - def __init__(self, api, key, *args, **kwargs): - """ Function __init__ - Represent the content of a foreman object as a dict - - @param api: The foreman api - @param key: The object Key - @param *args, **kwargs: the dict representation - @return RETURN: Itself - """ - ForemanItem.__init__(self, api, key, - self.objName, self.payloadObj, - *args, **kwargs) - self.update({'puppetclass_ids': - self.api.list('{}/{}/puppetclass_ids' - .format(self.objName, key))}) - self.update({'param_ids': - list(self.api.list('{}/{}/parameters' - .format(self.objName, key), - only_id=True) - .keys())}) - - - def getStatus(self): - """ Function getStatus - Get the status of an host - - @return RETURN: The host status - """ - return self.api.get('hosts', self.key, 'status')['status'] - - def powerOn(self): - """ Function powerOn - Power on a host - - @return RETURN: The API result - """ - return self.api.set('hosts', self.key, - {"power_action": "start"}, - 'power', async=self.async) - - def getParamFromEnv(self, var, default=''): - """ Function getParamFromEnv - Search a parameter in the host environment - - @param var: the var name - @param hostgroup: the hostgroup item linked to this host - @param default: default value - @return RETURN: the value - """ - if self.getParam(var): - return self.getParam(var) - if self.hostgroup: - if self.hostgroup.getParam(var): - return self.hostgroup.getParam(var) - if self.domain.getParam('password'): - return self.domain.getParam('password') - else: - return default - - def getUserData(self, - hostgroup, - domain, - defaultPwd='', - defaultSshKey='', - proxyHostname='', - tplFolder='templates_metadata/'): - """ Function getUserData - Generate a userdata script for metadata server from Foreman API - - @param domain: the domain item linked to this host - @param hostgroup: the hostgroup item linked to this host - @param defaultPwd: the default password if no password is specified - in the host>hostgroup>domain params - @param defaultSshKey: the default ssh key if no password is specified - in the host>hostgroup>domain params - @param proxyHostname: hostname of the smartproxy - @param tplFolder: the templates folder - @return RETURN: the user data - """ - if 'user-data' in self.keys(): - return self['user-data'] - else: - self.hostgroup = hostgroup - self.domain = domain - if proxyHostname == '': - proxyHostname = 'foreman.' + domain - password = self.getParamFromEnv('password', defaultPwd) - sshauthkeys = self.getParamFromEnv('global_sshkey', defaultSshKey) - with open(tplFolder+'puppet.conf', 'rb') as puppet_file: - p = MyTemplate(puppet_file.read()) - enc_puppet_file = base64.b64encode(p.substitute( - foremanHostname=proxyHostname)) - with open(tplFolder+'cloud-init.tpl', 'r') as content_file: - s = MyTemplate(content_file.read()) - if sshauthkeys: - sshauthkeys = ' - '+sshauthkeys - self.userdata = s.substitute( - password=password, - fqdn=self['name'], - sshauthkeys=sshauthkeys, - foremanurlbuilt="http://{}/unattended/built" - .format(proxyHostname), - puppet_conf_content=enc_puppet_file.decode('utf-8')) - return self.userdata - - -class MyTemplate(Template): - delimiter = '%' - idpattern = r'[a-z][_a-z0-9]*' diff --git a/opensteak/tools/opensteak/foreman_objects/itemHostsGroup.py b/opensteak/tools/opensteak/foreman_objects/itemHostsGroup.py deleted file mode 100644 index d6a641c..0000000 --- a/opensteak/tools/opensteak/foreman_objects/itemHostsGroup.py +++ /dev/null @@ -1,50 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.item import ForemanItem - - -class ItemHostsGroup(ForemanItem): - """ - ItemHostsGroup class - Represent the content of a foreman hostgroup as a dict - """ - - objName = 'hostgroups' - payloadObj = 'hostgroup' - - def __init__(self, api, key, *args, **kwargs): - """ Function __init__ - Represent the content of a foreman object as a dict - - @param api: The foreman api - @param key: The object Key - @param *args, **kwargs: the dict representation - @return RETURN: Itself - """ - ForemanItem.__init__(self, api, key, - self.objName, self.payloadObj, - *args, **kwargs) - self.update({'puppetclass_ids': - self.api.list('{}/{}/puppetclass_ids' - .format(self.objName, key))}) - self.update({'param_ids': - list(self.api.list('{}/{}/parameters' - .format(self.objName, key), - only_id=True) - .keys())}) diff --git a/opensteak/tools/opensteak/foreman_objects/itemOverrideValues.py b/opensteak/tools/opensteak/foreman_objects/itemOverrideValues.py deleted file mode 100644 index 936185e..0000000 --- a/opensteak/tools/opensteak/foreman_objects/itemOverrideValues.py +++ /dev/null @@ -1,61 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - - -from opensteak.foreman_objects.item import ForemanItem -from pprint import pprint as pp - -class ItemOverrideValues(ForemanItem): - """ - ItemOverrideValues class - Represent the content of a foreman smart class parameter as a dict - """ - - objName = 'override_values' - payloadObj = 'override_value' - - def __init__(self, api, key, parentName, parentKey, *args, **kwargs): - """ Function __init__ - Represent the content of a foreman object as a dict - - @param api: The foreman api - @param key: The object Key - @param parentName: The object parent name (eg: smart_class_parameter) - @param parentKey: The object parent key - @param *args, **kwargs: the dict representation - @return RETURN: Itself - """ - self.parentName = parentName - self.parentKey = parentKey - ForemanItem.__init__(self, api, key, - self.objName, self.payloadObj, - *args, **kwargs) - - def __setitem__(self, key, attributes): - """ Function __setitem__ - Set a parameter of a foreman object as a dict - - @param key: The key to modify - @param attribute: The data - @return RETURN: The API result - """ - payload = {self.payloadObj: {key: attributes}} - return self.api.set('{}/{}/{}'.format(self.parentName, - self.parentKey, - self.objName), - self.key, payload) diff --git a/opensteak/tools/opensteak/foreman_objects/itemSmartClassParameter.py b/opensteak/tools/opensteak/foreman_objects/itemSmartClassParameter.py deleted file mode 100644 index 2d7ca2a..0000000 --- a/opensteak/tools/opensteak/foreman_objects/itemSmartClassParameter.py +++ /dev/null @@ -1,62 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - - -from opensteak.foreman_objects.item import ForemanItem -from opensteak.foreman_objects.itemOverrideValues import ItemOverrideValues - - -class ItemSmartClassParameter(ForemanItem): - """ - ItemSmartClassParameter class - Represent the content of a foreman smart class parameter as a dict - """ - - objName = 'smart_class_parameters' - payloadObj = 'smart_class_parameter' - - def __init__(self, api, key, *args, **kwargs): - """ Function __init__ - Represent the content of a foreman object as a dict - - @param api: The foreman api - @param key: The object Key - @param *args, **kwargs: the dict representation - @return RETURN: Itself - """ - ForemanItem.__init__(self, api, key, - self.objName, self.payloadObj, - *args, **kwargs) - self.update({'override_values': - list(map(lambda x: ItemOverrideValues(self.api, - x['id'], - self.objName, - key, - x), - self['override_values']))}) - - def __setitem__(self, key, attributes): - """ Function __setitem__ - Set a parameter of a foreman object as a dict - - @param key: The key to modify - @param attribute: The data - @return RETURN: The API result - """ - payload = {self.payloadObj: {key: attributes}} - return self.api.set(self.objName, self.key, payload) diff --git a/opensteak/tools/opensteak/foreman_objects/objects.py b/opensteak/tools/opensteak/foreman_objects/objects.py deleted file mode 100644 index c20c5a1..0000000 --- a/opensteak/tools/opensteak/foreman_objects/objects.py +++ /dev/null @@ -1,136 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.item import ForemanItem - - -class ForemanObjects: - """ - ForemanObjects class - Parent class for Foreman Objects - """ - - def __init__(self, api, objName=None, payloadObj=None): - """ Function __init__ - Init the foreman object - - @param api: The foreman API - @param objName: The object name (linked with the Foreman API) - @param payloadObj: The object name inside the payload (in general - the singular of objName) - @return RETURN: Itself - """ - - self.api = api - if objName: - self.objName = objName - if payloadObj: - self.payloadObj = payloadObj - # For asynchronous creations - self.async = False - - def __iter__(self): - """ Function __iter__ - - @return RETURN: The iteration of objects list - """ - return iter(self.list()) - - def __getitem__(self, key): - """ Function __getitem__ - - @param key: The targeted object - @return RETURN: A ForemanItem - """ - return ForemanItem(self.api, - key, - self.objName, - self.payloadObj, - self.api.get(self.objName, key)) - - def __setitem__(self, key, attributes): - """ Function __setitem__ - - @param key: The targeted object - @param attributes: The attributes to apply to the object - @return RETURN: API result if the object was not present, or False - """ - if key not in self: - payload = {self.payloadObj: {'name': key}} - payload[self.payloadObj].update(attributes) - return self.api.create(self.objName, payload, async=self.async) - return False - - def __delitem__(self, key): - """ Function __delitem__ - - @return RETURN: API result - """ - return self.api.delete(self.objName, key) - - def __contains__(self, key): - """ Function __contains__ - - @param key: The targeted object - @return RETURN: True if the object exists - """ - return bool(key in self.listName().keys()) - - def getId(self, key): - """ Function getId - Get the id of an object - - @param key: The targeted object - @return RETURN: The ID - """ - return self.api.get_id_by_name(self.objName, key) - - def list(self, limit=20): - """ Function list - Get the list of all objects - - @param key: The targeted object - @param limit: The limit of items to return - @return RETURN: A ForemanItem list - """ - return list(map(lambda x: - ForemanItem(self.api, x['id'], - self.objName, self.payloadObj, - x), - self.api.list(self.objName, limit=limit))) - - def listName(self): - """ Function listName - Get the list of all objects name with Ids - - @param key: The targeted object - @return RETURN: A dict of obejct name:id - """ - return self.api.list(self.objName, limit=999999, only_id=True) - - def checkAndCreate(self, key, payload): - """ Function checkAndCreate - Check if an object exists and create it if not - - @param key: The targeted object - @param payload: The targeted object description - @return RETURN: The id of the object - """ - if key not in self: - self[key] = payload - return self[key]['id'] diff --git a/opensteak/tools/opensteak/foreman_objects/operatingsystems.py b/opensteak/tools/opensteak/foreman_objects/operatingsystems.py deleted file mode 100644 index 8cce606..0000000 --- a/opensteak/tools/opensteak/foreman_objects/operatingsystems.py +++ /dev/null @@ -1,66 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects -from opensteak.foreman_objects.item import ForemanItem - - -class OperatingSystems(ForemanObjects): - """ - OperatingSystems class - """ - objName = 'operatingsystems' - payloadObj = 'operatingsystem' - - def __getitem__(self, key): - """ Function __getitem__ - - @param key: The operating system id/name - @return RETURN: The item - """ - ret = self.api.list(self.objName, - filter='title = "{}"'.format(key)) - if len(ret): - return ForemanItem(self.api, key, - self.objName, self.payloadObj, - ret[0]) - else: - return None - - def __setitem__(self, key, attributes): - """ Function __getitem__ - - @param key: The operating system id/name - @param attributes: The content of the operating system to create - @return RETURN: The API result - """ - if key not in self: - payload = {self.payloadObj: {'title': key}} - payload[self.payloadObj].update(attributes) - return self.api.create(self.objName, payload) - return False - - def listName(self): - """ Function listName - Get the list of all objects name with Ids - - @param key: The targeted object - @return RETURN: A dict of obejct name:id - """ - return { x['title']: x['id'] for x in self.api.list(self.objName, - limit=999999)} diff --git a/opensteak/tools/opensteak/foreman_objects/puppetClasses.py b/opensteak/tools/opensteak/foreman_objects/puppetClasses.py deleted file mode 100644 index 7f397f2..0000000 --- a/opensteak/tools/opensteak/foreman_objects/puppetClasses.py +++ /dev/null @@ -1,46 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects -from opensteak.foreman_objects.item import ForemanItem -from pprint import pprint as pp - - -class PuppetClasses(ForemanObjects): - """ - OperatingSystems class - """ - objName = 'puppetclasses' - payloadObj = 'puppetclass' - - def list(self, limit=20): - """ Function list - Get the list of all objects - - @param key: The targeted object - @param limit: The limit of items to return - @return RETURN: A ForemanItem list - """ - puppetClassList = list() - for v in self.api.list(self.objName, limit=limit).values(): - puppetClassList.extend(v) - return list(map(lambda x: - ForemanItem(self.api, x['id'], - self.objName, self.payloadObj, - x), - puppetClassList)) diff --git a/opensteak/tools/opensteak/foreman_objects/smart_proxies.py b/opensteak/tools/opensteak/foreman_objects/smart_proxies.py deleted file mode 100644 index 2d6518b..0000000 --- a/opensteak/tools/opensteak/foreman_objects/smart_proxies.py +++ /dev/null @@ -1,36 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects - - -class SmartProxies(ForemanObjects): - """ - Domain class - """ - objName = 'smart_proxies' - payloadObj = 'smart_proxy' - - def importPuppetClasses(self, smartProxyId): - """ Function importPuppetClasses - Force the reload of puppet classes - - @param smartProxyId: smartProxy Id - @return RETURN: the API result - """ - return self.api.create('smart_proxies/{}/import_puppetclasses'.format(smartProxyId), '{}') diff --git a/opensteak/tools/opensteak/foreman_objects/subnets.py b/opensteak/tools/opensteak/foreman_objects/subnets.py deleted file mode 100644 index b1cac54..0000000 --- a/opensteak/tools/opensteak/foreman_objects/subnets.py +++ /dev/null @@ -1,67 +0,0 @@ -#!/usr/bin/python3 -# -*- coding: utf-8 -*- -# Licensed under the Apache License, Version 2.0 (the "License"); you may -# not use this file except in compliance with the License. You may obtain -# a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT -# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the -# License for the specific language governing permissions and limitations -# under the License. -# -# Authors: -# @author: David Blaisonneau <david.blaisonneau@orange.com> -# @author: Arnaud Morin <arnaud1.morin@orange.com> - -from opensteak.foreman_objects.objects import ForemanObjects - - -class Subnets(ForemanObjects): - """ - Subnets class - """ - objName = 'subnets' - payloadObj = 'subnet' - - def checkAndCreate(self, key, payload, domainId): - """ Function checkAndCreate - Check if a subnet exists and create it if not - - @param key: The targeted subnet - @param payload: The targeted subnet description - @param domainId: The domainId to be attached wiuth the subnet - @return RETURN: The id of the subnet - """ - if key not in self: - self[key] = payload - oid = self[key]['id'] - if not oid: - return False - #~ Ensure subnet contains the domain - subnetDomainIds = [] - for domain in self[key]['domains']: - subnetDomainIds.append(domain['id']) - if domainId not in subnetDomainIds: - subnetDomainIds.append(domainId) - self[key]["domain_ids"] = subnetDomainIds - if len(self[key]["domains"]) is not len(subnetDomainIds): - return False - return oid - - def removeDomain(self, subnetId, domainId): - """ Function removeDomain - Delete a domain from a subnet - - @param subnetId: The subnet Id - @param domainId: The domainId to be attached wiuth the subnet - @return RETURN: boolean - """ - subnetDomainIds = [] - for domain in self[subnetId]['domains']: - subnetDomainIds.append(domain['id']) - subnetDomainIds.remove(domainId) - self[subnetId]["domain_ids"] = subnetDomainIds - return len(self[subnetId]["domains"]) is len(subnetDomainIds) |