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

backend/app/tasks/cleanup.py

@@ -0,0 +1,38 @@
+import os
+from datetime import datetime, timedelta
+from sqlalchemy.orm import Session
+from app.database import SessionLocal
+from app.models.chat import Message, ChatFile
+
+def cleanup_old_messages():
+    """Удаляет сообщения и файлы старше 1 месяца"""
+    db = SessionLocal()
+    try:
+        # Определяем дату, до которой нужно удалить сообщения
+        cutoff_date = datetime.utcnow() - timedelta(days=30)
+
+        # Получаем файлы, которые нужно удалить
+        files_to_delete = db.query(ChatFile)\
+            .join(Message)\
+            .filter(Message.created_at < cutoff_date)\
+            .all()
+
+        # Удаляем физические файлы
+        for file in files_to_delete:
+            try:
+                if os.path.exists(file.file_path):
+                    os.remove(file.file_path)
+            except Exception as e:
+                print(f"Error deleting file {file.file_path}: {e}")
+
+        # Удаляем старые сообщения (каскадно удалятся и записи о файлах)
+        db.query(Message)\
+            .filter(Message.created_at < cutoff_date)\
+            .delete(synchronize_session=False)
+
+        db.commit()
+    except Exception as e:
+        print(f"Error during cleanup: {e}")
+        db.rollback()
+    finally:
+        db.close()