unread notifs

routes/contacts.py

@@ -1,11 +1,11 @@
-from flask import Blueprint, render_template, redirect, url_for, flash, request
+from flask import Blueprint, render_template, redirect, url_for, flash, request, jsonify
 from flask_login import login_required, current_user
-from models import db, User
+from models import db, User, Notif
 from forms import UserForm
 from flask import abort
 from sqlalchemy import or_
 from routes.auth import require_password_change
-from utils import log_event, create_notification
+from utils import log_event, create_notification, get_unread_count
 import json
 import os
 from werkzeug.utils import secure_filename
@@ -17,6 +17,13 @@ UPLOAD_FOLDER = os.path.join(os.getcwd(), 'uploads', 'profile_pics')
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
 
+@contacts_bp.context_processor
+def inject_unread_notifications():
+    if current_user.is_authenticated:
+        unread_count = get_unread_count(current_user.id)
+        return {'unread_notifications': unread_count}
+    return {'unread_notifications': 0}
+
 def admin_required():
     if not current_user.is_authenticated:
         return redirect(url_for('auth.login'))