import base64 import datetime import os import shutil import threading from PyQt5 import QtCore from PyQt5.QtCore import QBasicTimer from PyQt5.QtWidgets import * from PyQt5.QtGui import * import FileOperator import TestFileSystem from MainWindow import Ui_MainWindow from icon import img import TitleSpider global canRun def GetSeries(dataList): return int(dataList.split('_')[1]) class MainApp(QMainWindow, Ui_MainWindow): def __init__(self): global canRun QMainWindow.__init__(self) if not self.SetUI(): canRun = False return else: canRun = True self.setupUi(self) self.SetBaseInfo() self.InitMenuBar() self.HandleButtons() self.setFixedSize(self.width(), self.height()) self.SetLogText() self.SetProgressBar() self.progressBar.setValue(0) self.InitCheckBox() self.path = os.getcwd() self.saveName = "set.ini" self.joinedPath = os.path.join(self.path, self.saveName) # 配置文件的位置 self.isTextFileExists = False self.isTextFirstColumnHaveContent = False self.isTextSecondColumnHaveContent = False self.Log( "欢迎使用本工具，本工具发布于B站@落点dev。\n温馨提示：\n使用程序处理5GB以上的文件夹，出现未响应是正常现象。响应时间和硬盘读写速度挂钩，请耐心等待程序响应即可。\n\n期间你可以打开输出目录以观察处理进度，程序出现其它问题可以b站评论区评论或私信，如需快速回复可联系qq：3152319989") self.InitOutPutPath() # 发布绿色版时注释 def SetProgressBar(self): self.progressBar.setRange(0, 1000) # 1000的原因是防止进度条出现小于0的数 self.progressBar.setValue(0) def InitOutPutPath(self): self.isTextFileExists = FileOperator.SaveForOutput(self.path, self.saveName) if self.isTextFileExists is False: # 如果没有文件，在已经创建文件的前提下，等待用户输入手动填入OutputText. 相关事件在button事件实现 # line = FileOperator.ReadForOutput() # pass else: # 存在文件 则读取文件,并自动赋给文本框 self.lines = FileOperator.ReadForOutput(self.joinedPath) if len(self.lines) >= 2: if self.lines[0] != '': if os.path.isdir(self.lines[0].strip()): self.downloadDirEdit.setText(self.lines[0].strip()) self.isTextFirstColumnHaveContent = True if self.lines[1] != '': if os.path.isdir(self.lines[1].strip()): self.outputDirEdit.setText(self.lines[1].strip()) self.isTextSecondColumnHaveContent = True else: pass # self.Log("检测到配置文件set.ini 为空") # QMessageBox.critical(self, "错误", "配置文件set.ini ！") def InitCheckBox(self): # self.txtFileCheckBox.setChecked(True) # 默认保存Txt # self.deleteFileCheckBox.setChecked(False) # 默认保留源目录 # self.copyToOutput.setChecked(True) # 默认使用复制的方式 self.txtFileCheckBox.setChecked(False) # 默认不保存Txt self.deleteFileCheckBox.setChecked(True) # 默认不保留源目录 self.moveToOutput.setChecked(True) # 默认不使用复制的方式 self.localMode.setChecked(True) def MutiThreadCopy(self, mp4List, outputPath, videoCount): t = threading.Thread(target=self.CopyFile, args=(mp4List, outputPath, videoCount)) t.start() t.join() def MutiThreadMove(self, mp4List, outputPath, videoCount): t = threading.Thread(target=self.MoveFile, args=(mp4List, outputPath, videoCount)) t.start() t.join() def CheckIsChecked(self): # 按下按钮的事件里调用，检查checkbox状态，并提示。最后给对应的bool变量赋值 self.isSaveTxt = self.txtFileCheckBox.isChecked() self.isDeleteDir = self.deleteFileCheckBox.isChecked() if self.copyToOutput.isChecked() or self.moveToOutput.isChecked(): pass else: self.Log("错误：请至少勾选一种输出方式！") return False # QMessageBox.critical(self, "错误", "请至少勾选一种输出方式！") if self.localMode.isChecked() or self.spiderMode.isChecked(): pass else: self.Log("错误：请至少勾选一种处理模式（本地模式 或 爬虫模式）！") return False # QMessageBox.critical(self, "错误", "请至少勾选一种处理模式（本地模式 或 爬虫模式）！") if self.copyToOutput.isChecked(): self.isCopyOutput = True else: self.isCopyOutput = False if self.moveToOutput.isChecked(): self.isMoveOutput = True else: self.isMoveOutput = False if self.localMode.isChecked(): self.isLocalMode = True else: self.isLocalMode = False if self.spiderMode.isChecked(): self.Log("警告：" + "爬虫模式依赖网络，代理模式下程序不会正常运行") # QMessageBox.warning(self, "警告", "爬虫模式依赖网络，关闭本窗口前请确保代理服务是关闭状态") self.isSpiderMode = True else: self.isSpiderMode = False return True def SetLogText(self): self.activityLogEdit.setReadOnly(True) def Log(self, msg): self.statusbar.showMessage(msg) self.activityLogEdit.appendPlainText('[{0}]'.format(str(datetime.datetime.now())[0:19])) self.activityLogEdit.appendPlainText(msg) self.activityLogEdit.appendPlainText('') self.activityLogEdit.moveCursor(QTextCursor.End) def LogOnBar(self, msg): self.statusbar.showMessage(msg) def HandleButtons(self): self.downloadDirButton.clicked.connect(self.OpenDownloadDir) self.outputDirButton.clicked.connect(self.OpenOutputDir) self.renameButton.clicked.connect(self.MutiThreadRenameFile) # 多线程会导致 newMessageBox 的提示无法正常弹出 # self.renameButton.clicked.connect(self.RenameFile) self.copyToOutput.clicked.connect(self.DisableMove) self.moveToOutput.clicked.connect(self.DisableCopy) self.localMode.clicked.connect(self.DisableSpiderMode) self.spiderMode.clicked.connect(self.DisableLocalMode) def MutiThreadRenameFile(self): t = threading.Thread(target=self.RenameFile) t.start() t.join() # 处理checkbox冲突 def DisableCopy(self): if self.copyToOutput.isChecked(): self.copyToOutput.setChecked(False) def DisableMove(self): if self.moveToOutput.isChecked(): self.moveToOutput.setChecked(False) def DisableSpiderMode(self): if self.spiderMode.isChecked(): self.spiderMode.setChecked(False) self.renameButton.setText("一键解密+整理+重命名") def DisableLocalMode(self): if self.localMode.isChecked(): self.localMode.setChecked(False) self.renameButton.setText("一键解密+爬取+整理+重命名") def InitMenuBar(self): # 添加menu“帮助”的事件 aboutAction = QAction('&关于', self) # aboutAction.setStatusTip('关于') aboutAction.triggered.connect(self.ShowAboutDialog) # 已有菜单栏，此处只需要添加菜单 mainPageMenu = self.menubar.addMenu('&主页') helpMenu = self.menubar.addMenu('&帮助') # 菜单绑定之前添加的事件 helpMenu.addAction(aboutAction) # 设置UI def SetUI(self): tmp = None try: tmp = open('tmp.png', "wb+") tmp.write(base64.b64decode(img)) except Exception as e: # self.Log("错误："+"您当前安装在了C盘，软件权限不足：\n解决方案一：请关闭本程序并每次使用管理员权限运行本软件 \n解决方案二：卸载本软件再重新安装至其它非系统盘 比如：D盘、E盘！") QMessageBox.critical(self, "错误", "您当前安装在了C盘，软件权限不足：\n解决方案一：请关闭本程序并每次使用管理员权限运行本软件 \n解决方案二：卸载本软件再重新安装至其它非系统盘 比如：D盘、E盘！") return False finally: if tmp: tmp.close() icon = QIcon('tmp.png') os.remove("tmp.png") self.setWindowIcon(icon) return True def ShowAboutDialog(self): about_text = "

