04/28 Updated sprint 4 & 5

routes/employees.py

@@ -6,7 +6,7 @@ Employee CRUD and search routes.
 Routes: /employees, /employees/create, /employees/<id>/edit,
         /employees/<id>/delete, /api/employees/search, /employees/<id>
 """
-from flask import Blueprint, render_template, request, redirect, flash, session, jsonify, url_for
+from flask import abort, Blueprint, render_template, request, redirect, flash, session, jsonify, url_for
 from datetime import datetime, date
 
 from extensions import db, logger_handler
@@ -162,7 +162,9 @@ def edit_employee(employee_index):
     """Edit existing employee (Admin only)"""
     try:
         # Get employee by index (primary key)
-        employee = Employee.query.get_or_404(employee_index)
+        employee = db.session.get(Employee, employee_index)
+        if employee is None:
+            abort(404)
         
         if request.method == 'POST':
             # Get form data
@@ -244,7 +246,9 @@ def delete_employee(employee_index):
         )
 
         # Get employee by index (primary key)
-        employee = Employee.query.get_or_404(employee_index)
+        employee = db.session.get(Employee, employee_index)
+        if employee is None:
+            abort(404)
 
         # Store employee data for logging before deletion
         employee_data = {