Update record_attendance.py

record_attendance.py

@@ -8,18 +8,19 @@ def record_attendance_new(present_students: list, subject: str, max_roll_number=
     date = datetime.now().strftime("%Y-%m-%D")  
     length = len(present_students)  
     ty = type(present_students).__name__  # Get the name of the type as a string  
-    st.success(f"{length} students' attendance is going to complete {ty}.")
-    
+    st.success(f"{length} students' attendance is going to complete {ty}.")  
+
     # Convert present_students to integers  
     present_students = list(map(int, present_students))  
     print("Present Students:", present_students)  # Debugging line  
     length = len(present_students)  
     ty = type(present_students).__name__  # Get the name of the type as a string  
-    st.success(f"{length} students' attendance is going to complete {ty}.") 
+    st.success(f"{length} students' attendance is going to complete {ty}.")  
+
     # Get the path of the uploaded CSV file  
     csv_file_path = f'{subject}.csv'  
     print("CSV File Path:", csv_file_path)  # Debugging line  
-    
+
     # Read the existing CSV file  
     if os.path.exists(csv_file_path):  
         with open(csv_file_path, 'r', newline='') as csvfile:  
@@ -28,7 +29,7 @@ def record_attendance_new(present_students: list, subject: str, max_roll_number=
     else:  
         # If file doesn't exist, create a new one with header and roll numbers  
         rows = [['Roll Number']]  
-        for roll_number in range(1, 26 + 1):  
+        for roll_number in range(1, max_roll_number + 1):  
             rows.append([str(roll_number)])  
 
     # Add the new date column  
@@ -36,15 +37,16 @@ def record_attendance_new(present_students: list, subject: str, max_roll_number=
     date_col_index = len(rows[0]) - 1  
 
     # Create a list to hold the attendance status for each roll number  
-    attendance_status = ['Absent'] * 26 
-    for i in range(1, 26 + 1):  
-        if i in present_students:  
+    attendance_status = ['Absent'] * max_roll_number  
+    for i in present_students:  
+        if 1 <= i <= max_roll_number:  
             attendance_status[i - 1] = 'Present'  
 
     # Update the attendance status in the CSV data  
     for i, row in enumerate(rows[1:]):  
         roll_number = int(row[0])  
-        row[date_col_index] = attendance_status[roll_number - 1]  
+        if 1 <= roll_number <= max_roll_number:  
+            row[date_col_index] = attendance_status[roll_number - 1]  
 
     # Write the updated data back to the CSV file  
     with open(csv_file_path, 'w', newline='') as csvfile:  
@@ -54,7 +56,7 @@ def record_attendance_new(present_students: list, subject: str, max_roll_number=
     # Create a download button  
     with open(csv_file_path, 'r') as file:  
         csv_contents = file.read()  
-    
+
     st.download_button(  
         label="Download updated attendance file",  
         data=csv_contents,  
@@ -63,4 +65,5 @@ def record_attendance_new(present_students: list, subject: str, max_roll_number=
     )  
 
     print(f"Attendance recorded for {len(present_students)} students on {date} in {subject}.")
+