add websockets suppor93

backend/app/models/__init__.py

@@ -1,5 +1,6 @@
 """Models initialization"""
 from .employee import Employee
 from .request import Request
+from .token import Token
 
-__all__ = ['Employee', 'Request'] 
+__all__ = ["Employee", "Request", "Token"] 

backend/app/models/employee.py

@@ -1,10 +1,11 @@
-"""Employee model"""
-from sqlalchemy import Column, Integer, String, DateTime
-from sqlalchemy.sql import func
+"""Employee models"""
+from sqlalchemy import Column, Integer, String
 from sqlalchemy.orm import relationship
-from app.db.base_class import Base
+from ..db.base import Base
+
 
 class Employee(Base):
+    """Employee model"""
     __tablename__ = "employees"
 
     id = Column(Integer, primary_key=True, index=True)
@@ -13,7 +14,6 @@ class Employee(Base):
     department = Column(String, nullable=False)
     office = Column(String, nullable=False)
     hashed_password = Column(String, nullable=False)
-    created_at = Column(DateTime(timezone=True), server_default=func.now())
 
     # Определяем отношение к Request
     requests = relationship("Request", back_populates="employee", cascade="all, delete-orphan")

backend/app/models/request.py

@@ -2,7 +2,8 @@
 from enum import Enum
 from sqlalchemy import Column, Integer, String, ForeignKey, DateTime, Enum as SQLAlchemyEnum
 from sqlalchemy.sql import func
-from ..database import Base
+from sqlalchemy.orm import relationship
+from ..db.base import Base
 
 
 class RequestStatus(str, Enum):
@@ -33,3 +34,6 @@ class Request(Base):
     employee_id = Column(Integer, ForeignKey("employees.id", ondelete="CASCADE"), nullable=False)
     created_at = Column(DateTime(timezone=True), server_default=func.now(), nullable=False)
     updated_at = Column(DateTime(timezone=True), onupdate=func.now())
+
+    # Определяем отношение к Employee
+    employee = relationship("Employee", back_populates="requests")