Christian Krakau-Louis
99 lines
4.3 KiB
Python
99 lines
4.3 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import logging
|
|
from app.config import settings
|
|
from app.tasks.retry_config import BaseTaskWithRetry
|
|
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_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
|
|
from app.tasks.upload_to_onedrive import upload_to_onedrive
|
|
from app.tasks.upload_to_s3 import upload_to_s3
|
|
from app.celery_app import celery
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
@celery.task(base=BaseTaskWithRetry)
|
|
def send_to_all_destinations(file_path: str):
|
|
"""Distribute a file to all configured storage destinations."""
|
|
|
|
if not os.path.exists(file_path):
|
|
raise FileNotFoundError(f"File not found: {file_path}")
|
|
|
|
logger.info(f"Sending {file_path} to all configured destinations")
|
|
results = {}
|
|
|
|
# Send to Dropbox if configured
|
|
if settings.dropbox_app_key and settings.dropbox_app_secret and settings.dropbox_refresh_token:
|
|
logger.info(f"Queueing {file_path} for Dropbox upload")
|
|
task = upload_to_dropbox.delay(file_path)
|
|
results["dropbox_task_id"] = task.id
|
|
|
|
# Send to Nextcloud if configured
|
|
if settings.nextcloud_upload_url and settings.nextcloud_username and settings.nextcloud_password:
|
|
logger.info(f"Queueing {file_path} for Nextcloud upload")
|
|
task = upload_to_nextcloud.delay(file_path)
|
|
results["nextcloud_task_id"] = task.id
|
|
|
|
# Send to Paperless if configured
|
|
if settings.paperless_ngx_api_token and settings.paperless_host:
|
|
logger.info(f"Queueing {file_path} for Paperless upload")
|
|
task = upload_to_paperless.delay(file_path)
|
|
results["paperless_task_id"] = task.id
|
|
|
|
# Send to Google Drive if configured
|
|
if settings.google_drive_credentials_json:
|
|
logger.info(f"Queueing {file_path} for Google Drive upload")
|
|
task = upload_to_google_drive.delay(file_path)
|
|
results["google_drive_task_id"] = task.id
|
|
|
|
# Send to WebDAV if configured
|
|
if settings.webdav_url and settings.webdav_username and settings.webdav_password:
|
|
logger.info(f"Queueing {file_path} for WebDAV upload")
|
|
task = upload_to_webdav.delay(file_path)
|
|
results["webdav_task_id"] = task.id
|
|
|
|
# Send to FTP if configured
|
|
if settings.ftp_host and settings.ftp_username and settings.ftp_password:
|
|
logger.info(f"Queueing {file_path} for FTP upload")
|
|
task = upload_to_ftp.delay(file_path)
|
|
results["ftp_task_id"] = task.id
|
|
|
|
# Send to SFTP if configured
|
|
if settings.sftp_host and settings.sftp_username and (settings.sftp_password or settings.sftp_private_key):
|
|
logger.info(f"Queueing {file_path} for SFTP upload")
|
|
task = upload_to_sftp.delay(file_path)
|
|
results["sftp_task_id"] = task.id
|
|
|
|
# Send via email if configured
|
|
if settings.email_host and settings.email_username and settings.email_password and settings.email_default_recipient:
|
|
logger.info(f"Queueing {file_path} for email delivery")
|
|
task = upload_to_email.delay(file_path)
|
|
results["email_task_id"] = task.id
|
|
|
|
# Send to OneDrive if configured
|
|
if settings.onedrive_client_id and settings.onedrive_client_secret and (
|
|
settings.onedrive_refresh_token or
|
|
(settings.onedrive_tenant_id and settings.onedrive_tenant_id != "common")
|
|
):
|
|
logger.info(f"Queueing {file_path} for OneDrive upload")
|
|
task = upload_to_onedrive.delay(file_path)
|
|
results["onedrive_task_id"] = task.id
|
|
|
|
# Send to Amazon S3 if configured
|
|
if settings.s3_bucket_name and settings.aws_access_key_id and settings.aws_secret_access_key:
|
|
logger.info(f"Queueing {file_path} for S3 upload")
|
|
task = upload_to_s3.delay(file_path)
|
|
results["s3_task_id"] = task.id
|
|
|
|
return {
|
|
"status": "Queued",
|
|
"file_path": file_path,
|
|
"tasks": results
|
|
}
|