PEP-518 support: configure bandit via pyproject.toml

bandit/core/config.py

@@ -49,13 +49,15 @@ def __init__(self, config_file=None):
             if config_file.endswith('.toml'):
                 import toml
                 try:
-                    self._config = toml.load(f)['tool']['bandit']
+                    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:
-                    self._config = yaml.safe_load(f)
+                    with f:
+                        self._config = yaml.safe_load(f)
                 except yaml.YAMLError as err:
                     LOG.error(err)
                     raise utils.ConfigError("Error parsing file.", config_file)

setup.py

@@ -26,4 +26,8 @@
 
 setuptools.setup(
     setup_requires=['pbr>=2.0.0'],
+    extras_require={
+        'yaml': ['PyYAML'],
+        'toml': ['toml'],
+    },
     pbr=True)