mirror of
https://github.com/ansible/awx.git
synced 2026-05-14 12:57:40 -02:30
[feature] Keep pod_spec_override-provided pod labels
- Write a deepmerge() implementation, keeping only the test suite of https://stackoverflow.com/a/20666342/435004 - Use it to deep-merge pod['metadata'] with user input, instead of replacing fields in it
This commit is contained in:
@@ -12,6 +12,22 @@ from awx.main.utils.common import parse_yaml_or_json
|
|||||||
logger = logging.getLogger('awx.main.scheduler')
|
logger = logging.getLogger('awx.main.scheduler')
|
||||||
|
|
||||||
|
|
||||||
|
def deepmerge(a, b):
|
||||||
|
"""
|
||||||
|
>>> a = { 'first' : { 'all_rows' : { 'pass' : 'dog', 'number' : '1' } } }
|
||||||
|
>>> b = { 'first' : { 'all_rows' : { 'fail' : 'cat', 'number' : '5' } } }
|
||||||
|
>>> deepmerge(b, a) == { 'first' : { 'all_rows' : { 'pass' : 'dog', 'fail' : 'cat', 'number' : '5' } } }
|
||||||
|
True
|
||||||
|
"""
|
||||||
|
if isinstance(a, dict) and isinstance(b, dict):
|
||||||
|
return dict([(k, deepmerge(a.get(k), b.get(k)))
|
||||||
|
for k in set(a.keys()).union(b.keys())])
|
||||||
|
elif b is None:
|
||||||
|
return a
|
||||||
|
else:
|
||||||
|
return b
|
||||||
|
|
||||||
|
|
||||||
class PodManager(object):
|
class PodManager(object):
|
||||||
|
|
||||||
def __init__(self, task=None):
|
def __init__(self, task=None):
|
||||||
@@ -128,11 +144,13 @@ class PodManager(object):
|
|||||||
pod_spec = {**default_pod_spec, **pod_spec_override}
|
pod_spec = {**default_pod_spec, **pod_spec_override}
|
||||||
|
|
||||||
if self.task:
|
if self.task:
|
||||||
pod_spec['metadata']['name'] = self.pod_name
|
pod_spec['metadata'] = deepmerge(
|
||||||
pod_spec['metadata']['labels'] = {
|
pod_spec.get('metadata', {}),
|
||||||
'ansible-awx': settings.INSTALL_UUID,
|
dict(name=self.pod_name,
|
||||||
'ansible-awx-job-id': str(self.task.id)
|
labels={
|
||||||
}
|
'ansible-awx': settings.INSTALL_UUID,
|
||||||
|
'ansible-awx-job-id': str(self.task.id)
|
||||||
|
}))
|
||||||
pod_spec['spec']['containers'][0]['name'] = self.pod_name
|
pod_spec['spec']['containers'][0]['name'] = self.pod_name
|
||||||
|
|
||||||
return pod_spec
|
return pod_spec
|
||||||
|
|||||||
Reference in New Issue
Block a user