import os

import shutil

def createDirectoriesfunc():

    def make_if_dont_exist(folder_path,overwrite):

        """

        creates a folder if it does not exists

        input:

        folder_path : relative path of the folder which needs to be created

        over_write :(default: False) if True overwrite the existing folder

        """

        if os.path.exists(folder_path):

            if not overwrite:

                print(f"{folder_path} exists.")

            else:

                print(f"{folder_path} overwritten")

                shutil.rmtree(folder_path)

                os.makedirs(folder_path)

        else:

            os.makedirs(folder_path)

            print(f"{folder_path} created!")

    base_directory = 'D:/MRI - Tairawhiti (User POV)'

    new_folder_name = ['nnUNet Data', 'Raw NIFITI Segmentation Masks (3D Slicer Output)', 'Pre-Trained Models (Google Colab)', 'Model Code', 'Raw DICOM MRI Scans', 'Patient Segmentation Tasks (Google Colab)']

    overwrite = False

    for new_fname in new_folder_name:

        folder_path = os.path.join(base_directory, new_fname)

        make_if_dont_exist(folder_path, overwrite)

    base_directory = 'D:/MRI - Tairawhiti (User POV)/nnUNet Data'

    new_folder_name = ['masks', 'multiclass_masks', 'scans']

    for new_fname in new_folder_name:

        folder_path = os.path.join(base_directory, new_fname)

        make_if_dont_exist(folder_path, overwrite)

    base_directory = 'D:/MRI - Tairawhiti (User POV)/nnUNet Data/masks'

    new_folder_name = ['FEMUR', 'FIBULA', 'TIBIA', 'PELVIS']

    for new_fname in new_folder_name:

        folder_path = os.path.join(base_directory, new_fname)

        make_if_dont_exist(folder_path, overwrite)