mirror of
https://github.com/Yuvi9587/Kemono-Downloader.git
synced 2025-12-29 16:14:44 +00:00
105 lines
4.4 KiB
Python
105 lines
4.4 KiB
Python
import os
|
|
import time
|
|
import requests
|
|
from PyQt5.QtCore import QThread, pyqtSignal
|
|
|
|
from ...core.saint2_client import fetch_saint2_data
|
|
|
|
class Saint2DownloadThread(QThread):
|
|
"""A dedicated QThread for handling saint2.su downloads."""
|
|
progress_signal = pyqtSignal(str)
|
|
file_progress_signal = pyqtSignal(str, object)
|
|
finished_signal = pyqtSignal(int, int, bool) # dl_count, skip_count, cancelled
|
|
|
|
def __init__(self, url, output_dir, parent=None):
|
|
super().__init__(parent)
|
|
self.saint2_url = url
|
|
self.output_dir = output_dir
|
|
self.is_cancelled = False
|
|
|
|
def run(self):
|
|
download_count = 0
|
|
skip_count = 0
|
|
self.progress_signal.emit("=" * 40)
|
|
self.progress_signal.emit(f"🚀 Starting Saint2.su Download for: {self.saint2_url}")
|
|
|
|
album_name, files_to_download = fetch_saint2_data(self.saint2_url, self.progress_signal.emit)
|
|
|
|
if not files_to_download:
|
|
self.progress_signal.emit("❌ Failed to extract file information from Saint2. Aborting.")
|
|
self.finished_signal.emit(0, 0, self.is_cancelled)
|
|
return
|
|
|
|
album_path = os.path.join(self.output_dir, album_name)
|
|
try:
|
|
os.makedirs(album_path, exist_ok=True)
|
|
self.progress_signal.emit(f" Saving to folder: '{album_path}'")
|
|
except OSError as e:
|
|
self.progress_signal.emit(f"❌ Critical error creating directory: {e}")
|
|
self.finished_signal.emit(0, len(files_to_download), self.is_cancelled)
|
|
return
|
|
|
|
total_files = len(files_to_download)
|
|
session = requests.Session()
|
|
|
|
for i, file_data in enumerate(files_to_download):
|
|
if self.is_cancelled:
|
|
self.progress_signal.emit(" Download cancelled by user.")
|
|
skip_count = total_files - download_count
|
|
break
|
|
|
|
filename = file_data.get('filename', f'untitled_{i+1}.mp4')
|
|
file_url = file_data.get('url')
|
|
headers = file_data.get('headers')
|
|
filepath = os.path.join(album_path, filename)
|
|
|
|
if os.path.exists(filepath):
|
|
self.progress_signal.emit(f" -> Skip ({i+1}/{total_files}): '{filename}' already exists.")
|
|
skip_count += 1
|
|
continue
|
|
|
|
self.progress_signal.emit(f" Downloading ({i+1}/{total_files}): '{filename}'...")
|
|
|
|
try:
|
|
response = session.get(file_url, stream=True, headers=headers, timeout=60)
|
|
response.raise_for_status()
|
|
|
|
total_size = int(response.headers.get('content-length', 0))
|
|
downloaded_size = 0
|
|
last_update_time = time.time()
|
|
|
|
with open(filepath, 'wb') as f:
|
|
for chunk in response.iter_content(chunk_size=8192):
|
|
if self.is_cancelled:
|
|
break
|
|
if chunk:
|
|
f.write(chunk)
|
|
downloaded_size += len(chunk)
|
|
current_time = time.time()
|
|
if total_size > 0 and (current_time - last_update_time) > 0.5:
|
|
self.file_progress_signal.emit(filename, (downloaded_size, total_size))
|
|
last_update_time = current_time
|
|
|
|
if self.is_cancelled:
|
|
if os.path.exists(filepath): os.remove(filepath)
|
|
continue
|
|
|
|
if total_size > 0:
|
|
self.file_progress_signal.emit(filename, (total_size, total_size))
|
|
|
|
download_count += 1
|
|
except requests.exceptions.RequestException as e:
|
|
self.progress_signal.emit(f" ❌ Failed to download '{filename}'. Error: {e}")
|
|
if os.path.exists(filepath): os.remove(filepath)
|
|
skip_count += 1
|
|
except Exception as e:
|
|
self.progress_signal.emit(f" ❌ An unexpected error occurred with '{filename}': {e}")
|
|
if os.path.exists(filepath): os.remove(filepath)
|
|
skip_count += 1
|
|
|
|
self.file_progress_signal.emit("", None)
|
|
self.finished_signal.emit(download_count, skip_count, self.is_cancelled)
|
|
|
|
def cancel(self):
|
|
self.is_cancelled = True
|
|
self.progress_signal.emit(" Cancellation signal received by Saint2 thread.") |