38c8a59bb8
Toogle enabled vanue in manifest file to true or false Make some refactoring in parser class Change-Id: I3c970316f6e9b7a0132919ad97810caf004e27a8
106 lines
4.5 KiB
Python
106 lines
4.5 KiB
Python
# Copyright (c) 2013 Mirantis, Inc.
|
|
#
|
|
# 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.
|
|
|
|
import os
|
|
import yaml
|
|
from oslo.config import cfg
|
|
import logging as log
|
|
from muranorepository.manifest import Manifest
|
|
from muranorepository.consts import DATA_TYPES, MANIFEST
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class ManifestParser(object):
|
|
def __init__(self, manifest_directory=None):
|
|
if not manifest_directory:
|
|
manifest_directory = CONF.manifests
|
|
self.manifest_directory = manifest_directory
|
|
|
|
def _validate_manifest(self, file, service_manifest_data):
|
|
for key, value in service_manifest_data.iteritems():
|
|
valid_file_info = True
|
|
if key in DATA_TYPES:
|
|
if key != MANIFEST:
|
|
root_directory = os.path.join(self.manifest_directory,
|
|
getattr(CONF, key))
|
|
else:
|
|
root_directory = self.manifest_directory
|
|
|
|
if not isinstance(value, list):
|
|
log.error("{0} section should represent a file"
|
|
" listing in manifest {1}"
|
|
"".format(root_directory, file))
|
|
valid_file_info = False
|
|
continue
|
|
for filename in value:
|
|
absolute_path = os.path.join(root_directory,
|
|
filename)
|
|
|
|
if not os.path.exists(absolute_path):
|
|
valid_file_info = False
|
|
log.warning(
|
|
"File {0} specified in manifest {1} "
|
|
"doesn't exist at {2}".format(filename,
|
|
file,
|
|
absolute_path))
|
|
return valid_file_info
|
|
|
|
def parse(self):
|
|
manifests = []
|
|
for file in os.listdir(self.manifest_directory):
|
|
manifest_file = os.path.join(self.manifest_directory, file)
|
|
if os.path.isfile(manifest_file):
|
|
if not file.endswith(".yaml"):
|
|
log.warning("Extension of {0} file is not yaml. "
|
|
"Only yaml file supported for "
|
|
"service manifest files.".format(file))
|
|
continue
|
|
|
|
try:
|
|
with open(manifest_file) as stream:
|
|
manifest_data = yaml.load(stream)
|
|
except yaml.YAMLError, exc:
|
|
log.warn("Failed to load manifest file. {0}. "
|
|
"The reason: {1!s}".format(manifest_file,
|
|
exc))
|
|
continue
|
|
manifest_data['manifest_file_name'] = file
|
|
manifest_is_valid = self._validate_manifest(file,
|
|
manifest_data)
|
|
manifest_data["valid"] = manifest_is_valid
|
|
|
|
manifests.append(Manifest(manifest_data))
|
|
return manifests
|
|
|
|
def toggle_enabled(self, service_name):
|
|
manifests = self.parse()
|
|
path_to_manifest = None
|
|
for manifest in manifests:
|
|
if manifest.full_service_name == service_name:
|
|
path_to_manifest = os.path.join(self.manifest_directory,
|
|
manifest.manifest_file_name)
|
|
break
|
|
if not path_to_manifest:
|
|
log.error('There is no manifest '
|
|
'file for {0} service'.format(service_name))
|
|
return False
|
|
with open(path_to_manifest) as stream:
|
|
service_manifest_data = yaml.load(stream)
|
|
service_manifest_data['enabled'] = \
|
|
not service_manifest_data.get('enabled')
|
|
with open(path_to_manifest, 'w') as manifest_file:
|
|
manifest_file.write(yaml.dump(service_manifest_data,
|
|
default_flow_style=False))
|
|
return True
|