space/big-file-gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add --force flag to overwrite existing output files

Browse Source
This commit is contained in:
Luna
2026-04-08 00:33:33 +02:00
parent 31b46d0dd5
commit ea08675f9a
2 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
big_file_gen.py
View File

@@ -102,11 +102,20 @@ def _maybe_log_throughput(prefix: str, bytes_done: int, started_at: float, state
return ProgressState(state.total_bytes, state.last_reported_percent, now, bytes_done)
def create_file(output: str | Path, total_bytes: int, chunk_size: int, quiet: bool = False, sparse: bool = False) -> int:
def create_file(
output: str | Path,
total_bytes: int,
chunk_size: int,
quiet: bool = False,
sparse: bool = False,
force: bool = False,
) -> int:
output_path = Path(output)
existed_before = output_path.exists()
if output_path.exists():
if existed_before and not force:
print(f"Error: file already exists: {output_path}", file=sys.stderr)
print("Use --force to overwrite.", file=sys.stderr)
return 1
_ensure_parent_dir(output_path)
@@ -166,7 +175,7 @@ def create_file(output: str | Path, total_bytes: int, chunk_size: int, quiet: bo
except KeyboardInterrupt:
print("Interrupted, cleaning up partial file", file=sys.stderr)
try:
if output_path.exists():
if output_path.exists() and not existed_before:
output_path.unlink()
except OSError:
pass
@@ -174,7 +183,7 @@ def create_file(output: str | Path, total_bytes: int, chunk_size: int, quiet: bo
except OSError as exc:
print(f"Error writing file: {exc}", file=sys.stderr)
try:
if output_path.exists():
if output_path.exists() and not existed_before:
output_path.unlink()
except OSError:
pass
@@ -267,6 +276,7 @@ def build_create_parser(prog: str) -> argparse.ArgumentParser:
parser.add_argument("size", help="Target size, for example 15GB or 1.5TiB")
parser.add_argument("--chunk-size", default="64MB", help="Write chunk size (default: 64MB)")
parser.add_argument("--sparse", action="store_true", help="Create a sparse file instead of writing zeros")
parser.add_argument("--force", "-f", action="store_true", help="Overwrite output file if it already exists")
parser.add_argument("--quiet", "-q", action="store_true", help="Suppress progress output")
parser.add_argument("--version", action="version", version=f"{prog} {VERSION}")
return parser
@@ -301,7 +311,7 @@ def create_main(argv: Optional[Iterable[str]] = None) -> int:
except ValueError as exc:
print(f"Error: {exc}", file=sys.stderr)
return 1
return create_file(args.output, total_bytes, chunk_size, args.quiet, args.sparse)
return create_file(args.output, total_bytes, chunk_size, args.quiet, args.sparse, args.force)
def read_main(argv: Optional[Iterable[str]] = None) -> int: