lint: fix pbar counts

This commit is contained in:
Willi Ballenthin
2023-01-21 20:16:49 +01:00
parent 67cfefd2df
commit 61112c2527

View File

@@ -917,12 +917,11 @@ def lint(ctx: Context):
"""
ret = {}
with tqdm.contrib.logging.tqdm_logging_redirect(ctx.rules.rules.items(), unit="rule") as pbar:
source_rules = [rule for rule in ctx.rules.rules.values() if not rule.is_subscope_rule()]
with tqdm.contrib.logging.tqdm_logging_redirect(source_rules, unit="rule") as pbar:
with redirecting_print_to_tqdm():
for name, rule in pbar:
if rule.is_subscope_rule():
continue
for rule in pbar:
name = rule.name
pbar.set_description(width("linting rule: %s" % (name), 48))
ret[name] = lint_rule(ctx, rule)
@@ -999,7 +998,7 @@ def main(argv=None):
try:
rules = capa.main.get_rules(args.rules)
logger.info("successfully loaded %s rules", len(rules))
logger.info("successfully loaded %s rules", rules.source_rule_count)
if args.tag:
rules = rules.filter_rules_by_meta(args.tag)
logger.debug("selected %s rules", len(rules))