Miscellaneous fixed, cleanup and linting.

This commit is contained in:
2026-01-02 23:51:43 +00:00
parent 52cfcb9872
commit 017095df36
6 changed files with 258 additions and 115 deletions
+9 -5
View File
@@ -8,7 +8,6 @@ import sys
import time
import aeg
import tracerite
from randquik.benchmark import run_benchmark
from randquik.crypto import derive_key, generate_random_seed
@@ -24,8 +23,6 @@ from randquik.workers import (
MmapProducer,
)
tracerite.load()
__all__ = ["main"]
# Disable GC for performance
@@ -261,15 +258,20 @@ def _main():
producer.join()
finally:
progress.stop()
producer.cleanup()
elapsed = time.perf_counter() - start_time
if not args.quiet:
print_summary(producer.written, elapsed, "wrote", seed=seed_for_display)
if args.verbose:
stats = producer.get_combined_stats()
print(stats.format_report(f"Workers (×{workers})"), file=sys.stderr)
producer.cleanup()
return
# Standard file mode with ring buffers
with open_fd(args.output, total_bytes, dry=args.dry, oseek=oseek) as fd:
producer = FdProducer(workers, key, ciph, total_bytes, fd, dry=args.dry, iseek=iseek)
producer = FdProducer(
workers, key, ciph, total_bytes, fd, dry=args.dry, iseek=iseek, profile=args.verbose
)
if args.verbose:
dry_str = " (dry run)" if args.dry else ""
@@ -292,6 +294,8 @@ def _main():
elapsed = time.perf_counter() - start_time
if not args.quiet:
print_summary(producer.written, elapsed, "wrote", seed=seed_for_display)
if args.verbose:
print(producer.format_stats_report(), file=sys.stderr)
def main():