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