Fix custom format argument handling (#380)

* Fix incorrect parameter handling
* Remove custom format report limit

Signed-off-by: Antoine Salon <asalon@vmware.com>

bandit/core/manager.py

@@ -158,8 +158,7 @@ def output_results(self, lines, sev_level, conf_level, output_file,
             report_func = formatter.plugin
             if output_format == 'custom':
                 report_func(self, fileobj=output_file, sev_level=sev_level,
-                            conf_level=conf_level, lines=lines,
-                            template=template)
+                            conf_level=conf_level, template=template)
             else:
                 report_func(self, fileobj=output_file, sev_level=sev_level,
                             conf_level=conf_level, lines=lines)

bandit/formatters/custom.py

@@ -50,14 +50,13 @@ def __missing__(cls, key):
 
 
 @test_properties.accepts_baseline
-def report(manager, fileobj, sev_level, conf_level, lines=-1, template=None):
+def report(manager, fileobj, sev_level, conf_level, template=None):
     """Prints issues in custom format
 
     :param manager: the bandit manager object
     :param fileobj: The output file object, which may be sys.stdout
     :param sev_level: Filtering severity level
     :param conf_level: Filtering confidence level
-    :param lines: Number of lines to report, -1 for all
     :param template: Output template with non-terminal tags <N>
                     (default: '{abspath}:{line}:
                     {test_id}[bandit]: {severity}: {msg}')
@@ -149,9 +148,8 @@ def get_similar_tag(tag):
 
     msg_parsed_template = "".join([item for lst in msg_parsed_template_list
                                    for item in lst]) + "\n"
-    limit = lines if lines > 0 else None
     with fileobj:
-        for defect in results[:limit]:
+        for defect in results:
             evaluated_tags = SafeMapper(
                 (k, v(defect)) for k, v in tag_mapper.items()
             )