import os, time, hashlib, PIL
import tkinter as tk
from tkinter import filedialog, messagebox, simpledialog
from PIL import Image

def md5(fpath):
    hash_md5 = hashlib.md5()
    with open(fpath, "rb") as f:
        for chunk in iter(lambda: f.read(4096), b""):
            hash_md5.update(chunk)
    return hash_md5.hexdigest()


root = tk.Tk()
root.withdraw()

# Message with requested parameters
messagebox.showinfo(title="Ready to start", message="Pay attention to the following things: \n1) The RAW-recovery from r-Studio should be in the first parsed folder. \n2) Within the images-filder the \"JPEGs from camera\" folder should be before the folder with the normal JPEGs!")

# Select base folder
base_folder = filedialog.askdirectory(parent=root, initialdir="/", title='Please select the base-directory')

output_folder_name = "000_merged"
logfile_name = "000_merged.log"
found_types = []
md5_list = []
mctr = 0
sctr = 0

if base_folder is not None:
    # Check if output-folder exists and create it if not
    output_folder = os.path.join(base_folder, output_folder_name)
    if not os.path.isdir(output_folder):
        os.mkdir(output_folder)

    # read md4 values from previous run
    logfile_path = os.path.join(base_folder, logfile_name)
    if os.path.isfile(logfile_path):
        with open(logfile_path) as f:
            for line in f:
                md5_list.append(line.strip())

    # Write logfile
    with open(logfile_path, "a") as f:
        # iterate over folders
        for subdir, dirs, files in os.walk(base_folder):
            for filename in files:
                fpath = os.path.join(subdir, filename)
                # don't process output folder
                if not output_folder_name in fpath:
                    ftype = fpath.split(".")[-1]
                    ftype = ftype.lower()
                    ftype_folder = ftype

                    if ftype == "jpeg":
                        ftype_folder = "jpg"

                    # read EXIF data from JPGs
                    if ftype_folder == "jpg":
                        try:
                            exif = Image.open(fpath)._getexif()
                            try:
                                if str(exif[271]) != "" and str(exif[272]) != "":
                                    ftype_folder = "jpg_camera"
                                    ftype = "jpg_camera"
                                    
                            except (KeyError, TypeError) as e:
                                pass
                                
                        except PIL.UnidentifiedImageError:
                            pass

                    # File without extention
                    if not "." in fpath:
                        ftype_folder = "unknown"
                        ftype = "unknown"

                    # Assemble folder-path for the filetype
                    type_folder = os.path.join(output_folder, ftype_folder)

                    # remember new found types and create folder
                    if not ftype in found_types:
                        found_types.append(ftype)
                        if not os.path.isdir(type_folder):
                            os.mkdir(type_folder)

                    # get md5 value of file
                    file_md5 = md5(fpath)

                    # prepare output
                    out = fpath[0:100] + " "
                    out += "." * (102 - len(out))

                    # check if file was peviously sorted
                    if file_md5 not in md5_list:
                        md5_list.append(file_md5)
                        f.write(file_md5 + "\n")
                        mctr += 1

                        # Check if file exsts 
                        if os.path.isfile(os.path.join(type_folder, filename)):
                            filename = filename.replace("." + ftype, "_" + str(mctr) + "." + ftype)
                            print(out + ".. RENAMED")
                        
                        # Move file
                        try:
                            tpath = os.path.join(type_folder, filename)
                            os.rename(fpath, tpath)
                            print(out + ".. ADDED")
                        except FileExistsError:
                            tpath = tpath + str(time.time()) + "." + ftype
                            os.rename(fpath, tpath)
                            print(out + ".. ADDED")
                        except FileNotFoundError:
                            print(tpath + ".. NOT FOUND")

                    # if was perv. sorted just skip file
                    else:
                        print(out + ".. SKIPPED")
                        sctr += 1
                    
print("UNIQ FILES:    " + str(mctr))
print("SKIPPED FILES: " + str(sctr))