First commit

app/schemas.py

@@ -0,0 +1,219 @@
+from pydantic import BaseModel, field_validator
+from typing import Optional, List
+from datetime import datetime
+
+# Group Schemas
+class GroupBase(BaseModel):
+    id_telegram: int
+    name: str
+    description: Optional[str] = None
+    type: str
+    message_position: int
+
+class GroupCreate(GroupBase):
+    pass
+
+class GroupResponse(GroupBase):
+    # Eliminar la lista de messages para evitar recursión
+    class Config:
+        from_attributes = True
+
+class GroupWithMessages(GroupBase):
+    # Solo incluir información básica de los mensajes, no completa
+    messages: List["MessageSimple"] = []
+
+    class Config:
+        from_attributes = True
+
+class GroupUpdatePosition(BaseModel):
+    message_position: int
+
+class GroupUpdate(BaseModel):
+    name: str
+    description: Optional[str] = None
+    type: str
+
+# Sender Schemas
+class SenderBase(BaseModel):
+    id_telegram: int
+    type: str
+    username: Optional[str] = None
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
+    phone: Optional[str] = None
+
+class SenderCreate(SenderBase):
+    pass
+
+class SenderResponse(SenderBase):
+    # Eliminar la lista de messages para evitar recursión
+    class Config:
+        from_attributes = True
+
+class SenderWithMessages(SenderBase):
+    # Solo incluir información básica de los mensajes
+    messages: List["MessageSimple"] = []
+
+    class Config:
+        from_attributes = True
+
+# Message Schemas - Versiones simples para evitar recursión
+class MessageSimple(BaseModel):
+    id_mess_g: int
+    group_id: int
+    content: str
+    date: datetime
+
+    class Config:
+        from_attributes = True
+
+class MessageBase(BaseModel):
+    id_mess_g: int
+    content: str
+    date: datetime
+    sender_id: int
+    group_id: int
+
+class MessageCreate(MessageBase):
+    pass
+
+class MessageResponse(MessageBase):
+    sender: Optional[SenderResponse] = None
+    group: Optional[GroupResponse] = None  
+    attachments: List["AttachmentResponse"] = []
+
+    class Config:
+        from_attributes = True
+
+class MessageWithRelations(MessageBase):
+    # Versión con relaciones completas (usar con cuidado)
+    sender: Optional[SenderResponse] = None
+    group: Optional[GroupResponse] = None
+    attachments: List["AttachmentResponse"] = []
+    alerts: List["AlertResponse"] = []
+
+    class Config:
+        from_attributes = True
+
+class AttachmentBase(BaseModel):
+    type: str
+    description: Optional[str] = None
+    isDownloaded: bool
+
+class AttachmentCreate(AttachmentBase):
+    message_id: int
+    group_id: int  # necesario para la FK compuesta
+
+class AttachmentResponse(AttachmentBase):
+    id: int
+    message_id: int
+    group_id: int
+
+    class Config:
+        from_attributes = True
+
+# Rule Schemas
+class RuleBase(BaseModel):
+    description: str
+    regex: str
+    severity: str = "media"
+    is_active: bool = True
+ 
+    @field_validator("description")
+    @classmethod
+    def description_not_empty(cls, v: str) -> str:
+        if not v or not v.strip():
+            raise ValueError("La descripción no puede estar vacía")
+        return v.strip()
+ 
+    @field_validator("regex")
+    @classmethod
+    def regex_not_empty(cls, v: str) -> str:
+        if not v or not v.strip():
+            raise ValueError("La regex no puede estar vacía")
+        import re
+        try:
+            re.compile(v)
+        except re.error as e:
+            raise ValueError(f"La regex no es válida: {e}")
+        return v.strip()
+ 
+    @field_validator("severity")
+    @classmethod
+    def severity_valid(cls, v: str) -> str:
+        allowed = {"baja", "media", "alta"}
+        if v.lower() not in allowed:
+            raise ValueError(f"Severidad debe ser una de: {', '.join(allowed)}")
+        return v.lower()
+
+class RuleCreate(RuleBase):
+    apply_to_history: bool = False  # nuevo campo
+    pass
+
+class RuleResponse(RuleBase):
+    id: int
+    class Config:
+        from_attributes = True
+
+# Alert Schemas
+class AlertBase(BaseModel):
+    message_id: int
+    group_id: int
+    rule_id: int
+    status: str = "open"
+    notes: Optional[str] = None
+
+class AlertCreate(AlertBase):
+    pass
+
+class AlertResponse(AlertBase):
+    id: int
+    created_at: datetime
+    message: Optional[MessageSimple] = None
+    rule: Optional[RuleResponse] = None
+    notes_list: List["NoteResponse"] = []
+
+    class Config:
+        from_attributes = True
+
+# Note Schemas
+class NoteBase(BaseModel):
+    alert_id: int
+    user_id: int
+    content: str
+    creation_date: datetime
+
+class NoteCreate(BaseModel):
+    alert_id: int
+    user_id: int
+    content: str
+
+class NoteResponse(NoteBase):
+    id: int
+
+    class Config:
+        from_attributes = True
+
+class AuditLogResponse(BaseModel):
+    id:           int
+    entity_type:  str
+    entity_id:    str
+    action:       str
+    user_id:      Optional[int] = None
+    before_value: Optional[str] = None
+    after_value:  Optional[str] = None
+    timestamp:    datetime
+    ip_address:   Optional[str] = None
+ 
+    class Config:
+        from_attributes = True
+
+# Resolver referencias circulares
+MessageSimple.model_rebuild()
+MessageResponse.model_rebuild()
+MessageWithRelations.model_rebuild()
+AttachmentResponse.model_rebuild()
+AlertResponse.model_rebuild()
+GroupWithMessages.model_rebuild()
+SenderWithMessages.model_rebuild()
+NoteResponse.model_rebuild()