412 lines
16 KiB
Python
412 lines
16 KiB
Python
# coding:utf-8
|
||
from PyQt5.QtCore import Qt, pyqtSignal, QObject, QThread, QRegExp
|
||
from PyQt5.QtGui import QIcon, QFont, QTextCursor, QPixmap, QColor,QRegExpValidator
|
||
from PyQt5.QtWidgets import QApplication, QFrame, QGridLayout, QFileDialog
|
||
from qfluentwidgets import (setTheme, Theme, PushSettingCard, SettingCardGroup, ExpandLayout, TextEdit, ImageLabel, LineEdit, PushButton, Theme, ProgressRing, setTheme, Theme, OptionsSettingCard, OptionsConfigItem, OptionsValidator, FluentWindow, SubtitleLabel, NavigationItemPosition, setThemeColor, qconfig, ComboBox, SwitchSettingCard, BoolValidator, MessageBox)
|
||
from qfluentwidgets import FluentIcon as FIF
|
||
import sys
|
||
import base64
|
||
import shutil
|
||
from resource.logo import logo_base64
|
||
from resource.book import book_base64
|
||
from copymanga import *
|
||
from enum import Enum
|
||
|
||
font_label = QFont('微软雅黑', 18)
|
||
font_msg = QFont('微软雅黑', 11)
|
||
|
||
class MainThread(QThread):
|
||
def __init__(self, parent):
|
||
super().__init__()
|
||
self.parent = parent
|
||
|
||
def run(self):
|
||
self.parent.clear_signal.emit('')
|
||
try:
|
||
comic_no = self.parent.editline_book.text()
|
||
chap_no = self.parent.editline_volumn.text()
|
||
downloader_router(self.parent.parent.out_path, comic_no, chap_no, self.parent.parent.url_prev, self.parent.parent.high_quality, True, self.parent.parent.multi_thread, self.parent.hang_signal, self.parent.progressring_signal, self.parent.cover_signal, self.parent.editline_hang)
|
||
self.parent.end_signal.emit('')
|
||
except Exception as e:
|
||
self.parent.end_signal.emit('')
|
||
print('错误，请检查网络情况或确认输入是否正确')
|
||
print('错误信息：')
|
||
print(e)
|
||
def terminate(self) -> None:
|
||
result = super().terminate()
|
||
return result
|
||
|
||
class EmittingStr(QObject):
|
||
textWritten = pyqtSignal(str) # 定义一个发送str的信号
|
||
def write(self, text):
|
||
self.textWritten.emit(str(text))
|
||
def flush(self):
|
||
pass
|
||
def isatty(self):
|
||
pass
|
||
|
||
|
||
class UrlPrev(Enum):
|
||
""" Theme enumeration """
|
||
|
||
SITE = ".site"
|
||
COM= ".com"
|
||
ORG = ".org"
|
||
NET = ".net"
|
||
TV = ".tv"
|
||
INFO = ".info"
|
||
|
||
class SettingWidget(QFrame):
|
||
def __init__(self, text: str, parent=None):
|
||
super().__init__(parent=parent)
|
||
|
||
self.parent = parent
|
||
self.expandLayout = ExpandLayout(self)
|
||
self.setObjectName(text.replace(' ', '-'))
|
||
self.setting_group = SettingCardGroup(self.tr("下载设置"), self)
|
||
|
||
self.download_path_card = PushSettingCard(
|
||
self.tr('选择文件夹'),
|
||
FIF.DOWNLOAD,
|
||
self.tr("下载目录"),
|
||
self.parent.out_path,
|
||
self.setting_group
|
||
)
|
||
self.themeMode = OptionsConfigItem(None, "ThemeMode", Theme.DARK, OptionsValidator(Theme), None)
|
||
self.urlMode = OptionsConfigItem(None, "urlMode", UrlPrev.SITE, OptionsValidator(UrlPrev), None)
|
||
|
||
self.threadMode = OptionsConfigItem(None, "ThreadMode", True, BoolValidator())
|
||
self.qualityMode = OptionsConfigItem(None, "QualityMode", True, BoolValidator())
|
||
|
||
self.theme_card = OptionsSettingCard(
|
||
self.themeMode,
|
||
FIF.BRUSH,
|
||
self.tr('应用主题'),
|
||
self.tr("更改外观"),
|
||
texts=[
|
||
self.tr('亮'), self.tr('暗'),
|
||
self.tr('跟随系统设置')
|
||
],
|
||
parent=self.parent
|
||
)
|
||
|
||
self.url_card = OptionsSettingCard(
|
||
self.urlMode,
|
||
FIF.VPN,
|
||
self.tr('漫画域名后缀'),
|
||
self.tr("漫画域名切换"),
|
||
texts=[
|
||
self.tr('.tv'), self.tr('.org'),
|
||
self.tr('.com'), self.tr('.info'),
|
||
self.tr('.site'), self.tr('.net')
|
||
],
|
||
parent=self.parent
|
||
)
|
||
|
||
self.thread_card = SwitchSettingCard(
|
||
FIF.SPEED_HIGH,
|
||
self.tr('多线程缓存'),
|
||
self.tr('开启后理论上会加快下载速度，但可能会增加服务端压力'),
|
||
parent=self.parent,
|
||
configItem=self.threadMode
|
||
)
|
||
|
||
self.quality_card = SwitchSettingCard(
|
||
FIF.LEAF,
|
||
self.tr('下载高质量图片'),
|
||
self.tr('开启后会提高漫画清晰度，但会增加下载漫画的体积'),
|
||
parent=self.parent,
|
||
configItem=self.qualityMode
|
||
)
|
||
self.thread_card.setValue(True)
|
||
self.quality_card.setValue(True)
|
||
self.thread_changed()
|
||
self.quality_changed()
|
||
|
||
self.setting_group.addSettingCard(self.download_path_card)
|
||
self.setting_group.addSettingCard(self.thread_card)
|
||
self.setting_group.addSettingCard(self.quality_card)
|
||
self.setting_group.addSettingCard(self.url_card)
|
||
self.setting_group.addSettingCard(self.theme_card)
|
||
|
||
self.expandLayout.setSpacing(28)
|
||
self.expandLayout.setContentsMargins(20, 10, 20, 0)
|
||
self.expandLayout.addWidget(self.setting_group)
|
||
|
||
self.download_path_card.clicked.connect(self.download_path_changed)
|
||
self.theme_card.optionChanged.connect(self.theme_changed)
|
||
self.url_card.optionChanged.connect(self.url_changed)
|
||
self.thread_card.checkedChanged.connect(self.thread_changed)
|
||
self.quality_card.checkedChanged.connect(self.quality_changed)
|
||
|
||
def download_path_changed(self):
|
||
""" download folder card clicked slot """
|
||
self.parent.out_path = QFileDialog.getExistingDirectory(
|
||
self, self.tr("Choose folder"), self.parent.out_path)
|
||
self.download_path_card.contentLabel.setText(self.parent.out_path)
|
||
|
||
def theme_changed(self):
|
||
theme_name = self.theme_card.choiceLabel.text()
|
||
self.parent.set_theme(theme_name)
|
||
if os.path.exists('./config'):
|
||
shutil.rmtree('./config')
|
||
|
||
def url_changed(self):
|
||
self.parent.url_prev = self.url_card.choiceLabel.text()
|
||
if os.path.exists('./config'):
|
||
shutil.rmtree('./config')
|
||
|
||
def thread_changed(self):
|
||
is_checked = self.thread_card.isChecked()
|
||
self.thread_card.switchButton.setText(
|
||
self.tr('开') if is_checked else self.tr('关'))
|
||
self.parent.multi_thread = is_checked
|
||
if os.path.exists('./config'):
|
||
shutil.rmtree('./config')
|
||
def quality_changed(self):
|
||
is_checked = self.quality_card.isChecked()
|
||
self.quality_card.switchButton.setText(
|
||
self.tr('高') if is_checked else self.tr('低'))
|
||
self.parent.high_quality = is_checked
|
||
if os.path.exists('./config'):
|
||
shutil.rmtree('./config')
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
class HomeWidget(QFrame):
|
||
|
||
progressring_signal = pyqtSignal(object)
|
||
end_signal = pyqtSignal(object)
|
||
hang_signal = pyqtSignal(object)
|
||
clear_signal = pyqtSignal(object)
|
||
cover_signal = pyqtSignal(object)
|
||
|
||
def __init__(self, text: str, parent=None):
|
||
super().__init__(parent=parent)
|
||
self.setObjectName(text)
|
||
self.parent = parent
|
||
self.label_book = SubtitleLabel('名称：', self)
|
||
self.label_volumn = SubtitleLabel('卷号：', self)
|
||
self.editline_book = LineEdit(self)
|
||
self.editline_volumn = LineEdit(self)
|
||
|
||
|
||
# self.editline_book.setText('yaoyeluying')
|
||
# self.editline_volumn.setText('3')
|
||
|
||
self.book_icon = QPixmap()
|
||
self.book_icon.loadFromData(base64.b64decode(book_base64))
|
||
self.cover_w, self.cover_h = 152, 230
|
||
|
||
self.label_cover = ImageLabel(self.book_icon, self)
|
||
self.label_cover.setBorderRadius(8, 8, 8, 8)
|
||
self.label_cover.setFixedSize(self.cover_w, self.cover_h)
|
||
|
||
self.text_screen = TextEdit()
|
||
self.text_screen.setReadOnly(True)
|
||
self.text_screen.setFixedHeight(self.cover_h)
|
||
|
||
self.progressRing = ProgressRing(self)
|
||
self.progressRing.setValue(0)
|
||
self.progressRing.setTextVisible(True)
|
||
self.progressRing.setFixedSize(50, 50)
|
||
|
||
self.btn_run = PushButton('确定', self)
|
||
self.btn_run.setShortcut(Qt.Key_Return)
|
||
self.btn_stop = PushButton('取消', self)
|
||
self.btn_hang = PushButton('确定', self)
|
||
|
||
self.editline_hang = ComboBox(self)
|
||
self.gridLayout = QGridLayout(self)
|
||
self.screen_layout = QGridLayout()
|
||
self.btn_layout = QGridLayout()
|
||
self.hang_layout = QGridLayout()
|
||
|
||
self.label_book.setFont(font_label)
|
||
self.label_volumn.setFont(font_label)
|
||
self.editline_book.setFont(font_label)
|
||
self.editline_volumn.setFont(font_label)
|
||
self.text_screen.setFont(font_msg)
|
||
self.editline_hang.setFont(font_msg)
|
||
|
||
self.gridLayout.addWidget(self.editline_book, 0, 1)
|
||
self.gridLayout.addWidget(self.editline_volumn, 1, 1)
|
||
self.gridLayout.addWidget(self.label_book, 0, 0)
|
||
self.gridLayout.addWidget(self.label_volumn, 1, 0)
|
||
|
||
self.gridLayout.addLayout(self.btn_layout, 2, 1, 1, 1)
|
||
self.btn_layout.addWidget(self.btn_run, 2, 1)
|
||
self.btn_layout.addWidget(self.btn_stop, 2, 2)
|
||
|
||
self.gridLayout.addLayout(self.screen_layout, 3, 0, 2, 2)
|
||
|
||
self.screen_layout.addWidget(self.progressRing, 0, 0, Qt.AlignRight|Qt.AlignBottom)
|
||
self.screen_layout.addWidget(self.text_screen, 0, 0)
|
||
self.screen_layout.addWidget(self.label_cover, 0, 1)
|
||
|
||
|
||
|
||
self.gridLayout.addLayout(self.hang_layout, 5, 0, 1, 2)
|
||
self.hang_layout.addWidget(self.editline_hang, 0, 0)
|
||
self.hang_layout.addWidget(self.btn_hang, 0, 1)
|
||
|
||
self.screen_layout.setContentsMargins(0,0,0,0)
|
||
self.btn_layout.setContentsMargins(0,0,0,0)
|
||
self.gridLayout.setContentsMargins(20, 10, 20, 10)
|
||
|
||
self.btn_run.clicked.connect(self.process_start)
|
||
self.btn_stop.clicked.connect(self.process_stop)
|
||
self.btn_hang.clicked.connect(self.process_continue)
|
||
|
||
self.progressring_signal.connect(self.progressring_msg)
|
||
self.end_signal.connect(self.process_end)
|
||
self.hang_signal.connect(self.process_hang)
|
||
self.clear_signal.connect(self.clear_screen)
|
||
self.cover_signal.connect(self.display_cover)
|
||
|
||
self.progressRing.hide()
|
||
self.btn_hang.hide()
|
||
self.editline_hang.hide()
|
||
self.btn_stop.setEnabled(False)
|
||
|
||
sys.stdout = EmittingStr(textWritten=self.outputWritten)
|
||
sys.stderr = EmittingStr(textWritten=self.outputWritten)
|
||
self.text_screen.setText(self.parent.welcome_text)
|
||
|
||
def process_start(self):
|
||
self.label_cover.setImage(self.book_icon)
|
||
self.label_cover.setFixedSize(self.cover_w, self.cover_h)
|
||
self.btn_run.setEnabled(False)
|
||
self.btn_run.setText('正在下载')
|
||
self.btn_stop.setEnabled(True)
|
||
self.main_thread = MainThread(self)
|
||
self.main_thread.start()
|
||
|
||
def process_end(self, input=None):
|
||
self.btn_run.setEnabled(True)
|
||
self.btn_run.setText('开始下载')
|
||
self.btn_run.setShortcut(Qt.Key_Return)
|
||
self.btn_stop.setEnabled(False)
|
||
self.progressRing.hide()
|
||
self.btn_hang.hide()
|
||
self.editline_hang.clear()
|
||
self.editline_hang.hide()
|
||
if input=='refresh':
|
||
self.label_cover.setImage(self.book_icon)
|
||
self.label_cover.setFixedSize(self.cover_w, self.cover_h)
|
||
self.clear_signal.emit('')
|
||
self.text_screen.setText(self.parent.welcome_text)
|
||
|
||
def outputWritten(self, text):
|
||
cursor = self.text_screen.textCursor()
|
||
scrollbar=self.text_screen.verticalScrollBar()
|
||
is_bottom = (scrollbar.value()>=scrollbar.maximum() - 15)
|
||
cursor.movePosition(QTextCursor.End)
|
||
cursor.insertText(text)
|
||
if is_bottom:
|
||
self.text_screen.setTextCursor(cursor)
|
||
# self.text_screen.ensureCursorVisible()
|
||
|
||
def clear_screen(self):
|
||
self.text_screen.clear()
|
||
|
||
def display_cover(self, signal_msg):
|
||
filepath, img_h, img_w = signal_msg
|
||
self.label_cover.setImage(filepath)
|
||
self.label_cover.setFixedSize(int(img_w*self.cover_h/img_h), self.cover_h)
|
||
|
||
def progressring_msg(self, input):
|
||
if input == 'start':
|
||
self.progressRing.setValue(0)
|
||
self.progressRing.show()
|
||
elif input == 'end':
|
||
self.progressRing.hide()
|
||
self.progressRing.setValue(0)
|
||
else:
|
||
self.progressRing.setValue(input)
|
||
|
||
def process_hang(self, input=None):
|
||
self.btn_hang.setEnabled(True)
|
||
self.btn_hang.setShortcut(Qt.Key_Return)
|
||
self.btn_hang.show()
|
||
self.editline_hang.show()
|
||
|
||
def process_continue(self, input=None):
|
||
self.btn_hang.hide()
|
||
self.btn_hang.setEnabled(False)
|
||
self.editline_hang.hide()
|
||
|
||
|
||
def process_stop(self):
|
||
self.main_thread.terminate()
|
||
self.end_signal.emit('refresh')
|
||
|
||
|
||
|
||
|
||
class Window(FluentWindow):
|
||
update_signal = pyqtSignal(object)
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
self.out_path = os.path.join(os.path.expanduser('~'), 'Downloads')
|
||
self.url_prev = '.tv'
|
||
self.head = 'https://www.copymanga.tv'
|
||
split_str = '**************************************\n '
|
||
self.welcome_text = f'使用说明（共4条，记得下拉）：\n{split_str}1.拷贝漫画{self.head}，根据书籍网址输入漫画名以及下载的卷号。\n{split_str}2.例如漫画网址是{self.head}/comic/yaoyeluying，则漫画名输入yaoyeluying。\n{split_str}3.要查询漫画卷号卷名等信息，则可以只输入漫画名不输入卷号，点击确定会返回漫画卷名称和对应的卷号。\n{split_str}4.根据上一步返回的信息确定自己想下载的卷号，要下载编号[2]对应卷，则卷号输入2。想下载多卷比如[1]至[3]对应卷，则卷号输入1-3或1,2,3（英文逗号分隔，编号也可以不连续）并点击确定。'
|
||
self.homeInterface = HomeWidget('Home Interface', self)
|
||
self.settingInterface = SettingWidget('Setting Interface', self)
|
||
self.initNavigation()
|
||
self.initWindow()
|
||
self.multi_thread = True
|
||
self.high_quality = True
|
||
|
||
def initNavigation(self):
|
||
self.addSubInterface(self.homeInterface, FIF.HOME, '主界面')
|
||
self.addSubInterface(self.settingInterface, FIF.SETTING, '设置', NavigationItemPosition.BOTTOM)
|
||
|
||
def initWindow(self):
|
||
self.resize(700, 460)
|
||
pixmap = QPixmap()
|
||
pixmap.loadFromData(base64.b64decode(logo_base64))
|
||
self.setWindowIcon(QIcon(pixmap))
|
||
self.setWindowTitle('拷贝漫画下载器')
|
||
self.setFont(font_label)
|
||
|
||
desktop = QApplication.desktop().availableGeometry()
|
||
w, h = desktop.width(), desktop.height()
|
||
self.move(w//2 - self.width()//2, h//2 - self.height()//2)
|
||
|
||
def set_theme(self, mode=None):
|
||
if mode=='亮':
|
||
setTheme(Theme.LIGHT)
|
||
elif mode=='暗':
|
||
setTheme(Theme.DARK)
|
||
elif mode=='跟随系统设置':
|
||
setTheme(Theme.AUTO)
|
||
theme = qconfig.theme
|
||
if theme == Theme.DARK:
|
||
self.homeInterface.label_book.setTextColor(QColor(255,255,255))
|
||
self.homeInterface.label_volumn.setTextColor(QColor(255,255,255))
|
||
elif theme == Theme.LIGHT:
|
||
self.homeInterface.label_book.setTextColor(QColor(0,0,0))
|
||
self.homeInterface.label_volumn.setTextColor(QColor(0,0,0))
|
||
|
||
|
||
|
||
if __name__ == '__main__':
|
||
QApplication.setHighDpiScaleFactorRoundingPolicy(Qt.HighDpiScaleFactorRoundingPolicy.PassThrough)
|
||
QApplication.setAttribute(Qt.AA_EnableHighDpiScaling)
|
||
QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps)
|
||
|
||
setTheme(Theme.DARK)
|
||
setThemeColor('#1E90FF')
|
||
app = QApplication(sys.argv)
|
||
w = Window()
|
||
w.show()
|
||
app.exec_()
|