From 69760d415dddb0325bd0eb6bb607624747a2da6e Mon Sep 17 00:00:00 2001 From: Ansuel Smith Date: Tue, 6 Jul 2021 03:25:06 +0200 Subject: [PATCH] scripts: assume dl/ the default dir for dl_cleanup script Assume dl/ the default dl dir and make it configurable if someone have that in a different place. Signed-off-by: Ansuel Smith --- scripts/dl_cleanup.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/scripts/dl_cleanup.py b/scripts/dl_cleanup.py index d22762136a..c3b427c436 100755 --- a/scripts/dl_cleanup.py +++ b/scripts/dl_cleanup.py @@ -185,6 +185,9 @@ def usage(): print(" -d|--dry-run Do a dry-run. Don't delete any files") print(" -B|--show-blacklist Show the blacklist and exit") print(" -w|--whitelist ITEM Remove ITEM from blacklist") + print( + " -D|--download-dir Provide path to dl dir to clean also the build directory" + ) def main(argv): @@ -193,25 +196,20 @@ def main(argv): try: (opts, args) = getopt.getopt( argv[1:], - "hdBw:", + "hdBwD:", [ "help", "dry-run", "show-blacklist", "whitelist=", + "download-dir=", ], ) - if len(args) != 1: - usage() - return 1 except getopt.GetoptError as e: usage() return 1 - directory = args[0] - if not os.path.exists(directory): - print("Can't find dl path", directory) - return 1 + directory = "dl/" for (o, v) in opts: if o in ("-h", "--help"): @@ -235,6 +233,12 @@ def main(argv): sep = "\t" print("%s%s(%s)" % (name, sep, regex.pattern)) return 0 + if o in ("-D", "--download-dir"): + directory = v + + if not os.path.exists(directory): + print("Can't find dl path", directory) + return 1 # Create a directory listing and parse the file names. entries = [] -- 2.30.2