mirror of
https://github.com/zylon-ai/private-gpt.git
synced 2025-12-22 23:22:57 +01:00
25 lines
No EOL
873 B
Python
25 lines
No EOL
873 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()
|
|
|
|
def remove_user(
|
|
self, db: Session, *, user_id: int
|
|
)-> Optional[UserRole]:
|
|
return db.query(UserRole).filter(UserRole.user_id == user_id).delete()
|
|
|
|
# def update_user_role(
|
|
# self, db: Session, *, user_id: int, role_id: int
|
|
# ) -> Optional[UserRole]:
|
|
# return self.update(db, )
|
|
|
|
user_role = CRUDUserRole(UserRole) |