Update record_attendance.py

record_attendance.py

@@ -1,20 +1,28 @@
 import csv
+import os
 from datetime import datetime
 import streamlit as st
-import os
 
-def record_attendance_new(present_students: list, subject: str):
+def record_attendance_new(present_students: list, subject: str, max_roll_number=26):
     # Automatically get the current date in YYYY-MM-DD format
     date = datetime.now().strftime("%Y-%m-%d")
-    csv_file_path = f'{subject}_attendance.csv'
 
-    # Read existing data or create new if file doesn't exist
+    # Convert present_students to integers
+    present_students = list(map(int, present_students))
+    print("Present Students:", present_students)  # Debugging line
+
+    # 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 file:
-            reader = csv.reader(file)
+        with open(csv_file_path, 'r', newline='') as csvfile:
+            reader = csv.reader(csvfile)
             rows = list(reader)
     else:
-        rows = [['Roll Number']]  # Header only for new file
+        # If file doesn't exist, create a new one with header
+        rows = [['Roll Number']]
 
     # Check if the date column already exists
     if date not in rows[0]:
@@ -22,23 +30,32 @@ def record_attendance_new(present_students: list, subject: str):
 
     date_col_index = rows[0].index(date)
 
-    # Update or add roll numbers for the given date
-    for roll_number in present_students:
+    # Update or add attendance for each roll number
+    for roll_number in range(1, max_roll_number + 1):
         found = False
         for row in rows[1:]:
-            if row[0] == str(roll_number):
+            if int(row[0]) == roll_number:
                 found = True
                 if len(row) < date_col_index + 1:
                     row.extend([''] * (date_col_index - len(row) + 1))
-                row[date_col_index] = 'Present'
+                
+                if roll_number in present_students:
+                    row[date_col_index] = 'Present'
+                    print(f"Updating {roll_number}: Present")  # Debugging line
+                else:
+                    row[date_col_index] = 'Absent'
+                    print(f"Updating {roll_number}: Absent")  # Debugging line
                 break
+        
         if not found:
-            new_row = [str(roll_number)] + [''] * (date_col_index - 1) + ['Present']
+            new_row = [str(roll_number)] + [''] * (date_col_index - 1)
+            new_row.append('Present' if roll_number in present_students else 'Absent')
             rows.append(new_row)
+            print(f"Adding new row for {roll_number}: {'Present' if roll_number in present_students else 'Absent'}")
 
     # Write the updated data back to the CSV file
-    with open(csv_file_path, 'w', newline='') as file:
-        writer = csv.writer(file)
+    with open(csv_file_path, 'w', newline='') as csvfile:
+        writer = csv.writer(csvfile)
         writer.writerows(rows)
 
     # Create a download button
@@ -52,5 +69,4 @@ def record_attendance_new(present_students: list, subject: str):
         mime='text/csv'
     )
 
-    print(f"Attendance recorded for {len(present_students)} students on {date} in {subject}.")
-
+    print(f"Attendance recorded for {len(present_students)} students on {date} in {subject}.")