22 lines
546 B
Python
22 lines
546 B
Python
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")
|