Commit

src/ui/dialogs/MultipartScopeDialog.py

@@ -0,0 +1,118 @@
+# multipart_scope_dialog.py
+from PyQt5.QtWidgets import (
+    QDialog, QVBoxLayout, QGroupBox, QRadioButton, QDialogButtonBox, QButtonGroup,
+    QLabel, QLineEdit, QHBoxLayout, QFrame
+)
+from PyQt5.QtGui import QIntValidator
+from PyQt5.QtCore import Qt
+
+# It's good practice to get this constant from the source
+# but for this example, we will define it here.
+MAX_PARTS = 16 
+
+class MultipartScopeDialog(QDialog):
+    """
+    A dialog to let the user select the scope, number of parts, and minimum size for multipart downloads.
+    """
+    SCOPE_VIDEOS = 'videos'
+    SCOPE_ARCHIVES = 'archives'
+    SCOPE_BOTH = 'both'
+
+    def __init__(self, current_scope='both', current_parts=4, current_min_size_mb=100, parent=None):
+        super().__init__(parent)
+        self.setWindowTitle("Multipart Download Options")
+        self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint)
+        self.setMinimumWidth(350)
+
+        # Main Layout
+        layout = QVBoxLayout(self)
+
+        # --- Options Group for Scope ---
+        self.options_group_box = QGroupBox("Apply multipart downloads to:")
+        options_layout = QVBoxLayout()
+        # ... (Radio buttons and button group code remains unchanged) ...
+        self.radio_videos = QRadioButton("Videos Only")
+        self.radio_archives = QRadioButton("Archives Only (.zip, .rar, etc.)")
+        self.radio_both = QRadioButton("Both Videos and Archives")
+
+        if current_scope == self.SCOPE_VIDEOS:
+            self.radio_videos.setChecked(True)
+        elif current_scope == self.SCOPE_ARCHIVES:
+            self.radio_archives.setChecked(True)
+        else:
+            self.radio_both.setChecked(True)
+
+        self.button_group = QButtonGroup(self)
+        self.button_group.addButton(self.radio_videos)
+        self.button_group.addButton(self.radio_archives)
+        self.button_group.addButton(self.radio_both)
+
+        options_layout.addWidget(self.radio_videos)
+        options_layout.addWidget(self.radio_archives)
+        options_layout.addWidget(self.radio_both)
+        self.options_group_box.setLayout(options_layout)
+        layout.addWidget(self.options_group_box)
+
+        # --- START: MODIFIED Download Settings Group ---
+        self.settings_group_box = QGroupBox("Download settings:")
+        settings_layout = QVBoxLayout()
+
+        # Layout for Parts count
+        parts_layout = QHBoxLayout()
+        self.parts_label = QLabel("Number of download parts per file:")
+        self.parts_input = QLineEdit(str(current_parts))
+        self.parts_input.setValidator(QIntValidator(2, MAX_PARTS, self))
+        self.parts_input.setFixedWidth(40)
+        self.parts_input.setToolTip(f"Set the number of concurrent connections per file (2-{MAX_PARTS}).")
+        parts_layout.addWidget(self.parts_label)
+        parts_layout.addStretch()
+        parts_layout.addWidget(self.parts_input)
+        settings_layout.addLayout(parts_layout)
+
+        # Layout for Minimum Size
+        size_layout = QHBoxLayout()
+        self.size_label = QLabel("Minimum file size for multipart (MB):")
+        self.size_input = QLineEdit(str(current_min_size_mb))
+        self.size_input.setValidator(QIntValidator(10, 10000, self)) # Min 10MB, Max ~10GB
+        self.size_input.setFixedWidth(40)
+        self.size_input.setToolTip("Files smaller than this will use a normal, single-part download.")
+        size_layout.addWidget(self.size_label)
+        size_layout.addStretch()
+        size_layout.addWidget(self.size_input)
+        settings_layout.addLayout(size_layout)
+
+        self.settings_group_box.setLayout(settings_layout)
+        layout.addWidget(self.settings_group_box)
+        # --- END: MODIFIED Download Settings Group ---
+
+        # OK and Cancel Buttons
+        self.button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
+        self.button_box.accepted.connect(self.accept)
+        self.button_box.rejected.connect(self.reject)
+        layout.addWidget(self.button_box)
+
+        self.setLayout(layout)
+
+    def get_selected_scope(self):
+        # ... (This method remains unchanged) ...
+        if self.radio_videos.isChecked():
+            return self.SCOPE_VIDEOS
+        if self.radio_archives.isChecked():
+            return self.SCOPE_ARCHIVES
+        return self.SCOPE_BOTH
+
+    def get_selected_parts(self):
+        # ... (This method remains unchanged) ...
+        try:
+            parts = int(self.parts_input.text())
+            return max(2, min(parts, MAX_PARTS))
+        except (ValueError, TypeError):
+            return 4
+
+    def get_selected_min_size(self):
+        """Returns the selected minimum size in MB as an integer."""
+        try:
+            size = int(self.size_input.text())
+            return max(10, min(size, 10000)) # Enforce valid range
+        except (ValueError, TypeError):
+            return 100 # Return a safe default

