Создание чата9testt

backend/app/schemas/chat.py

@@ -1,74 +1,54 @@
-"""Chat schemas"""
+"""Chat schemas."""
 from pydantic import BaseModel, ConfigDict
 from datetime import datetime
 from typing import Optional, List
-from app.models.user import User
-
-class ChatBase(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-    employee_id: int
-
-class ChatCreate(ChatBase):
-    pass
-
-class MessageBase(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-    content: str
-    chat_id: int
-    sender_id: int
-
-class MessageCreate(MessageBase):
-    pass
 
 class ChatFileBase(BaseModel):
-    model_config = ConfigDict(arbitrary_types_allowed=True)
+    """Chat file base schema."""
+    model_config = ConfigDict(from_attributes=True)
     filename: str
     file_path: str
-    message_id: int
 
 class ChatFileCreate(ChatFileBase):
-    pass
+    """Chat file create schema."""
+    message_id: int
 
 class ChatFile(ChatFileBase):
+    """Chat file schema."""
     id: int
+    message_id: int
     created_at: datetime
 
-    class Config:
-        from_attributes = True
+class MessageBase(BaseModel):
+    """Message base schema."""
+    model_config = ConfigDict(from_attributes=True)
+    content: str
+
+class MessageCreate(MessageBase):
+    """Message create schema."""
+    chat_id: int
 
 class Message(MessageBase):
+    """Message schema."""
     id: int
+    chat_id: int
+    sender_id: int
     is_read: bool
     created_at: datetime
     files: List[ChatFile] = []
 
-    class Config:
-        from_attributes = True
+class ChatBase(BaseModel):
+    """Chat base schema."""
+    model_config = ConfigDict(from_attributes=True)
+    employee_id: int
+
+class ChatCreate(ChatBase):
+    """Chat create schema."""
+    pass
 
 class Chat(ChatBase):
+    """Chat schema."""
     id: int
-    employee: User
-    messages: List[Message] = []
     created_at: datetime
     updated_at: Optional[datetime] = None
-
-    class Config:
-        from_attributes = True
-
-# Схемы для WebSocket сообщений
-class WSMessage(BaseModel):
-    type: str
-    content: Optional[str] = None
-    message_ids: Optional[List[int]] = None
-    files: Optional[List[dict]] = None
-
-class WSResponse(BaseModel):
-    type: str
-    id: Optional[int] = None
-    sender_id: Optional[int] = None
-    content: Optional[str] = None
-    created_at: Optional[datetime] = None
-    is_read: Optional[bool] = None
-    message_ids: Optional[List[int]] = None
-    files: Optional[List[ChatFile]] = None
-    error: Optional[str] = None 
+    messages: List[Message] = []