copilot-swe-agent[bot]
9c4cf3d70e
Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
from app.config import settings
|
|
from app.tasks.retry_config import BaseTaskWithRetry
|
|
# Import the shared Celery instance
|
|
from app.celery_app import celery
|
|
|
|
# 1) Import the aggregator task
|
|
from app.tasks.send_to_all import send_to_all_destinations, get_configured_services_from_validator
|
|
|
|
# Import notification utility
|
|
from app.utils.notification import notify_file_processed
|
|
|
|
|
|
@celery.task(base=BaseTaskWithRetry)
|
|
def finalize_document_storage(original_file: str, processed_file: str, metadata: dict):
|
|
"""
|
|
Final storage step after embedding metadata.
|
|
We will now call 'send_to_all_destinations' to push the final PDF to Dropbox/Nextcloud/Paperless.
|
|
After uploading, send a notification about the processed file.
|
|
"""
|
|
print(f"[INFO] Finalizing document storage for {processed_file}")
|
|
|
|
# Determine which destinations are configured
|
|
configured_destinations = []
|
|
try:
|
|
configured_services = get_configured_services_from_validator()
|
|
# Get list of service names that are configured
|
|
for service_name, is_configured in configured_services.items():
|
|
if is_configured:
|
|
# Format service names for display
|
|
display_name = service_name.replace('_', ' ').title()
|
|
configured_destinations.append(display_name)
|
|
except Exception as e:
|
|
print(f"[WARNING] Could not determine configured destinations: {e}")
|
|
configured_destinations = ["configured destinations"]
|
|
|
|
# 2) Enqueue uploads to all destinations (Dropbox, Nextcloud, Paperless)
|
|
# Note: send_to_all_destinations is asynchronous and queues upload tasks
|
|
send_to_all_destinations.delay(processed_file)
|
|
|
|
# 3) Send notification about successful file processing
|
|
# Note: This notification is sent after processing is complete but while uploads
|
|
# are being queued. The message reflects that uploads are being initiated.
|
|
try:
|
|
# Get file information
|
|
file_size = os.path.getsize(processed_file) if os.path.exists(processed_file) else 0
|
|
filename = os.path.basename(processed_file)
|
|
|
|
notify_file_processed(
|
|
filename=filename,
|
|
file_size=file_size,
|
|
metadata=metadata,
|
|
destinations=configured_destinations
|
|
)
|
|
except Exception as e:
|
|
print(f"[WARNING] Failed to send file processed notification: {e}")
|
|
|
|
return {
|
|
"status": "Completed",
|
|
"file": processed_file
|
|
}
|