#importing libraries
import face_recognition
import os
from Student import Student
from datetime import datetime


#method for encoding the image
#it takes path as argument, which is the address of the photo we want to encode
def encoding(path):
    image = face_recognition.load_image_file(path)
    imageEncoding = face_recognition.face_encodings(image)[0]
    return imageEncoding


#this mehod is for comparing the photo from unkwon folder with all photos in known folder
#it takes 2 arguments, unkownPerson is the path of photo in unknown folder, while nameOfPerson is simple name of photo
def compare(unkownPerson,nameOfPerson):
    isAttendant = False     #it's used to check if the photo from unkown folder is already in known folder
    for image in os.listdir('/Users/yermukhanbet/PycharmProjects/dataBase/venv/lib/python3.7/known'): #iterates through photos in the known folder
        if image == '.DS_Store': #checks if the image is the hidden file ".DS_Store" and if yes then it passes and goes to the next file in the folder
            continue
        else:
            known_image_path = '/Users/yermukhanbet/PycharmProjects/dataBase/venv/lib/python3.7/known/'+image
            result = face_recognition.compare_faces([encoding(known_image_path)],encoding(unkownPerson), tolerance=0.45)
            print(image, str(result))
            if result[0] == True:
                print("Your face matched! You're attended")
                os.remove(unkownPerson)
                isAttendant = True
                break
    if isAttendant == False:
        print("Oops, it seems that you are not in this class!")
        id = input("enter your id ")
        os.rename(unkownPerson, "/Users/yermukhanbet/PycharmProjects/dataBase/venv/lib/python3.7/known/"+str(id)+".jpg")
        addNewStudentToList(id)
        print("You're added successfully!")

def addNewStudentToList(id):
    firstName = input("Enter your first name: ")
    lastName = input("Enter your last name: ")
    checkInTime = getTheCurrentTime()
    newStudent = Student(firstName,lastName,id,checkInTime)
    listOfStudents.append(newStudent)

def getTheCurrentTime():
    now = datetime.now()
    currentTime = now.strftime("%H:%M:%S")
    return currentTime

def printPresentStudents():
    for student in listOfStudents:
        print(student)

def mainCompare():
    if len(os.listdir('/Users/yermukhanbet/PycharmProjects/dataBase/venv/lib/python3.7/unknown')) - 1 == 0:
        print("There are no pictures")
    else:
        for image in os.listdir('/Users/yermukhanbet/PycharmProjects/dataBase/venv/lib/python3.7/unknown'):
            if image == '.DS_Store':
                continue
            else:
                return compare('/Users/yermukhanbet/PycharmProjects/dataBase/venv/lib/python3.7/unknown/' + image, image)


listOfStudents = []
mainCompare()

printPresentStudents()