Update record_attendance.py

record_attendance.py

@@ -1,61 +1,67 @@
-import csv
-import os
-from datetime import datetime
-import streamlit as st
-
-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")
-
-    # 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 csvfile:
-            reader = csv.reader(csvfile)
-            rows = list(reader)
-    else:
-        # If file doesn't exist, create a new one with header and roll numbers
-        rows = [['Roll Number']]
-        for roll_number in range(1, max_roll_number + 1):
-            rows.append([str(roll_number)])
-
-    # Add the new date column
-    rows[0].append(date)
-    date_col_index = len(rows[0]) - 1
-
-    # Update attendance for each roll number
-    for row in rows[1:]:
-        roll_number = int(row[0])
-        if roll_number in present_students:
-            row.append('Present')
-            print(f"Marking {roll_number}: Present")  # Debugging line
-        else:
-            row.append('Absent')
-            print(f"Marking {roll_number}: Absent")  # Debugging line
-
-    # Write the updated data back to the CSV file
-    with open(csv_file_path, 'w', newline='') as csvfile:
-        writer = csv.writer(csvfile)
-        writer.writerows(rows)
-
-    # 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,
-        file_name=csv_file_path,
-        mime='text/csv'
-    )
+import csv  
+import os  
+from datetime import datetime  
+import streamlit as st  
+
+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")  
+
+    # Convert present_students to integers  
+    present_students = list(map(int, present_students))  
+    print("Present Students:", present_students)  # Debugging line  
+
+    # Get the path of the CSV file for the specified subject  
+    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:  
+            reader = csv.reader(csvfile)  
+            rows = list(reader)  
+    else:  
+        # If file doesn't exist, create a new one with header and roll numbers  
+        rows = [['Roll Number'] + [date]]  # Start with header row  
+        for roll_number in range(1, max_roll_number + 1):  
+            rows.append([str(roll_number)])  
+
+    # Check if the date column already exists  
+    if date not in rows[0]:  
+        rows[0].append(date)  # Add the new date column if it doesn't exist  
 
-    print(f"Attendance recorded for {len(present_students)} students on {date} in {subject}.")
+    # Get the index of the date column  
+    date_col_index = rows[0].index(date)  
+
+    # Update attendance for each roll number  
+    for row in rows[1:]:  
+        roll_number = int(row[0])  
+        # Ensure the row has enough columns for the date  
+        if len(row) <= date_col_index:  
+            row.extend([''] * (date_col_index - len(row) + 1))  # Fill with empty strings if needed  
+
+        # Mark attendance  
+        if roll_number in present_students:  
+            row[date_col_index] = 'Present'  
+            print(f"Marking {roll_number}: Present")  # Debugging line  
+        else:  
+            row[date_col_index] = 'Absent'  
+            print(f"Marking {roll_number}: Absent")  # Debugging line  
+
+    # Write the updated data back to the CSV file  
+    with open(csv_file_path, 'w', newline='') as csvfile:  
+        writer = csv.writer(csvfile)  
+        writer.writerows(rows)  
+
+    # 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,  
+        file_name=csv_file_path,  
+        mime='text/csv'  
+    )  
 
-    print(f"Attendance recorded for {len(present_students)} students in {subject} on {date}.")
+    print(f"Attendance recorded for {len(present_students)} students on {date} in {subject}.")