airship
/
deckhand
Code Issues Proposed changes

Merge "Add explicit start/end to Deckhand response middleware"

This commit is contained in:
Zuul 2018-10-08 14:48:23 +00:00 committed by Gerrit Code Review
parent f9e4b5993f 88e1c12b23
commit 0462b7b929
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
deckhand/control/middleware.py
View File

@ -163,6 +163,11 @@ class YAMLTranslator(HookableMiddlewareMixin, object):
if resp.status != '204 No Content':
resp.set_header('Content-Type', 'application/x-yaml')
kwargs = {
"explicit_start": True,
"explicit_end": True
}
for attr in ('body', 'data'):
if not hasattr(resp, attr):
continue
@ -170,6 +175,6 @@ class YAMLTranslator(HookableMiddlewareMixin, object):
resp_attr = getattr(resp, attr)
if isinstance(resp_attr, dict):
setattr(resp, attr, yaml.safe_dump(resp_attr))
setattr(resp, attr, yaml.safe_dump(resp_attr, **kwargs))
elif isinstance(resp_attr, (list, tuple)):
setattr(resp, attr, yaml.safe_dump_all(resp_attr))
setattr(resp, attr, yaml.safe_dump_all(resp_attr, **kwargs))