226 lines
7.6 KiB
Python
226 lines
7.6 KiB
Python
from . import exceptions, logging, validation
|
|
import copy
|
|
import jinja2
|
|
import jsonpath_ng
|
|
import requests
|
|
import yaml
|
|
|
|
__all__ = ['Configuration']
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class Configuration:
|
|
def __init__(self, *, documents, debug=False, substitute=True):
|
|
if substitute:
|
|
documents = _substitute(documents)
|
|
self.debug = debug
|
|
self.documents = documents
|
|
|
|
@classmethod
|
|
def from_streams(cls, *, streams, **kwargs):
|
|
documents = []
|
|
for stream in streams:
|
|
stream_name = getattr(stream, 'name')
|
|
if stream_name is not None:
|
|
LOG.info('Loading documents from %s', stream_name)
|
|
stream_documents = list(yaml.safe_load_all(stream))
|
|
validation.check_schemas(stream_documents)
|
|
if stream_name is not None:
|
|
LOG.info('Successfully validated documents from %s',
|
|
stream_name)
|
|
documents.extend(stream_documents)
|
|
|
|
return cls(documents=documents, **kwargs)
|
|
|
|
@classmethod
|
|
def from_design_ref(cls, design_ref):
|
|
response = requests.get(design_ref)
|
|
response.raise_for_status()
|
|
|
|
documents = list(yaml.safe_load_all(response.text))
|
|
validation.check_schemas(documents)
|
|
|
|
return cls(documents=documents)
|
|
|
|
def __getitem__(self, path):
|
|
value = self.get_path(path)
|
|
if value:
|
|
return value
|
|
else:
|
|
return jinja2.StrictUndefined('No match found for path %s' % path)
|
|
|
|
def get_first(self, *paths):
|
|
result = self._get_first(*paths)
|
|
if result:
|
|
return result
|
|
else:
|
|
return jinja2.StrictUndefined(
|
|
'Nothing found matching paths: %s' % ','.join(paths))
|
|
|
|
def get(self, *, kind=None, name=None, schema=None):
|
|
result = _get(self.documents, kind=kind, schema=schema, name=name)
|
|
|
|
if result:
|
|
return result['data']
|
|
else:
|
|
return jinja2.StrictUndefined(
|
|
'No document found matching kind=%s schema=%s name=%s' %
|
|
(kind, schema, name))
|
|
|
|
def iterate(self, *, kind=None, schema=None, labels=None):
|
|
if kind is not None:
|
|
if schema is not None:
|
|
raise AssertionError(
|
|
'Logic error: specified both kind and schema')
|
|
schema = 'promenade/%s/v1' % kind
|
|
|
|
for document in self.documents:
|
|
if _matches_filter(document, schema=schema, labels=labels):
|
|
yield document
|
|
|
|
def extract_genesis_config(self):
|
|
LOG.debug('Extracting genesis config.')
|
|
documents = []
|
|
for document in self.documents:
|
|
if document['schema'] != 'promenade/KubernetesNode/v1':
|
|
documents.append(document)
|
|
else:
|
|
LOG.debug('Excluding schema=%s metadata.name=%s',
|
|
document['schema'], _mg(document, 'name'))
|
|
return Configuration(
|
|
debug=self.debug, documents=documents, substitute=False)
|
|
|
|
def extract_node_config(self, name):
|
|
LOG.debug('Extracting node config for %s.', name)
|
|
documents = []
|
|
for document in self.documents:
|
|
schema = document['schema']
|
|
if schema == 'promenade/Genesis/v1':
|
|
LOG.debug('Excluding schema=%s metadata.name=%s', schema,
|
|
_mg(document, 'name'))
|
|
continue
|
|
elif schema == 'promenade/KubernetesNode/v1' and _mg(
|
|
document, 'name') != name:
|
|
LOG.debug('Excluding schema=%s metadata.name=%s', schema,
|
|
_mg(document, 'name'))
|
|
continue
|
|
else:
|
|
documents.append(document)
|
|
return Configuration(
|
|
debug=self.debug, documents=documents, substitute=False)
|
|
|
|
@property
|
|
def kubelet_name(self):
|
|
for document in self.iterate(kind='Genesis'):
|
|
return 'genesis'
|
|
|
|
for document in self.iterate(kind='KubernetesNode'):
|
|
return document['data']['hostname']
|
|
|
|
return jinja2.StrictUndefined(
|
|
'No Genesis or KubernetesNode found while getting kubelet name')
|
|
|
|
def _get_first(self, *paths):
|
|
for path in paths:
|
|
value = self.get_path(path)
|
|
if value:
|
|
return value
|
|
|
|
def get_path(self, path, default=None):
|
|
kind, jsonpath = path.split(':')
|
|
document = _get(self.documents, kind=kind)
|
|
if document:
|
|
data = _extract(document['data'], jsonpath)
|
|
if data:
|
|
return data
|
|
return default
|
|
|
|
def append(self, item):
|
|
validation.check_schema(item)
|
|
self.documents.append(item)
|
|
|
|
|
|
def _matches_filter(document, *, schema, labels):
|
|
matches = True
|
|
if schema is not None and not document.get('schema',
|
|
'').startswith(schema):
|
|
matches = False
|
|
|
|
if labels is not None:
|
|
document_labels = _mg(document, 'labels', [])
|
|
for key, value in labels.items():
|
|
if key not in document_labels:
|
|
matches = False
|
|
else:
|
|
if document_labels[key] != value:
|
|
matches = False
|
|
|
|
return matches
|
|
|
|
|
|
def _get(documents, kind=None, schema=None, name=None):
|
|
if kind is not None:
|
|
if schema is not None:
|
|
msg = "Only kind or schema may be specified, not both"
|
|
raise exceptions.ValidationException(msg)
|
|
schema = 'promenade/%s/v1' % kind
|
|
|
|
for document in documents:
|
|
if (schema == document.get('schema')
|
|
and (name is None or name == _mg(document, 'name'))):
|
|
return document
|
|
|
|
|
|
def _substitute(documents):
|
|
result = []
|
|
|
|
for document in documents:
|
|
dest_schema = document.get('schema')
|
|
dest_name = _mg(document, 'name')
|
|
LOG.debug('Checking for substitutions in schema=%s metadata.name=%s',
|
|
dest_schema, dest_name)
|
|
final_doc = copy.deepcopy(document)
|
|
for substitution in _mg(document, 'substitutions', []):
|
|
source_schema = substitution['src']['schema']
|
|
source_name = substitution['src']['name']
|
|
source_path = substitution['src']['path']
|
|
dest_path = substitution['dest']['path']
|
|
LOG.debug('Substituting from schema=%s name=%s src_path=%s '
|
|
'into dest_path=%s', source_schema, source_name,
|
|
source_path, dest_path)
|
|
source_document = _get(
|
|
documents, schema=source_schema, name=source_name)
|
|
if source_document is None:
|
|
msg = 'Failed to find source document for subsitution. ' \
|
|
'dest_schema=%s dest_name=%s ' \
|
|
'source_schema=%s source_name=%s' \
|
|
% (dest_schema, dest_name, source_schema, source_name)
|
|
LOG.critical(msg)
|
|
raise exceptions.ValidationException(msg)
|
|
|
|
source_value = _extract(source_document['data'],
|
|
substitution['src']['path'])
|
|
final_doc['data'] = _replace(final_doc['data'], source_value,
|
|
substitution['dest']['path'])
|
|
|
|
result.append(final_doc)
|
|
|
|
return result
|
|
|
|
|
|
def _extract(document, jsonpath):
|
|
p = jsonpath_ng.parse(jsonpath)
|
|
matches = p.find(document)
|
|
if matches:
|
|
return matches[0].value
|
|
|
|
|
|
def _replace(document, value, jsonpath):
|
|
p = jsonpath_ng.parse(jsonpath)
|
|
return p.update(document, value)
|
|
|
|
|
|
def _mg(document, field, default=None):
|
|
return document.get('metadata', {}).get(field, default)
|