Merge pull request #19 from rad10/Python-Experimental

1.6.2 release

main.py

@@ -0,0 +1,270 @@
+import json
+import logging
+import os
+import re
+
+from modules.corrections import JSON, connectDict, correctValue
+from modules.gui import InstallError, PopupTag, mainGUI
+from modules.imageScraper import imageScraper
+from modules.sanity import checkBlankRow, sanityName
+
+# if opencv isnt installed, it'll install it for you
+try:
+    import numpy as nm
+    import cv2
+except ImportError:
+    if(os.system("pip install opencv-python")):
+        os.system("pip install --user opencv-python")
+try:
+    from PIL import Image, ImageTk
+except ModuleNotFoundError:
+    if(os.system("pip install pillow")):
+        os.system("pip install --user pillow")
+except ImportError:
+    import Image
+    import ImageTk
+
+# if tesseract isnt installed, itll install it for you
+try:
+    import pytesseract as tess
+except ImportError:
+    if(os.system("pip install pytesseract")):
+        os.system("pip install --user pytesseract")
+    import pytesseract as tess
+# installing pdf to image libraries
+try:
+    from pdf2image import convert_from_path
+except ImportError:
+    if(os.system("pip install pdf2image")):
+        os.system("pip install --user pdf2image")
+    from pdf2image import convert_from_path
+
+# Checking that external software is installed and ready to use
+# check if tesseract exists
+if os.system("tesseract --help"):
+    if os.path.exists("C:\\Program Files\\Tesseract-OCR\\tesseract.exe"):
+        tess.pytesseract.tesseract_cmd = r'C:\\Program Files\\Tesseract-OCR\\tesseract'
+    else:
+        InstallError(
+            "Tesseract", "https://github.com/UB-Mannheim/tesseract/releases", "tesseract.exe").run()
+# check if poppler exists
+if os.system("pdfimages -help"):
+    InstallError("Poppler", "https://poppler.freedesktop.org/",
+                 "pdfimages.exe").run()
+
+
+# Functions
+
+
+logging.getLogger().setLevel(logging.WARNING)
+if "info" in os.sys.argv:
+    logging.basicConfig(format="%(asctime)s: INFO %(message)s",
+                        datefmt="%H:%M:%S", level=logging.INFO)
+elif "debug" in os.sys.argv:
+    logging.basicConfig(format="%(asctime)s: DEBUG %(message)s",
+                        datefmt="%H:%M:%S", level=logging.DEBUG)
+    if not os.path.exists("debugOutput/."):
+        os.makedirs("debugOutput/dictionary", exist_ok=True)
+        os.makedirs("debugOutput/scrapper", exist_ok=True)
+    else:
+        os.system("del /s debugOutput\\*.jpg")
+
+JSONFile = open("./aliases.json", "r")
+connectDict(json.load(JSONFile))
+JSONFile.close()
+JSONChange = False  # this is only used when the database is updated
+mainDisplay = None
+
+
+def debug(label: str, content: list):
+    logging.debug("%s:", label)
+    if(logging.getLogger().level <= logging.DEBUG):
+        for i in content:
+            print(i)
+
+
+def debugImageDictionary(diction):
+    if (logging.getLogger().level <= logging.INFO):
+        debugOutput = "Sheet | SheetLen | TableRow | TableCol\n"
+        for sheet in range(len(diction)):
+            debugOutput += "{ind: 5d} | {slen: 8d} | {trow: 8d} | {tcol: 8d}\n".format(ind=sheet, slen=len(
+                diction[sheet]), trow=len(diction[sheet][1]), tcol=len(diction[sheet][1][0]))
+        logging.info(debugOutput)
+        exportToFile("debugOutput/dictionaryStats.txt", debugOutput)
+        for sheet in range(len(diction)):
+            for dates in range(len(diction[sheet][0])):
+                cv2.imwrite("debugOutput/dictionary/sheet{sheet}date{date}.jpg".format(
+                    sheet=sheet, date=dates), diction[sheet][0][dates])
+            for row in range(len(diction[sheet][1])):
+                for col in range(len(diction[sheet][1][row])):
+                    cv2.imwrite("debugOutput/dictionary/sheet{sheet}table{row}{col}.jpg".format(
+                        sheet=sheet, row=row, col=col), diction[sheet][1][row][col])
+
+
+def exportToFile(dir, content):
+    open(dir, "w").write(content)
+
+
+def appendToFile(dir, content):
+    try:
+        inside = open(dir, "r").read()
+        open(dir, "w").write(inside + content)
+    except:
+        open(dir, "w").write(content)
+
+
+def TranslateDictionary(sheetsDict, gui=False, outputDict=None):
+    """ Phase two of plan. This function goes through the image dictionary passed 
+    to it and creates a matrix of the dictionary in text.\n
+    @param sheetsDict: a matrix of images made from a table.\n
+    @param gui: whether to switch on global gui manipulation for the progress bar.\n
+    @param outputDict: a variable passed by reference instead of using return.\n
+    @return a matrix of strings that represents the text in the image dictionary.
+    """
+    global JSON
+    global JSONChange
+    results = [[] for x in sheetsDict]  # results the size of pages in dict
+
+    # GUI widgets to manipulate while in middle of function
+    if(gui):
+        sheetMax = len(sheetsDict)
+        sheetInd = 0
+        rowInd = 0
+        progressMax = 1
+
+        # Gui Texts
+        textScan = "Scanning\tSheet: {sInd} of {sMax}\tRow: {rInd} of {rMax}"
+        textSanitize = "Sanitizing\tSheet: {sInd} of {sMax}\tRow: {rInd} of {rMax}"
+
+        # Getting max for progress bar
+        for sheet in sheetsDict:
+            progressMax += len(sheet[1]) - 1
+        mainDisplay.progressBar.configure(
+            mode="determinate", maximum=progressMax)
+
+    # Collecting data to database
+    for sheet in range(len(sheetsDict)):
+        if gui:
+            sheetInd += 1
+            rowMax = len(sheetsDict[sheet][1]) - 1
+        # Collecting dates on page first
+        dates = []
+        dformat = re.compile(r'\d{1,2}\/\d{1,2}\/(\d{4}|\d{2})')
+        dstr = ""
+        for date in sheetsDict[sheet][0]:
+            dstr = tess.image_to_string(date).replace(
+                "\n", "").replace(" ", "")
+            if (bool(dformat.match(dstr))):
+                dates.insert(0, (dstr, 1, True))
+            else:
+                dates.append((dstr, 1, True))
+
+        # | Full name | Time in | Time out | hours (possibly blank) | purpose | date | day (possibly blank) |
+        # skips first row which is dummy
+        for row in range(1, len(sheetsDict[sheet][1])):
+            if gui:
+                rowInd += 1
+                mainDisplay.progressBar.step()
+                mainDisplay.sheetStatus.configure(
+                    text=textScan.format(sInd=sheetInd, sMax=sheetMax, rInd=rowInd, rMax=rowMax))
+                mainDisplay.root.update_idletasks()
+            results[sheet].append([None for x in range(5)])  # array of 5 slots
+            # skip first col which is dummy
+            for col in range(1, len(sheetsDict[sheet][1][row])):
+                logging.info("Sheet[%d]: [%d, %d]", int(
+                    sheetInd), int(rowInd), int(col))
+                results[sheet][row - 1][col -
+                                        1] = correctValue(sheetsDict[sheet][1][row][col], col)
+            results[sheet][-1].extend(dates)
+        if (logging.getLogger().level <= logging.DEBUG):
+            for e in range(len(results)):
+                debug("Results Sheet[" + str(e) + "]", results[e])
+
+    # Checking names for repetitions
+    results = sanityName(results)
+
+    # Analysis
+    for sheet in range(len(results)):
+        # Iterating through results to see where errors occured
+        for row in range(len(results[sheet])):
+            for col in range(len(results[sheet][row][:-len(dates)])):
+                mainDisplay.sheetStatus.configure(
+                    text=textSanitize.format(sInd=sheet + 1, sMax=len(results), rInd=row + 1, rMax=len(results[sheet])))
+                if (results[sheet][row][col][2] == False):
+                    results[sheet][row][col] = mainDisplay.requestCorrection(
+                        sheetsDict[sheet][1][row + 1][col + 1], results[sheet][row][col][0])
+                    if (col + 1 in [1, 5]):
+                        for entry in JSON["names"][str(col + 1)]:
+                            if (results[sheet][row][col][0].lower() == entry):
+                                break
+                        else:
+                            JSONChange = True
+                            # if the name possibly entered in by the user doesnt exist in the database, add it
+                            JSON["names"][str(
+                                col + 1)].append(results[sheet][row][col][0].lower())
+
+        # Checking if any rows are blank
+        for row in range(len(results[sheet])-1, -1, -1):
+            if checkBlankRow(results[sheet][row]):
+                results[sheet].pop(row)
+
+    if(outputDict == None):
+        return results
+    else:
+        globals()[outputDict] = results.copy()
+        return
+
+
+def arrayToCsv(directory):
+    """takes a matrix and returns a string in CSV format.
+    var directory: a string[][] matrix that contains the information of people at the center.
+    returns: a string that contains all the information in CSV format.
+    """
+    cvarray = ''
+    for i in range(len(directory)):
+        for e in range(len(directory[i])-1):
+            cvarray += (directory[i][e][0]+",")
+        cvarray += (directory[i][-1][0]+"\n")
+    logging.debug("cvarray:\n%s", cvarray)
+    return (cvarray+"\n")
+
+
+def main():
+    ##########################################
+    ## Phase 3: Hooking everything together ##
+    ##########################################
+
+    try:
+        signinsheet = mainDisplay.signinsheet
+        outputCSV = mainDisplay.outputCSV
+        imageDictionary = imageScraper(signinsheet)
+        debugImageDictionary(imageDictionary)
+        textDictionary = TranslateDictionary(imageDictionary, gui=True)
+        csvString = ""
+        for sheet in textDictionary:
+            csvString += arrayToCsv(sheet)
+        exportToFile(mainDisplay.outputCSV, csvString)
+        mainDisplay.errorLabel.configure(text="All finished.")
+    except BaseException:
+        import traceback
+        PopupTag(mainDisplay, "Error", "Looks like something went wrong.\n" +
+                 str(os.sys.exc_info())+"\n"+str(traceback.format_exc()), "#ff0000").run()
+        raise
+    PopupTag(mainDisplay, "Done",
+             "Congrats! its all finished.\nLook at your csv and see if it looks alright.").run()
+    if (JSONChange):
+        JSON["names"]["1"].sort()  # Sorting new libraries for optimization
+        JSON["names"]["5"].sort()
+        JSONFile = open("aliases.json", "w")
+        json.dump(JSON, JSONFile, indent=4, separators=(
+            ",", ": "), ensure_ascii=True, sort_keys=True)
+        JSONFile.close()
+
+    # Cleaning old ocr files from tmp
+    os.system("del /s /q %tmp%\\tess_*.hocr")
+    return
+
+
+mainDisplay = mainGUI(main)
+if __name__ == "__main__":
+    mainDisplay.run()