adding current revision information

2026-01-02 08:29:16 -05:00
parent 789465483c
commit 920947e556
64 changed files with 5963 additions and 18 deletions
@@ -263,4 +263,58 @@ class DuplicatesManager:

        return {"report_id": report.id, "status": "completed"}

+    def get_reports(self):
+        return self.db.query(models.DuplicateReport).all()
+
+    def get_duplicate_report(self, report_id: int):
+        report = self.db.query(models.DuplicateReport).filter(models.DuplicateReport.id == report_id).first()
+        if not report:
+            return {"error": "Report not found"}
+
+        groups = []
+        for group in report.groups:
+            files = []
+            for duplicate_file in group.files:
+                files.append(duplicate_file.video_file)
+            groups.append({
+                "group_id": group.id,
+                "score": group.score,
+                "files": files,
+            })
+        
+        return {
+            "report_id": report.id,
+            "created_at": report.created_at,
+            "status": report.status,
+            "groups": groups,
+        }
+
+    def delete_files(self, filepaths: list[str]):
+        samba_manager = SambaManager(
+            self.settings.samba_server_ip,
+            "isolation", # Assuming all duplicates are in the isolation share
+            self.settings.samba_username,
+            self.settings.samba_password,
+        )
+        try:
+            for filepath in filepaths:
+                # Delete from Samba
+                samba_manager.delete_file(filepath)
+
+                # Delete from database
+                video_file = self.db.query(models.VideoFile).filter_by(filepath=filepath).first()
+                if video_file:
+                    # Delete associations in DuplicateFile
+                    self.db.query(models.DuplicateFile).filter_by(video_file_id=video_file.id).delete()
+                    self.db.delete(video_file)
+            
+            self.db.commit()
+            return {"status": "success"}
+        except Exception as e:
+            self.db.rollback()
+            logger.error(f"Error deleting files: {e}", exc_info=True)
+            return {"status": "error", "message": str(e)}
+        finally:
+            samba_manager.close()
+

@@ -1,13 +1,18 @@
 from fastapi import APIRouter, Depends, BackgroundTasks
 from sqlalchemy.orm import Session
+from pydantic import BaseModel
 from ..duplicates_manager import DuplicatesManager
 from ..config import settings, Settings
 from ..database import get_db, SessionLocal
 from ..samba_manager import SambaManager
 import os
+from typing import List

 router = APIRouter()

+class DeleteFilesRequest(BaseModel):
+    filepaths: List[str]
+
 def run_scan_videos_background():
    db = SessionLocal()
    app_settings = Settings()
@@ -55,3 +60,18 @@ def find_duplicates(threshold: float = 0.95, db: Session = Depends(get_db)):
    manager = DuplicatesManager(settings, db)
    result = manager.find_duplicates(threshold)
    return result
+
+@router.get("/reports")
+def get_reports(db: Session = Depends(get_db)):
+    manager = DuplicatesManager(settings, db)
+    return manager.get_reports()
+
+@router.get("/report/{report_id}")
+def get_report(report_id: int, db: Session = Depends(get_db)):
+    manager = DuplicatesManager(settings, db)
+    return manager.get_duplicate_report(report_id)
+
+@router.delete("/files")
+def delete_files(request: DeleteFilesRequest, db: Session = Depends(get_db)):
+    manager = DuplicatesManager(settings, db)
+    return manager.delete_files(request.filepaths)