zack3d/pylingual
1
0
Fork 0
mirror of https://github.com/syssec-utd/pylingual.git synced 2026-05-10 18:39:03 -07:00
Code Issues Packages Projects Releases Wiki Activity

running progress bar for success rate

Browse Source
This commit is contained in:
Joel Flores
2025-06-03 13:44:21 -05:00
parent 10d938cc76
commit 3fccaf03fe
1 changed files with 26 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dev_scripts/cflow.py
+26 -5
View File
@@ -13,7 +13,7 @@ from tempfile import TemporaryDirectory
import click import click
import rich import rich
from rich.progress import track from rich.progress import Progress, BarColumn, TextColumn, TimeRemainingColumn, TimeElapsedColumn, MofNCompleteColumn
from pylingual.control_flow_reconstruction.cfg import CFG from pylingual.control_flow_reconstruction.cfg import CFG
from pylingual.control_flow_reconstruction.structure import bc_to_cft from pylingual.control_flow_reconstruction.structure import bc_to_cft
@@ -134,6 +134,18 @@ def get_unused(a: Path, _=True):
def main(input: Path, output: str, version: PythonVersion, graph: str | None, prefix: Path, processes: int, graph_format: str): def main(input: Path, output: str, version: PythonVersion, graph: str | None, prefix: Path, processes: int, graph_format: str):
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
print = rich.get_console().print print = rich.get_console().print
progress_columns = [
TextColumn("[progress.description]{task.description}"),
BarColumn(),
MofNCompleteColumn(),
TextColumn("[progress.percentage]{task.percentage:>3.1f}%"),
TextColumn("•"),
TimeElapsedColumn(),
TextColumn("<"),
TimeRemainingColumn(),
TextColumn("• [green]:{task.fields[success]}/{task.fields[processed]}"), # success rate
TextColumn("• [green]Success Rate:{task.fields[srate]:>3.2f}%"), # success rate
]
if graph: if graph:
CFG.enable_graphing(prefix / graph, graph_format) CFG.enable_graphing(prefix / graph, graph_format)
if input.is_file() and input.suffix == ".py": if input.is_file() and input.suffix == ".py":
@@ -167,16 +179,25 @@ def main(input: Path, output: str, version: PythonVersion, graph: str | None, pr
pool = contextlib.nullcontext(NoPool) pool = contextlib.nullcontext(NoPool)
dir_map = {} dir_map = {}
with pool as p: with pool as p:
for result, input, od in track(p.imap_unordered(f, files), total=len(files)): with Progress(*progress_columns, console=rich.get_console(), transient=False) as progress_bar:
results[result].append(input) task_id = progress_bar.add_task("Evaluating Control Flow", total=len(files), success=0, processed=0, srate=0.0)
dir_map[str(input)] = str(od) succs = 0
processed = 0
for result, input, od in p.imap_unordered(f, files):
results[result].append(input)
processed += 1
dir_map[str(input)] = str(od)
if result == Result.Success:
succs += 1
progress_bar.update(task_id, advance=1, success=succs, processed=processed, srate=(succs / processed * 100))
for res in Result: for res in Result:
print(f"{res}: {len(results[res])}") print(f"{res}: {len(results[res])}")
total = sum(len(x) for x in results.values()) total = sum(len(x) for x in results.values())
if total: if total:
print(f"{len(results[Result.Success])} / {total} succeeded ({len(results[Result.Success]) / total:.3%})") print(f"{len(results[Result.Success])} / {total} succeeded ({len(results[Result.Success]) / total:.3%})")
res = json.dumps({k.value: list(map(str, v)) for k, v in results.items()} | {'map': dir_map}) res = json.dumps({k.value: list(map(str, v)) for k, v in results.items()} | {"map": dir_map})
(out_dir / "results.json").write_text(res) (out_dir / "results.json").write_text(res)