#?It?imports?all?the?classes,?attributes,?and?methods?of?the?PyQt5.QtCore?module?into?the?global?symbol?table.
from?PyQt5.QtCore?import?*

#?It?imports?all?the?classes,?attributes,?and?methods?of?the?PyQt5.QtWidgets?module?into?the?global?symbol?table.
from?PyQt5.QtWidgets?import?QApplication,?QWidget,?QHBoxLayout,?QVBoxLayout,?QTextBrowser,?QLineEdit,?QPushButton,?\
????QFormLayout,?QFileDialog,?QLabel

#?It?imports?all?the?classes,?attributes,?and?methods?of?the?PyQt5.QtGui?module?into?the?global?symbol?table.
from?PyQt5.QtGui?import?QIcon,?QFont,?QTextCursor

#?It?imports?the?sys?module.
import?sys

#?It?imports?the?os?module.
import?os

#?It?imports?the?logger?from?the?loguru?module.
from?loguru?import?logger

#?It?imports?the?add_mark?function?from?the?marker?module?in?the?watermarker?package.
from?watermarker.marker?import?add_mark

#?This?class?is?a?widget?that?contains?a?QLabel?and?a?QPushButton
class?PicWaterUI(QWidget):
????def?__init__(self):
????????"""
????????A?constructor.?It?is?called?when?an?object?is?created?from?a?class?and?it?allows?the?class?to?initialize?the
????????attributes?of?a?class.
????????"""
????????super(PicWaterUI,?self).__init__()
????????self.init_ui()

????def?init_ui(self):
????????"""
????????This?function?initializes?the?UI.
????????"""
????????self.setWindowTitle('批量圖片水印添加器??公眾號(hào)：Python 集中營！')
????????self.setWindowIcon(QIcon('water.ico'))

????????self.brower?=?QTextBrowser()
????????self.brower.setFont(QFont('宋體',?8))
????????self.brower.setReadOnly(True)
????????self.brower.setPlaceholderText('處理進(jìn)程展示區(qū)域...')
????????self.brower.ensureCursorVisible()

????????self.pic_file_path?=?QLineEdit()
????????self.pic_file_path.setPlaceholderText('源批量圖片路徑')
????????self.pic_file_path.setReadOnly(True)

????????self.pic_file_path_btn?=?QPushButton()
????????self.pic_file_path_btn.setText('源圖片路徑')
????????self.pic_file_path_btn.clicked.connect(self.pic_file_path_btn_click)

????????self.new_pic_file_path?=?QLineEdit()
????????self.new_pic_file_path.setPlaceholderText('新圖片存儲(chǔ)路徑')
????????self.new_pic_file_path.setReadOnly(True)

????????self.new_pic_file_path_btn?=?QPushButton()
????????self.new_pic_file_path_btn.setText('保存路徑')
????????self.new_pic_file_path_btn.clicked.connect(self.new_pic_file_path_btn_click)

????????self.water_current_label?=?QLabel()
????????self.water_current_label.setText('水印內(nèi)容：')

????????self.water_current_in?=?QLineEdit()
????????self.water_current_in.setPlaceholderText('Python?集中營')

????????self.water_angle_label?=?QLabel()
????????self.water_angle_label.setText('水印角度：')

????????self.water_angle_in?=?QLineEdit()
????????self.water_angle_in.setPlaceholderText('30')

????????self.water_back_label?=?QLabel()
????????self.water_back_label.setText('水印透明度：')

????????self.water_back_in?=?QLineEdit()
????????self.water_back_in.setPlaceholderText('0.3')

