Merge pull request #123 from christianlouis/copilot/add-files-detail-view-features
Add branching visualization and per-subtask retry to file detail view
This commit is contained in:
@@ -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)}")
|
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")
|
@router.get("/files/{file_id}/preview")
|
||||||
@require_login
|
@require_login
|
||||||
def get_file_preview(request: Request, file_id: int, version: str = Query("original", description="original or processed"), db: Session = Depends(get_db)):
|
def get_file_preview(request: Request, file_id: int, version: str = Query("original", description="original or processed"), db: Session = Depends(get_db)):
|
||||||
|
|||||||
+128
-6
@@ -200,13 +200,17 @@ def file_detail_page(request: Request, file_id: int, db: Session = Depends(get_d
|
|||||||
# Compute processing flow for visualization
|
# Compute processing flow for visualization
|
||||||
flow_data = _compute_processing_flow(logs)
|
flow_data = _compute_processing_flow(logs)
|
||||||
|
|
||||||
|
# Compute step-aligned summary
|
||||||
|
step_summary = _compute_step_summary(logs)
|
||||||
|
|
||||||
return templates.TemplateResponse("file_detail.html", {
|
return templates.TemplateResponse("file_detail.html", {
|
||||||
"request": request,
|
"request": request,
|
||||||
"file": file_record,
|
"file": file_record,
|
||||||
"logs": logs,
|
"logs": logs,
|
||||||
"file_exists": file_exists,
|
"file_exists": file_exists,
|
||||||
"processed_exists": processed_exists,
|
"processed_exists": processed_exists,
|
||||||
"flow_data": flow_data
|
"flow_data": flow_data,
|
||||||
|
"step_summary": step_summary
|
||||||
})
|
})
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"Error retrieving file details: {str(e)}")
|
logger.error(f"Error retrieving file details: {str(e)}")
|
||||||
@@ -221,8 +225,9 @@ def _compute_processing_flow(logs):
|
|||||||
Compute the processing flow structure from logs for visualization.
|
Compute the processing flow structure from logs for visualization.
|
||||||
|
|
||||||
Returns a structured representation of the processing pipeline with branches.
|
Returns a structured representation of the processing pipeline with branches.
|
||||||
|
Detects upload sub-tasks and organizes them as branches under the parent upload stage.
|
||||||
"""
|
"""
|
||||||
# Define the processing stages and their relationships
|
# Define the main processing stages
|
||||||
stages = {
|
stages = {
|
||||||
"hash_file": {"label": "File Upload & Hash", "next": ["create_file_record"]},
|
"hash_file": {"label": "File Upload & Hash", "next": ["create_file_record"]},
|
||||||
"create_file_record": {"label": "Create File Record", "next": ["check_text"]},
|
"create_file_record": {"label": "Create File Record", "next": ["check_text"]},
|
||||||
@@ -231,14 +236,55 @@ def _compute_processing_flow(logs):
|
|||||||
"process_with_azure_document_intelligence": {"label": "OCR Processing (Azure)", "next": ["extract_metadata_with_gpt"]},
|
"process_with_azure_document_intelligence": {"label": "OCR Processing (Azure)", "next": ["extract_metadata_with_gpt"]},
|
||||||
"extract_metadata_with_gpt": {"label": "Extract Metadata (GPT)", "next": ["embed_metadata_into_pdf"]},
|
"extract_metadata_with_gpt": {"label": "Extract Metadata (GPT)", "next": ["embed_metadata_into_pdf"]},
|
||||||
"embed_metadata_into_pdf": {"label": "Embed Metadata into PDF", "next": ["finalize_document_storage"]},
|
"embed_metadata_into_pdf": {"label": "Embed Metadata into PDF", "next": ["finalize_document_storage"]},
|
||||||
"finalize_document_storage": {"label": "Finalize & Queue Distribution", "next": ["upload_destinations"]},
|
"finalize_document_storage": {"label": "Finalize & Queue Distribution", "next": ["send_to_all_destinations"]},
|
||||||
"upload_destinations": {"label": "Upload to Destinations", "next": []}
|
"send_to_all_destinations": {"label": "Upload to Destinations", "next": [], "has_branches": True}
|
||||||
|
}
|
||||||
|
|
||||||
|
# Define upload sub-tasks (branches)
|
||||||
|
upload_tasks = {
|
||||||
|
"upload_to_dropbox": "Dropbox",
|
||||||
|
"upload_to_nextcloud": "Nextcloud",
|
||||||
|
"upload_to_paperless": "Paperless-ngx",
|
||||||
|
"upload_to_google_drive": "Google Drive",
|
||||||
|
"upload_to_onedrive": "OneDrive",
|
||||||
|
"upload_to_s3": "S3 Storage",
|
||||||
|
"upload_to_webdav": "WebDAV",
|
||||||
|
"upload_to_ftp": "FTP Storage",
|
||||||
|
"upload_to_sftp": "SFTP Storage",
|
||||||
|
"upload_to_email": "Email",
|
||||||
|
"queue_dropbox": "Dropbox",
|
||||||
|
"queue_nextcloud": "Nextcloud",
|
||||||
|
"queue_paperless": "Paperless-ngx",
|
||||||
|
"queue_google_drive": "Google Drive",
|
||||||
|
"queue_onedrive": "OneDrive",
|
||||||
|
"queue_s3": "S3 Storage",
|
||||||
|
"queue_webdav": "WebDAV",
|
||||||
|
"queue_ftp": "FTP Storage",
|
||||||
|
"queue_sftp": "SFTP Storage",
|
||||||
|
"queue_email": "Email"
|
||||||
}
|
}
|
||||||
|
|
||||||
# Create a map of step names to their log entries
|
# Create a map of step names to their log entries
|
||||||
step_map = {}
|
step_map = {}
|
||||||
|
upload_branches = {}
|
||||||
|
|
||||||
for log in logs:
|
for log in logs:
|
||||||
step_name = log.step_name
|
step_name = log.step_name
|
||||||
|
|
||||||
|
# Check if this is an upload sub-task
|
||||||
|
if step_name in upload_tasks:
|
||||||
|
# Extract the actual upload task name (remove queue_ prefix if present)
|
||||||
|
upload_key = step_name.replace("queue_", "upload_to_")
|
||||||
|
if upload_key not in upload_branches:
|
||||||
|
upload_branches[upload_key] = []
|
||||||
|
upload_branches[upload_key].append({
|
||||||
|
"status": log.status,
|
||||||
|
"message": log.message,
|
||||||
|
"timestamp": log.timestamp,
|
||||||
|
"task_id": log.task_id
|
||||||
|
})
|
||||||
|
else:
|
||||||
|
# Regular processing step
|
||||||
if step_name not in step_map:
|
if step_name not in step_map:
|
||||||
step_map[step_name] = []
|
step_map[step_name] = []
|
||||||
step_map[step_name].append({
|
step_map[step_name].append({
|
||||||
@@ -266,14 +312,90 @@ def _compute_processing_flow(logs):
|
|||||||
timestamp = None
|
timestamp = None
|
||||||
task_id = None
|
task_id = None
|
||||||
|
|
||||||
flow.append({
|
stage_data = {
|
||||||
"key": stage_key,
|
"key": stage_key,
|
||||||
"label": stage_info["label"],
|
"label": stage_info["label"],
|
||||||
"status": status,
|
"status": status,
|
||||||
"message": message,
|
"message": message,
|
||||||
"timestamp": timestamp,
|
"timestamp": timestamp,
|
||||||
"task_id": task_id,
|
"task_id": task_id,
|
||||||
"can_retry": status == "failure"
|
"can_retry": status == "failure",
|
||||||
|
"is_branch_parent": stage_info.get("has_branches", False)
|
||||||
|
}
|
||||||
|
|
||||||
|
# If this is the upload stage, add branches
|
||||||
|
if stage_info.get("has_branches") and upload_branches:
|
||||||
|
branches = []
|
||||||
|
for upload_key, upload_logs in upload_branches.items():
|
||||||
|
latest_upload = upload_logs[-1]
|
||||||
|
upload_name = upload_tasks.get(upload_key, upload_key.replace("upload_to_", "").title())
|
||||||
|
|
||||||
|
branches.append({
|
||||||
|
"key": upload_key,
|
||||||
|
"label": upload_name,
|
||||||
|
"status": latest_upload["status"],
|
||||||
|
"message": latest_upload["message"],
|
||||||
|
"timestamp": latest_upload["timestamp"],
|
||||||
|
"task_id": latest_upload["task_id"],
|
||||||
|
"can_retry": latest_upload["status"] == "failure"
|
||||||
})
|
})
|
||||||
|
stage_data["branches"] = branches
|
||||||
|
|
||||||
|
flow.append(stage_data)
|
||||||
|
|
||||||
return flow
|
return flow
|
||||||
|
|
||||||
|
|
||||||
|
def _compute_step_summary(logs):
|
||||||
|
"""
|
||||||
|
Compute a step-aligned summary from logs showing queued, success, and failure counts.
|
||||||
|
|
||||||
|
Returns a dictionary with main step counts and upload branch counts.
|
||||||
|
"""
|
||||||
|
# Count statuses for main processing steps (not uploads)
|
||||||
|
main_steps = [
|
||||||
|
"hash_file", "create_file_record", "check_text", "extract_text",
|
||||||
|
"process_with_azure_document_intelligence", "extract_metadata_with_gpt",
|
||||||
|
"embed_metadata_into_pdf", "finalize_document_storage", "send_to_all_destinations"
|
||||||
|
]
|
||||||
|
|
||||||
|
upload_prefixes = ["upload_to_", "queue_"]
|
||||||
|
|
||||||
|
main_counts = {"queued": 0, "in_progress": 0, "success": 0, "failure": 0}
|
||||||
|
upload_counts = {"queued": 0, "in_progress": 0, "success": 0, "failure": 0}
|
||||||
|
|
||||||
|
# Track which steps we've seen
|
||||||
|
main_steps_seen = set()
|
||||||
|
upload_tasks_seen = {}
|
||||||
|
|
||||||
|
for log in logs:
|
||||||
|
step_name = log.step_name
|
||||||
|
status = log.status.lower()
|
||||||
|
|
||||||
|
# Normalize status
|
||||||
|
if status == "pending":
|
||||||
|
status = "queued"
|
||||||
|
|
||||||
|
# Check if it's an upload task
|
||||||
|
is_upload = any(step_name.startswith(prefix) for prefix in upload_prefixes)
|
||||||
|
|
||||||
|
if is_upload:
|
||||||
|
# Track latest status for each unique upload task
|
||||||
|
upload_tasks_seen[step_name] = status
|
||||||
|
elif step_name in main_steps:
|
||||||
|
# Track latest status for main steps
|
||||||
|
main_steps_seen.add(step_name)
|
||||||
|
if status in main_counts:
|
||||||
|
main_counts[status] += 1
|
||||||
|
|
||||||
|
# Count upload task statuses
|
||||||
|
for task_status in upload_tasks_seen.values():
|
||||||
|
if task_status in upload_counts:
|
||||||
|
upload_counts[task_status] += 1
|
||||||
|
|
||||||
|
return {
|
||||||
|
"main": main_counts,
|
||||||
|
"uploads": upload_counts,
|
||||||
|
"total_main_steps": len(main_steps_seen),
|
||||||
|
"total_upload_tasks": len(upload_tasks_seen)
|
||||||
|
}
|
||||||
|
|||||||
@@ -85,7 +85,71 @@
|
|||||||
color: #991B1B;
|
color: #991B1B;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Processing logs */
|
/* Step summary section */
|
||||||
|
.step-summary {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
|
||||||
|
gap: 1rem;
|
||||||
|
margin-bottom: 1.5rem;
|
||||||
|
}
|
||||||
|
.summary-card {
|
||||||
|
background-color: #f7fafc;
|
||||||
|
border-radius: 0.5rem;
|
||||||
|
padding: 1rem;
|
||||||
|
border-left: 4px solid #e2e8f0;
|
||||||
|
}
|
||||||
|
.summary-card.main-steps {
|
||||||
|
border-left-color: #4299e1;
|
||||||
|
}
|
||||||
|
.summary-card.upload-steps {
|
||||||
|
border-left-color: #48bb78;
|
||||||
|
}
|
||||||
|
.summary-title {
|
||||||
|
font-weight: 600;
|
||||||
|
color: #2d3748;
|
||||||
|
margin-bottom: 0.5rem;
|
||||||
|
font-size: 0.875rem;
|
||||||
|
text-transform: uppercase;
|
||||||
|
}
|
||||||
|
.summary-counts {
|
||||||
|
display: flex;
|
||||||
|
gap: 1rem;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
}
|
||||||
|
.summary-count {
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 0.25rem;
|
||||||
|
font-size: 0.875rem;
|
||||||
|
}
|
||||||
|
.count-badge {
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
width: 1.5rem;
|
||||||
|
height: 1.5rem;
|
||||||
|
border-radius: 50%;
|
||||||
|
font-weight: 600;
|
||||||
|
font-size: 0.75rem;
|
||||||
|
}
|
||||||
|
.count-badge.success {
|
||||||
|
background-color: #48bb78;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.count-badge.failure {
|
||||||
|
background-color: #f56565;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.count-badge.in-progress {
|
||||||
|
background-color: #4299e1;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.count-badge.queued {
|
||||||
|
background-color: #ecc94b;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Processing logs - collapsible */
|
||||||
.timeline {
|
.timeline {
|
||||||
position: relative;
|
position: relative;
|
||||||
padding-left: 2rem;
|
padding-left: 2rem;
|
||||||
@@ -170,6 +234,29 @@
|
|||||||
color: #a0aec0;
|
color: #a0aec0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.logs-toggle {
|
||||||
|
cursor: pointer;
|
||||||
|
user-select: none;
|
||||||
|
display: inline-flex;
|
||||||
|
align-items: center;
|
||||||
|
gap: 0.5rem;
|
||||||
|
color: #3182ce;
|
||||||
|
font-weight: 600;
|
||||||
|
margin-bottom: 1rem;
|
||||||
|
}
|
||||||
|
.logs-toggle:hover {
|
||||||
|
color: #2c5aa0;
|
||||||
|
}
|
||||||
|
.logs-content {
|
||||||
|
max-height: 0;
|
||||||
|
overflow: hidden;
|
||||||
|
transition: max-height 0.3s ease-out;
|
||||||
|
}
|
||||||
|
.logs-content.expanded {
|
||||||
|
max-height: 5000px;
|
||||||
|
transition: max-height 0.5s ease-in;
|
||||||
|
}
|
||||||
|
|
||||||
.no-logs {
|
.no-logs {
|
||||||
text-align: center;
|
text-align: center;
|
||||||
padding: 3rem;
|
padding: 3rem;
|
||||||
@@ -206,6 +293,178 @@
|
|||||||
background-color: #FEE2E2;
|
background-color: #FEE2E2;
|
||||||
color: #991B1B;
|
color: #991B1B;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* Flow visualization with branches */
|
||||||
|
.flow-stage {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
margin-bottom: 1.5rem;
|
||||||
|
}
|
||||||
|
.flow-indicator {
|
||||||
|
width: 40px;
|
||||||
|
height: 40px;
|
||||||
|
border-radius: 50%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
margin-right: 1rem;
|
||||||
|
flex-shrink: 0;
|
||||||
|
}
|
||||||
|
.flow-indicator.success {
|
||||||
|
background-color: #48bb78;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.flow-indicator.failure {
|
||||||
|
background-color: #f56565;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.flow-indicator.in_progress {
|
||||||
|
background-color: #4299e1;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.flow-indicator.pending, .flow-indicator.queued {
|
||||||
|
background-color: #ecc94b;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.flow-indicator.not_run {
|
||||||
|
background-color: #e2e8f0;
|
||||||
|
color: #718096;
|
||||||
|
}
|
||||||
|
.flow-content {
|
||||||
|
flex: 1;
|
||||||
|
background-color: #f7fafc;
|
||||||
|
padding: 1rem;
|
||||||
|
border-radius: 0.5rem;
|
||||||
|
border-left: 3px solid #e2e8f0;
|
||||||
|
}
|
||||||
|
.flow-content.success {
|
||||||
|
border-left-color: #48bb78;
|
||||||
|
background-color: #f0fff4;
|
||||||
|
}
|
||||||
|
.flow-content.failure {
|
||||||
|
border-left-color: #f56565;
|
||||||
|
background-color: #fff5f5;
|
||||||
|
}
|
||||||
|
.flow-content.in_progress {
|
||||||
|
border-left-color: #4299e1;
|
||||||
|
background-color: #ebf8ff;
|
||||||
|
}
|
||||||
|
.flow-content.pending, .flow-content.queued {
|
||||||
|
border-left-color: #ecc94b;
|
||||||
|
background-color: #fffff0;
|
||||||
|
}
|
||||||
|
.flow-title {
|
||||||
|
font-weight: 600;
|
||||||
|
color: #2d3748;
|
||||||
|
margin-bottom: 0.25rem;
|
||||||
|
}
|
||||||
|
.flow-message {
|
||||||
|
color: #4a5568;
|
||||||
|
font-size: 0.875rem;
|
||||||
|
}
|
||||||
|
.flow-timestamp {
|
||||||
|
color: #718096;
|
||||||
|
font-size: 0.75rem;
|
||||||
|
margin-top: 0.25rem;
|
||||||
|
}
|
||||||
|
.flow-connector {
|
||||||
|
width: 2px;
|
||||||
|
height: 20px;
|
||||||
|
background-color: #e2e8f0;
|
||||||
|
margin-left: 19px;
|
||||||
|
margin-bottom: 0.5rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Branch visualization */
|
||||||
|
.flow-branches {
|
||||||
|
margin-left: 56px;
|
||||||
|
margin-top: 1rem;
|
||||||
|
padding-left: 1rem;
|
||||||
|
border-left: 2px dashed #cbd5e0;
|
||||||
|
}
|
||||||
|
.branch-item {
|
||||||
|
display: flex;
|
||||||
|
align-items: flex-start;
|
||||||
|
margin-bottom: 1rem;
|
||||||
|
position: relative;
|
||||||
|
}
|
||||||
|
.branch-item::before {
|
||||||
|
content: '';
|
||||||
|
position: absolute;
|
||||||
|
left: -1rem;
|
||||||
|
top: 20px;
|
||||||
|
width: 1rem;
|
||||||
|
height: 2px;
|
||||||
|
background-color: #cbd5e0;
|
||||||
|
}
|
||||||
|
.branch-indicator {
|
||||||
|
width: 30px;
|
||||||
|
height: 30px;
|
||||||
|
border-radius: 50%;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
margin-right: 0.75rem;
|
||||||
|
flex-shrink: 0;
|
||||||
|
font-size: 0.875rem;
|
||||||
|
}
|
||||||
|
.branch-indicator.success {
|
||||||
|
background-color: #48bb78;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.branch-indicator.failure {
|
||||||
|
background-color: #f56565;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.branch-indicator.in_progress {
|
||||||
|
background-color: #4299e1;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.branch-indicator.pending, .branch-indicator.queued {
|
||||||
|
background-color: #ecc94b;
|
||||||
|
color: white;
|
||||||
|
}
|
||||||
|
.branch-content {
|
||||||
|
flex: 1;
|
||||||
|
background-color: white;
|
||||||
|
padding: 0.75rem;
|
||||||
|
border-radius: 0.5rem;
|
||||||
|
border: 1px solid #e2e8f0;
|
||||||
|
}
|
||||||
|
.branch-content.failure {
|
||||||
|
border-color: #f56565;
|
||||||
|
background-color: #fffafa;
|
||||||
|
}
|
||||||
|
.branch-title {
|
||||||
|
font-weight: 600;
|
||||||
|
color: #2d3748;
|
||||||
|
font-size: 0.875rem;
|
||||||
|
margin-bottom: 0.25rem;
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
.branch-message {
|
||||||
|
color: #4a5568;
|
||||||
|
font-size: 0.75rem;
|
||||||
|
}
|
||||||
|
.retry-btn {
|
||||||
|
background-color: #f56565;
|
||||||
|
color: white;
|
||||||
|
border: none;
|
||||||
|
padding: 0.25rem 0.75rem;
|
||||||
|
border-radius: 0.25rem;
|
||||||
|
font-size: 0.75rem;
|
||||||
|
cursor: pointer;
|
||||||
|
font-weight: 600;
|
||||||
|
}
|
||||||
|
.retry-btn:hover {
|
||||||
|
background-color: #e53e3e;
|
||||||
|
}
|
||||||
|
.retry-btn:disabled {
|
||||||
|
background-color: #cbd5e0;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
</style>
|
</style>
|
||||||
<script>
|
<script>
|
||||||
// JavaScript for handling retry functionality
|
// JavaScript for handling retry functionality
|
||||||
@@ -257,6 +516,79 @@
|
|||||||
button.innerHTML = '<i class="fas fa-redo"></i> Retry Processing';
|
button.innerHTML = '<i class="fas fa-redo"></i> Retry Processing';
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// JavaScript for handling per-subtask retry
|
||||||
|
async function retrySubtask(subtaskName, buttonId) {
|
||||||
|
const fileId = {{ file.id | tojson }};
|
||||||
|
const button = document.getElementById(buttonId);
|
||||||
|
const statusDiv = document.getElementById('subtask-status-' + subtaskName);
|
||||||
|
|
||||||
|
// Disable button and show loading
|
||||||
|
button.disabled = true;
|
||||||
|
button.innerHTML = '<i class="fas fa-spinner fa-spin"></i> Retrying...';
|
||||||
|
if (statusDiv) {
|
||||||
|
statusDiv.innerHTML = '';
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await fetch(`/api/files/${fileId}/retry-subtask?subtask_name=${subtaskName}`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json'
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
const data = await response.json();
|
||||||
|
|
||||||
|
if (response.ok) {
|
||||||
|
if (statusDiv) {
|
||||||
|
statusDiv.innerHTML = `
|
||||||
|
<div style="background-color: #D1FAE5; color: #065F46; padding: 0.5rem; border-radius: 0.25rem; margin-top: 0.5rem; font-size: 0.75rem;">
|
||||||
|
<i class="fas fa-check-circle"></i> Queued for retry. Task ID: ${data.task_id}
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
// Reload page after 2 seconds
|
||||||
|
setTimeout(() => window.location.reload(), 2000);
|
||||||
|
} else {
|
||||||
|
if (statusDiv) {
|
||||||
|
statusDiv.innerHTML = `
|
||||||
|
<div style="background-color: #FEE2E2; color: #991B1B; padding: 0.5rem; border-radius: 0.25rem; margin-top: 0.5rem; font-size: 0.75rem;">
|
||||||
|
<i class="fas fa-exclamation-triangle"></i> Error: ${data.detail || 'Failed to retry'}
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
button.disabled = false;
|
||||||
|
button.innerHTML = '<i class="fas fa-redo"></i> Retry';
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
if (statusDiv) {
|
||||||
|
statusDiv.innerHTML = `
|
||||||
|
<div style="background-color: #FEE2E2; color: #991B1B; padding: 0.5rem; border-radius: 0.25rem; margin-top: 0.5rem; font-size: 0.75rem;">
|
||||||
|
<i class="fas fa-exclamation-triangle"></i> Network error: ${error.message}
|
||||||
|
</div>
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
button.disabled = false;
|
||||||
|
button.innerHTML = '<i class="fas fa-redo"></i> Retry';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Toggle logs visibility
|
||||||
|
function toggleLogs() {
|
||||||
|
const logsContent = document.getElementById('logs-content');
|
||||||
|
const toggleIcon = document.getElementById('logs-toggle-icon');
|
||||||
|
|
||||||
|
if (logsContent.classList.contains('expanded')) {
|
||||||
|
logsContent.classList.remove('expanded');
|
||||||
|
toggleIcon.classList.remove('fa-chevron-up');
|
||||||
|
toggleIcon.classList.add('fa-chevron-down');
|
||||||
|
} else {
|
||||||
|
logsContent.classList.add('expanded');
|
||||||
|
toggleIcon.classList.remove('fa-chevron-down');
|
||||||
|
toggleIcon.classList.add('fa-chevron-up');
|
||||||
|
}
|
||||||
|
}
|
||||||
</script>
|
</script>
|
||||||
{% endblock %}
|
{% endblock %}
|
||||||
|
|
||||||
@@ -322,7 +654,77 @@
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<!-- Processing History Card -->
|
<!-- Step Summary Card -->
|
||||||
|
{% if step_summary %}
|
||||||
|
<div class="detail-card">
|
||||||
|
<h3>Processing Status Summary</h3>
|
||||||
|
<div class="step-summary">
|
||||||
|
<div class="summary-card main-steps">
|
||||||
|
<div class="summary-title">Main Processing Steps</div>
|
||||||
|
<div class="summary-counts">
|
||||||
|
{% if step_summary.main.success > 0 %}
|
||||||
|
<div class="summary-count">
|
||||||
|
<span class="count-badge success">{{ step_summary.main.success }}</span>
|
||||||
|
<span>Success</span>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
{% if step_summary.main.failure > 0 %}
|
||||||
|
<div class="summary-count">
|
||||||
|
<span class="count-badge failure">{{ step_summary.main.failure }}</span>
|
||||||
|
<span>Failed</span>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
{% if step_summary.main.in_progress > 0 %}
|
||||||
|
<div class="summary-count">
|
||||||
|
<span class="count-badge in-progress">{{ step_summary.main.in_progress }}</span>
|
||||||
|
<span>In Progress</span>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
{% if step_summary.main.queued > 0 %}
|
||||||
|
<div class="summary-count">
|
||||||
|
<span class="count-badge queued">{{ step_summary.main.queued }}</span>
|
||||||
|
<span>Queued</span>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{% if step_summary.total_upload_tasks > 0 %}
|
||||||
|
<div class="summary-card upload-steps">
|
||||||
|
<div class="summary-title">Upload Destinations</div>
|
||||||
|
<div class="summary-counts">
|
||||||
|
{% if step_summary.uploads.success > 0 %}
|
||||||
|
<div class="summary-count">
|
||||||
|
<span class="count-badge success">{{ step_summary.uploads.success }}</span>
|
||||||
|
<span>Success</span>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
{% if step_summary.uploads.failure > 0 %}
|
||||||
|
<div class="summary-count">
|
||||||
|
<span class="count-badge failure">{{ step_summary.uploads.failure }}</span>
|
||||||
|
<span>Failed</span>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
{% if step_summary.uploads.in_progress > 0 %}
|
||||||
|
<div class="summary-count">
|
||||||
|
<span class="count-badge in-progress">{{ step_summary.uploads.in_progress }}</span>
|
||||||
|
<span>In Progress</span>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
{% if step_summary.uploads.queued > 0 %}
|
||||||
|
<div class="summary-count">
|
||||||
|
<span class="count-badge queued">{{ step_summary.uploads.queued }}</span>
|
||||||
|
<span>Queued</span>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<!-- Processing History Card (Collapsible) -->
|
||||||
<div class="detail-card">
|
<div class="detail-card">
|
||||||
<div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 1rem;">
|
<div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 1rem;">
|
||||||
<h3 style="margin: 0;">Processing History</h3>
|
<h3 style="margin: 0;">Processing History</h3>
|
||||||
@@ -334,6 +736,11 @@
|
|||||||
</div>
|
</div>
|
||||||
<div id="reprocess-status"></div>
|
<div id="reprocess-status"></div>
|
||||||
{% if logs %}
|
{% if logs %}
|
||||||
|
<div class="logs-toggle" onclick="toggleLogs()">
|
||||||
|
<i id="logs-toggle-icon" class="fas fa-chevron-down"></i>
|
||||||
|
<span>View Full Processing Logs ({{ logs|length }} entries)</span>
|
||||||
|
</div>
|
||||||
|
<div id="logs-content" class="logs-content">
|
||||||
<div class="timeline">
|
<div class="timeline">
|
||||||
{% for log in logs %}
|
{% for log in logs %}
|
||||||
<div class="timeline-item">
|
<div class="timeline-item">
|
||||||
@@ -355,6 +762,7 @@
|
|||||||
</div>
|
</div>
|
||||||
{% endfor %}
|
{% endfor %}
|
||||||
</div>
|
</div>
|
||||||
|
</div>
|
||||||
{% else %}
|
{% else %}
|
||||||
<div class="no-logs">
|
<div class="no-logs">
|
||||||
<i class="fas fa-clipboard-list"></i>
|
<i class="fas fa-clipboard-list"></i>
|
||||||
@@ -369,52 +777,77 @@
|
|||||||
<div class="detail-card">
|
<div class="detail-card">
|
||||||
<h3>Process Flow Visualization</h3>
|
<h3>Process Flow Visualization</h3>
|
||||||
<div style="padding: 1rem 0;">
|
<div style="padding: 1rem 0;">
|
||||||
<div style="position: relative;">
|
|
||||||
{% for stage in flow_data %}
|
{% for stage in flow_data %}
|
||||||
<div style="display: flex; align-items: center; margin-bottom: 1.5rem;">
|
<div class="flow-stage">
|
||||||
<!-- Status indicator -->
|
<!-- Status indicator -->
|
||||||
<div style="width: 40px; height: 40px; border-radius: 50%; display: flex; align-items: center; justify-content: center; margin-right: 1rem;
|
<div class="flow-indicator {{ stage.status.lower().replace(' ', '_') }}">
|
||||||
{% if stage.status == 'success' %}background-color: #48bb78; color: white;
|
|
||||||
{% elif stage.status == 'failure' %}background-color: #f56565; color: white;
|
|
||||||
{% elif stage.status == 'in_progress' %}background-color: #4299e1; color: white;
|
|
||||||
{% elif stage.status == 'pending' %}background-color: #ecc94b; color: white;
|
|
||||||
{% else %}background-color: #e2e8f0; color: #718096;{% endif %}
|
|
||||||
">
|
|
||||||
{% if stage.status == 'success' %}<i class="fas fa-check"></i>
|
{% if stage.status == 'success' %}<i class="fas fa-check"></i>
|
||||||
{% elif stage.status == 'failure' %}<i class="fas fa-times"></i>
|
{% elif stage.status == 'failure' %}<i class="fas fa-times"></i>
|
||||||
{% elif stage.status == 'in_progress' %}<i class="fas fa-spinner fa-spin"></i>
|
{% elif stage.status == 'in_progress' %}<i class="fas fa-spinner fa-spin"></i>
|
||||||
{% elif stage.status == 'pending' %}<i class="fas fa-clock"></i>
|
{% elif stage.status in ['pending', 'queued'] %}<i class="fas fa-clock"></i>
|
||||||
{% else %}<i class="fas fa-circle"></i>{% endif %}
|
{% else %}<i class="fas fa-circle"></i>{% endif %}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<!-- Stage info -->
|
<!-- Stage content -->
|
||||||
<div style="flex: 1; background-color: #f7fafc; padding: 1rem; border-radius: 0.5rem; border-left: 3px solid
|
<div class="flow-content {{ stage.status.lower().replace(' ', '_') }}">
|
||||||
{% if stage.status == 'success' %}#48bb78
|
<div class="flow-title">{{ stage.label }}</div>
|
||||||
{% elif stage.status == 'failure' %}#f56565
|
|
||||||
{% elif stage.status == 'in_progress' %}#4299e1
|
|
||||||
{% elif stage.status == 'pending' %}#ecc94b
|
|
||||||
{% else %}#e2e8f0{% endif %};">
|
|
||||||
<div style="font-weight: 600; color: #2d3748; margin-bottom: 0.25rem;">{{ stage.label }}</div>
|
|
||||||
{% if stage.message %}
|
{% if stage.message %}
|
||||||
<div style="color: #4a5568; font-size: 0.875rem;">{{ stage.message }}</div>
|
<div class="flow-message">{{ stage.message }}</div>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
{% if stage.status == 'not_run' %}
|
{% if stage.status == 'not_run' %}
|
||||||
<div style="color: #718096; font-size: 0.875rem; font-style: italic;">Not executed</div>
|
<div style="color: #718096; font-size: 0.875rem; font-style: italic;">Not executed</div>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
{% if stage.timestamp %}
|
{% if stage.timestamp %}
|
||||||
<div style="color: #718096; font-size: 0.75rem; margin-top: 0.25rem;">
|
<div class="flow-timestamp">
|
||||||
{{ stage.timestamp.strftime('%Y-%m-%d %H:%M:%S') }}
|
{{ stage.timestamp.strftime('%Y-%m-%d %H:%M:%S') }}
|
||||||
</div>
|
</div>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
<!-- Connector line (except for last item) -->
|
<!-- Upload branches (if this stage has them) -->
|
||||||
{% if not loop.last %}
|
{% if stage.is_branch_parent and stage.branches %}
|
||||||
<div style="width: 2px; height: 20px; background-color: #e2e8f0; margin-left: 19px; margin-bottom: 0.5rem;"></div>
|
<div class="flow-branches">
|
||||||
|
{% for branch in stage.branches %}
|
||||||
|
<div class="branch-item">
|
||||||
|
<div class="branch-indicator {{ branch.status.lower().replace(' ', '_') }}">
|
||||||
|
{% if branch.status == 'success' %}<i class="fas fa-check"></i>
|
||||||
|
{% elif branch.status == 'failure' %}<i class="fas fa-times"></i>
|
||||||
|
{% elif branch.status == 'in_progress' %}<i class="fas fa-spinner fa-spin"></i>
|
||||||
|
{% else %}<i class="fas fa-clock"></i>{% endif %}
|
||||||
|
</div>
|
||||||
|
<div class="branch-content {% if branch.status == 'failure' %}failure{% endif %}">
|
||||||
|
<div class="branch-title">
|
||||||
|
<span>{{ branch.label }}</span>
|
||||||
|
{% if branch.can_retry and branch.status == 'failure' %}
|
||||||
|
<button
|
||||||
|
id="retry-btn-{{ branch.key }}"
|
||||||
|
class="retry-btn"
|
||||||
|
onclick="retrySubtask('{{ branch.key }}', 'retry-btn-{{ branch.key }}')">
|
||||||
|
<i class="fas fa-redo"></i> Retry
|
||||||
|
</button>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
{% if branch.message %}
|
||||||
|
<div class="branch-message">{{ branch.message }}</div>
|
||||||
|
{% endif %}
|
||||||
|
{% if branch.timestamp %}
|
||||||
|
<div style="color: #718096; font-size: 0.65rem; margin-top: 0.25rem;">
|
||||||
|
{{ branch.timestamp.strftime('%Y-%m-%d %H:%M:%S') }}
|
||||||
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
<div id="subtask-status-{{ branch.key }}"></div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
{% endfor %}
|
{% endfor %}
|
||||||
</div>
|
</div>
|
||||||
|
{% endif %}
|
||||||
|
|
||||||
|
<!-- Connector line (except for last item) -->
|
||||||
|
{% if not loop.last %}
|
||||||
|
<div class="flow-connector"></div>
|
||||||
|
{% endif %}
|
||||||
|
{% endfor %}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
|
|||||||
@@ -71,6 +71,55 @@ class TestFileReprocessing:
|
|||||||
assert "not found on disk" in response.json()["detail"].lower()
|
assert "not found on disk" in response.json()["detail"].lower()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.integration
|
||||||
|
class TestSubtaskRetry:
|
||||||
|
"""Tests for per-subtask retry endpoint."""
|
||||||
|
|
||||||
|
def test_retry_subtask_invalid_file(self, client: TestClient):
|
||||||
|
"""Test retrying a subtask for nonexistent file."""
|
||||||
|
response = client.post("/api/files/99999/retry-subtask?subtask_name=upload_to_dropbox")
|
||||||
|
assert response.status_code == 404
|
||||||
|
assert "not found" in response.json()["detail"].lower()
|
||||||
|
|
||||||
|
def test_retry_subtask_invalid_task_name(self, client: TestClient, db_session, sample_pdf_path):
|
||||||
|
"""Test retrying with invalid subtask name."""
|
||||||
|
# Create a file record
|
||||||
|
file_record = FileRecord(
|
||||||
|
filehash="retry123",
|
||||||
|
original_filename="retry.pdf",
|
||||||
|
local_filename=sample_pdf_path,
|
||||||
|
file_size=1024,
|
||||||
|
mime_type="application/pdf"
|
||||||
|
)
|
||||||
|
db_session.add(file_record)
|
||||||
|
db_session.commit()
|
||||||
|
db_session.refresh(file_record)
|
||||||
|
|
||||||
|
# Test with invalid subtask name
|
||||||
|
response = client.post(f"/api/files/{file_record.id}/retry-subtask?subtask_name=invalid_task")
|
||||||
|
assert response.status_code == 400
|
||||||
|
assert "invalid subtask name" in response.json()["detail"].lower()
|
||||||
|
|
||||||
|
def test_retry_subtask_missing_processed_file(self, client: TestClient, db_session, sample_pdf_path):
|
||||||
|
"""Test retrying when processed file is missing."""
|
||||||
|
# Create a file record
|
||||||
|
file_record = FileRecord(
|
||||||
|
filehash="retry456",
|
||||||
|
original_filename="retry2.pdf",
|
||||||
|
local_filename=sample_pdf_path,
|
||||||
|
file_size=1024,
|
||||||
|
mime_type="application/pdf"
|
||||||
|
)
|
||||||
|
db_session.add(file_record)
|
||||||
|
db_session.commit()
|
||||||
|
db_session.refresh(file_record)
|
||||||
|
|
||||||
|
# Test retry (processed file won't exist)
|
||||||
|
response = client.post(f"/api/files/{file_record.id}/retry-subtask?subtask_name=upload_to_dropbox")
|
||||||
|
assert response.status_code == 400
|
||||||
|
assert "processed file not found" in response.json()["detail"].lower()
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.integration
|
@pytest.mark.integration
|
||||||
class TestFilePreview:
|
class TestFilePreview:
|
||||||
"""Tests for file preview endpoint."""
|
"""Tests for file preview endpoint."""
|
||||||
@@ -193,8 +242,146 @@ class TestFileDetailView:
|
|||||||
assert b"detail.pdf" in response.content
|
assert b"detail.pdf" in response.content
|
||||||
assert b"Processing History" in response.content
|
assert b"Processing History" in response.content
|
||||||
|
|
||||||
|
def test_file_detail_view_with_upload_branches(self, client: TestClient, db_session, sample_pdf_path):
|
||||||
|
"""Test file detail view with upload subtask branches."""
|
||||||
|
# Create a file record
|
||||||
|
file_record = FileRecord(
|
||||||
|
filehash="branch123",
|
||||||
|
original_filename="branches.pdf",
|
||||||
|
local_filename=sample_pdf_path,
|
||||||
|
file_size=1024,
|
||||||
|
mime_type="application/pdf"
|
||||||
|
)
|
||||||
|
db_session.add(file_record)
|
||||||
|
db_session.commit()
|
||||||
|
db_session.refresh(file_record)
|
||||||
|
|
||||||
|
# Add processing logs including upload branches
|
||||||
|
logs = [
|
||||||
|
ProcessingLog(
|
||||||
|
file_id=file_record.id,
|
||||||
|
task_id="task-1",
|
||||||
|
step_name="send_to_all_destinations",
|
||||||
|
status="success",
|
||||||
|
message="Queued uploads"
|
||||||
|
),
|
||||||
|
ProcessingLog(
|
||||||
|
file_id=file_record.id,
|
||||||
|
task_id="task-2",
|
||||||
|
step_name="upload_to_dropbox",
|
||||||
|
status="success",
|
||||||
|
message="Uploaded to Dropbox"
|
||||||
|
),
|
||||||
|
ProcessingLog(
|
||||||
|
file_id=file_record.id,
|
||||||
|
task_id="task-3",
|
||||||
|
step_name="upload_to_s3",
|
||||||
|
status="failure",
|
||||||
|
message="S3 connection error"
|
||||||
|
),
|
||||||
|
ProcessingLog(
|
||||||
|
file_id=file_record.id,
|
||||||
|
task_id="task-4",
|
||||||
|
step_name="upload_to_nextcloud",
|
||||||
|
status="success",
|
||||||
|
message="Uploaded to Nextcloud"
|
||||||
|
)
|
||||||
|
]
|
||||||
|
for log in logs:
|
||||||
|
db_session.add(log)
|
||||||
|
db_session.commit()
|
||||||
|
|
||||||
|
# Test detail view
|
||||||
|
response = client.get(f"/files/{file_record.id}/detail")
|
||||||
|
assert response.status_code == 200
|
||||||
|
# Check that response contains branching visualization elements
|
||||||
|
assert b"Process Flow Visualization" in response.content
|
||||||
|
assert b"Processing Status Summary" in response.content
|
||||||
|
# Should have upload branches
|
||||||
|
assert b"Dropbox" in response.content or b"dropbox" in response.content
|
||||||
|
|
||||||
def test_file_detail_view_nonexistent(self, client: TestClient):
|
def test_file_detail_view_nonexistent(self, client: TestClient):
|
||||||
"""Test file detail view for nonexistent file."""
|
"""Test file detail view for nonexistent file."""
|
||||||
response = client.get("/files/99999/detail")
|
response = client.get("/files/99999/detail")
|
||||||
assert response.status_code == 200 # Returns page with error message
|
assert response.status_code == 200 # Returns page with error message
|
||||||
assert b"not found" in response.content.lower()
|
assert b"not found" in response.content.lower()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.unit
|
||||||
|
class TestProcessingFlowComputation:
|
||||||
|
"""Tests for the _compute_processing_flow function."""
|
||||||
|
|
||||||
|
def test_flow_with_upload_branches(self, db_session):
|
||||||
|
"""Test that upload tasks are properly grouped as branches."""
|
||||||
|
from app.views.files import _compute_processing_flow
|
||||||
|
|
||||||
|
# Create mock logs
|
||||||
|
class MockLog:
|
||||||
|
def __init__(self, step_name, status, message, timestamp, task_id):
|
||||||
|
self.step_name = step_name
|
||||||
|
self.status = status
|
||||||
|
self.message = message
|
||||||
|
self.timestamp = timestamp
|
||||||
|
self.task_id = task_id
|
||||||
|
|
||||||
|
logs = [
|
||||||
|
MockLog("hash_file", "success", "Hashed", None, "task-1"),
|
||||||
|
MockLog("send_to_all_destinations", "success", "Queued", None, "task-2"),
|
||||||
|
MockLog("upload_to_dropbox", "success", "Uploaded", None, "task-3"),
|
||||||
|
MockLog("upload_to_s3", "failure", "Failed", None, "task-4"),
|
||||||
|
]
|
||||||
|
|
||||||
|
flow = _compute_processing_flow(logs)
|
||||||
|
|
||||||
|
# Find the upload stage
|
||||||
|
upload_stage = None
|
||||||
|
for stage in flow:
|
||||||
|
if stage.get("is_branch_parent"):
|
||||||
|
upload_stage = stage
|
||||||
|
break
|
||||||
|
|
||||||
|
assert upload_stage is not None
|
||||||
|
assert "branches" in upload_stage
|
||||||
|
assert len(upload_stage["branches"]) == 2
|
||||||
|
|
||||||
|
# Check branch details
|
||||||
|
branches = {b["key"]: b for b in upload_stage["branches"]}
|
||||||
|
assert "upload_to_dropbox" in branches
|
||||||
|
assert branches["upload_to_dropbox"]["status"] == "success"
|
||||||
|
assert "upload_to_s3" in branches
|
||||||
|
assert branches["upload_to_s3"]["status"] == "failure"
|
||||||
|
assert branches["upload_to_s3"]["can_retry"] is True
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.unit
|
||||||
|
class TestStepSummary:
|
||||||
|
"""Tests for the _compute_step_summary function."""
|
||||||
|
|
||||||
|
def test_summary_with_mixed_statuses(self):
|
||||||
|
"""Test step summary with various statuses."""
|
||||||
|
from app.views.files import _compute_step_summary
|
||||||
|
|
||||||
|
# Create mock logs
|
||||||
|
class MockLog:
|
||||||
|
def __init__(self, step_name, status):
|
||||||
|
self.step_name = step_name
|
||||||
|
self.status = status
|
||||||
|
|
||||||
|
logs = [
|
||||||
|
MockLog("hash_file", "success"),
|
||||||
|
MockLog("create_file_record", "success"),
|
||||||
|
MockLog("extract_metadata_with_gpt", "failure"),
|
||||||
|
MockLog("upload_to_dropbox", "success"),
|
||||||
|
MockLog("upload_to_s3", "failure"),
|
||||||
|
MockLog("upload_to_nextcloud", "in_progress"),
|
||||||
|
]
|
||||||
|
|
||||||
|
summary = _compute_step_summary(logs)
|
||||||
|
|
||||||
|
assert "main" in summary
|
||||||
|
assert "uploads" in summary
|
||||||
|
assert summary["total_main_steps"] == 3
|
||||||
|
assert summary["total_upload_tasks"] == 3
|
||||||
|
assert summary["uploads"]["success"] == 1
|
||||||
|
assert summary["uploads"]["failure"] == 1
|
||||||
|
assert summary["uploads"]["in_progress"] == 1
|
||||||
|
|||||||
Reference in New Issue
Block a user