src/ui/dialogs/SinglePDF.py

@@ -3,8 +3,27 @@ import re
 try:
     from fpdf import FPDF
     FPDF_AVAILABLE = True
+
+    # --- FIX: Move the class definition inside the try block ---
+    class PDF(FPDF):
+        """Custom PDF class to handle headers and footers."""
+        def header(self):
+            pass 
+
+        def footer(self):
+            self.set_y(-15)
+            if self.font_family:
+                 self.set_font(self.font_family, '', 8)
+            else:
+                 self.set_font('Arial', '', 8)
+            self.cell(0, 10, 'Page ' + str(self.page_no()), 0, 0, 'C')
+
 except ImportError:
     FPDF_AVAILABLE = False
+    # If the import fails, FPDF and PDF will not be defined,
+    # but the program won't crash here.
+    FPDF = None 
+    PDF = None
 
 def strip_html_tags(text):
     if not text:
@@ -12,19 +31,6 @@ def strip_html_tags(text):
     clean = re.compile('<.*?>')
     return re.sub(clean, '', text)
 
-class PDF(FPDF):
-    """Custom PDF class to handle headers and footers."""
-    def header(self):
-        pass 
-
-    def footer(self):
-        self.set_y(-15)
-        if self.font_family:
-             self.set_font(self.font_family, '', 8)
-        else:
-             self.set_font('Arial', '', 8)
-        self.cell(0, 10, 'Page ' + str(self.page_no()), 0, 0, 'C')
-
 def create_single_pdf_from_content(posts_data, output_filename, font_path, logger=print):
     """
     Creates a single, continuous PDF, correctly formatting both descriptions and comments.
@@ -68,7 +74,7 @@ def create_single_pdf_from_content(posts_data, output_filename, font_path, logge
                 pdf.ln(10)
 
         pdf.set_font(default_font_family, 'B', 16)
-        pdf.multi_cell(w=0, h=10, text=post.get('title', 'Untitled Post'), align='L')
+        pdf.multi_cell(w=0, h=10, txt=post.get('title', 'Untitled Post'), align='L')
         pdf.ln(5)
 
         if 'comments' in post and post['comments']:
@@ -89,7 +95,7 @@ def create_single_pdf_from_content(posts_data, output_filename, font_path, logge
                 pdf.ln(10)
 
                 pdf.set_font(default_font_family, '', 11)
-                pdf.multi_cell(0, 7, body)
+                pdf.multi_cell(w=0, h=7, txt=body)
                 
                 if comment_index < len(comments_list) - 1:
                     pdf.ln(3)
@@ -97,7 +103,7 @@ def create_single_pdf_from_content(posts_data, output_filename, font_path, logge
                     pdf.ln(3)
         elif 'content' in post:
             pdf.set_font(default_font_family, '', 12)
-            pdf.multi_cell(w=0, h=7, text=post.get('content', 'No Content'))
+            pdf.multi_cell(w=0, h=7, txt=post.get('content', 'No Content'))
     
     try:
         pdf.output(output_filename)
@@ -105,4 +111,4 @@ def create_single_pdf_from_content(posts_data, output_filename, font_path, logge
         return True
     except Exception as e:
         logger(f"❌ A critical error occurred while saving the final PDF: {e}")
-        return False
+        return False