import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QLabel, QPushButton, QFileDialog
from PyQt5.QtCore import Qt
from PyPDF2 import PdfFileWriter, PdfFileReader


class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("水印程序")
        self.setGeometry(100, 100, 400, 200)

        self.upload_button = QPushButton("上传文件", self)
        self.upload_button.setGeometry(50, 50, 100, 30)
        self.upload_button.clicked.connect(self.open_file_dialog)

        self.save_label = QLabel("保存路径:", self)
        self.save_label.setGeometry(50, 100, 300, 30)

    def open_file_dialog(self):
        options = QFileDialog.Options()
        file_path, _ = QFileDialog.getOpenFileName(
            self, "选择文件", "", "PDF文件 (*.pdf)", options=options)
        if file_path:
            self.add_watermark(file_path)

    def add_watermark(self, file_path):
        # 添加水印的逻辑
        output_path = file_path.replace(".pdf", "_watermarked.pdf")
        # 这里可以调用添加水印的函数，将file_path的PDF文件打上水印并保存到output_path

        self.save_label.setText(f"保存路径: {output_path}")


if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())