描述：这是一款致力于解决BiliBili UWP版下载后的视频加密、命名信息丢失和存放位置不合理等痛点的软件

版本：4.5

@Author：LZY

@github：love" \ "-in-cpp

" QMessageBox.about(self, '说明', about_text) def OpenDownloadDir(self): if self.isTextFirstColumnHaveContent is False: dName = QFileDialog.getExistingDirectory(self, '选择下载文件夹', '/') self.downloadDirEdit.setText(dName) else: dName = QFileDialog.getExistingDirectory(self, '选择下载文件夹', self.lines[0].strip()) self.downloadDirEdit.setText(dName) def OpenOutputDir(self): if self.isTextSecondColumnHaveContent is False: dName = QFileDialog.getExistingDirectory(self, '选择输出文件夹', '/') self.outputDirEdit.setText(dName) else: dName = QFileDialog.getExistingDirectory(self, '选择输出文件夹', self.lines[1].strip()) self.outputDirEdit.setText(dName) def SetBaseInfo(self): self.setWindowTitle('BiliBili UWP版视频解密整理工具') self.downloadDirEdit.setToolTip(r"例如：E:\BiliDownload\44938322") self.downloadDirEdit.setPlaceholderText("路径请具体到单个数字名称的文件夹，暂不支持文件夹的批量处理") self.outputDirEdit.setPlaceholderText("您希望处理后的文件被保存到的地方") # def FindFiles(self,downloadPath): def RenameFile(self): if self.CheckIsChecked() is False: return self.progressBar.setValue(0) # 进入目录查找dvi文件 downloadPath = self.downloadDirEdit.toPlainText() outputPath = self.outputDirEdit.toPlainText() if os.path.isdir(downloadPath) is False or os.path.isdir(self.downloadDirEdit.toPlainText().strip()) is False: self.Log('UWP下载目录的路径存在非法输入！') else: self.Log("进入目录：{0}".format(downloadPath)) dviInfoList = FileOperator.GetDviInfo(downloadPath) # 获取dvi文件信息 if dviInfoList[0] is False: self.Log( '没有找到.dvi文件！请检查下载目录后重试！请确保使用的是「bilibili uwp客户端」下载的视频而非「桌面客户端」或其它客户端，请谨慎甄别，本工具不能处理其他客户端下载的视频；如果确认使用的是uwp客户端下载的视频，请仔细查看视频 1分20秒 「选择下载目录」的片段') else: # 在outputDir下新建名为dvi[3]文件夹 try: outputPath = FileOperator.MakeDir(outputPath, dviInfoList[3]) except Exception as e: self.Log("错误" + "已经存在同名文件夹！ Error：" + str(e)) # QMessageBox.critical(self, "错误", "已经存在同名文件夹！ Error：" + str(e)) return if self.isSpiderMode: self.Log("开始爬取BV:{0}, 标题:{1} 的所有视频标题,请稍后...".format(dviInfoList[1], dviInfoList[3])) self.progressBar.setValue(10) try: TitleSpider.GetTxt(dviInfoList[1], outputPath) except Exception as e: self.Log("错误" + "请检查网络后重试 Error：" + str(e)) self.progressBar.setValue(0) # QMessageBox.critical(self, "错误", "请检查网络后重试 Error：" + str(e)) return # 调用爬虫产生.txt global fileName fileName = TitleSpider.fileName self.LogOnBar('已成功爬取文件: {0} ！ 注：只显示部分文件名'.format(fileName[0:35])) self.Log('已成功爬取文件: {0} ！'.format(fileName)) self.progressBar.setValue(20) elif self.isLocalMode: self.Log("开始遍历获取BV:{0}, 标题:{1} 的所有视频标题,请稍后...".format(dviInfoList[1], dviInfoList[3])) localVideoTitleList = FileOperator.GetLocalVideoTitle(downloadPath, dviInfoList[2]) fileName = FileOperator.GetTxt(localVideoTitleList, dviInfoList[3], outputPath) self.progressBar.setValue(10) self.Log('已成功获取文件: {0} ！'.format(fileName)) self.progressBar.setValue(20) else: self.Log("impossible") # 找到所有downloadPath的.mp4文件 mp4List = FileOperator.FindSpecialMp4Files(downloadPath, dviInfoList[2])[0] # mp4真正在的地方 # Log mp4nameList = FileOperator.FindSpecialMp4Files(downloadPath, dviInfoList[2])[1] try: mp4nameList.sort(key=GetSeries) except Exception as e: self.Log("错误" + "存在干扰文件！排序错误，请联系作者！" + str(e)) self.progressBar.setValue(0) # QMessageBox.critical(self, "错误", "存在干扰文件！排序错误，请联系作者！" + str(e)) return s = "查询到以下mp4文件：\n" videoCount = 0 for item in mp4nameList: s += (item + '\n') videoCount += 1 self.Log(s) self.progressBar.setValue(30) if os.path.isdir(outputPath) is False or os.path.isdir( self.outputDirEdit.toPlainText().strip()) is False: self.Log('输出目录的路径存在非法输入！') self.progressBar.setValue(0) else: # 记忆输出目录 FileOperator.WriteForOutput(self.joinedPath, os.path.dirname(downloadPath), self.outputDirEdit.toPlainText()) # 发布绿色版时注释 # 解密 self.Log("开始解密...") FileOperator.DecryptMp4(downloadPath, dviInfoList[2]) self.progressBar.setValue(40) self.Log("解密完毕！") self.progressBar.setValue(50) # 复制 todo # self.SetProgressBar(self.progressBar.value(), self.progressBar.value() + videoCount + 2) # 适配进度条 self.CopyOrMove(self.isCopyOutput, mp4List, outputPath, videoCount) # 重命名 self.Log("开始重命名...") self.progressBar.setValue(998) FileOperator.DoRename(outputPath, fileName, dviInfoList[2], self.isLocalMode) self.Log("重命名完毕！") self.progressBar.setValue(1000) # 进度条100％ # self.progressBar.setValue(100) # 是否保存.txt文件 if self.isSaveTxt is True: pass else: self.Log("正在删除程序运行过程中产生的.txt文件") FileOperator.DeleteTxt(outputPath, fileName) self.Log("删除.txt文件成功！") # 是否删除源文件夹 if self.isDeleteDir is True: self.Log("正在删除源文件夹") FileOperator.DeleteDir(downloadPath) self.Log("删除源文件夹成功！") else: pass # 重命名输出文件夹 搁置 def CopyFile(self, srcFileList, dstFolder, videoCount): for file in srcFileList: shutil.copy(file, dstFolder) self.progressBar.setValue(self.progressBar.value() + int(950/videoCount)) def MoveFile(self, srcFileList, dstFolder, videoCount): for file in srcFileList: shutil.move(file, dstFolder) self.progressBar.setValue(self.progressBar.value() + int(950/videoCount)) # 输出方式：复制或移动 def CopyOrMove(self, isCopyTo, mp4List, outputPath, videoCount): if isCopyTo is True: self.Log("进入目录：{0}".format(outputPath)) self.Log("正在复制... 这可能需要一段时间...") self.MutiThreadCopy(mp4List, outputPath, videoCount) # 多线程复制 # self.CopyFile(mp4List, outputPath, videoCount); self.Log("复制完毕！") else: self.Log("进入目录：{0}".format(outputPath)) self.Log("正在移动... 这可能需要一段时间...") self.MutiThreadMove(mp4List, outputPath, videoCount) # 多线程移动 # self.MoveFile(mp4List, outputPath, videoCount) self.Log("移动完毕！") def DSpiderMode(self): pass def DoLocalMode(self): pass if __name__ == '__main__': QtCore.QCoreApplication.setAttribute(QtCore.Qt.AA_EnableHighDpiScaling) app = QApplication([]) window = MainApp() if canRun: window.show() app.exec_()