Add work queue icon to navigation bar

[cheatdb.git] / app / template_filters.py

diff --git a/app/template_filters.py b/app/template_filters.py
index 574c1b1f00f975c1f1516ade5289767a830e647b..39d38065e454ec21850cb8ce0e8d281927e87396 100644 (file)
--- a/app/template_filters.py
+++ b/app/template_filters.py
@@ -1,6 +1,7 @@
 from . import app
-from .models import Permission
+from .models import Permission, Package, PackageState, PackageRelease
 from .utils import abs_url_for, url_set_query
+from flask_user import current_user
 from urllib.parse import urlparse
 
 @app.context_processor
@@ -12,6 +13,15 @@ def inject_functions():
        check_global_perm = Permission.checkPerm
        return dict(abs_url_for=abs_url_for, url_set_query=url_set_query, check_global_perm=check_global_perm)
 
+@app.context_processor
+def inject_todo():
+       todo_list_count = None
+       if current_user.is_authenticated and current_user.canAccessTodoList():
+               todo_list_count = Package.query.filter_by(state=PackageState.READY_FOR_REVIEW).count()
+               todo_list_count += PackageRelease.query.filter_by(approved=False, task_id=None).count()
+
+       return dict(todo_list_count=todo_list_count)
+
 @app.template_filter()
 def throw(err):
        raise Exception(err)