import os
import pathlib
import platform
import sys
import exiftool


def exif_rename(dir: str):
    base_directory = pathlib.Path("/pool1/clarkzjw/Media/Photo")
    directory = str(base_directory.joinpath(dir).absolute()) + "/"

    files = os.listdir(directory)
    for file in files:
        filename = directory + file
        ext = file.split(".")[-1]
        if platform.system() == "Windows":
            old_filename = filename.split("\\")[-1].split(".")[0]
        else:
            old_filename = filename.split("/")[-1].split(".")[0]
        img = filename

        if (file.startswith("IMG_") and (file.endswith(".CR2") or file.endswith(".CR3"))) or \
                (file.startswith("R") and (file.endswith(".JPG") or file.endswith(".DNG"))) or \
                (file.startswith("PXL_") and (file.endswith(".jpg") or file.endswith(".dng"))) or \
                (file.startswith("DSC") and file.endswith(".ARW")):
            with exiftool.ExifToolHelper() as et:
                for d in et.get_tags(filename, tags="EXIF:DateTimeOriginal"):
                    if "EXIF:DateTimeOriginal" in d.keys():
                        timestamp = d["EXIF:DateTimeOriginal"].replace(":", "-")
                        filename = directory + timestamp + "-{}.{}".format(old_filename, ext)
                        print(filename)
                        os.rename(img, filename)
                        break


if __name__ == "__main__":
    if len(sys.argv) <= 1:
        exit(0)
    exif_rename(sys.argv[1])