eafbc1ac52
The crawler subdirectory was the only active project. Moving it to the repo root simplifies paths and removes the unnecessary nesting. The vqa/ and immoweb/ directories were legacy/unused and have been removed. Updated .drone.yml, .gitignore, .claude/ docs, and skills to reflect the new flat structure.
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
from models.passkey_credential import PasskeyCredential
|
|
from models.user import User
|
|
from sqlalchemy import Engine
|
|
from sqlmodel import Session, select
|
|
|
|
|
|
class UserRepository:
|
|
engine: Engine
|
|
|
|
def __init__(self, engine: Engine) -> None:
|
|
self.engine = engine
|
|
|
|
def get_user_by_email(self, email: str) -> User | None:
|
|
with Session(self.engine) as session:
|
|
statement = select(User).where(User.email == email)
|
|
return session.exec(statement).first()
|
|
|
|
def get_user_by_id(self, user_id: int) -> User | None:
|
|
with Session(self.engine) as session:
|
|
return session.get(User, user_id)
|
|
|
|
def create_user(self, email: str) -> User:
|
|
with Session(self.engine) as session:
|
|
user = User(email=email) # type: ignore[call-arg]
|
|
session.add(user)
|
|
session.commit()
|
|
session.refresh(user)
|
|
return user
|
|
|
|
def get_credentials_for_user(self, user_id: int) -> list[PasskeyCredential]:
|
|
with Session(self.engine) as session:
|
|
statement = select(PasskeyCredential).where(
|
|
PasskeyCredential.user_id == user_id
|
|
)
|
|
return list(session.exec(statement).all())
|
|
|
|
def get_credential_by_id(self, credential_id: str) -> PasskeyCredential | None:
|
|
with Session(self.engine) as session:
|
|
statement = select(PasskeyCredential).where(
|
|
PasskeyCredential.credential_id == credential_id
|
|
)
|
|
return session.exec(statement).first()
|
|
|
|
def save_credential(self, credential: PasskeyCredential) -> PasskeyCredential:
|
|
with Session(self.engine) as session:
|
|
session.add(credential)
|
|
session.commit()
|
|
session.refresh(credential)
|
|
return credential
|
|
|
|
def update_credential_sign_count(
|
|
self, credential_id: str, new_sign_count: int
|
|
) -> None:
|
|
with Session(self.engine) as session:
|
|
statement = select(PasskeyCredential).where(
|
|
PasskeyCredential.credential_id == credential_id
|
|
)
|
|
credential = session.exec(statement).first()
|
|
if credential:
|
|
credential.sign_count = new_sign_count
|
|
session.add(credential)
|
|
session.commit()
|