Initial commit
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
from starlette.requests import Request
|
||||
|
||||
from core.app import app
|
||||
from core.utils.auth import UserModel
|
||||
|
||||
|
||||
def use_request() -> Request:
|
||||
try:
|
||||
return app.state.request.state
|
||||
except AttributeError:
|
||||
raise AttributeError("No request object found in app.state")
|
||||
|
||||
|
||||
def use_user(user=None) -> UserModel:
|
||||
try:
|
||||
if user is None:
|
||||
return use_request().user
|
||||
else:
|
||||
app.state.request.state.user = user
|
||||
except AttributeError:
|
||||
raise AttributeError("No user object found in app.state")
|
||||
Reference in New Issue
Block a user