This commit is contained in:
2026-07-17 17:59:43 +03:00
parent 507f4db7fd
commit 4512eb6e5c
4 changed files with 109 additions and 8 deletions

View File

@@ -1,6 +1,7 @@
from sqlalchemy import Table, create_engine, String, Boolean, MetaData, Column, ForeignKey
from sqlalchemy import TIMESTAMP, Table, create_engine, String, Boolean, MetaData, Column, ForeignKey, func
from sqlalchemy.orm import Mapped, mapped_column, DeclarativeBase, relationship
from uuid import UUID, uuid4
from datetime import datetime
engine = create_engine("sqlite:///DB/database.db", echo=True)
'''remember as a boilerplate, or just cp/pst'''
@@ -27,6 +28,8 @@ class User(Model):
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")
refresh_token:Mapped[list['RefreshTokens']]=relationship(back_populates="user")
def __repr__(self) -> str:
return f"ID: {self.id}, Name: {self.first_name}, Status: {self.status}"
@@ -73,4 +76,25 @@ group_permission = Table(
Model.metadata,
Column("group_id", ForeignKey("groups_of_permissions.id"), primary_key=True),
Column("permission_id", ForeignKey("permissions.id"), primary_key=True),
)
)
class RefreshTokens(Model):
__tablename__= "refresh_tokens"
id:Mapped[int]=mapped_column(primary_key=True, index=True)
user_id:Mapped[UUID]=mapped_column(ForeignKey("users.id", ondelete="CASCADE"), index=True)
token_hash:Mapped[str]=mapped_column(String(255), unique=True)
device_info:Mapped[str]=mapped_column(String(255))
ip_address:Mapped[str]=mapped_column(String(45))
is_revoked:Mapped[bool]=mapped_column(Boolean, default=False)
expires_at:Mapped[datetime]=mapped_column(TIMESTAMP)
created_at:Mapped[datetime]=mapped_column(TIMESTAMP, server_default=func.now())
replaced_by:Mapped[int|None]=mapped_column(ForeignKey("refresh_tokens.id"), nullable=True, default=None)
user:Mapped["User"]=relationship(back_populates="refresh_token")
def __repr__(self)->str:
return f'ID: {self.id}, USER_ID: {self.user_id}, Is_revoked: {self.is_revoked}, Token_Hash: {self.token_hash}'

View File

@@ -1,10 +1,12 @@
from dataclasses import replace
from datetime import datetime
from typing import Annotated
from pydantic import BaseModel, EmailStr, Field
from uuid import UUID
class Base(BaseModel):
pass
model_config = {"from_attributes": True}
class UserCreate(Base):
@@ -28,7 +30,6 @@ class UserOut(Base):
permissions:Annotated[list[str], Field(..., description="permissions of the user")]
permission_groups:Annotated[list[str], Field(..., description="permissions groups of the user")]
model_config = {"from_attributes": True}
class UserUpdate(Base):
@@ -42,11 +43,30 @@ class UserUpdate(Base):
class Permissions(Base):
id:int
permission_name:Annotated[str, Field(..., max_length=30, description="permission name")]
permission:Annotated[str, Field(..., max_length=30, description="permission name")]
model_config = {"from_attributes": True}
class PermissionsOut(Base):
permission_name:Annotated[str, Field(..., max_length=30, description="permission name")]
permission:Annotated[str, Field(..., max_length=30, description="permission name")]
model_config = {"from_attributes": True}
class PermissionsGroups(Base):
id:int
group:Annotated[str, Field(..., max_length=255, description="group name for the permissions")]
class PermissionsGroupsOut(Base):
group:Annotated[str, Field(..., max_length=255, description="group name for the permissions")]
class RefreshTokens(Base):
id:int
user_id:Annotated[UUID, Field(..., description="foreign key for the user")]
token_hash:Annotated[str, Field(..., description="token hash")]
device_info:Annotated[str, Field(..., description="User device info")]
ip_address:Annotated[str, Field(..., description="ip v4/v6 of the user")]
is_revoked:Annotated[bool|None, Field(None, description="revoke token if logout was made")]
expires_at:Annotated[datetime, Field(..., description="when token is going to be expired")]