Add branching visualization, per-subtask retry, and step summary to file detail view

Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
2026-02-07 22:08:09 +00:00
parent d71cd5034e
commit 910fe86603
3 changed files with 734 additions and 75 deletions
@@ -412,6 +412,110 @@ def reprocess_single_file(request: Request, file_id: int, db: Session = Depends(
        raise HTTPException(status_code=500, detail=f"Error reprocessing file: {str(e)}")


+@router.post("/files/{file_id}/retry-subtask")
+@require_login
+def retry_subtask(
+    request: Request, 
+    file_id: int, 
+    subtask_name: str = Query(..., description="Name of the upload subtask to retry (e.g., 'upload_to_dropbox')"),
+    db: Session = Depends(get_db)
+):
+    """
+    Retry a specific failed upload subtask for a file.
+    
+    Args:
+        file_id: ID of the file
+        subtask_name: Name of the upload task (e.g., upload_to_dropbox, upload_to_s3)
+        
+    Returns:
+        Task ID and status information
+    """
+    try:
+        # Find the file record
+        file_record = db.query(FileRecord).filter(FileRecord.id == file_id).first()
+        
+        if not file_record:
+            raise HTTPException(status_code=404, detail=f"File with ID {file_id} not found")
+        
+        # Check for processed file (upload tasks work with processed files)
+        workdir = settings.workdir
+        processed_dir = os.path.join(workdir, "processed")
+        
+        # Try to find the processed file
+        base_filename = os.path.splitext(file_record.original_filename)[0]
+        potential_paths = [
+            os.path.join(processed_dir, f"{file_record.filehash}.pdf"),
+            os.path.join(processed_dir, f"{base_filename}_processed.pdf"),
+            os.path.join(processed_dir, file_record.original_filename),
+        ]
+        
+        file_path = None
+        for path in potential_paths:
+            if os.path.exists(path):
+                file_path = path
+                break
+        
+        if not file_path:
+            raise HTTPException(
+                status_code=400,
+                detail="Processed file not found. Cannot retry upload."
+            )
+        
+        # Map subtask names to their corresponding Celery tasks
+        from app.tasks.upload_to_dropbox import upload_to_dropbox
+        from app.tasks.upload_to_nextcloud import upload_to_nextcloud
+        from app.tasks.upload_to_paperless import upload_to_paperless
+        from app.tasks.upload_to_google_drive import upload_to_google_drive
+        from app.tasks.upload_to_onedrive import upload_to_onedrive
+        from app.tasks.upload_to_s3 import upload_to_s3
+        from app.tasks.upload_to_webdav import upload_to_webdav
+        from app.tasks.upload_to_ftp import upload_to_ftp
+        from app.tasks.upload_to_sftp import upload_to_sftp
+        from app.tasks.upload_to_email import upload_to_email
+        
+        task_map = {
+            "upload_to_dropbox": upload_to_dropbox,
+            "upload_to_nextcloud": upload_to_nextcloud,
+            "upload_to_paperless": upload_to_paperless,
+            "upload_to_google_drive": upload_to_google_drive,
+            "upload_to_onedrive": upload_to_onedrive,
+            "upload_to_s3": upload_to_s3,
+            "upload_to_webdav": upload_to_webdav,
+            "upload_to_ftp": upload_to_ftp,
+            "upload_to_sftp": upload_to_sftp,
+            "upload_to_email": upload_to_email
+        }
+        
+        if subtask_name not in task_map:
+            raise HTTPException(
+                status_code=400,
+                detail=f"Invalid subtask name: {subtask_name}. Must be one of: {', '.join(task_map.keys())}"
+            )
+        
+        # Queue the specific upload task
+        upload_task = task_map[subtask_name]
+        task = upload_task.delay(file_path, file_id)
+        
+        logger.info(
+            f"Retrying upload subtask: FileID={file_record.id}, "
+            f"Subtask={subtask_name}, TaskID={task.id}"
+        )
+        
+        return {
+            "status": "success",
+            "message": f"Upload task {subtask_name} queued for retry",
+            "file_id": file_record.id,
+            "subtask_name": subtask_name,
+            "task_id": task.id
+        }
+        
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.exception(f"Error retrying subtask {subtask_name} for file {file_id}: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Error retrying subtask: {str(e)}")
+
+
@router.get("/files/{file_id}/preview")
@require_login
 def get_file_preview(request: Request, file_id: int, version: str = Query("original", description="original or processed"), db: Session = Depends(get_db)):