04/28 Updated sprint 4 & 5

routes/projects.py

@@ -6,7 +6,7 @@ Project CRUD and related API routes.
 Routes: /projects, /projects/create, /projects/<id>/edit,
         /projects/<id>/toggle, /api/projects/active
 """
-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
 import json
 from datetime import datetime
 
@@ -76,7 +76,9 @@ def create_project():
 def edit_project(project_id):
     """Edit existing project"""
     try:
-        project = Project.query.get_or_404(project_id)
+        project = db.session.get(Project, project_id)
+        if project is None:
+            abort(404)
 
         if request.method == 'POST':
             old_name = project.name
@@ -114,7 +116,9 @@ def edit_project(project_id):
 def toggle_project(project_id):
     """Toggle project active status"""
     try:
-        project = Project.query.get_or_404(project_id)
+        project = db.session.get(Project, project_id)
+        if project is None:
+            abort(404)
         old_status = project.active_status
         project.active_status = not project.active_status