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()
+