Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Binary file modified app/ui/pdf_tools/__pycache__/pdf_tool.cpython-313.pyc
Binary file not shown.
20 changes: 19 additions & 1 deletion app/ui/pdf_tools/pdf_tool.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
from app.ui.pdf_tools.pdf_tool_ui import Ui_Form
from app.ui.components.router import Router
from app.ui.pdf_tools.blank_pages.blank_pages import BlankPagesControl
from app.ui.pdf_tools.watermark.watermark import WatermarkControl


class PDFToolControl(QWidget, Ui_Form, QCursorGif):
Expand Down Expand Up @@ -36,7 +37,7 @@ def __init__(self, router: Router, parent=None):
self.commandLinkButton_encrypt.clicked.connect(self.encrypt_pdf)
self.commandLinkButton_decrypt.clicked.connect(self.decrypt_pdf)
self.commandLinkButton_delete_blank_pages.clicked.connect(self.delete_blank_pages)
self.commandLinkButton_add_watermark.clicked.connect(globalSignals.not_support)
self.commandLinkButton_add_watermark.clicked.connect(self.add_watermark)

def init_ui(self):
if not self.parent():
Expand Down Expand Up @@ -110,6 +111,17 @@ def delete_blank_pages(self):
else:
self.router.navigate(self.blank_pages_view.router_path)

def add_watermark(self):
if not hasattr(self, 'watermark_view') or not self.watermark_view:
self.watermark_view = WatermarkControl(router=self.router, parent=self if self.parent() else None)
self.watermark_view.okSignal.connect(self.watermark_finish)
self.router.add_route(self.watermark_view.router_path, self.watermark_view)
self.child_routes[self.watermark_view.router_path] = 0
self.childRouterSignal.emit(self.watermark_view.router_path)
self.router.navigate(self.watermark_view.router_path)
else:
self.router.navigate(self.watermark_view.router_path)

def merge_finish(self, s=None):
if self.merge_view and self.merge_view.router_path in self.child_routes:
self.child_routes.pop(self.merge_view.router_path)
Expand Down Expand Up @@ -140,6 +152,12 @@ def blank_pages_finish(self, s=None):
self.router.remove_route(self.blank_pages_view.router_path)
self.blank_pages_view = None

def watermark_finish(self, s=None):
if hasattr(self, 'watermark_view') and self.watermark_view and self.watermark_view.router_path in self.child_routes:
self.child_routes.pop(self.watermark_view.router_path)
self.router.remove_route(self.watermark_view.router_path)
self.watermark_view = None

def __del__(self):
self.merge_view = None

Expand Down
Loading