Update record_attendance.py

record_attendance.py

@@ -1,76 +1,60 @@
-import csv
-import os
-from datetime import datetime
-import streamlit as st
+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 it exists, otherwise create a new one
-    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)])
-    
-    # Check if the date column already exists
-    if date not in rows[0]:
-        # Add the new date column if it doesn't exist
-        rows[0].append(date)
-    
-    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
-        while len(row) <= date_col_index:
-            row.append('')
-        
-        if roll_number in present_students:
-            row[date_col_index] = 'Present'
-            print(f"Marking {roll_number}: Present")  # Debugging line
-        elif row[date_col_index] == '':
-            # Only mark as absent if the cell is empty
-            row[date_col_index] = 'Absent'
-            print(f"Marking {roll_number}: Absent")  # Debugging line
-    
-    # Cross-check function to ensure all present students are marked correctly
-    def cross_check_attendance():
-        for row in rows[1:]:
-            roll_number = int(row[0])
-            if roll_number in present_students and row[date_col_index] != 'Present':
-                row[date_col_index] = 'Present'
-                print(f"Cross-check: Correcting {roll_number} to Present")
+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")  
 
-    # Run the cross-check
-    cross_check_attendance()
-    
-    # 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 on {date} in {subject}.")
+    # 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[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 on {date} in {subject}.")
+