main_file.py

import os
import sys
import cv2
import excel
import tkinter
from capture import capture
from tkinter import messagebox
from face_detect import FaceDetect
from students_list import StudentsList

root = tkinter.Tk()
root.withdraw()


class MainFile:
    def __init__(self, class_name):
        self.i = 0
        self.class_name = class_name

    def capture_and_mark(self):
        sl = StudentsList(self.class_name)
        students, roll_numbers = sl.load_pkl_file()

        FaceDetectObj = FaceDetect(self.class_name)

        Yes = True
        No = False
        Cancel = None

        i = 0
        while i <= 2:
            captured_image = None
            frame = None

            students_present = []
            while len(students_present) == 0:
                captured_image, frame = capture()
                students_present = FaceDetectObj.recognize(captured_image, roll_numbers)
                if students_present == "No Training Data":
                    return

            try:
                name_student_present = students[roll_numbers.index(students_present[0])]
            except:
                messagebox.showerror(
                    "Error",
                    "Recognized student not in database\nUnable to mark attendance",
                )
                return

            response = messagebox.askyesnocancel(
                "Confirm your identity",
                students_present[0] + "\n" + name_student_present,
            )

            if response is Yes:
                wb = excel.attendance_workbook(self.class_name)
                excel.mark_present(wb, students_present, self.class_name)
                img_path = os.path.join(
                    os.getcwd(),
                    "images",
                    self.class_name,
                    "s" + students_present[0][-2:],
                    os.path.basename(captured_image),
                )
                cv2.imwrite(img_path, frame)
                os.remove(captured_image)
                messagebox.showinfo(
                    "Attendance Confirmation", "Your attendance is marked!"
                )
                break
            elif response is Cancel:
                break
            elif response is No:
                if i == 2:
                    img_path = os.path.join(
                        os.getcwd(),
                        "images",
                        self.class_name,
                        "unrecognized students",
                        os.path.basename(captured_image),
                    )
                    cv2.imwrite(img_path, frame)
                    messagebox.showinfo(
                        "Unrecognized Student",
                        "You were not recognized as any student of this class.\nYour attendance will be marked later if you really are",
                    )
                    cv2.imwrite(img_path, frame)
                os.remove(captured_image)

            i += 1


if __name__ == "__main__":
    pass