paunch: support a directory for "config" parameter

Before, the "config" parameter would only support a JSON file, which
would be sent to Paunch via YAML.
Now, we accept a directory path, where all .json files in that directory
will be sent as a single dict into Paunch via YAML.

It's part of the effort to break down the container config into
individual files per step and per container.

Story: 2006732
Task: 37162
Change-Id: Ie8bda685c499ff5ef1dc2799a85287662b5af407
This commit is contained in:
Emilien Macchi 2019-10-15 20:37:53 -04:00
parent 18ece9fd5f
commit 6992d5e880

View File

@ -20,8 +20,10 @@ __metaclass__ = type
from ansible.module_utils.basic import AnsibleModule from ansible.module_utils.basic import AnsibleModule
import yaml import json
import os
import paunch as p import paunch as p
import yaml
from paunch import runner as prunner from paunch import runner as prunner
from paunch.builder import compose1 as pcompose1 from paunch.builder import compose1 as pcompose1
from paunch.builder import podman as ppodman from paunch.builder import podman as ppodman
@ -50,7 +52,7 @@ description:
options: options:
config: config:
description: description:
- YAML or JSON file containing configuration data - JSON file or directory of JSON files containing configuration data
config_id: config_id:
description: description:
- ID to assign to containers - ID to assign to containers
@ -143,6 +145,18 @@ class PaunchManager:
log_file=self.log_file) log_file=self.log_file)
if self.config: if self.config:
if os.path.isdir(self.config):
container_configs = {}
config_files = [c_json for c_json in
os.listdir(self.config)
if c_json.endswith('.json')]
for cf in config_files:
with open(os.path.join(self.config, cf), 'r') as f:
c = os.path.splitext(cf)[0]
container_configs[c] = {}
container_configs[c].update(yaml.safe_load(f))
self.config_yaml = container_configs
else:
with open(self.config, 'r') as f: with open(self.config, 'r') as f:
self.config_yaml = yaml.safe_load(f) self.config_yaml = yaml.safe_load(f)