From 9d2dabb5f9cf032b76f1298ba75adb759d073e29 Mon Sep 17 00:00:00 2001 From: Adam Williamson Date: Mon, 27 Apr 2020 15:06:34 -0700 Subject: [PATCH] Add a helper script for moving needle files I actually wrote this a while back but have just been keeping it on my local machine till now; figured I'd clean it up a bit and add it to the repo. It's just a little helper for moving needles into subdirectories, it's useful if you do a big needle retake and have dozens of needles you need to put in the right subdirs, saves figuring them out for yourself. Signed-off-by: Adam Williamson --- move-needles.py | 101 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 101 insertions(+) create mode 100755 move-needles.py diff --git a/move-needles.py b/move-needles.py new file mode 100755 index 00000000..ad44dbc9 --- /dev/null +++ b/move-needles.py @@ -0,0 +1,101 @@ +#!/bin/python3 + +# Copyright (C) 2020 Red Hat +# +# This file is part of os-autoinst-distri-fedora. +# +# os-autoinst-distri-fedora is free software; you can redistribute it +# and/or modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation, either version 2 of +# the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . +# +# Author: Adam Williamson + +"""This is a helper script for moving needles to appropriate subdirectories, by examining their +filenames and looking for existing needles with similar filenames. It's convenient if you have +to update a bunch of needles for a font change, or something, and don't want to figure out where +each one goes by hand. +""" + +import argparse +import datetime +import glob +import os +import shutil +import sys + +def process_needles(date, move=False): + """Main function that does all the work. date is the date to look + for needles from (as a string in YYYYMMDD format), move (boolean) + is whether to actually move them or just do a dry run. + """ + # we assume all non-template JSON files in the current directory + # are needles + needles = glob.glob('*json') + needles = [needle for needle in needles if not needle.startswith('templates')] + for needle in needles: + cands = [] + # strip off the date + name = str(needle).replace(f"-{date}.json", "") + # now look for any other directory that has needle(s) starting + # with the stripped name + for (root, dirs, files) in os.walk('.'): + if any(file.startswith(name) for file in files) and root != '.': + cands.append(root) + # if we found exactly one potential target dir, go ahead + if len(cands) == 1: + print(f"Move {needle} to {cands[0]}") + if move: + shutil.move(needle, cands[0]) + sshot = needle.replace('json', 'png') + shutil.move(sshot, cands[0]) + # if we found more than one, just explain + elif len(cands) > 1: + print(f"Multiple candidates found for {needle}!") + for (idx, cand) in enumerate(cands, 1): + print(f"{str(idx)}: {cand}") + +def parse_args(args): + """Parse arguments with argparse.""" + parser = argparse.ArgumentParser( + description=( + "Helper script for moving needles to appropriate subdirectories." + ) + ) + parser.add_argument( + "-d", "--date", help="Date to work with, in YYYYMMDD format. Script will look for needle " + "files with this date in their name and strip it to form the base name to look for other " + "instances of). If not specified, script will use today's date in the local timezone", + default=datetime.date.today().strftime("%Y%m%d") + ) + parser.add_argument( + "-m", "--move", help="If set, actually move files; if not set, do a dry run and only " + "report what files would be moved", action="store_true" + ) + return parser.parse_args(args) + +def run(args): + """Parse args and call main function with appropriate options.""" + args = parse_args(args) + process_needles(args.date, args.move) + +def main(): + """Main loop.""" + try: + run(args=sys.argv[1:]) + except KeyboardInterrupt: + sys.stderr.write("Interrupted, exiting...\n") + sys.exit(1) + +if __name__ == '__main__': + main() + +# vim: set textwidth=100 ts=8 et sw=4: