fix(FastAPI): access AccountType ID directly

Signed-off-by: Steven Guikal <void@fluix.one>
This commit is contained in:
Steven Guikal 2021-10-21 17:48:29 -04:00
parent a6ac5f0dbf
commit ecbab8546b

View file

@ -244,7 +244,7 @@ def account_type_required(one_of: set):
def decorator(func):
@functools.wraps(func)
async def wrapper(request: fastapi.Request, *args, **kwargs):
if request.user.AccountType.ID not in one_of:
if request.user.AccountTypeID not in one_of:
return RedirectResponse("/",
status_code=int(HTTPStatus.SEE_OTHER))
return await func(request, *args, **kwargs)