0.1.2

src/models/database_models/model.py

@@ -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}'