Christian Krakau-Louis
44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import requests
|
|
from app.config import settings
|
|
from app.tasks.retry_config import BaseTaskWithRetry
|
|
from app.celery_app import celery
|
|
from app.utils import task_logger, log_task
|
|
|
|
@celery.task(base=BaseTaskWithRetry)
|
|
@log_task("upload_to_nextcloud")
|
|
def upload_to_nextcloud(file_path: str):
|
|
"""Uploads a file to Nextcloud in the configured folder."""
|
|
|
|
if not os.path.exists(file_path):
|
|
task_logger(f"File not found: {file_path}", level="error", step_name="nextcloud_upload")
|
|
raise FileNotFoundError(f"File not found: {file_path}")
|
|
|
|
# Extract filename
|
|
filename = os.path.basename(file_path)
|
|
|
|
# Construct the full upload URL
|
|
nextcloud_url = f"{settings.nextcloud_upload_url}/{settings.nextcloud_folder}/{filename}"
|
|
|
|
task_logger(f"Starting upload of {filename} to Nextcloud", step_name="nextcloud_upload")
|
|
|
|
# Read file content
|
|
with open(file_path, "rb") as file_data:
|
|
response = requests.put(
|
|
nextcloud_url,
|
|
auth=(settings.nextcloud_username, settings.nextcloud_password),
|
|
data=file_data
|
|
)
|
|
|
|
# Check if upload was successful
|
|
if response.status_code in (200, 201):
|
|
task_logger(f"Successfully uploaded {filename} to Nextcloud at {nextcloud_url}",
|
|
step_name="nextcloud_upload", status="success")
|
|
return {"status": "Completed", "file": file_path}
|
|
else:
|
|
error_msg = f"Failed to upload {filename} to Nextcloud: {response.status_code} - {response.text}"
|
|
task_logger(error_msg, level="error", step_name="nextcloud_upload", status="failure")
|
|
raise Exception(error_msg)
|