Add --timestamp-format option

tests/filegenerator.py

@@ -10,6 +10,7 @@ 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")
@@ -45,14 +46,26 @@ parser.add_argument(
     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("%Y%m%d")
+    timestamp = (date.today() - timedelta(days=i)).strftime(timestamp_format)
     filename = f"{prefix}{timestamp}{suffix}"
     Path(os.path.join(path, filename)).touch()