Skip to content

Commit

Permalink
Improves the output of verify cpu models
Browse files Browse the repository at this point in the history
  • Loading branch information
mvaltas committed May 24, 2023
1 parent b86a154 commit 64b0de6
Showing 1 changed file with 10 additions and 8 deletions.
18 changes: 10 additions & 8 deletions ccfcoef/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -75,20 +75,22 @@ def verify_cpu_models():
"""
Check for duplication of CPU models across the families files.
"""
duplicated_models = {}
model_families = {}
for family in CPU_FAMILIES:
family_info = CPUInfo.instantiate(DATA_DIR.joinpath(f"{family.short}.csv"))
for model in family_info.cpus:
if model in duplicated_models:
duplicated_models[model].append(family.short)
if model in model_families:
model_families[model].append(family.short)
else:
duplicated_models[model] = [family.short]
model_families[model] = [family.short]

if duplicated_models:
duplicated = {model: family for model, family in model_families.items() if len(family) > 1}
if duplicated:
click.secho('Duplicated CPU models:', fg='red')
for model, families in duplicated_models.items():
if len(families) > 1:
click.secho(f'{click.style(model, fg="white")} is present in {click.style(families, fg="yellow")}')
for model, families in duplicated.items():
click.secho(f'{click.style(model, fg="white")} is present in {click.style(families, fg="yellow")}')
else:
click.secho('No duplications on CPU models found.', fg='green')

@cli.command()
def show_constants():
Expand Down

0 comments on commit 64b0de6

Please sign in to comment.