PEP-518 support: configure bandit via pyproject.toml

bandit/core/config.py

@@ -36,13 +36,23 @@ def __init__(self, config_file=None):
                 raise utils.ConfigError("Could not read config file.",
                                         config_file)
 
-            try:
-                with f:
-                    self._config = yaml.safe_load(f)
-                self.validate(config_file)
-            except yaml.YAMLError as err:
-                LOG.error(err)
-                raise utils.ConfigError("Error parsing file.", config_file)
+            if config_file.endswith('.toml'):
+                import toml
+                try:
+                    with f:
+                        self._config = toml.load(f)['tool']['bandit']
+                except toml.TomlDecodeError as err:
+                    LOG.error(err)
+                    raise utils.ConfigError("Error parsing file.", config_file)
+            else:
+                try:
+                    with f:
+                        self._config = yaml.safe_load(f)
+                except yaml.YAMLError as err:
+                    LOG.error(err)
+                    raise utils.ConfigError("Error parsing file.", config_file)
+
+            self.validate(config_file)
 
             # valid config must be a dict
             if not isinstance(self._config, dict):

doc/source/config.rst

@@ -37,6 +37,45 @@ several config files and pick from them using `-c`. If you only wish to control
 the specific tests that are to be run (and not their parameters) then using
 `-s` or `-t` on the command line may be more appropriate.
 
+Also you can configure bandit via
+`pyproject.toml <https://www.python.org/dev/peps/pep-0518/>`_ file. In this
+case you would explicitly specify the path to configuration via `-s` too.
+For example:
+
+.. code-block:: python
+    [tool.bandit.profiles.test_1]
+    include = [
+      "any_other_function_with_shell_equals_true",
+      "assert_used",
+    ]
+
+    [tool.bandit.profiles.test_2]
+    include = ["blacklist_calls"]
+
+    [tool.bandit.profiles.test_3]
+    include = ["blacklist_imports"]
+
+    [tool.bandit.profiles.test_4]
+    exclude = ["assert_used"]
+
+    [tool.bandit.profiles.test_5]
+    exclude = ["blacklist_calls", "blacklist_imports"]
+
+    [tool.bandit.profiles.test_6]
+    include = ["blacklist_calls"]
+    exclude = ["blacklist_imports"]
+
+    [[tool.bandit.blacklist_calls.bad_name_sets]]
+      [tool.bandit.blacklist_calls.bad_name_sets.pickle]
+      qualnames = ["pickle.loads"]
+      message = "{func} library appears to be in use."
+
+    [[tool.bandit.blacklist_imports.bad_import_sets]]
+      [tool.bandit.blacklist_imports.bad_import_sets.telnet]
+      imports = ["telnetlib"]
+      level = "HIGH"
+      message = "{module} is considered insecure."
+
 Skipping Tests
 --------------
 The bandit config may contain optional lists of test IDs to either include

setup.cfg

@@ -26,7 +26,13 @@ classifier =
 project_urls =
     Release notes = https://github.com/PyCQA/bandit/releases
 
-[entry_points]
+[options.extras_require]
+yaml =
+    PyYAML
+toml =
+    toml
+
+[options.entry_points]
 console_scripts =
     bandit = bandit.cli.main:main
     bandit-config-generator = bandit.cli.config_generator:main

test-requirements.txt

@@ -8,5 +8,6 @@ mock>=3.0.5 # BSD
 stestr>=2.5.0 # Apache-2.0
 testscenarios>=0.5.0 # Apache-2.0/BSD
 testtools>=2.3.0 # MIT
+toml # MIT
 beautifulsoup4>=4.8.0 # MIT
 pylint==1.9.4 # GPLv2

tests/unit/core/test_config.py

@@ -16,14 +16,16 @@
 
 
 class TempFile(fixtures.Fixture):
-    def __init__(self, contents=None):
+    def __init__(self, contents=None, suffix='.yaml'):
         super(TempFile, self).__init__()
         self.contents = contents
+        self.suffix = suffix
 
     def setUp(self):
         super(TempFile, self).setUp()
 
-        with tempfile.NamedTemporaryFile(mode='wt', delete=False) as f:
+        with tempfile.NamedTemporaryFile(suffix=self.suffix, mode='wt',
+                                         delete=False) as f:
             if self.contents:
                 f.write(self.contents)
 
@@ -112,7 +114,7 @@ def test_not_exist(self):
 
 
 class TestConfigCompat(testtools.TestCase):
-    sample_yaml = textwrap.dedent("""
+    sample = textwrap.dedent("""
         profiles:
             test_1:
                 include:
@@ -157,10 +159,11 @@ class TestConfigCompat(testtools.TestCase):
                     level: HIGH
                     message: "{module} is considered insecure."
         """)
+    suffix = '.yaml'
 
     def setUp(self):
         super(TestConfigCompat, self).setUp()
-        f = self.useFixture(TempFile(self.sample_yaml))
+        f = self.useFixture(TempFile(self.sample, suffix=self.suffix))
         self.config = config.BanditConfig(f.name)
 
     def test_converted_include(self):
@@ -252,3 +255,41 @@ def test_bad_yaml(self):
             self.config = config.BanditConfig(f.name)
         except utils.ConfigError as e:
             self.assertIn("Error parsing file.", e.message)
+
+
+class TestTomlConfig(TestConfigCompat):
+    sample = textwrap.dedent("""
+        [tool.bandit.profiles.test_1]
+        include = [
+            "any_other_function_with_shell_equals_true",
+            "assert_used",
+        ]
+
+        [tool.bandit.profiles.test_2]
+        include = ["blacklist_calls"]
+
+        [tool.bandit.profiles.test_3]
+        include = ["blacklist_imports"]
+
+        [tool.bandit.profiles.test_4]
+        exclude = ["assert_used"]
+
+        [tool.bandit.profiles.test_5]
+        exclude = ["blacklist_calls", "blacklist_imports"]
+
+        [tool.bandit.profiles.test_6]
+        include = ["blacklist_calls"]
+        exclude = ["blacklist_imports"]
+
+        [[tool.bandit.blacklist_calls.bad_name_sets]]
+            [tool.bandit.blacklist_calls.bad_name_sets.pickle]
+            qualnames = ["pickle.loads"]
+            message = "{func} library appears to be in use."
+
+        [[tool.bandit.blacklist_imports.bad_import_sets]]
+            [tool.bandit.blacklist_imports.bad_import_sets.telnet]
+            imports = ["telnetlib"]
+            level = "HIGH"
+            message = "{module} is considered insecure."
+        """)
+    suffix = '.toml'