97c01e48ed
Currently we have all our elements and library files in a top-level directory and install them into <root>/share/diskimage-builder/[elements|lib] (where root is either / or the root of a virtualenv). The problem with this is that editable/development installs (pip -e) do *not* install data_files. Thus we have no canonical location to look for elements -- leading to the various odd things we do such as a whole bunch of guessing at the top of disk-image-create and having a special test-loader in tests/test_elements.py so we can run python unit tests on those elements that have it. data_files is really the wrong thing to use for what are essentially assets of the program. data_files install works well for things like config-files, init.d files or dropping documentation files. By moving the elements under the diskimage_builder package, we always know where they are relative to where we import from. In fact, pkg_resources has an api for this which we wrap in the new diskimage_builder/paths.py helper [1]. We use this helper to find the correct path in the couple of places we need to find the base-elements dir, and for the paths to import the library shell functions. Elements such as svc-map and pkg-map include python unit-tests, which we do not need tests/test_elements.py to special-case load any more. They just get found automatically by the normal subunit loader. I have a follow-on change (I69ca3d26fede0506a6353c077c69f735c8d84d28) to move disk-image-create to a regular python entry-point. Unfortunately, this has to move to work with setuptools. You'd think a symlink under diskimage_builder/[elements|lib] would work, but it doesn't. [1] this API handles stuff like getting files out of .zip archive modules, which we don't do. Essentially for us it's returning __file__. Change-Id: I5e3e3c97f385b1a4ff2031a161a55b231895df5b
88 lines
2.8 KiB
Python
Executable File
88 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
#
|
|
# 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 sys
|
|
|
|
import six
|
|
import yaml
|
|
|
|
|
|
def os_family(distro):
|
|
"Given a distribution, returns a operating system family."
|
|
family = None
|
|
if distro in ['centos', 'centos7', 'fedora', 'rhel', 'rhel7']:
|
|
family = 'redhat'
|
|
elif distro in ['debian', 'ubuntu']:
|
|
family = 'debian'
|
|
elif distro == 'opensuse':
|
|
family = 'suse'
|
|
return family
|
|
|
|
|
|
def merge_data(source, destination, distro):
|
|
"""Merges two dictionaries and filters on distro.
|
|
|
|
Merges two dictionaries and filters on distro family, or default (in order)
|
|
"""
|
|
result = dict()
|
|
result.update(destination)
|
|
family = os_family(distro)
|
|
for servicename, mapping in six.iteritems(source):
|
|
if servicename in result:
|
|
raise Exception("%s already found in services list" % servicename)
|
|
if distro in mapping:
|
|
result[servicename] = mapping.get(distro)
|
|
elif family in mapping:
|
|
result[servicename] = mapping.get(family)
|
|
elif 'default' in mapping:
|
|
result[servicename] = mapping.get('default')
|
|
else:
|
|
result[servicename] = servicename
|
|
|
|
return result
|
|
|
|
|
|
def write_data_to_file(data, service_file_path):
|
|
"""Writes yaml data to a specified path."""
|
|
with open(service_file_path, 'w') as destination:
|
|
yaml.dump(data, destination, default_flow_style=False)
|
|
|
|
|
|
def main():
|
|
elements = yaml.safe_load(os.environ.get("IMAGE_ELEMENT_YAML"))
|
|
service_names = dict()
|
|
for element, path in elements.items():
|
|
data_path = os.path.join(path, "svc-map")
|
|
if os.path.exists(data_path):
|
|
with open(data_path, 'r') as dataFile:
|
|
data = yaml.load(dataFile.read())
|
|
try:
|
|
service_names = merge_data(
|
|
data,
|
|
service_names,
|
|
os.environ.get("DISTRO_NAME"))
|
|
except Exception as err:
|
|
print("%s. Check %s for duplicate"
|
|
" service name." % (err, element))
|
|
sys.exit(1)
|
|
|
|
write_data_to_file(
|
|
service_names,
|
|
os.path.join(os.environ['TMP_MOUNT_PATH'], 'tmp', 'svc-map-services'))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|