Updated functionalities

app/routes/audit.py

@@ -1,6 +1,6 @@
 import logging
 from datetime import datetime, timedelta
-from flask import Blueprint, render_template, request, redirect, url_for, flash
+from flask import Blueprint, render_template, request, redirect, url_for, flash, abort
 from flask_login import login_required
 from app import db
 from app.models.audit import AuditLog
@@ -88,7 +88,9 @@ def index():
 @login_required
 @admin_required
 def view(log_id):
-    entry = AuditLog.query.get_or_404(log_id)
+    entry = db.session.get(AuditLog, log_id)
+    if entry is None:
+        abort(404)
     return render_template('audit/view.html', entry=entry)