| import logging |
| import os |
| import re |
| from urllib.parse import quote_plus |
| from typing import List, Union |
| import requests |
|
|
| from plugins.common import DownloadCancelled |
| from plugins.stream import download_stream_url |
|
|
| logger = logging.getLogger("worker.plugins.threads") |
|
|
|
|
| def download_threads( |
| url: str, |
| output_dir: str, |
| progress_fn=None, |
| cancel_event=None |
| ) -> Union[str, List[str]]: |
| """ |
| Descarga fotos o videos de posts de Threads con fallback automático. |
| """ |
| if cancel_event and cancel_event.is_set(): |
| raise DownloadCancelled() |
|
|
| os.makedirs(output_dir, exist_ok=True) |
| delirius_key = os.getenv("DELIRIUS_API_KEY", "") |
|
|
| try: |
| enc_url = quote_plus(url.strip()) |
| api_url = f"https://api.delirius.online/download/threads?url={enc_url}" |
| if delirius_key: |
| api_url += f"&apikey={delirius_key}" |
|
|
| resp = requests.get(api_url, timeout=25) |
| if resp.status_code == 200: |
| json_data = resp.json() |
| if json_data.get("status") and "data" in json_data: |
| data = json_data["data"] |
| media_list = [] |
|
|
| if isinstance(data, list): |
| for idx, item in enumerate(data, 1): |
| d_url = item.get("url") or item.get("download_url") or item.get("dl_url") |
| m_type = item.get("type", "video" if ".mp4" in str(d_url) else "photo") |
| ext = ".mp4" if m_type == "video" else ".jpg" |
| if d_url: |
| media_list.append((d_url, f"threads_{idx}{ext}")) |
| elif isinstance(data, dict): |
| media_arr = data.get("media") or data.get("result") or [] |
| if isinstance(media_arr, list) and media_arr: |
| for idx, item in enumerate(media_arr, 1): |
| d_url = item.get("url") or item.get("download_url") if isinstance(item, dict) else str(item) |
| ext = ".mp4" if ".mp4" in str(d_url) else ".jpg" |
| if d_url: |
| media_list.append((d_url, f"threads_{idx}{ext}")) |
| else: |
| d_url = data.get("url") or data.get("download") or data.get("dl_url") |
| m_type = data.get("type", "video" if ".mp4" in str(d_url) else "photo") |
| ext = ".mp4" if m_type == "video" else ".jpg" |
| if d_url: |
| media_list.append((d_url, f"threads_1{ext}")) |
|
|
| if media_list: |
| downloaded = [] |
| for idx, (m_url, m_name) in enumerate(media_list): |
| if cancel_event and cancel_event.is_set(): |
| raise DownloadCancelled() |
| target = os.path.join(output_dir, m_name) |
| out_path, _ = download_stream_url(m_url, target, progress_fn=progress_fn, cancel_event=cancel_event) |
| if os.path.exists(out_path): |
| downloaded.append(out_path) |
| if downloaded: |
| return downloaded[0] if len(downloaded) == 1 else downloaded |
| except Exception as e: |
| logger.warning(f"Descarga falló para Threads: {e}") |
|
|
| raise RuntimeError("No se pudo obtener el archivo multimedia de Threads para este enlace.") |
|
|
