refactor: clean up code by adding missing commas and improving import order
This commit is contained in:
@@ -4,7 +4,7 @@ from functools import wraps
|
||||
from typing import Any
|
||||
|
||||
from flask import jsonify, request
|
||||
from flask_jwt_extended import get_jwt, get_jwt_identity, verify_jwt_in_request
|
||||
from flask_jwt_extended import get_jwt_identity, verify_jwt_in_request
|
||||
|
||||
|
||||
def get_user_from_jwt() -> dict[str, Any] | None:
|
||||
@@ -109,7 +109,7 @@ def require_auth(f):
|
||||
if not user:
|
||||
return (
|
||||
jsonify(
|
||||
{"error": "Authentication required (JWT or API token)"}
|
||||
{"error": "Authentication required (JWT or API token)"},
|
||||
),
|
||||
401,
|
||||
)
|
||||
@@ -133,8 +133,8 @@ def require_role(required_role: str):
|
||||
return (
|
||||
jsonify(
|
||||
{
|
||||
"error": f"Access denied. {required_role.title()} role required"
|
||||
}
|
||||
"error": f"Access denied. {required_role.title()} role required",
|
||||
},
|
||||
),
|
||||
403,
|
||||
)
|
||||
@@ -152,8 +152,8 @@ def require_credits(credits_needed: int):
|
||||
def decorator(f):
|
||||
@wraps(f)
|
||||
def wrapper(*args, **kwargs):
|
||||
from app.models.user import User
|
||||
from app.database import db
|
||||
from app.models.user import User
|
||||
|
||||
# First check authentication
|
||||
user_data = get_current_user()
|
||||
@@ -170,8 +170,8 @@ def require_credits(credits_needed: int):
|
||||
return (
|
||||
jsonify(
|
||||
{
|
||||
"error": f"Insufficient credits. Required: {credits_needed}, Available: {user.credits}"
|
||||
}
|
||||
"error": f"Insufficient credits. Required: {credits_needed}, Available: {user.credits}",
|
||||
},
|
||||
),
|
||||
402, # Payment Required status code
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user