56 lines
1.1 KiB
Python
56 lines
1.1 KiB
Python
"""
|
|
Python representation of `user_capability` table.
|
|
"""
|
|
|
|
from enum import Enum
|
|
from typing import TYPE_CHECKING
|
|
|
|
from sqlmodel import Field, Relationship, SQLModel
|
|
|
|
if TYPE_CHECKING:
|
|
from .user import User
|
|
|
|
|
|
class UserCapabilityType(Enum):
|
|
"""
|
|
Allowed values for capabilities
|
|
"""
|
|
|
|
admin = "admin"
|
|
login = "login"
|
|
issue = "issue"
|
|
renew = "renew"
|
|
|
|
def __repr__(self) -> str:
|
|
return self.value
|
|
|
|
|
|
class UserCapabilityBase(SQLModel):
|
|
"""
|
|
Common to all representations of capabilities
|
|
"""
|
|
|
|
capability_name: str = Field(primary_key=True)
|
|
|
|
@property
|
|
def _(self) -> UserCapabilityType:
|
|
"""
|
|
Transform into a `Capability`.
|
|
"""
|
|
|
|
return UserCapabilityType(self.capability_name)
|
|
|
|
def __repr__(self) -> str:
|
|
return self.capability_name
|
|
|
|
|
|
class UserCapability(UserCapabilityBase, table=True):
|
|
"""
|
|
Representation of `user_capability` table
|
|
"""
|
|
|
|
user_name: str = Field(primary_key=True, foreign_key="user.name")
|
|
|
|
user: "User" = Relationship(
|
|
back_populates="capabilities",
|
|
)
|