private-gpt/private_gpt/users/models/user_role.py

26 lines
No EOL
678 B
Python

from private_gpt.users.db.base_class import Base
from sqlalchemy import Column, ForeignKey, UniqueConstraint, Integer
from sqlalchemy.orm import relationship
class UserRole(Base):
__tablename__ = "user_roles"
user_id = Column(
Integer,
ForeignKey("users.id"),
primary_key=True,
nullable=False,
)
role_id = Column(
Integer,
ForeignKey("roles.id"),
primary_key=True,
nullable=False,
)
role = relationship("Role")
user = relationship("User", back_populates="user_role", uselist=False)
__table_args__ = (
UniqueConstraint("user_id", "role_id", name="unique_user_role"),
)