import DBHelper
import shutil
from joblib import Parallel, delayed
import multiprocessing
import os

import Facial_Recognition_Enrollment


def remove_your_face(firstname, lastname, email, phone):
    users = DBHelper.db.child("Users").get()
    try:
        count = 0
        count2 = 0
        for user in users.each():
            count += 1
        print("The specified user will be removed...")
        for user in users.each():
            count2 += 1
            if DBHelper.get_firstname("User_" + str(count2)) == firstname and DBHelper.get_lastname(
                    "User_" + str(count2)) == lastname and DBHelper.get_email(
                "User_" + str(count2)) == email and DBHelper.get_phone("User_" + str(count2)) == phone:
                Parallel(n_jobs=multiprocessing.cpu_count())(
                    delayed(remove_parallel_user_photos)(i, count2) for i in range(50))
                DBHelper.remove_data("User_" + str(count2))
                shutil.rmtree("Facial_images/face_rec/train/User_" + str(count2))
                print("Successfully removed the User.")
                break
        print("Reorganizing the Users...")
        if count2 != count and count - count2 - 1 != 0:
            for x in range(count - count2):
                if not os.path.isdir("Facial_images/face_rec/train/User_" + str(count2)):
                    os.makedirs("Facial_images/face_rec/train/User_" + str(count2))
                Parallel(n_jobs=multiprocessing.cpu_count())(
                    delayed(update_parallel_user_photos)(i, count2) for i in range(50))
                DBHelper.upload_data("User_" + str(count2), DBHelper.get_firstname("User_" + str(count2 + 1)),
                                     DBHelper.get_lastname("User_" + str(count2 + 1)),
                                     DBHelper.get_email("User_" + str(count2 + 1)),
                                     DBHelper.get_phone("User_" + str(count2 + 1)))
                count2 += 1
            DBHelper.remove_data("User_" + str(count))
            shutil.rmtree("Facial_images/face_rec/train/User_" + str(count))
        elif count2 != count and count - count2 - 1 == 0:
            if not os.path.isdir("Facial_images/face_rec/train/User_" + str(count2)):
                os.makedirs("Facial_images/face_rec/train/User_" + str(count2))
            Parallel(n_jobs=multiprocessing.cpu_count())(
                delayed(update_parallel_user_photos)(i, count2) for i in range(50))
            DBHelper.upload_data("User_" + str(count2), DBHelper.get_firstname("User_" + str(count2 + 1)),
                                 DBHelper.get_lastname("User_" + str(count2 + 1)),
                                 DBHelper.get_email("User_" + str(count2 + 1)),
                                 DBHelper.get_phone("User_" + str(count2 + 1)))
            DBHelper.remove_data("User_" + str(count))
            shutil.rmtree("Facial_images/face_rec/train/User_" + str(count))
        print("Data saved! Starting enrollment...")
        Facial_Recognition_Enrollment.enroll_face_dataset()
        print("Face removal completed!")
        print("Success.")
    except:
        print("No Users exist for User Removal.")


def remove_parallel_user_photos(i, count2):
    DBHelper.delete_user_photo("User_" + str(count2) + "/" + str(i) + ".jpg")


def update_parallel_user_photos(i, count2):
    DBHelper.download_user_photo_other("User_" + str(count2 + 1) + "/" + str(i) + ".jpg",
                                       "User_" + str(count2) + "/" + str(i) + ".jpg")
    DBHelper.upload_user_photo("User_" + str(count2) + "/" + str(i) + ".jpg")
    DBHelper.delete_user_photo("User_" + str(count2 + 1) + "/" + str(i) + ".jpg")


if __name__ == "__main__":
    f = input('Enter your First Name:')
    l = input('Enter your Last Name:')
    e = input('Enter your E-Mail:')
    p = input('Enter your Phone:')
    remove_your_face(f, l, e, p)