# 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_()