From a38056fafaead26e1b7198523b20f323130fa262 Mon Sep 17 00:00:00 2001 From: Eric Brown Date: Mon, 20 Feb 2017 13:42:59 -0800 Subject: [PATCH] Dump bandit config file lists vertically Currently when using the bandit-config-generator to dump out a config file, it looks rather messy because config option values that are lists are dumped onto one long line. So rather than dumping on one line, use the vertical yaml list format by specifying default_flow_style=False. Change-Id: Ic0dc97f19d067471b507421dcb98ac749874e49c --- bandit/cli/config_generator.py | 2 +- tests/unit/cli/test_config_generator.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/bandit/cli/config_generator.py b/bandit/cli/config_generator.py index e5d232cf..a48c5187 100644 --- a/bandit/cli/config_generator.py +++ b/bandit/cli/config_generator.py @@ -124,7 +124,7 @@ def get_config_settings(): if hasattr(fn_module, 'gen_config'): config[fn_name] = fn_module.gen_config(function._takes_config) - return yaml.safe_dump(config) + return yaml.safe_dump(config, default_flow_style=False) def main(): diff --git a/tests/unit/cli/test_config_generator.py b/tests/unit/cli/test_config_generator.py index f85c0a94..1431ab56 100644 --- a/tests/unit/cli/test_config_generator.py +++ b/tests/unit/cli/test_config_generator.py @@ -85,7 +85,8 @@ class BanditConfigGeneratorTests(testtools.TestCase): config[plugin.name] = module.gen_config( function._takes_config) settings = config_generator.get_config_settings() - self.assertEqual(yaml.safe_dump(config), settings) + self.assertEqual(yaml.safe_dump(config, default_flow_style=False), + settings) @mock.patch('sys.argv', ['bandit-config-generator', '--show-defaults']) def test_main_show_defaults(self):