????????self.water_font_label?=?QLabel()
????????self.water_font_label.setText('水印字體大?。?)

????????self.water_font_in?=?QLineEdit()
????????self.water_font_in.setPlaceholderText('30')

????????self.water_space_label?=?QLabel()
????????self.water_space_label.setText('水印間隔：')

????????self.water_space_in?=?QLineEdit()
????????self.water_space_in.setPlaceholderText('40')

????????self.water_color_label?=?QLabel()
????????self.water_color_label.setText('水印顏色：')

????????self.water_color_in?=?QLineEdit()
????????self.water_color_in.setPlaceholderText('#8B8B1B')

????????self.start_btn?=?QPushButton()
????????self.start_btn.setText('開始添加水印')
????????self.start_btn.clicked.connect(self.start_btn_click)

????????hbox?=?QHBoxLayout()
????????hbox.addWidget(self.brower)

????????fbox?=?QFormLayout()
????????fbox.addRow(self.pic_file_path,?self.pic_file_path_btn)
????????fbox.addRow(self.new_pic_file_path,?self.new_pic_file_path_btn)
????????fbox.addRow(self.water_current_label,?self.water_current_in)
????????fbox.addRow(self.water_angle_label,?self.water_angle_in)
????????fbox.addRow(self.water_back_label,?self.water_back_in)
????????fbox.addRow(self.water_font_label,?self.water_font_in)
????????fbox.addRow(self.water_space_label,?self.water_space_in)
????????fbox.addRow(self.water_color_label,?self.water_color_in)

????????v_vbox?=?QVBoxLayout()
????????v_vbox.addWidget(self.start_btn)

????????vbox?=?QVBoxLayout()
????????vbox.addLayout(fbox)
????????vbox.addLayout(v_vbox)

????????hbox.addLayout(vbox)

????????self.thread_?=?PicWaterThread(self)
????????self.thread_.message.connect(self.show_message)
????????self.thread_.finished.connect(self.finshed)

????????self.setLayout(hbox)

????def?show_message(self,?text):
????????"""
????????It?shows?a?message

????????:param?text:?The?text?to?be?displayed
????????"""
????????cursor?=?self.brower.textCursor()
????????cursor.movePosition(QTextCursor.End)
????????self.brower.append(text)
????????self.brower.setTextCursor(cursor)
????????self.brower.ensureCursorVisible()

????def?pic_file_path_btn_click(self):
????????"""
????????It?opens?a?file?dialog?box?and?allows?the?user?to?select?a?file.
????????"""
????????pic_file_path?=?QFileDialog.getExistingDirectory(self,?'選擇文件夾',?os.getcwd())
????????self.pic_file_path.setText(pic_file_path)

????def?new_pic_file_path_btn_click(self):
????????"""
????????This?function?opens?a?file?dialog?box?and?allows?the?user?to?select?a?file?to?save?the?output?to.
????????"""
????????new_pic_file_path?=?QFileDialog.getExistingDirectory(self,?'選擇文件夾',?os.getcwd())
????????self.new_pic_file_path.setText(new_pic_file_path)

????def?start_btn_click(self):
????????"""
????????A?function?that?is?called?when?the?start?button?is?clicked.
????????"""
????????self.thread_.start()
????????self.start_btn.setEnabled(False)

????def?finshed(self,?finished):
????????"""
????????:param?finished:?A?boolean?value?that?is?True?if?the?download?is?finished,?False?otherwise
????????"""
????????if?finished?is?True:
????????????self.start_btn.setEnabled(True)

#?This?class?is?a?subclass?of?QThread,?and?it's?used?to?watermark?pictures
class?PicWaterThread(QThread):
????#?A?signal?that?is?emitted?when?a?message?is?received.
????message?=?pyqtSignal(str)
????#?A?signal?that?is?emitted?when?the?download?is?finished.
????finished?=?pyqtSignal(bool)

????def?__init__(self,?parent=None):
????????"""
????????A?constructor?that?initializes?the?class.

????????:param?parent:?The?parent?widget
????????"""
????????super(PicWaterThread,?self).__init__(parent)
????????self.working?=?True
????????self.parent?=?parent

????def?__del__(self):
????????"""
????????A?destructor.?It?is?called?when?the?object?is?destroyed.
????????"""
????????self.working?=?True
????????self.wait()

????def?run(self)?->?None:
????????"""
????????>?This?function?runs?the?program
????????"""
????????try:
????????????directory?=?self.parent.pic_file_path.text().strip()
????????????water_name?=?self.parent.water_current_in.text().strip()
????????????new_directory?=?self.parent.new_pic_file_path.text().strip()
????????????water_angle_in?=?self.parent.water_angle_in.text().strip()
????????????water_back_in?=?self.parent.water_back_in.text().strip()
????????????water_font_in?=?self.parent.water_font_in.text().strip()
????????????water_space_in?=?self.parent.water_space_in.text().strip()
????????????color?=?self.parent.water_color_in.text().strip()

????????????self.message.emit('源文件路徑：{}'.format(directory))
????????????self.message.emit('水印內(nèi)容：{}'.format(water_name))
????????????self.message.emit('保存文件路徑：{}'.format(new_directory))
????????????self.message.emit('水印角度：{}'.format(water_angle_in))
????????????self.message.emit('水印透明度：{}'.format(water_back_in))
????????????self.message.emit('水印字體大?。簕}'.format(water_font_in))
????????????self.message.emit('水印間隔：{}'.format(water_space_in))
????????????self.message.emit('水印顏色：{}'.format(color))

????????????if?directory?is?None?or?water_name?is?None:
????????????????logger.info('文件夾地址或水印名稱不能為空！')
????????????????return
????????????for?file_name?in?os.listdir(directory):
????????????????logger.info('當(dāng)前文件名稱：{0}，即將開始添加水印操作！'.format(file_name))
????????????????self.message.emit('當(dāng)前文件名稱：{0}，即將開始添加水印操作！'.format(file_name))
????????????????add_mark(file=os.path.join(directory,?file_name),?out=new_directory,?mark=water_name,
?????????????????????????opacity=float(water_back_in),?angle=int(water_angle_in),?space=int(water_space_in),
?????????????????????????size=int(water_font_in),?color=color)
????????????????self.message.emit('當(dāng)前文件名稱：{0}，已經(jīng)完成添加水印操作！'.format(file_name))
????????????????logger.info('當(dāng)前文件名稱：{0}，已經(jīng)完成添加水印操作！'.format(file_name))
????????????self.finished.emit(True)
????????except?Exception?as?e:
????????????self.message.emit('文件內(nèi)容讀取或格式化發(fā)生異常！')
????????????self.finished.emit(True)

if?__name__?==?'__main__':
????app?=?QApplication(sys.argv)
????main?=?PicWaterUI()
????main.show()
????sys.exit(app.exec_())