backupcleaner/tests/filegenerator.py

#!/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"
DEFAULT_TIMESTAMP_FORMAT = "%Y%m%d"

# Argument parser
parser = argparse.ArgumentParser(
    description="Cleanup old backups",
    epilog="For a complete timestamp format description, see the python strftime() " +
           "documentation: https://docs.python.org/3/library/datetime.html" +
           "#strftime-strptime-behavior"
)
# 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}"
)
# timestamp format
parser.add_argument(
    "-t",
    "--timestamp-format",
    type=str,
    default=DEFAULT_TIMESTAMP_FORMAT,
    metavar="FORMAT",
    help=f"format of timestamp, default: {DEFAULT_TIMESTAMP_FORMAT}".replace(r"%", r"%%")
)

args = parser.parse_args()
path = args.path[0]
number_of_files = args.number_of_files
prefix = args.prefix
suffix = args.suffix
timestamp_format = args.timestamp_format

for i in range(number_of_files):
    timestamp = (date.today() - timedelta(days=i)).strftime(timestamp_format)
    filename = f"{prefix}{timestamp}{suffix}"
    Path(os.path.join(path, filename)).touch()