from os.path import exists
from shutil import copytree, rmtree

from flask_admin import __file__ as flaskAdminPath


def copyAdminTemplates():
    src = flaskAdminPath.removesuffix("__init__.py") + "templates/bootstrap4/admin"
    if not exists(src):
        print("Templates could not be found in", src)
        print("You can also copy them manually.")
        return False

    dist = "advlabdb/templates/admin"
    if exists(dist):
        while True:
            ans = input(
                f"The directory {dist} already exists. Enter 'o' to override it and update the templates or enter 's' to stop the process: "
            ).lower()
            if ans == "s":
                print("Process stopped!")
                return False
            elif ans == "o":
                break
        rmtree(dist)
        print("Old templates deleted!")

    copytree(src, dist)
    print("Copied", src, "->", dist)
    return True


if __name__ == "__main__":
    if copyAdminTemplates():
        print("Done!")
    else:
        print("Did not copy!")