From 8c8c4d9063794490a736d0353ed48e134221fd56 Mon Sep 17 00:00:00 2001 From: Christian Krakau-Louis Date: Wed, 26 Mar 2025 03:11:29 +0100 Subject: [PATCH] made auth.py more robust --- app/auth.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/app/auth.py b/app/auth.py index c6a9f18b..5cbf9cf5 100644 --- a/app/auth.py +++ b/app/auth.py @@ -1,5 +1,5 @@ -# app/auth.py import os +import inspect from functools import wraps from authlib.integrations.starlette_client import OAuth @@ -37,7 +37,11 @@ def require_login(func): if not request.session.get("user"): request.session["redirect_after_login"] = str(request.url) return RedirectResponse(url="/login", status_code=status.HTTP_302_FOUND) - return await func(request, *args, **kwargs) + # Check if the wrapped function is a coroutine function + if inspect.iscoroutinefunction(func): + return await func(request, *args, **kwargs) + else: + return func(request, *args, **kwargs) return wrapper