private-gpt/private_gpt/users/crud/user_role_crud.py

16 lines
No EOL
532 B
Python

from typing import Optional
from private_gpt.users.crud.base import CRUDBase
from private_gpt.users.models.user_role import UserRole
from private_gpt.users.schemas.user_role import UserRoleCreate, UserRoleUpdate
from sqlalchemy.orm import Session
class CRUDUserRole(CRUDBase[UserRole, UserRoleCreate, UserRoleUpdate]):
def get_by_user_id(
self, db: Session, *, user_id: int
) -> Optional[UserRole]:
return db.query(UserRole).filter(UserRole.user_id == user_id).first()
user_role = CRUDUserRole(UserRole)