From 9d2dabb5f9cf032b76f1298ba75adb759d073e29 Mon Sep 17 00:00:00 2001
From: Adam Williamson <awilliam@redhat.com>
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 <awilliam@redhat.com>
---
 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 <http://www.gnu.org/licenses/>.
+#
+# Author: Adam Williamson <awilliam@redhat.com>
+
+"""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: