Upload 5 files

utils/fileSelect.py

@@ -0,0 +1,11 @@
+from tkinter import Tk
+from tkinter.filedialog import askdirectory
+
+
+def getDirectory(**args):
+    title = args.get("title", None)
+    Tk().withdraw()
+    return askdirectory(title=title)
+
+
+# getDirectory()

utils/files.py

@@ -0,0 +1,61 @@
+from os import listdir, makedirs
+from os.path import isdir, isfile, join
+from re import search, IGNORECASE
+from shutil import copy
+
+
+def getFilesWith(path: str, reg: str):
+    if not isdir(path):
+        print(path, "is not a valid path")
+        return None
+    content = listdir(path)
+    if len(content) == 0:
+        print(path, "has no content")
+        return None
+    files = [f for f in content if isfile(join(path, f)) and search(reg, f, IGNORECASE)]
+    if len(files) == 0:
+        print(path, "contains no", reg)
+        return None
+    return files
+
+
+def createNewFolders(dirs: list):
+    for d in dirs:
+        if not isdir(d):
+            makedirs(d)
+        else:
+            print("directory already exists", d)
+
+
+def createNewTemplates(objs, templatesDir, regTemplate, root):
+    templatefiles = getFilesWith(templatesDir, regTemplate)
+    for k in objs:
+        regPhase = r""
+        match k:
+            case "(i)SAT":
+                regPhase = r"sat"
+            case "iFAT":
+                regPhase = r"fat"
+
+        files = [f for f in templatefiles if search(regPhase, f, IGNORECASE)]
+        if len(files) == 0:
+            print("phase %s has no templates" % k)
+            continue
+
+        for o in objs[k]:
+            targetLocation = join(root, o)
+            tlFiles = getFilesWith(targetLocation, regPhase)
+
+            if tlFiles:
+                print(k, "files already exist in:", targetLocation)
+                print("--------------------")
+                [print("|-",f) for f in tlFiles]
+                print("--------------------")
+                continue
+
+            for f in files:
+                templatepath = join(templatesDir, f)
+                targetpath = targetLocation
+                if search(r"hut_\d{4}[a-zA-Z]{2}", f, IGNORECASE):
+                    targetpath = join(targetLocation, f[:4] + o + f[10:])
+                copy(templatepath, targetpath)

utils/input.py

@@ -0,0 +1,40 @@
+from utils.options import *
+
+keys = {
+    "ifat": 0,
+    "if": 0,
+    "f": 0,
+    "isat": 1,
+    "(i)sat": 1,
+    "is": 1,
+    "s": 1,
+    "all": 2,
+    "a": 2,
+}
+
+
+def validatedPhaseInput():
+    inputPhase = None
+    while inputPhase is None:
+        printOptions()
+        inputPhase = input()
+
+        if inputPhase.isnumeric():
+            inputPhase = int(inputPhase)
+            if inputPhase not in range(len(inputPhases)):
+                print("\n", inputPhase, "is not a valid option")
+                inputPhase = None
+            else:
+                return inputPhases[inputPhase]
+        else:
+            inputPhase = inputPhase.lower()
+            if inputPhase not in keys:
+                print("\n", inputPhase, "is not a valid option")
+                inputPhase = None
+            else:
+                return inputPhases[keys[inputPhase]]
+
+    print(
+        "Something went seriously wrong, please consult the maintainer of the codebase."
+    )
+    return inputPhase

utils/options.py

@@ -0,0 +1,44 @@
+from re import search
+
+phases = ["iFAT", "(i)SAT"]
+
+inputPhases = {**{i: k for i, k in enumerate(phases)}, **{len(phases): "All"}}
+
+exitinput = {"no", "n", "0"}
+
+
+def printOptions():
+    print("\nchoose one of the following options;\n")
+    for key in inputPhases:
+        print("[%d] %s" % (key, inputPhases[key]))
+    print()
+
+
+def getObjectsPerPhase(phase: str = "All"):
+    f = open("./objecten.txt", "r")
+    t = f.read().split("\n\n")
+    f.close()
+
+    objs = {p: [] for p in phases}
+    if phase in phases:
+        objs = {phase:[]}
+
+    regObject = r"\d{4}[a-zA-Z]{2}"
+    for g in t:
+        ls = g.split("\n")
+        k = ls[0]
+        if k in objs:
+            objs[k] = ls[1::]
+        else:
+            print("key [%s] is not recognized" % k)
+
+    objs = {k: objs[k] for k in objs if objs[k]}
+
+    for k in objs:
+        for i, o in enumerate(objs[k]):
+            m = search(regObject, o)
+            if not m:
+                continue
+            objs[k][i] = m.group(0)
+
+    return objs

utils/steps.py

@@ -0,0 +1,39 @@
+from utils.options import *
+from utils.fileSelect import *
+from utils.files import *
+from utils.input import *
+
+from os.path import join, realpath
+from os import startfile
+
+regTemplate = r'template'
+
+def choosePhase(): return validatedPhaseInput()
+
+def getTemplatesDir():
+    p = getDirectory(title='Select the template directory')
+    while not p:
+        print('input a valid directory')
+        if input('continue?').lower() in exitinput: quit()
+        p = getDirectory(title='Select the template directory')
+    
+    f = getFilesWith(p, regTemplate)
+    if not f:
+        print('no templates found')
+        if input('continue?').lower() in exitinput: quit()
+    return p
+    
+def getRoot(): return getDirectory(title='Select the root directory')
+
+def copyPasteTemplates(root:str, phase:str, templatesDir:str):
+    objs = getObjectsPerPhase(phase)
+    
+    objectslist = list(set([o for p in [objs[k] for k in objs] for o in p]))
+    
+    createNewFolders([join(root,o) for o in objectslist])
+    print("directories created")
+
+    createNewTemplates(objs, templatesDir, regTemplate, root)
+    print("templates ready")
+
+    startfile(realpath(root))