76 lines
3.2 KiB
Python
76 lines
3.2 KiB
Python
from sqlalchemy import Table, create_engine, String, Boolean, MetaData, Column, ForeignKey
|
|
from sqlalchemy.orm import Mapped, mapped_column, DeclarativeBase, relationship
|
|
from uuid import UUID, uuid4
|
|
engine = create_engine("sqlite:///DB/database.db", echo=True)
|
|
|
|
'''remember as a boilerplate, or just cp/pst'''
|
|
class Model(DeclarativeBase):
|
|
metadata = MetaData(naming_convention={
|
|
"ix": "ix_%(column_0_label)s",
|
|
"uq": "uq_%(table_name)s_%(column_0_name)s",
|
|
"ck": "ck_%(table_name)s_%(constraint_name)s",
|
|
"fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
|
|
"pk": "pk_%(table_name)s",
|
|
})
|
|
|
|
class User(Model):
|
|
__tablename__ = "users"
|
|
|
|
id:Mapped[UUID] = mapped_column(default=uuid4,primary_key=True)
|
|
first_name:Mapped[str] = mapped_column(String(64), index=True)
|
|
last_name:Mapped[str]=mapped_column(String(64), index=True)
|
|
middle_name:Mapped[str]=mapped_column(String(64), index=True)
|
|
email:Mapped[str]=mapped_column(String(255), index=True, unique=True)
|
|
status:Mapped[bool]=mapped_column(Boolean, default=True)
|
|
hashed_password:Mapped[str]=mapped_column(String(255))
|
|
|
|
group:Mapped[list['PermissionsGroups']]=relationship(secondary="user_group", back_populates="user", lazy="selectin")
|
|
|
|
direct_permissions:Mapped[list['Permissions']]=relationship(secondary="user_direct_permissions", back_populates="users_direct")
|
|
def __repr__(self) -> str:
|
|
return f"ID: {self.id}, Name: {self.first_name}, Status: {self.status}"
|
|
|
|
class PermissionsGroups(Model):
|
|
__tablename__="groups_of_permissions"
|
|
|
|
id:Mapped[int]=mapped_column(primary_key=True, index=True)
|
|
group:Mapped[str]=mapped_column(String(255), unique=True)
|
|
|
|
user:Mapped[list['User']]=relationship(secondary="user_group", back_populates="group")
|
|
permissions:Mapped[list["Permissions"]]=relationship(secondary="group_permission", back_populates="group", lazy="selectin")
|
|
|
|
def __repr__(self)->str:
|
|
return f"ID: {self.id}, Permissions: {self.group}"
|
|
|
|
class Permissions(Model):
|
|
__tablename__ = "permissions"
|
|
|
|
id:Mapped[int]=mapped_column(primary_key=True, index=True)
|
|
permission:Mapped[str]=mapped_column(String(255), unique=True)
|
|
|
|
group:Mapped[list['PermissionsGroups']]=relationship(secondary="group_permission", back_populates="permissions")
|
|
users_direct:Mapped[list["User"]]=relationship(secondary="user_direct_permissions", back_populates="direct_permissions")
|
|
|
|
def __repr__(self)->str:
|
|
return f"ID: {self.id}, Permissions: {self.permission}"
|
|
|
|
user_group_of_permissions=Table(
|
|
"user_group",
|
|
Model.metadata,
|
|
Column("user_id", ForeignKey("users.id"), primary_key=True),
|
|
Column("permission_group_id", ForeignKey("groups_of_permissions.id"), primary_key=True)
|
|
)
|
|
|
|
user_permission=Table(
|
|
"user_direct_permissions",
|
|
Model.metadata,
|
|
Column("user_id",ForeignKey("users.id"), primary_key=True),
|
|
Column("permission_id", ForeignKey("permissions.id"), primary_key=True)
|
|
)
|
|
|
|
group_permission = Table(
|
|
"group_permission",
|
|
Model.metadata,
|
|
Column("group_id", ForeignKey("groups_of_permissions.id"), primary_key=True),
|
|
Column("permission_id", ForeignKey("permissions.id"), primary_key=True),
|
|
) |