58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""
|
||
|
|
Generate files for tests.
|
||
|
|
"""
|
||
|
|
import argparse
|
||
|
|
from datetime import date, timedelta
|
||
|
|
from pathlib import Path
|
||
|
|
import os
|
||
|
|
|
||
|
|
DEFAULT_NUMBER_OF_FILES = 200
|
||
|
|
DEFAULT_PREFIX = "testfile-"
|
||
|
|
DEFAULT_SUFFIX = ".bak"
|
||
|
|
|
||
|
|
# Argument parser
|
||
|
|
parser = argparse.ArgumentParser(description="Cleanup old backups")
|
||
|
|
# path argument
|
||
|
|
parser.add_argument(
|
||
|
|
"path",
|
||
|
|
metavar = "PATH",
|
||
|
|
type = str,
|
||
|
|
nargs = 1,
|
||
|
|
help = "directory path"
|
||
|
|
)
|
||
|
|
# number of files
|
||
|
|
parser.add_argument(
|
||
|
|
"-n", "--number-of-files",
|
||
|
|
type = int,
|
||
|
|
default = DEFAULT_NUMBER_OF_FILES,
|
||
|
|
metavar = "N",
|
||
|
|
help = f"generate N files, default: {DEFAULT_NUMBER_OF_FILES}"
|
||
|
|
)
|
||
|
|
# prefix argument
|
||
|
|
parser.add_argument(
|
||
|
|
"-p", "--prefix",
|
||
|
|
type = str,
|
||
|
|
default = DEFAULT_PREFIX,
|
||
|
|
metavar = "PREFIX",
|
||
|
|
help = f"use PREFIX as file name prefix, default: {DEFAULT_PREFIX}"
|
||
|
|
)
|
||
|
|
# suffix argument
|
||
|
|
parser.add_argument(
|
||
|
|
"-s", "--suffix",
|
||
|
|
type = str,
|
||
|
|
default = DEFAULT_SUFFIX,
|
||
|
|
metavar = "SUFFIX",
|
||
|
|
help = f"use SUFFIX as file name suffix, default: {DEFAULT_SUFFIX}"
|
||
|
|
)
|
||
|
|
args = parser.parse_args()
|
||
|
|
path = args.path[0]
|
||
|
|
number_of_files = args.number_of_files
|
||
|
|
prefix = args.prefix
|
||
|
|
suffix = args.suffix
|
||
|
|
|
||
|
|
for i in range(number_of_files):
|
||
|
|
timestamp = (date.today() - timedelta(days=i)).strftime("%Y%m%d")
|
||
|
|
filename = f"{prefix}{timestamp}{suffix}"
|
||
|
|
Path(os.path.join(path, filename)).touch()
|
||
|
|
|