PyQt6实例_批量下载pdf工具_主线程启用线程池

目录

前置：

代码：

视频：

---

前置：

1 本系列将以 “PyQt6实例_批量下载pdf工具”开头，放在 【PyQt6实例】 专栏
2 本系列涉及到的PyQt6知识点：
线程池：QThreadPool,QRunnable；
信号与槽：pyqtSignal,pyqtSlot；
界面：QTextEdit,QLabel,QLineText,QPushButton,QMainWindow,QWidget;
布局：QHBoxLayout,QVBoxLayout;
弹框：QFileDialog,QMessageBox。
3 本系列后续会在B站录制视频，到时会在文末贴出链接。本人还是建议先看博文，不懂的再看视频，这样效率高，节约时间。

代码：

    def execute_btn_clicked(self):txt_dir = self.txtdir_lineedit.text()if txt_dir is None or len(txt_dir.strip())<=0:self.information_dialog('请先选择txt所在目录')returntxt_list = os.listdir(txt_dir)if len(txt_list)<=0:self.information_dialog('txt所在目录为空')returnpdf_dir = self.savedir_lineedit.text()if pdf_dir is None or len(pdf_dir.strip())<=0:self.information_dialog('请设置pdf存储目录')returnanswer = QMessageBox.question(self,'确认启动？',f'如果确定启动，程序将把任务分成 {self.max_thread_count} 个线程执行。执行过程将占用设备资源。',QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No)if answer == QMessageBox.StandardButton.Yes:self.this_time_start_yeah = Trueself.thread_finished_count = 0self.txtdir_lineedit.setDisabled(True)self.savedir_lineedit.setDisabled(True)self.choicedir_btn.setDisabled(True)self.savedir_btn.setDisabled(True)self.execute_btn.setDisabled(True)if self.last_time_executed_tickers is not None:for one in self.last_time_executed_tickers:txt_list.remove(f'{one}.txt')pass# 分发任务interval = len(txt_list)//self.max_thread_countif interval == 0:self.max_thread_count = 1self.insert_executelog('需要执行的内容很少，只开启一个线程')passfor i in range(0,self.max_thread_count):if i == self.max_thread_count-1:node_txt_list = txt_list[i*interval:]else:node_txt_list = txt_list[i*interval:(i+1)*interval]task_data = {'txt_dir':txt_dir,'pdf_dir':pdf_dir,'txt_list':node_txt_list,'temp_dict':self.last_time_data}worker = Worker(i,task_data)worker.signals.result.connect(self.thread_result_fn)worker.signals.finished.connect(self.thread_finished_fn)worker.signals.error.connect(self.thread_error_fn)self.runner_list.append(worker)self.insert_otherlog(f'线程 {i} 启动。')self.threadpool.start(worker)passelse:returnpassdef thread_result_fn(self,res:tuple):# (thread_num,stoped,ticker,executed_url_list,excuted_ticker_list)# (thread_num,success,None,None,excuted_ticker_list)thread_num = res[0]status = res[1]if status == 'stoped':self.insert_otherlog(f'线程 {thread_num} 停止.')if res[2] is not None:self.pre_last_time_data[res[2]] = res[3]self.pre_last_time_executed_tickers.extend(res[4])passelse:self.insert_otherlog(f'线程 {thread_num} 正常结束。')passdef thread_finished_fn(self,res:int):self.thread_finished_count += 1res_str = f'线程 {res} 结束.'self.insert_otherlog(res_str)if self.thread_finished_count == self.max_thread_count:temp_str = '上次执行正常结束'if self.pre_last_time_data:temp_str = '上次被强制停止'with open(os.path.join(basedir,'data','temp.json'),'w',encoding='utf-8') as fw:json.dump(self.pre_last_time_data,fw)passif len(self.pre_last_time_executed_tickers)>0:temp_str = '上次被强制停止'tickers_str = '\n'.join(self.pre_last_time_executed_tickers)with open(os.path.join(basedir,'data','executed.txt'),'w',encoding='utf-8') as fw:fw.write(tickers_str)passpre_str = f"{self.txtdir_lineedit.text()};{self.savedir_lineedit.text()};{temp_str}"with open(os.path.join(basedir,'data','params.txt'),'w',encoding='utf-8') as fw:fw.write(pre_str)if self.waitting_close:self.close()else:self.txtdir_lineedit.setDisabled(False)self.savedir_lineedit.setDisabled(False)self.choicedir_btn.setDisabled(False)self.savedir_btn.setDisabled(False)self.execute_btn.setDisabled(False)self.infomation_dialog('所有工作线程停止完毕')passpassdef thread_error_fn(self,res:tuple):error_str = f"线程 {res[0]} 报错。报错类型：{res[1]}。值：{res[2]}。异常栈：{res[3]}"self.insert_executelog(error_str)pass

视频：

https://www.bilibili.com/video/BV1zeZcYQEax/
https://www.bilibili.com/video/BV1BeZcYQEZq/
https://www.bilibili.com/video/BV1VSZAYJEUf/
https://www.bilibili.com/video/BV1KKZPYBEJV/
https://www.bilibili.com/video/BV1KKZPYBEG2/