18 lines
437 B
Python
18 lines
437 B
Python
|
from models.user import User
|
||
|
|
from sqlalchemy import Engine
|
||
|
|
from sqlmodel import Session, select
|
||
|
|
|
||
|
|
|
||
|
|
class UserRepository:
|
||
|
|
engine: Engine
|
||
|
|
|
||
|
|
def __init__(self, engine: Engine):
|
||
|
|
self.engine = engine
|
||
|
|
|
||
|
|
async def get_user_from_token(self, token: str) -> User | None:
|
||
|
|
raise NotImplementedError()
|
||
|
|
query = select(User)
|
||
|
|
|
||
|
|
with Session(self.engine) as session:
|
||
|
|
return session.exec(query).first()
|