def 网盘检查(self): try: lzy = LanZouCloud() fileURL = '' folder_info = lzy.get_folder_info_by_url( 'https://pan.lanzous.com/b01bfj76f') if folder_info.code != LanZouCloud.SUCCESS: self.网盘链接 = '' self.网盘报错 = 1 return # resp = urllib.request.urlopen('http://dnf.17173.com/jsq/instructions.html?j') for file in folder_info.files: if file.name.startswith("DNF计算器"): self.云端版本 = file.name.replace(".zip", " 17173DNF.exe") fileURL = file.url if file.name.replace("DNF计算器", "").replace(".zip", "").replace( "-", ".") == self.计算器版本[5:]: self.网盘链接 = '' return self.网盘链接 = fileURL except Exception as error: self.网盘链接 = '' self.网盘报错 = 1 return
def set_values(self, text): '''获取分享链接信息''' if not text: return for share_url, _, pwd in re.findall(self._pat, text): if LanZouCloud.is_file_url(share_url): # 文件链接 is_file = True is_folder = False self.msg.emit("正在获取文件链接信息……", 0) elif LanZouCloud.is_folder_url(share_url): # 文件夹链接 is_folder = True is_file = False self.msg.emit("正在获取文件夹链接信息,可能需要几秒钟,请稍后……", 0) else: self.msg.emit(f"{share_url} 为非法链接!", 0) self.btn_extract.setEnabled(True) self.line_share_url.setEnabled(True) return self.update.emit() # 清理旧的显示信息 self.share_url = share_url self.pwd = pwd self.is_file = is_file self.is_folder = is_folder self.start() break
class GetMoreInfoWorker(QThread): '''获取文件直链、文件(夹)提取码描述,用于登录后显示更多信息''' infos = pyqtSignal(object) msg = pyqtSignal(str, int) def __init__(self, parent=None): super(GetMoreInfoWorker, self).__init__(parent) self._disk = LanZouCloud() self.old_infos = None self._mutex = QMutex() self._is_work = False def set_values(self, infos, disk=None): if disk: # 登录情况 self._disk = disk self.old_infos = infos self.start() def __del__(self): self.wait() def stop(self): self._mutex.lock() self._is_work = False self._mutex.unlock() def run(self): # infos: ID/None,文件名,大小,日期,下载次数(dl_count),提取码(pwd),描述(desc),|链接(share-url),直链 if not self._is_work and self.old_infos: self._mutex.lock() self._is_work = True try: self.msg.emit("网络请求中,请稍后……", 0) if self.old_infos[0]: # 从 disk 运行 if self.old_infos[2]: # 文件 _info = self._disk.get_share_info(self.old_infos[0], is_file=True) else: # 文件夹 _info = self._disk.get_share_info(self.old_infos[0], is_file=False) self.old_infos[5] = _info['pwd'] self.old_infos.append(_info['url']) if self.old_infos[2]: # 是文件,解析下载直链 res = self._disk.get_file_info_by_url(self.old_infos[-1], self.old_infos[5]) if res["code"] == LanZouCloud.SUCCESS: self.old_infos.append("{}".format(res["durl"] or "无")) # 下载直链 elif res["code"] == LanZouCloud.NETWORK_ERROR: self.old_infos.append("网络错误!获取失败") # 下载直链 else: self.old_infos.append("其它错误!") # 下载直链 else: self.old_infos.append("无") # 下载直链 self.old_infos[5] = self.old_infos[5] or "无" # 提取码 self.infos.emit(self.old_infos) self.msg.emit("", 0) # 删除提示信息 except TimeoutError: self.msg.emit("网络超时!稍后重试", 6000) self._is_work = False self._mutex.unlock() else: self.msg.emit("后台正在运行,请稍后重试!", 2000)
class Downloader(QThread): '''单个文件下载线程''' download_proc = pyqtSignal(str) download_failed = pyqtSignal(str) def __init__(self, parent=None): super(Downloader, self).__init__(parent) self._stopped = True self._mutex = QMutex() self._disk = LanZouCloud() self.name = "" self.url = "" self.pwd = "" self.save_path = "" # if os.name == 'nt': # self._disk.set_rar_tool("./rar.exe") # else: # self._disk.set_rar_tool("/usr/bin/rar") def stop(self): self._mutex.lock() self._stopped = True self._mutex.unlock() def _show_progress(self, file_name, total_size, now_size): """显示进度条的回调函数""" msg = show_progress(file_name, total_size, now_size) self.download_proc.emit(msg) def _show_down_failed(self, code, file): """显示下载失败的回调函数""" msg = show_down_failed(code, file) self.download_failed.emit(msg) def __del__(self): self.wait() def set_values(self, name, url, pwd, save_path): self.name = name self.url = url self.pwd = pwd self.save_path = save_path def run(self): if is_file_url(self.url): # 下载文件 self._disk.down_file_by_url(self.url, self.pwd, self.save_path, self._show_progress) elif is_folder_url(self.url): # 下载文件夹 folder_path = self.save_path + os.sep + self.name os.makedirs(folder_path, exist_ok=True) self.save_path = folder_path self._disk.down_dir_by_url(self.url, self.pwd, self.save_path, self._show_progress, mkdir=True, failed_callback=self._show_down_failed)
def __init__(self, parent=None): super(GetMoreInfoWorker, self).__init__(parent) self._disk = LanZouCloud() self.emit_infos = None self._url = '' self._pwd = '' self._mutex = QMutex() self._is_work = False
def 自动更新(self, fileURL): path = os.getcwd() + "/download" lzy = LanZouCloud() lzy.down_file_by_url(fileURL, '', path, callback=self.show_progress, downloaded_handler=self.after_downloaded)
def __init__(self, parent=None): super(Downloader, self).__init__(parent) self._stopped = True self._mutex = QMutex() self._disk = LanZouCloud() self.name = "" self.url = "" self.pwd = "" self.save_path = ""
def __init__(self, parent=None): super(GetSharedInfo, self).__init__(parent) self._disk = LanZouCloud() self.share_url = "" self.pwd = "" self.is_file = "" self.is_folder = "" self._mutex = QMutex() self._is_work = False self._pat = r"(https?://(www\.)?lanzous.com/[bi][a-z0-9]+)[^0-9a-z]*([a-z0-9]+)?"
def 网盘检查(self): lzy = LanZouCloud() fileURL = '' folder_info = lzy.get_folder_info_by_url('https://wws.lanzous.com/b01bfj76f') for file in folder_info.files: if file.name.startswith("DNF计算器"): self.云端版本 = file.name.replace(".zip",".exe") fileURL = file.url if file.name.replace("DNF计算器","").replace(".zip","").replace("-",".") == self.计算器版本[5:]: return '' return fileURL
def run(self): path = os.getcwd() + "/download" try: lzy = LanZouCloud() lzy.down_file_by_url(self.fileURL, '', path, callback=self.show_progress, downloaded_handler=self.after_downloaded) except Exception as error: self.sinOut.emit(100)
def init_variables(self): self._disk = LanZouCloud() self._config_file = "./config.pkl" self._folder_list = {} self._file_list = {} self._path_list = {} self._path_list_old = {} self._locs = {} self._parent_id = -1 # --> .. self._work_name = "" # share disk rec, not use now self._work_id = -1 # disk folder id self._old_work_id = self._work_id # 用于上传完成后判断是否需要更新disk界面 self.load_settings()
def login(self): lzy = LanZouCloud() if self.get_cookie(): cookie = {'ylogin': self.ylogin, 'phpdisk_info': self.phpdisk_info} if lzy.login_by_cookie(cookie) == LanZouCloud.SUCCESS: print('与云服务器连接成功...') return lzy else: print('你可能输入了错误的cookie地址') return False else: print(登录失败) return False
def __init__(self, parent=None): super(Downloader, self).__init__(parent) self._stopped = True self._mutex = QMutex() self._disk = LanZouCloud() self.name = "" self.url = "" self.pwd = "" self.save_path = "" if os.name == 'nt': self._disk.set_rar_tool("./rar.exe") else: self._disk.set_rar_tool("/usr/bin/rar")
def init_variable(self): self._disk = LanZouCloud() self._config = "./config.pkl" self._folder_list = {} self._file_list = {} self._path_list = {} self._path_list_old = {} self._locs = {} self._parent_id = -1 # --> .. self._work_name = "" # share disk rec, not use now self._work_id = -1 # disk folder id self._old_work_id = self._work_id # 用于上传完成后判断是否需要更新disk界面 self.load_settings() if os.name == 'nt': self._disk.set_rar_tool("./rar.exe") else: self._disk.set_rar_tool("/usr/bin/rar") # 登录器 self.login_luncher = LoginLuncher(self._disk) self.login_luncher.code.connect(self.login_update_ui) # 下载器 self.download_manager = DownloadManager(self._disk) self.download_manager.downloaders_msg.connect(self.show_status) self.download_manager.download_mgr_msg.connect(self.show_status) self.download_manager.finished.connect( lambda: self.show_status("所有下载任务已完成!", 7000)) # 上传器,信号在登录更新界面设置 self.upload_dialog = UploadDialog() self.upload_dialog.new_infos.connect(self.call_upload) # 文件描述更新器 self.desc_fetcher = DescFetcher(self._disk) self.desc_fetcher.desc.connect(self.call_update_desc) # 设置 tab self.tabWidget.setCurrentIndex(0) self.tabWidget.removeTab(2) self.tabWidget.removeTab(1) self.disk_tab.setEnabled(False) self.rec_tab.setEnabled(False) # 状态栏 self._msg_label = QLabel() self._msg_label.setObjectName("msg_label") self.statusbar.addWidget(self._msg_label) # 重命名、修改简介与新建文件夹对话框 self.rename_dialog = RenameDialog() self.rename_dialog.out.connect(self.rename_set_desc_and_mkdir) # 菜单栏关于 self.about_dialog = AboutDialog() self.about_dialog.set_values(self.__version__)
def show_share_url_file_lists(self, infos): if infos["code"] == LanZouCloud.SUCCESS: file_count = len(infos["info"].keys()) self.model_share.setHorizontalHeaderLabels( ["文件{}个".format(file_count), "大小", "时间"]) for infos in infos["info"].values(): name = QStandardItem(self.set_file_icon(infos[1]), infos[1]) name.setData(infos) time = QStandardItem(LanZouCloud.time_format( infos[3])) if self.time_fmt else QStandardItem(infos[3]) self.model_share.appendRow( [name, QStandardItem(infos[2]), time]) for r in range(self.model_share.rowCount()): # 右对齐 self.model_share.item(r, 1).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.model_share.item(r, 2).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.table_share.setDisabled(False) self.btn_share_select_all.setDisabled(False) self.btn_share_select_all.setToolTip("按下 Ctrl/Alt + A 全选或则取消全选") self.btn_share_dl.setDisabled(False) else: self.btn_share_select_all.setDisabled(True) self.btn_share_select_all.setToolTip("") self.btn_share_dl.setDisabled(True) self.table_share.setDisabled(True)
def __init__(self): self.lzy = LanZouCloud() self.lzy._host_url = 'https://pan.lanzouo.com' self.lzy.available_domains = [ 'lanzouo.com', # 2021-09-15 鲁ICP备15001327号-8 'lanzouw.com', # 2021-09-02 鲁ICP备15001327号-7 'lanzoui.com', # 2020-06-09 鲁ICP备15001327号-6 'lanzoux.com', # 2020-06-09 鲁ICP备15001327号-5 ]
def init_variable(self): self._disk = LanZouCloud() self._config = "./config.pkl" self._folder_list = {} self._file_list = {} self._path_list = {} self._path_list_old = {} self._locs = {} self._parent_id = -1 # --> .. self._work_name = "" # share disk rec, not use now self._work_id = -1 # disk folder id self._old_work_id = self._work_id # 用于上传完成后判断是否需要更新disk界面 self.download_threads = 3 # 同时三个下载任务 self.load_settings() if os.name == 'nt': self._disk.set_rar_tool("./rar.exe") else: self._disk.set_rar_tool("/usr/bin/rar")
def show_file_and_folder_lists(self): """显示文件和文件夹列表""" self.model_disk.removeRows(0, self.model_disk.rowCount()) # 清理旧的内容 file_count = len(self._file_list.keys()) folder_count = len(self._folder_list.keys()) name_header = [ "文件夹{}个".format(folder_count), ] if folder_count else [] if file_count: name_header.append("文件{}个".format(file_count)) self.model_disk.setHorizontalHeaderLabels( ["/".join(name_header), "大小", "时间"]) folder_ico = QIcon("./icon/folder.gif") pwd_ico = QIcon("./icon/keys.ico") # infos: ID/None,文件名,大小,日期,下载次数(dl_count),提取码(pwd),描述(desc),|链接(share-url),直链 if self._work_id != -1: _back = QStandardItem(folder_ico, "..") _back.setToolTip("双击返回上层文件夹,选中无效") self.model_disk.appendRow( [_back, QStandardItem(""), QStandardItem("")]) for infos in self._folder_list.values(): # 文件夹 name = QStandardItem(folder_ico, infos[1]) name.setData(infos) tips = "" if infos[5] is not False: tips = "有提取码" if infos[6] is not False: tips = tips + ",描述:" + str(infos[6]) elif infos[6] is not False: tips = "描述:" + str(infos[6]) name.setToolTip(tips) size_ = QStandardItem(pwd_ico, "") if infos[5] else QStandardItem( "") # 提取码+size self.model_disk.appendRow([name, size_, QStandardItem("")]) for infos in self._file_list.values(): # 文件 name = QStandardItem(self.set_file_icon(infos[1]), infos[1]) name.setData(infos) tips = "" if infos[5] is not False: tips = "有提取码" if infos[6] is not False: tips = tips + ",描述:" + str(infos[6]) elif infos[6] is not False: tips = "描述:" + str(infos[6]) name.setToolTip(tips) size_ = QStandardItem(pwd_ico, infos[2]) if infos[5] else QStandardItem( infos[2]) # 提取码+size time_ = QStandardItem(LanZouCloud.time_format( infos[3])) if self.time_fmt else QStandardItem(infos[3]) self.model_disk.appendRow([name, size_, time_]) for row in range(self.model_disk.rowCount()): # 右对齐 self.model_disk.item(row, 1).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.model_disk.item(row, 2).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter)
class Uploader: folder_dnf_calc = Folder("魔改计算器", "1810329") folder_djc_helper = Folder("蚊子腿小助手", "2290618") folder_history_files = Folder("历史版本", "2303716") history_version_prefix = "DNF蚊子腿小助手_v" def __init__(self, cookie): self.lzy = LanZouCloud() self.login_ok = self.lzy.login_by_cookie(cookie) == LanZouCloud.SUCCESS def upload_to_lanzouyun(self, filepath, target_folder): logger.warning("开始上传 {} 到 {}".format(os.path.basename(filepath), target_folder.name)) run_start_time = datetime.now() def on_uploaded(fid, is_file): if not is_file: return logger.info("下载完成,fid={}".format(fid)) files = self.lzy.get_file_list(target_folder.id) for file in files: if file.name.startswith(self.history_version_prefix): self.lzy.move_file(file.id, self.folder_history_files.id) logger.info("将{}移动到目录({})".format( file.name, self.folder_history_files.name)) logger.info("将文件移到目录({})中".format(target_folder.name)) self.lzy.move_file(fid, target_folder.id) # 上传到指定的文件夹中 retCode = self.lzy.upload_file(filepath, -1, callback=self.show_progress, uploaded_handler=on_uploaded) if retCode != LanZouCloud.SUCCESS: logger.error("上传失败,retCode={}".format(retCode)) return False logger.warning("上传当前文件总计耗时{}".format(datetime.now() - run_start_time)) return True def show_progress(self, file_name, total_size, now_size): """显示进度的回调函数""" percent = now_size / total_size bar_len = 40 # 进度条长总度 bar_str = '>' * round(bar_len * percent) + '=' * round(bar_len * (1 - percent)) print('\r{:.2f}%\t[{}] {:.1f}/{:.1f}MB | {} '.format( percent * 100, bar_str, now_size / 1048576, total_size / 1048576, file_name), end='') if total_size == now_size: print('') # 下载完成换行
def download(self): start = time.time() # 下载文件名 # print(os.path.split(self.api_file_url)) # print(os.path.basename(self.api_file_url)) # 获取文件名 e6a59548da20bcf3.mp4?sign=05e6532286c697b10dbcac6761dcac83&t=5bee2e2d # basename = self.getFile(self.api_file_url) # 获取?号的位置 # pos = basename.find('?') # 获取文件名 # file_name = basename[0:pos] # print('='*30 +'正在下载:'+file_name + '='*30) # file_name = os.getcwd() + '/' + basename # print(file_name) temp_size = 0 # 已经下载文件大小 res = requests.get(self.api_file_url, headers=self.headers) base_name = self.get_file_name(self.api_file_url, res) file_name = os.getcwd() + "/" + base_name chunk_size = 1024 # 每次下载数据大小 # total_size = int(res.headers.get("Content-Length")) print(file_name) if res.status_code == 200: # print('[文件大小]:%0.2f MB' %(total_size / chunk_size /1024)) #换算单位并打印 # 保存下载文件 with open(file_name, 'wb') as f: for chunk in res.iter_content(chunk_size=chunk_size): if chunk: temp_size += len(chunk) f.write(chunk) f.flush() # ############花哨的下载进度部分############## # # done = int(50 * temp_size / total_size) # 调用标准输出刷新命令行,看到\r 回车符了吧 # 相当于把每一行重新刷新 print() # 避免上面\r 回车符,执行完后需要换行了,不然都在一行显示 end = time.time() # 结束时间 print('全部下载完成!用时%.2f 秒' % (end - start)) print("文件大小:%.2f MB" % (temp_size / 1024 / 1024)) lzy = LanZouCloud() cookie2 = eval(self.cookie) print(lzy.login_by_cookie(cookie2)) lzy.ignore_limits() code = lzy.upload_file(file_name, -1, callback=None, uploaded_handler=None) print("上传状态:") print(code) # self.upload2lanzous(file_name,file_name,self.cookie) else: print(res.status_code)
import sys import os import requests import time import json from urllib import parse from lanzou.api import LanZouCloud lzy = LanZouCloud() cookie = json.loads(os.getenv('LANZOU_COOKIE')) server_key = os.getenv('SERVER_KEY') if lzy.login_by_cookie(cookie) != LanZouCloud.SUCCESS: url = 'https://sc.ftqq.com/' + server_key + \ '.send?text=' + parse.quote('蓝奏云Cookie失效') + '&desp=' + \ time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) res = requests.get(url) sys.exit() def show_progress(file_name, total_size, now_size): """显示进度的回调函数""" percent = now_size / total_size bar_len = 40 # 进度条长总度 bar_str = '>' * round(bar_len * percent) + '=' * round(bar_len * (1 - percent)) print('\r{:.2f}%\t[{}] {:.1f}/{:.1f}MB | {} '.format( percent * 100, bar_str, now_size / 1048576, total_size / 1048576, file_name), end='') if total_size == now_size:
class Uploader: folder_dnf_calc = Folder("魔改计算器", "1810329") folder_djc_helper = Folder("蚊子腿小助手", "2290618") folder_history_files = Folder("历史版本", "2303716") folder_online_files = Folder("在线文件存储", "2866929") folder_online_files_history_files = Folder("历史版本", "2867307") history_version_prefix = "DNF蚊子腿小助手_v" history_patches_prefix = "DNF蚊子腿小助手_增量更新文件_" regex_version = r'DNF蚊子腿小助手_v(.+)_by风之凌殇.7z' regex_patches = r'DNF蚊子腿小助手_增量更新文件_v(.+)_to_v(.+).7z' # 保存购买了自动更新工具的用户信息 buy_auto_updater_users_filename = "buy_auto_updater_users.txt" # 保存用户的付费信息 user_monthly_pay_info_filename = "user_monthly_pay_info.txt" # 卡密操作的付费信息 cs_buy_auto_updater_users_filename = "cs_buy_auto_updater_users.txt" cs_user_monthly_pay_info_filename = "cs_user_monthly_pay_info.txt" def __init__(self, cookie): self.lzy = LanZouCloud() self.login_ok = self.lzy.login_by_cookie(cookie) == LanZouCloud.SUCCESS def upload_to_lanzouyun(self, filepath, target_folder, history_file_prefix=""): logger.warning( f"开始上传 {os.path.basename(filepath)} 到 {target_folder.name}") run_start_time = datetime.now() def on_uploaded(fid, is_file): if not is_file: return logger.info(f"上传完成,fid={fid}") prefix = history_file_prefix if prefix == "": prefix = self.history_version_prefix folder_history_files = self.folder_history_files if target_folder.id == self.folder_online_files.id: folder_history_files = self.folder_online_files_history_files files = self.lzy.get_file_list(target_folder.id) for file in files: if file.name.startswith(prefix): self.lzy.move_file(file.id, folder_history_files.id) logger.info( f"将{file.name}移动到目录({folder_history_files.name})") logger.info(f"将文件移到目录({target_folder.name})中") self.lzy.move_file(fid, target_folder.id) # 上传到指定的文件夹中 retCode = self.lzy.upload_file(filepath, -1, callback=self.show_progress, uploaded_handler=on_uploaded) if retCode != LanZouCloud.SUCCESS: logger.error(f"上传失败,retCode={retCode}") return False logger.warning(f"上传当前文件总计耗时{datetime.now() - run_start_time}") return True def latest_version(self) -> str: """ 返回形如"1.0.0"的最新版本信息 """ latest_version_file = self.find_latest_version() # DNF蚊子腿小助手_v4.6.6_by风之凌殇.7z match = re.search(self.regex_version, latest_version_file.name) if match is not None: latest_version = match.group(1) return latest_version # 保底返回1.0.0 return "1.0.0" def download_latest_version(self, download_dir) -> str: """ 下载最新版本压缩包到指定目录,并返回最终压缩包的完整路径 """ return self.download_file(self.find_latest_version(), download_dir) def find_latest_version(self): """ 查找最新版本,如找到,返回lanzouyun提供的file信息,否则抛出异常 """ files = self.lzy.get_file_list(self.folder_djc_helper.id) for file in files: if file.name.startswith(self.history_version_prefix): return file raise FileNotFoundError("latest version not found") def latest_patches_range(self): """ 返回形如("1.0.0", "1.1.2")的补丁范围 """ latest_patches_file = self.find_latest_patches() # DNF蚊子腿小助手_增量更新文件_v4.6.5_to_v4.6.6.7z match = re.search(self.regex_patches, latest_patches_file.name) if match is not None: version_left, version_right = match.group(1), match.group(2) return (version_left, version_right) # 保底返回 return ("1.0.0", "1.0.0") def download_latest_patches(self, download_dir) -> str: """ 下载最新版本压缩包到指定目录,并返回最终压缩包的完整路径 """ return self.download_file(self.find_latest_patches(), download_dir) def find_latest_patches(self): """ 查找最新版本的补丁,如找到,返回lanzouyun提供的file信息,否则抛出异常 """ files = self.lzy.get_file_list(self.folder_djc_helper.id) for file in files: if file.name.startswith(self.history_patches_prefix): return file raise FileNotFoundError("latest patches not found") def download_file_in_folder(self, folder, name, download_dir, overwrite=True, show_log=True) -> str: """ 下载网盘指定文件夹的指定文件到本地指定目录,并返回最终本地文件的完整路径 """ return self.download_file(self.find_file(folder, name), download_dir, overwrite=overwrite, show_log=show_log) def find_file(self, folder, name): """ 在对应目录查找指定名称的文件,如找到,返回lanzouyun提供的file信息,否则抛出异常 """ files = self.lzy.get_file_list(folder.id) for file in files: if file.name == name: return file raise FileNotFoundError( f"file={name} not found in folder={folder.name}") def download_file(self, fileinfo, download_dir, overwrite=True, show_log=True) -> str: """ 下载最新版本压缩包到指定目录,并返回最终压缩包的完整路径 """ if not os.path.isdir(download_dir): os.mkdir(download_dir) download_dir = os.path.realpath(download_dir) target_path = os.path.join(download_dir, fileinfo.name) def after_downloaded(file_name): """下载完成后的回调函数""" target_path = file_name if show_log: logger.info(f"最终下载文件路径为 {file_name}") if show_log: logger.info(f"即将开始下载 {target_path}") callback = None if show_log: callback = self.show_progress retCode = self.lzy.down_file_by_id(fileinfo.id, download_dir, callback=callback, downloaded_handler=after_downloaded, overwrite=overwrite) if retCode != LanZouCloud.SUCCESS: if show_log: logger.error(f"下载失败,retCode={retCode}") if retCode == LanZouCloud.NETWORK_ERROR: if show_log: logger.warning( color("bold_yellow") + ("蓝奏云api返回网络错误,这很可能是由于dns的问题导致的\n" "分别尝试在浏览器中访问下列两个网页,是否一个打的开一个打不开?\n" "https://fzls.lanzoux.com/s/djc-helper\n" "https://fzls.lanzous.com/s/djc-helper\n" "\n" "如果是这样,请按照下面这个链接,修改本机的dns,使用阿里、腾讯、百度、谷歌dns中的任意一个应该都可以解决。\n" "https://www.ypojie.com/9830.html\n" "\n" "如果两个都打不开,大概率是蓝奏云挂了-。-可选择忽略后面的弹框,继续运行旧版本,或者手动去QQ群或github下载最新版本" )) raise Exception("下载失败") return target_path def show_progress(self, file_name, total_size, now_size): """显示进度的回调函数""" percent = now_size / total_size bar_len = 40 # 进度条长总度 bar_str = '>' * round(bar_len * percent) + '=' * round(bar_len * (1 - percent)) show_percent = percent * 100 now_mb = now_size / 1048576 total_mb = total_size / 1048576 print( f'\r{show_percent:.2f}%\t[{bar_str}] {now_mb:.2f}/{total_mb:.2f}MB | {file_name} ', end='') if total_size == now_size: print('') # 下载完成换行
class MainWindow(QMainWindow, Ui_MainWindow): __version__ = 'v0.1.0' if not os.path.isdir("./src") or not os.path.isfile("./src/file.ico"): from src import release_src os.makedirs("./src", exist_ok=True) release_src() def __init__(self, parent=None): super(MainWindow, self).__init__(parent) self.setupUi(self) self.init_default_settings() self.init_variables() self.init_workers() self.init_menu() self.setWindowTitle("蓝奏云客户端 - {}".format(self.__version__)) self.set_window_at_center() self.init_extract_share_ui() self.init_disk_ui() self.init_rec_ui() self.call_login_luncher() self.create_left_menus() self.setStyleSheet(qssStyle) self.tabWidget.setStyleSheet("QTabBar{ background-color: #AEEEEE; }") self.check_update_worker.set_values(self.__version__, False) # 检测新版 # 系统托盘 self.tray = QSystemTrayIcon(QIcon('src/lanzou_logo2.png'), parent=self) show_action = QAction("显示窗口", self) hide_action = QAction("最小化到托盘", self) quit_action = QAction("退出程序", self) show_action.triggered.connect(self.show) hide_action.triggered.connect(self.hide) quit_action.triggered.connect(qApp.quit) show_action.setIcon(self.style().standardIcon(QStyle.SP_ComputerIcon)) hide_action.setIcon(self.style().standardIcon( QStyle.SP_TitleBarMinButton)) quit_action.setIcon(self.style().standardIcon( QStyle.SP_TitleBarCloseButton)) self.tray.activated.connect(self.show) #托盘点击事件 tray_menu = QMenu(QApplication.desktop()) tray_menu.addAction(show_action) tray_menu.addAction(hide_action) tray_menu.addAction(quit_action) self.tray.setContextMenu(tray_menu) self.tray.show() def init_menu(self): self.login.triggered.connect(self.show_login_dialog) # 登录 self.login.setIcon(QIcon("./src/login.ico")) self.login.setShortcut("Ctrl+L") self.toolbar.addAction(self.login) self.logout.triggered.connect( lambda: self.logout_worker.set_values(self._disk)) # 登出 self.logout.setIcon(QIcon("./src/logout.ico")) self.logout.setShortcut("Ctrl+Q") # 登出快捷键 self.download.setShortcut("Ctrl+J") self.download.setIcon(QIcon("./src/download.ico")) self.download.setEnabled(False) # 暂时不用 self.delete.setShortcut("Ctrl+D") self.delete.setIcon(QIcon("./src/delete.ico")) self.delete.setEnabled(False) # 暂时不用 self.how.setShortcut("F1") self.how.setIcon(QIcon("./src/help.ico")) self.how.triggered.connect(self.open_wiki_url) self.about.setShortcut("Ctrl+B") self.about.setIcon(QIcon("./src/about.ico")) self.about.triggered.connect(self.about_dialog.exec) self.upload.setIcon(QIcon("./src/upload.ico")) self.upload.setShortcut("Ctrl+U") # 上传快捷键 # 添加设置菜单,暂时放这里 self.setting_menu = QAction(self) # 设置菜单 self.setting_menu.setObjectName("setting_menu") self.setting_menu.setText("设置") self.files.addAction(self.setting_menu) self.setting_menu.setIcon(QIcon("./src/settings.ico")) self.setting_menu.triggered.connect(self.setting_dialog.open_dialog) self.setting_menu.setShortcut("Ctrl+P") # 设置快捷键 # tab 切换时更新 self.tabWidget.currentChanged.connect(self.call_change_tab) def init_default_settings(self): """初始化默认设置""" download_threads = 3 # 同时三个下载任务 max_size = 100 # 单个文件大小上限 MB timeout = 5 # 每个请求的超时 s(不包含下载响应体的用时) time_fmt = False # 是否使用年月日时间格式 to_tray = False # 关闭到系统托盘 dl_path = os.path.dirname( os.path.abspath(__file__)) + os.sep + "downloads" self._default_settings = { "download_threads": download_threads, "max_size": max_size, "to_tray": to_tray, "dl_path": dl_path, "timeout": timeout, "time_fmt": time_fmt } def init_variables(self): self._disk = LanZouCloud() self._config_file = "./config.pkl" self._folder_list = {} self._file_list = {} self._path_list = FolderList() self._path_list_old = FolderList() self._locs = {} self._parent_id = -1 # --> .. self._work_name = "" # share disk rec, not use now self._work_id = -1 # disk folder id self._old_work_id = self._work_id # 用于上传完成后判断是否需要更新disk界面 self.load_settings() def update_lanzoucloud_settings(self): """更新LanzouCloud实例设置""" self._disk.set_timeout(self.configs["settings"]["timeout"]) self._disk.set_max_size(self.configs["settings"]["max_size"]) self.download_threads = self.configs["settings"]["download_threads"] self.time_fmt = self.configs["settings"]["time_fmt"] # 时间显示格式 self.to_tray = self.configs["settings"][ "to_tray"] if "to_tray" in self.configs["settings"] else False def init_workers(self): # 登录器 self.login_luncher = LoginLuncher(self._disk) self.login_luncher.code.connect(self.login_update_ui) self.login_luncher.update_cookie.connect(self.call_update_cookie) # 登出器 self.logout_worker = LogoutWorker() self.logout_worker.successed.connect(self.call_logout_update_ui) # 下载器 self.download_manager = DownloadManager() self.download_manager.downloaders_msg.connect(self.show_status) self.download_manager.download_mgr_msg.connect(self.show_status) self.download_manager.finished.connect( lambda: self.show_status("所有下载任务已完成!", 2999)) # 获取更多信息,直链、下载次数等 self.info_dialog = InfoDialog() # 对话框 self.info_dialog.setWindowModality(Qt.ApplicationModal) # 窗口前置 self.more_info_worker = GetMoreInfoWorker() # 后台更新线程 self.more_info_worker.msg.connect(self.show_status) self.more_info_worker.infos.connect(self.info_dialog.set_values) self.more_info_worker.dl_link.connect( self.info_dialog.tx_dl_link.setText) self.info_dialog.get_dl_link.connect(self.more_info_worker.get_dl_link) # 登录文件列表更新器 self.list_refresher = ListRefresher(self._disk) self.list_refresher.err_msg.connect(self.show_status) self.list_refresher.infos.connect(self.update_disk_lists) self.list_refresher.infos.connect(lambda: self.show_status("")) # 获取所有文件夹fid,并移动 self.all_folders_worker = GetAllFoldersWorker() self.all_folders_worker.msg.connect(self.show_status) self.all_folders_worker.infos.connect(self.show_move_file_dialog) self.all_folders_worker.moved.connect( lambda: self.list_refresher.set_values(self._work_id, True, False, False)) # 更新文件列表 # 重命名、修改简介、新建文件夹 self.rename_mkdir_worker = RenameMkdirWorker() self.rename_mkdir_worker.msg.connect(self.show_status) self.rename_mkdir_worker.update.connect( self.list_refresher.set_values) # 更新界面 # 设置文件(夹)提取码 self.set_pwd_worker = SetPwdWorker() self.set_pwd_worker.msg.connect(self.show_status) self.set_pwd_worker.update.connect( self.list_refresher.set_values) # 更新界面 # 删除文件(夹) self.remove_files_worker = RemoveFilesWorker(self._disk) self.remove_files_worker.msg.connect(self.show_status) # 显示错误提示 self.remove_files_worker.finished.connect( lambda: self.list_refresher.set_values(self._work_id)) # 更新界面 # 上传器,信号在登录更新界面设置 self.upload_dialog = UploadDialog() self.upload_dialog.new_infos.connect(self.call_upload) # 文件描述与提取码更新器 self.desc_pwd_fetcher = DescPwdFetcher() self.desc_pwd_fetcher.desc.connect(self.call_update_desc_pwd) self.desc_pwd_fetcher.tasks.connect( self.call_download_manager_thread) # 连接下载管理器线程 # 设置 tab self.tabWidget.setCurrentIndex(0) self.tabWidget.removeTab(2) self.tabWidget.removeTab(1) self.disk_tab.setEnabled(False) self.rec_tab.setEnabled(False) # 状态栏 self._msg_label = QLabel() self._msg_movie_lb = QLabel() self._msg_movie = QMovie("src/loading_more.gif") self._msg_movie.setScaledSize(QSize(24, 24)) self._msg_movie_lb.setMovie(self._msg_movie) self._msg_label.setObjectName("msg_label") self._msg_movie_lb.setObjectName("msg_movie_lb") self.statusbar.addWidget(self._msg_movie_lb) self.statusbar.addWidget(self._msg_label) # 重命名、修改简介与新建文件夹对话框 self.rename_dialog = RenameDialog() self.rename_dialog.out.connect(self.call_rename_mkdir_worker) # 修改设置 提取码对话框 self.set_pwd_dialog = SetPwdDialog() self.set_pwd_dialog.new_infos.connect(self.set_passwd) # 菜单栏关于 self.about_dialog = AboutDialog() self.about_dialog.set_values(self.__version__) # 菜单栏设置 self.setting_dialog = SettingDialog(self._config_file, self._default_settings) self.setting_dialog.saved.connect( lambda: self.load_settings(ref_ui=True)) # 登录回收站信息更新器 self.get_rec_lists_worker = GetRecListsWorker(self._disk) self.get_rec_lists_worker.msg.connect(self.show_status) self.get_rec_lists_worker.infos.connect(self.update_rec_lists) self.get_rec_lists_worker.folders.connect( lambda: self.show_status('', 0)) self.get_rec_lists_worker.folders.connect( self.pop_up_rec_folder_dialog) # 回收站操作器 self.rec_manipulator = RecManipulator(self._disk) self.rec_manipulator.msg.connect(self.show_status) self.rec_manipulator.successed.connect(self.get_rec_lists_worker.start) # 检查软件版本 self.check_update_worker = CheckUpdateWorker() self.about_dialog.check_update.connect( self.check_update_worker.set_values) self.check_update_worker.infos.connect(self.about_dialog.show_update) self.check_update_worker.bg_update_infos.connect( self.show_new_version_msg) def show_login_dialog(self): """显示登录对话框""" login_dialog = LoginDialog(self._config_file) login_dialog.clicked_ok.connect(self.call_login_luncher) login_dialog.setWindowModality(Qt.ApplicationModal) login_dialog.exec() def show_upload_dialog(self, files): """显示上传文件对话框""" if len(self._path_list) > 0: self.upload_dialog.set_values(self._path_list[-1].name, files) else: self.show_status("等待文件列表更新...", 2000) def show_upload_dialog_menus(self): '''菜单栏显示上传对话框槽函数''' self.show_upload_dialog(None) def load_settings(self, ref_ui=False): """加载用户设置""" try: with open(self._config_file, "rb") as _file: self.configs = load(_file) except: self.configs = { "user": "", "pwd": "", "cookie": "", "settings": self._default_settings } with open(self._config_file, "wb") as _file: dump(self.configs, _file) # 兼容以前的平配置文件 if "settings" not in self.configs or not self.configs["settings"]: self.configs.update({"settings": self._default_settings}) update_settings(self._config_file, {"settings": self._default_settings}) self.update_lanzoucloud_settings() if ref_ui and self.tabWidget.currentIndex() == 1: # 更新文件界面的时间 self.show_file_and_folder_lists() def call_download_manager_thread(self, tasks): self.download_manager.set_values(tasks, self.configs["settings"]["dl_path"], self.download_threads) self.download_manager.start() def call_multi_manipulator(self, action): """批量操作器""" tab_page = self.tabWidget.currentIndex() if tab_page == 0: listview = self.table_share model = self.model_share elif tab_page == 1: listview = self.table_disk model = self.model_disk elif tab_page == 2: listview = self.table_rec model = self.model_rec else: return infos = [] _indexes = listview.selectionModel().selection().indexes() for i in _indexes: # 获取所选行号 info = model.item(i.row(), 0).data() if info and info not in infos: infos.append(info) if tab_page == 0 or tab_page == 1: if not infos: return self.desc_pwd_fetcher.set_values(self._disk, infos, download=True) elif tab_page == 2: if action == "recovery": title = "确定恢复选定文件(夹)?" elif action == "delete": title = "确定彻底删除选定文件(夹)?" elif action == "recovery_all": title = "确定还原全部文件(夹)?" msg = "提示: 恢复回收站中的文件将不可撤销,请确认。" elif action == "clean": title = "确定清除全部文件(夹)?" msg = "提示: 删除回收站中的文件将不可恢复,请确认。" if action == "recovery" or action == "delete": if not infos: self.show_status("请先选中需要操作的文件!", 2999) return msg = "\t\t列表:\n" for i in infos: msg += f"{i.time}\t{i.name}\t{i.size}\n" message_box = QMessageBox(self) message_box.setStyleSheet(btn_style) message_box.setWindowTitle(title) message_box.setText(msg) message_box.setStandardButtons(QMessageBox.Yes | QMessageBox.No) buttonY = message_box.button(QMessageBox.Yes) buttonY.setText('确定') buttonN = message_box.button(QMessageBox.No) buttonN.setText('取消') message_box.exec_() if message_box.clickedButton() == buttonY: self.rec_manipulator.set_values(infos, action) def call_logout_update_ui(self): """菜单栏、工具栏登出""" self.toolbar.removeAction(self.logout) self.tabWidget.setCurrentIndex(0) self.disk_tab.setEnabled(False) self.rec_tab.setEnabled(False) self.tabWidget.removeTab(2) self.tabWidget.removeTab(1) self.toolbar.removeAction(self.logout) # 登出工具 self.logout.setEnabled(False) self.toolbar.removeAction(self.upload) # 上传文件工具栏 self.upload.setEnabled(False) self.upload.triggered.disconnect(self.show_upload_dialog_menus) def login_update_ui(self, success, msg, duration): """根据登录是否成功更新UI""" if success: self.show_status(msg, duration) self.tabWidget.insertTab(1, self.disk_tab, "我的蓝奏云") self.tabWidget.insertTab(2, self.rec_tab, "回收站") self.disk_tab.setEnabled(True) self.rec_tab.setEnabled(True) # 更新快捷键与工具栏 self.toolbar.addAction(self.logout) # 添加登出工具栏 self.toolbar.addAction(self.upload) # 添加上传文件工具栏 # 菜单栏槽 self.logout.setEnabled(True) self.upload.setEnabled(True) self.upload.triggered.connect(self.show_upload_dialog_menus) # 设置当前显示 tab self.tabWidget.setCurrentIndex(1) QCoreApplication.processEvents() # 重绘界面 # 刷新文件列表 # self.list_refresher.set_values(self._work_id) else: self.show_status(msg, duration) self.tabWidget.setCurrentIndex(0) self.tabWidget.removeTab(2) self.tabWidget.removeTab(1) self.disk_tab.setEnabled(False) self.rec_tab.setEnabled(False) # 更新快捷键与工具栏 self.toolbar.removeAction(self.logout) # 登出工具栏 self.toolbar.removeAction(self.upload) # 上传文件工具栏 self.logout.setEnabled(False) self.upload.setEnabled(False) def call_login_luncher(self): """登录网盘""" self.load_settings() self.logout_worker.set_values(self._disk, update_ui=False) self.toolbar.removeAction(self.logout) try: username = self.configs["user"] password = self.configs["pwd"] cookie = self.configs["cookie"] self.login_luncher.set_values(username, password, cookie) self.login_luncher.start() except Exception as exp: print(exp) pass def call_update_cookie(self, cookie): """更新cookie至config文件""" up_info = {"cookie": cookie} update_settings(self._config_file, up_info) def show_file_and_folder_lists(self): """显示用户文件和文件夹列表""" self.model_disk.removeRows(0, self.model_disk.rowCount()) # 清理旧的内容 file_count = len(self._file_list.keys()) folder_count = len(self._folder_list.keys()) name_header = [ "文件夹{}个".format(folder_count), ] if folder_count else [] if file_count: name_header.append("文件{}个".format(file_count)) self.model_disk.setHorizontalHeaderLabels( ["/".join(name_header), "大小", "时间"]) folder_ico = QIcon("./src/folder.gif") pwd_ico = QIcon("./src/keys.ico") # infos: ID/None,文件名,大小,日期,下载次数(dl_count),提取码(pwd),描述(desc),|链接(share-url),直链 if self._work_id != -1: _back = QStandardItem(folder_ico, "..") _back.setToolTip("双击返回上层文件夹,选中无效") self.model_disk.appendRow( [_back, QStandardItem(""), QStandardItem("")]) for infos in self._folder_list.values(): # 文件夹 name = QStandardItem(folder_ico, infos[1]) name.setData(infos) tips = "" if infos[5] is not False: tips = "有提取码" if infos[6] is not False: tips = tips + ",描述:" + str(infos[6]) elif infos[6] is not False: tips = "描述:" + str(infos[6]) name.setToolTip(tips) size_ = QStandardItem(pwd_ico, "") if infos[5] else QStandardItem( "") # 提取码+size self.model_disk.appendRow([name, size_, QStandardItem("")]) for infos in self._file_list.values(): # 文件 name = QStandardItem(set_file_icon(infos[1]), infos[1]) name.setData(infos) tips = "" if infos[5] is not False: tips = "有提取码" if infos[6] is not False: tips = tips + ",描述:" + str(infos[6]) elif infos[6] is not False: tips = "描述:" + str(infos[6]) name.setToolTip(tips) size_ = QStandardItem(pwd_ico, infos[2]) if infos[5] else QStandardItem( infos[2]) # 提取码+size time_ = QStandardItem(time_format( infos[3])) if self.time_fmt else QStandardItem(infos[3]) self.model_disk.appendRow([name, size_, time_]) for row in range(self.model_disk.rowCount()): # 右对齐 self.model_disk.item(row, 1).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.model_disk.item(row, 2).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) def update_disk_lists(self, infos): """更新用户文件列表""" if not infos: return if infos['r']['files']: self._file_list = infos['file_list'] if infos['r']['folders']: self._folder_list = infos['folder_list'] self._path_list = infos['path_list'] self._work_id = self._path_list[-1].id if infos['r']['fid'] != -1: self._parent_id = self._path_list[-2].id self.show_file_and_folder_lists() if infos['r']['path']: self.show_full_path() def config_tableview(self, tab): """Tab 初始化""" if tab == "share": model = self.model_share table = self.table_share elif tab == "disk": model = self.model_disk table = self.table_disk elif tab == "rec": model = self.model_rec table = self.table_rec model.setHorizontalHeaderLabels(["文件名", "大小", "时间"]) table.setModel(model) # 是否显示网格线 table.setShowGrid(False) # 禁止编辑表格 table.setEditTriggers(QAbstractItemView.NoEditTriggers) # 隐藏水平表头 table.verticalHeader().setVisible(False) # 设置表头可以自动排序 table.setSortingEnabled(True) table.setMouseTracking(False) # 设置表头的背景色为绿色 table.horizontalHeader().setStyleSheet( "QHeaderView::section{background:lightgray}") # 设置 不可选择单个单元格,只可选择一行。 table.setSelectionBehavior(QAbstractItemView.SelectRows) # 设置第二三列的宽度 table.horizontalHeader().resizeSection(1, 90) table.horizontalHeader().resizeSection(2, 80) # 设置第一列宽度自动调整,充满屏幕 table.horizontalHeader().setSectionResizeMode(0, QHeaderView.Stretch) if tab != "rec": table.setContextMenuPolicy(Qt.CustomContextMenu) # 允许右键产生子菜单 table.customContextMenuRequested.connect(self.generateMenu) # 右键菜单 def create_left_menus(self): self.left_menus = QMenu() self.left_menu_share_url = self.left_menus.addAction("外链分享地址等") self.left_menu_share_url.setIcon(QIcon("./src/share.ico")) self.left_menu_rename_set_desc = self.left_menus.addAction("修改文件夹名与描述") self.left_menu_rename_set_desc.setIcon(QIcon("./src/desc.ico")) self.left_menu_set_pwd = self.left_menus.addAction("设置访问密码") self.left_menu_set_pwd.setIcon(QIcon("./src/password.ico")) self.left_menu_move = self.left_menus.addAction("移动(支持批量)") self.left_menu_move.setIcon(QIcon("./src/move.ico")) def call_rename_mkdir_worker(self, infos): """重命名、修改简介与新建文件夹""" self.rename_mkdir_worker.set_values(self._disk, infos, self._work_id, self._folder_list) def set_passwd(self, infos): """设置文件(夹)提取码""" self.set_pwd_worker.set_values(self._disk, infos, self._work_id) def call_mkdir(self): """弹出新建文件夹对话框""" self.rename_dialog.set_values(None) self.rename_dialog.exec() def call_remove_files(self): indexs = [] infos = [] _indexs = self.table_disk.selectionModel().selection().indexes() if not _indexs: return for i in _indexs: # 获取所选行号 indexs.append(i.row()) indexs = set(indexs) for index in indexs: info = self.model_disk.item(index, 0).data() # 用于提示删除的文件名 if info: infos.append(info[:3]) delete_dialog = DeleteDialog(infos) delete_dialog.new_infos.connect(self.remove_files_worker.set_values) delete_dialog.exec() def generateMenu(self, pos): """右键菜单""" row_nums = self.sender().selectionModel().selection().indexes() if not row_nums: # 如果没有选中行,什么也不做 return _model = self.sender().model() infos = [] # 多个选中的行,用于移动文件与... for one_row in row_nums: one_row_data = _model.item(one_row.row(), 0).data() if one_row_data and one_row_data not in infos: # 删掉 .. 行 infos.append(one_row_data) if not infos: return info = infos[0] # 取选中的第一行 # 通过是否有文件 ID 判断是登录界面还是提取界面 if info[0]: self.left_menu_rename_set_desc.setEnabled(True) self.left_menu_set_pwd.setEnabled(True) # 通过infos第3个字段 size 判断是否为文件夹,文件夹不能移动,设置不同的显示菜单名 if info[2]: self.left_menu_rename_set_desc.setText("修改文件描述") self.left_menu_move.setEnabled(True) else: self.left_menu_rename_set_desc.setText("修改文件夹名与描述") self.left_menu_move.setDisabled(True) else: self.left_menu_rename_set_desc.setDisabled(True) self.left_menu_move.setDisabled(True) self.left_menu_set_pwd.setDisabled(True) action = self.left_menus.exec_(self.sender().mapToGlobal(pos)) if action == self.left_menu_share_url: # 显示详细信息 # 后台跟新信息,并显示信息对话框 self.more_info_worker.set_values(info, self._disk) elif action == self.left_menu_move: # 移动文件 self.all_folders_worker.set_values(self._disk, infos) elif action == self.left_menu_set_pwd: # 修改提取码 self.desc_pwd_fetcher.set_values(self._disk, [ info, ]) # 兼容下载器,使用列表的列表 self.set_pwd_dialog.set_values(info) self.set_pwd_dialog.exec() elif action == self.left_menu_rename_set_desc: # 重命名与修改描述 self.desc_pwd_fetcher.set_values(self._disk, [ info, ]) # 兼容下载器,使用列表的列表 self.rename_dialog.set_values(info) self.rename_dialog.exec() def call_update_desc_pwd(self, desc, pwd, infos): '''更新 desc、pwd''' infos[6] = desc infos[5] = pwd self.rename_dialog.set_values(infos) self.set_pwd_dialog.set_values(infos) def show_move_file_dialog(self, infos, all_dirs_dict): '''显示移动文件对话框''' move_file_dialog = MoveFileDialog(infos, all_dirs_dict) move_file_dialog.new_infos.connect( self.all_folders_worker.move_file) # 调用移动线程 move_file_dialog.exec() def call_change_dir(self, folder_id=-1): """顶部路径按钮调用""" def callfunc(): self.list_refresher.set_values(folder_id) return callfunc def change_dir(self, dir_name): """双击切换工作目录""" dir_name = self.model_disk.item(dir_name.row(), 0).text() # 文件夹名 if dir_name == "..": # 返回上级路径 self.list_refresher.set_values(self._parent_id) elif dir_name in self._folder_list.keys(): folder_id = self._folder_list[dir_name][0] self.list_refresher.set_values(folder_id) def call_upload(self, infos): """上传文件(夹)""" self._old_work_id = self._work_id # 记录上传文件夹id self.upload_worker.set_values(self._disk, infos, self._old_work_id) self.upload_worker.start() def show_full_path(self): """路径框显示当前路径""" index = 1 for _ in iter(self._path_list_old): self._locs[index].clicked.disconnect() self.disk_loc.removeWidget(self._locs[index]) self._locs[index].deleteLater() self._locs[index] = None del self._locs[index] index += 1 index = 1 for item in iter(self._path_list): self._locs[index] = QPushButton(item.name, self.disk_tab) self._locs[index].setToolTip(f"fid:{item.id}") self._locs[index].setIcon(QIcon("./src/folder.gif")) self._locs[index].setStyleSheet( "QPushButton {border:none; background:transparent;}") self.disk_loc.insertWidget(index, self._locs[index]) self._locs[index].clicked.connect(self.call_change_dir(item.id)) index += 1 self._path_list_old = self._path_list def select_all_btn(self, action="reverse"): """默认反转按钮状态""" page = self.tabWidget.currentIndex() if page == 0: btn = self.btn_share_select_all table = self.table_share elif page == 1: btn = self.btn_disk_select_all table = self.table_disk elif page == 2: btn = self.btn_rec_select_all table = self.table_rec else: return if btn.isEnabled(): if action == "reverse": if btn.text() == "全选": table.selectAll() btn.setText("取消") btn.setIcon(QIcon("./src/select_none.ico")) elif btn.text() == "取消": table.clearSelection() btn.setText("全选") btn.setIcon(QIcon("./src/select_all.ico")) elif action == "cancel": # 点击列表其中一个就表示放弃全选 btn.setText("全选") btn.setIcon(QIcon("./src/select_all.ico")) else: table.selectAll() btn.setText("取消") btn.setIcon(QIcon("./src/select_none.ico")) def finished_upload(self): """上传完成调用""" if self._old_work_id == self._work_id: self.list_refresher.set_values(self._work_id, True, True, False) else: self._old_work_id = self._work_id self.show_status("上传完成!", 7000) # disk tab def init_disk_ui(self): self.model_disk = QStandardItemModel(1, 3) self.config_tableview("disk") self.btn_disk_delete.setIcon(QIcon("./src/delete.ico")) self.btn_disk_dl.setIcon(QIcon("./src/downloader.ico")) self.btn_disk_select_all.setIcon(QIcon("./src/select_all.ico")) self.btn_disk_select_all.setToolTip("按下 Ctrl/Alt + A 全选或则取消全选") self.btn_disk_select_all.clicked.connect( lambda: self.select_all_btn("reverse")) self.table_disk.clicked.connect(lambda: self.select_all_btn("cancel")) self.btn_disk_dl.clicked.connect( lambda: self.call_multi_manipulator("download")) self.btn_disk_mkdir.setIcon(QIcon("./src/add_folder.ico")) self.btn_disk_mkdir.clicked.connect(self.call_mkdir) self.btn_disk_delete.clicked.connect(self.call_remove_files) # 文件拖拽上传 self.table_disk.drop_files.connect(self.show_upload_dialog) self.table_disk.doubleClicked.connect(self.change_dir) # 上传器 self.upload_worker = UploadWorker() self.upload_worker.finished.connect(self.finished_upload) self.upload_worker.code.connect(self.show_status) # rec tab def pop_up_rec_folder_dialog(self, files): # 弹出回收站文件夹内容对话框 if files: rec_file_dialog = RecFolderDialog(files) rec_file_dialog.exec() else: self.show_status("文件夹为空!", 2999) def call_rec_folder_dialog(self, dir_name): # 显示弹出对话框 dir_data = self.model_rec.item(dir_name.row(), 0).data() # 文件夹信息 if isinstance(dir_data, RecFolder): self.show_status(f"正在获取文件夹 {dir_data.name} 信息,稍后", 10000) self.get_rec_lists_worker.set_values(dir_data.id) def update_rec_lists(self, dir_lists, file_lists): """显示回收站文件和文件夹列表""" self.model_rec.removeRows(0, self.model_rec.rowCount()) # 清理旧的内容 file_count = len(file_lists) folder_count = len(dir_lists) if ((not dir_lists) and (not file_lists)) or (file_count == 0 and folder_count == 0): self.show_status("回收站为空!", 4000) return name_header = [ "文件夹{}个".format(folder_count), ] if folder_count else [] if file_count: name_header.append("文件{}个".format(file_count)) self.model_rec.setHorizontalHeaderLabels( ["/".join(name_header), "大小", "时间"]) folder_ico = QIcon("./src/folder.gif") for item in iter(dir_lists): # 文件夹 name = QStandardItem(folder_ico, item.name) name.setData(item) name.setToolTip("双击查看详情") size_ = QStandardItem(item.size) time_ = QStandardItem(item.time) self.model_rec.appendRow([name, size_, time_]) for item in iter(file_lists): # 文件 name = QStandardItem(set_file_icon(item.name), item.name) name.setData(item) size_ = QStandardItem(item.size) time_ = QStandardItem(item.time) self.model_rec.appendRow([name, size_, time_]) for row in range(self.model_rec.rowCount()): # 右对齐 self.model_rec.item(row, 1).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.model_rec.item(row, 2).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) def init_rec_ui(self): """回收站ui初始化""" self.model_rec = QStandardItemModel(1, 3) self.config_tableview("rec") self.table_rec.doubleClicked.connect(self.call_rec_folder_dialog) self.btn_rec_select_all.setIcon(QIcon("./src/select_all.ico")) self.btn_rec_select_all.clicked.connect( lambda: self.select_all_btn("reverse")) self.btn_rec_delete.clicked.connect( lambda: self.call_multi_manipulator("delete")) self.btn_recovery.clicked.connect( lambda: self.call_multi_manipulator("recovery")) self.btn_rec_delete.setToolTip("彻底删除选中文件(夹)") self.btn_recovery.setToolTip("恢复选中文件(夹)") self.btn_recovery_all.clicked.connect( lambda: self.call_multi_manipulator("recovery_all")) self.btn_recovery_all.setToolTip("恢复全部") self.btn_rec_clean.clicked.connect( lambda: self.call_multi_manipulator("clean")) self.btn_rec_clean.setToolTip("清理回收站全部") # shared url def call_get_shared_info(self): if not self.get_shared_info_thread.isRunning(): # 防止快速多次调用 self.line_share_url.setEnabled(False) self.btn_extract.setEnabled(False) text = self.line_share_url.text().strip() self.get_shared_info_thread.set_values(text) def show_share_url_file_lists(self, infos): if infos["code"] == LanZouCloud.SUCCESS: file_count = len(infos["info"].keys()) self.model_share.setHorizontalHeaderLabels( ["文件{}个".format(file_count), "大小", "时间"]) for infos in infos["info"].values(): name = QStandardItem(set_file_icon(infos[1]), infos[1]) name.setData(infos) time = QStandardItem(time_format( infos[3])) if self.time_fmt else QStandardItem(infos[3]) self.model_share.appendRow( [name, QStandardItem(infos[2]), time]) for r in range(self.model_share.rowCount()): # 右对齐 self.model_share.item(r, 1).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.model_share.item(r, 2).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.table_share.setDisabled(False) self.btn_share_select_all.setDisabled(False) self.btn_share_select_all.setToolTip("按下 Ctrl/Alt + A 全选或则取消全选") self.btn_share_dl.setDisabled(False) else: self.btn_share_select_all.setDisabled(True) self.btn_share_select_all.setToolTip("") self.btn_share_dl.setDisabled(True) self.table_share.setDisabled(True) def set_download_path(self): """设置下载路径""" dl_path = QFileDialog.getExistingDirectory() dl_path = os.path.normpath(dl_path) # windows backslash if dl_path == self.configs["settings"]["dl_path"] or dl_path == ".": return if dl_path == "": dl_path = os.path.dirname( os.path.abspath(__file__)) + os.sep + "downloads" up_info = {"dl_path": dl_path} else: up_info = {"dl_path": dl_path} update_settings(self._config_file, up_info, is_settings=True) self.load_settings() self.line_dl_path.setText(self.configs["settings"]["dl_path"]) def init_extract_share_ui(self): self.btn_share_select_all.setDisabled(True) self.btn_share_dl.setDisabled(True) self.table_share.setDisabled(True) self.model_share = QStandardItemModel(1, 3) self.config_tableview("share") # 获取分享链接信息线程 self.get_shared_info_thread = GetSharedInfo() self.get_shared_info_thread.update.connect( lambda: self.model_share.removeRows(0, self.model_share.rowCount() )) # 清理旧的信息 self.get_shared_info_thread.msg.connect(self.show_status) # 提示信息 self.get_shared_info_thread.infos.connect( self.show_share_url_file_lists) # 内容信息 self.get_shared_info_thread.finished.connect( lambda: self.btn_extract.setEnabled(True)) self.get_shared_info_thread.finished.connect( lambda: self.line_share_url.setEnabled(True)) # 控件设置 self.line_share_url.setPlaceholderText( "蓝奏云链接,如有提取码,放后面,空格或汉字等分割,回车键提取") self.line_share_url.returnPressed.connect(self.call_get_shared_info) self.btn_extract.clicked.connect(self.call_get_shared_info) self.btn_share_dl.clicked.connect( lambda: self.call_multi_manipulator("download")) self.btn_share_dl.setIcon(QIcon("./src/downloader.ico")) self.btn_share_select_all.setIcon(QIcon("./src/select_all.ico")) self.btn_share_select_all.clicked.connect( lambda: self.select_all_btn("reverse")) self.table_share.clicked.connect( lambda: self.select_all_btn("cancel")) # 全选按钮 # 添加文件下载路径选择器 self.line_dl_path = MyLineEdit(self.share_tab) self.line_dl_path.setObjectName("line_dl_path") self.horizontalLayout_share_2.insertWidget(2, self.line_dl_path) self.line_dl_path.setText(self.configs["settings"]["dl_path"]) self.line_dl_path.clicked.connect(self.set_download_path) # QSS self.label_share_url.setStyleSheet( "#label_share_url {color: rgb(255,255,60);}") self.label_dl_path.setStyleSheet( "#label_dl_path {color: rgb(255,255,60);}") # others def clean_status(self): self._msg_label.setText("") self._msg_movie_lb.clear() self._msg_movie.stop() def show_status(self, msg, duration=0): self._msg_label.setText(msg) if msg and duration >= 3000: self._msg_movie_lb.setMovie(self._msg_movie) self._msg_movie.start() else: self._msg_movie_lb.clear() self._msg_movie.stop() if duration != 0: QTimer.singleShot(duration, self.clean_status) def keyPressEvent(self, e): if e.key() == Qt.Key_A: # Ctrl/Alt + A 全选 if e.modifiers() and Qt.ControlModifier: self.select_all_btn() elif e.key() == Qt.Key_F5: # 刷新 if self.tabWidget.currentIndex() == 1: # disk 界面 self.show_status("正在更新当前目录...", 10000) self.list_refresher.set_values(self._work_id) elif self.tabWidget.currentIndex() == 2: # rec 界面 self.show_status("正在更新回收站...", 10000) self.get_rec_lists_worker.start() def call_change_tab(self): """切换标签页 动作""" tab_index = self.tabWidget.currentIndex() if tab_index == 2: self.show_status("正在更新回收站...", 10000) self.get_rec_lists_worker.start() elif tab_index == 1: self.show_status("正在更新当前目录...", 10000) self.list_refresher.set_values(self._work_id) def set_window_at_center(self): screen = QDesktopWidget().screenGeometry() size = self.geometry() new_left = int((screen.width() - size.width()) / 2) new_top = int((screen.height() - size.height()) / 2) self.move(new_left, new_top) def open_wiki_url(self): # 打开使用说明页面 url = QUrl('https://github.com/rachpt/lanzou-gui/wiki') if not QDesktopServices.openUrl(url): self.show_status('Could not open wiki page!', 5000) def show_new_version_msg(self, ver, msg): message_box = QMessageBox(self) message_box.setStyleSheet(btn_style) message_box.setWindowTitle(f"检测到新版 {ver}") message_box.setText(msg) message_box.setStandardButtons(QMessageBox.Close) buttonC = message_box.button(QMessageBox.Close) buttonC.setText('关闭') message_box.exec() def closeEvent(self, event): if self.to_tray: event.ignore() self.hide() self.tray.showMessage("蓝奏云客户端", "应用已经最小化到托盘,如需退出请右击", QSystemTrayIcon.Information, 2500)
class MainWindow(QMainWindow, Ui_MainWindow): __version__ = 'v0.0.6' def __init__(self, parent=None): super(MainWindow, self).__init__(parent) self.setupUi(self) self.init_variable() self.init_workers() self.init_menu() self.setWindowTitle("蓝奏云客户端 - {}".format(self.__version__)) self.setWindowIcon(QIcon("./icon/lanzou-logo2.png")) self.set_window_at_center() self.init_extract_share_ui() self.init_disk_ui() self.call_login_luncher() self.create_left_menus() self.setObjectName("MainWindow") self.setStyleSheet(qssStyle) # self.disk_tab.setStyleSheet("* {background-color: rgba(255, 255, 255, 150);}") self.tabWidget.setStyleSheet("QTabBar{ background-color: #AEEEEE; }") def init_menu(self): self.login.triggered.connect(self.show_login_dialog) # 登录 self.login.setIcon(QIcon("./icon/login.ico")) self.login.setShortcut("Ctrl+L") self.toolbar.addAction(self.login) self.logout.triggered.connect( lambda: self.logout_worker.set_values(self._disk)) # 登出 self.logout.setIcon(QIcon("./icon/logout.ico")) self.logout.setShortcut("Ctrl+Q") # 登出快捷键 self.download.setShortcut("Ctrl+J") # 以下还未使用 self.download.setIcon(QIcon("./icon/download.ico")) self.download.setEnabled(False) # 暂时不用 self.delete.setShortcut("Ctrl+D") self.delete.setIcon(QIcon("./icon/delete.ico")) self.delete.setEnabled(False) # 暂时不用 # self.how.setShortcut("Ctrl+H") self.how.setIcon(QIcon("./icon/help.ico")) self.how.triggered.connect(self.open_wiki_url) # self.about.setShortcut("Ctrl+O") self.about.setIcon(QIcon("./icon/about.ico")) self.about.triggered.connect(self.about_dialog.exec) self.upload.setIcon(QIcon("./icon/upload.ico")) self.upload.setShortcut("Ctrl+U") # 上传快捷键 def init_variable(self): self._disk = LanZouCloud() self._config = "./config.pkl" self._folder_list = {} self._file_list = {} self._path_list = {} self._path_list_old = {} self._locs = {} self._parent_id = -1 # --> .. self._work_name = "" # share disk rec, not use now self._work_id = -1 # disk folder id self._old_work_id = self._work_id # 用于上传完成后判断是否需要更新disk界面 self.download_threads = 3 # 同时三个下载任务 self.load_settings() if os.name == 'nt': self._disk.set_rar_tool("./rar.exe") else: self._disk.set_rar_tool("/usr/bin/rar") def init_workers(self): # 登录器 self.login_luncher = LoginLuncher(self._disk) self.login_luncher.code.connect(self.login_update_ui) # 登出器 self.logout_worker = LogoutWorker() self.logout_worker.successed.connect(self.call_logout_update_ui) # 下载器 self.download_manager = DownloadManager() self.download_manager.downloaders_msg.connect(self.show_status) self.download_manager.download_mgr_msg.connect(self.show_status) self.download_manager.finished.connect( lambda: self.show_status("所有下载任务已完成!", 7000)) # 获取更多信息,直链、下载次数等 self.more_info_worker = GetMoreInfoWorker() self.more_info_worker.msg.connect(self.show_status) self.more_info_worker.infos.connect(self.show_info_dialog) # 登录文件列表更新器 self.list_refresher = ListRefresher(self._disk) self.list_refresher.err_msg.connect(self.show_status) self.list_refresher.infos.connect(self.update_lists) # 获取所有文件夹fid,并移动 self.all_folders_worker = GetAllFoldersWorker() self.all_folders_worker.msg.connect(self.show_status) self.all_folders_worker.infos.connect(self.show_move_file_dialog) self.all_folders_worker.moved.connect( lambda: self.list_refresher.set_values(self._work_id, False, True, False)) # 更新文件列表 # 重命名、修改简介、新建文件夹 self.rename_mkdir_worker = RenameMkdirWorker() self.rename_mkdir_worker.msg.connect(self.show_status) self.rename_mkdir_worker.update.connect( self.list_refresher.set_values) # 更新界面 # 设置文件(夹)提取码 self.set_pwd_worker = SetPwdWorker() self.set_pwd_worker.msg.connect(self.show_status) self.set_pwd_worker.update.connect( self.list_refresher.set_values) # 更新界面 # 删除文件(夹) self.remove_files_worker = RemoveFilesWorker(self._disk) self.remove_files_worker.msg.connect(self.show_status) # 显示错误提示 self.remove_files_worker.finished.connect( lambda: self.list_refresher.set_values(self._work_id)) # 更新界面 # 上传器,信号在登录更新界面设置 self.upload_dialog = UploadDialog() self.upload_dialog.new_infos.connect(self.call_upload) # 文件描述与提取码更新器 self.desc_pwd_fetcher = DescPwdFetcher() self.desc_pwd_fetcher.desc.connect(self.call_update_desc_pwd) self.desc_pwd_fetcher.tasks.connect( self.call_download_manager_thread) # 连接下载管理器线程 # 设置 tab self.tabWidget.setCurrentIndex(0) self.tabWidget.removeTab(2) self.tabWidget.removeTab(1) self.disk_tab.setEnabled(False) self.rec_tab.setEnabled(False) # 状态栏 self._msg_label = QLabel() self._msg_label.setObjectName("msg_label") self.statusbar.addWidget(self._msg_label) # 重命名、修改简介与新建文件夹对话框 self.rename_dialog = RenameDialog() self.rename_dialog.out.connect(self.call_rename_mkdir_worker) # 修改设置 提取码对话框 self.set_pwd_dialog = SetPwdDialog() self.set_pwd_dialog.new_infos.connect(self.set_passwd) # 菜单栏关于 self.about_dialog = AboutDialog() self.about_dialog.set_values(self.__version__) def show_login_dialog(self): """显示登录对话框""" login_dialog = LoginDialog(self._config) login_dialog.clicked_ok.connect(self.call_login_luncher) login_dialog.setWindowModality(Qt.ApplicationModal) login_dialog.exec() def show_upload_dialog(self): """显示上传文件对话框""" self.upload_dialog.set_values(list(self._path_list.keys())[-1]) self.upload_dialog.exec() def load_settings(self): try: with open(self._config, "rb") as _file: self.settings = load(_file) except Exception: dl_path = os.path.dirname( os.path.abspath(__file__)) + os.sep + "downloads" self.settings = {"user": "", "pwd": "", "path": dl_path} with open(self._config, "wb") as _file: dump(self.settings, _file) def call_download_manager_thread(self, tasks): self.download_manager.set_values(tasks, self.settings["path"], self.download_threads) self.download_manager.start() def call_downloader(self): tab_page = self.tabWidget.currentIndex() if tab_page == 0: listview = self.table_share model = self.model_share elif tab_page == 1: listview = self.table_disk model = self.model_disk else: return infos = [] _indexes = listview.selectionModel().selection().indexes() for i in _indexes: # 获取所选行号 info = model.item(i.row(), 0).data() if info and info not in infos: infos.append(info) if not infos: return self.desc_pwd_fetcher.set_values(self._disk, infos, download=True) def call_logout_update_ui(self): """菜单栏、工具栏登出""" self.toolbar.removeAction(self.logout) self.tabWidget.setCurrentIndex(0) self.disk_tab.setEnabled(False) self.rec_tab.setEnabled(False) self.tabWidget.removeTab(2) self.tabWidget.removeTab(1) self.toolbar.removeAction(self.logout) # 登出工具 self.logout.setEnabled(False) self.toolbar.removeAction(self.upload) # 上传文件工具栏 self.upload.setEnabled(False) self.upload.triggered.disconnect(self.show_upload_dialog) def login_update_ui(self, success, msg, duration): """根据登录是否成功更新UI""" if success: self.show_status(msg, duration) self.tabWidget.insertTab(1, self.disk_tab, "我的蓝奏云") self.tabWidget.insertTab(2, self.rec_tab, "回收站") self.disk_tab.setEnabled(True) self.rec_tab.setEnabled(True) # 更新快捷键与工具栏 self.toolbar.addAction(self.logout) # 添加登出工具栏 self.toolbar.addAction(self.upload) # 添加上传文件工具栏 # 菜单栏槽 self.logout.setEnabled(True) self.upload.setEnabled(True) self.upload.triggered.connect(self.show_upload_dialog) # 设置当前显示 tab self.tabWidget.setCurrentIndex(1) QCoreApplication.processEvents() # 重绘界面 # 刷新文件列表 self.list_refresher.set_values(self._work_id) else: self.show_status(msg, duration) self.tabWidget.setCurrentIndex(0) self.tabWidget.removeTab(2) self.tabWidget.removeTab(1) self.disk_tab.setEnabled(False) self.rec_tab.setEnabled(False) # 更新快捷键与工具栏 self.toolbar.removeAction(self.logout) # 登出工具栏 self.toolbar.removeAction(self.upload) # 上传文件工具栏 self.logout.setEnabled(False) self.upload.setEnabled(False) def call_login_luncher(self): """登录网盘""" self.load_settings() self.logout_worker.set_values(self._disk, update_ui=False) self.toolbar.removeAction(self.logout) try: username = self.settings["user"] password = self.settings["pwd"] cookie = self.settings["cookie"] self.login_luncher.set_values(username, password, cookie) self.login_luncher.start() except Exception as exp: print(exp) pass def set_file_icon(self, name): suffix = name.split(".")[-1] ico_path = "./icon/{}.gif".format(suffix) if os.path.isfile(ico_path): return QIcon(ico_path) else: return QIcon("./icon/file.ico") def show_file_and_folder_lists(self): """显示文件和文件夹列表""" self.model_disk.removeRows(0, self.model_disk.rowCount()) # 清理旧的内容 file_count = len(self._file_list.keys()) folder_count = len(self._folder_list.keys()) name_header = [ "文件夹{}个".format(folder_count), ] if folder_count else [] if file_count: name_header.append("文件{}个".format(file_count)) self.model_disk.setHorizontalHeaderLabels( ["/".join(name_header), "大小", "时间"]) folder_ico = QIcon("./icon/folder.gif") pwd_ico = QIcon("./icon/keys.ico") # infos: ID/None,文件名,大小,日期,下载次数(dl_count),提取码(pwd),描述(desc),|链接(share-url),直链 if self._work_id != -1: _back = QStandardItem(folder_ico, "..") _back.setToolTip("双击返回上层文件夹,选中无效") self.model_disk.appendRow( [_back, QStandardItem(""), QStandardItem("")]) for infos in self._folder_list.values(): # 文件夹 name = QStandardItem(folder_ico, infos[1]) name.setData(infos) tips = "" if infos[5] is not False: tips = "有提取码" if infos[6] is not False: tips = tips + ",描述:" + str(infos[6]) elif infos[6] is not False: tips = "描述:" + str(infos[6]) name.setToolTip(tips) size_ = QStandardItem(pwd_ico, "") if infos[5] else QStandardItem( "") # 提取码+size self.model_disk.appendRow([name, size_, QStandardItem("")]) for infos in self._file_list.values(): # 文件 name = QStandardItem(self.set_file_icon(infos[1]), infos[1]) name.setData(infos) tips = "" if infos[5] is not False: tips = "有提取码" if infos[6] is not False: tips = tips + ",描述:" + str(infos[6]) elif infos[6] is not False: tips = "描述:" + str(infos[6]) name.setToolTip(tips) size_ = QStandardItem(pwd_ico, infos[2]) if infos[5] else QStandardItem( infos[2]) # 提取码+size self.model_disk.appendRow([name, size_, QStandardItem(infos[3])]) for row in range(self.model_disk.rowCount()): # 右对齐 self.model_disk.item(row, 1).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.model_disk.item(row, 2).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) def update_lists(self, infos): if not infos: return if infos['r']['files']: self._file_list = infos['file_list'] if infos['r']['folders']: self._folder_list = infos['folder_list'] self._path_list = infos['path_list'] current_folder = list(self._path_list.keys())[-1] self._work_id = self._path_list.get(current_folder, -1) if infos['r']['fid'] != -1: parent_folder_name = list(self._path_list.keys())[-2] self._parent_id = self._path_list.get(parent_folder_name, -1) self.show_file_and_folder_lists() if infos['r']['path']: self.show_full_path() def config_tableview(self, tab): if tab == "share": model = self.model_share table = self.table_share elif tab == "disk": model = self.model_disk table = self.table_disk model.setHorizontalHeaderLabels(["文件名", "大小", "时间"]) table.setModel(model) # 是否显示网格线 table.setShowGrid(False) # 禁止编辑表格 table.setEditTriggers(QAbstractItemView.NoEditTriggers) # 隐藏水平表头 table.verticalHeader().setVisible(False) # 设置表头可以自动排序 table.setSortingEnabled(True) table.setMouseTracking(False) # 设置表头的背景色为绿色 table.horizontalHeader().setStyleSheet( "QHeaderView::section{background:lightgray}") # 设置 不可选择单个单元格,只可选择一行。 table.setSelectionBehavior(QAbstractItemView.SelectRows) # 设置第二三列的宽度 table.horizontalHeader().resizeSection(1, 90) table.horizontalHeader().resizeSection(2, 80) # 设置第一列宽度自动调整,充满屏幕 table.horizontalHeader().setSectionResizeMode(0, QHeaderView.Stretch) table.setContextMenuPolicy(Qt.CustomContextMenu) # 允许右键产生子菜单 table.customContextMenuRequested.connect(self.generateMenu) # 右键菜单 def create_left_menus(self): self.left_menus = QMenu() self.left_menu_share_url = self.left_menus.addAction("外链分享地址等") self.left_menu_share_url.setIcon(QIcon("./icon/share.ico")) self.left_menu_rename_set_desc = self.left_menus.addAction("修改文件夹名与描述") self.left_menu_rename_set_desc.setIcon(QIcon("./icon/desc.ico")) self.left_menu_set_pwd = self.left_menus.addAction("设置访问密码") self.left_menu_set_pwd.setIcon(QIcon("./icon/password.ico")) self.left_menu_move = self.left_menus.addAction("移动(支持批量)") self.left_menu_move.setIcon(QIcon("./icon/move.ico")) def call_rename_mkdir_worker(self, infos): """重命名、修改简介与新建文件夹""" self.rename_mkdir_worker.set_values(self._disk, infos, self._work_id) def set_passwd(self, infos): """设置文件(夹)提取码""" self.set_pwd_worker.set_values(self._disk, infos, self._work_id) def call_mkdir(self): """弹出新建文件夹对话框""" self.rename_dialog.set_values(None) self.rename_dialog.exec() def call_remove_files(self): indexs = [] infos = [] _indexs = self.table_disk.selectionModel().selection().indexes() if not _indexs: return for i in _indexs: # 获取所选行号 indexs.append(i.row()) indexs = set(indexs) for index in indexs: info = self.model_disk.item(index, 0).data() # 用于提示删除的文件名 if info: infos.append(info[:3]) delete_dialog = DeleteDialog(infos) delete_dialog.new_infos.connect(self.remove_files_worker.set_values) delete_dialog.exec() def generateMenu(self, pos): """右键菜单""" row_nums = self.sender().selectionModel().selection().indexes() if not row_nums: # 如果没有选中行,什么也不做 return _model = self.sender().model() infos = [] # 多个选中的行,用于移动文件与... for one_row in row_nums: one_row_data = _model.item(one_row.row(), 0).data() if one_row_data and one_row_data not in infos: # 删掉 .. 行 infos.append(one_row_data) if not infos: return info = infos[0] # 取选中的第一行 # 通过是否有文件 ID 判断是登录界面还是提取界面 if info[0]: self.left_menu_rename_set_desc.setEnabled(True) self.left_menu_set_pwd.setEnabled(True) # 通过infos第3个字段 size 判断是否为文件夹,文件夹不能移动,设置不同的显示菜单名 if info[2]: self.left_menu_rename_set_desc.setText("修改文件描述") self.left_menu_move.setEnabled(True) else: self.left_menu_rename_set_desc.setText("修改文件夹名与描述") self.left_menu_move.setDisabled(True) else: self.left_menu_rename_set_desc.setDisabled(True) self.left_menu_move.setDisabled(True) self.left_menu_set_pwd.setDisabled(True) action = self.left_menus.exec_(self.sender().mapToGlobal(pos)) if action == self.left_menu_share_url: # 显示详细信息 # 后台跟新信息,并显示信息对话框 self.more_info_worker.set_values(info, self._disk) elif action == self.left_menu_move: # 移动文件 self.all_folders_worker.set_values(self._disk, infos) elif action == self.left_menu_set_pwd: # 修改提取码 self.desc_pwd_fetcher.set_values(self._disk, [ info, ]) # 兼容下载器,使用列表的列表 self.set_pwd_dialog.set_values(info) self.set_pwd_dialog.exec() elif action == self.left_menu_rename_set_desc: # 重命名与修改描述 self.desc_pwd_fetcher.set_values(self._disk, [ info, ]) # 兼容下载器,使用列表的列表 self.rename_dialog.set_values(info) self.rename_dialog.exec() def call_update_desc_pwd(self, desc, pwd, infos): '''更新 desc、pwd''' infos[6] = desc infos[5] = pwd self.rename_dialog.set_values(infos) self.set_pwd_dialog.set_values(infos) def show_move_file_dialog(self, infos, all_dirs_dict): '''显示移动文件对话框''' move_file_dialog = MoveFileDialog(infos, all_dirs_dict) move_file_dialog.new_infos.connect( self.all_folders_worker.move_file) # 调用移动线程 move_file_dialog.exec() def show_info_dialog(self, infos): '''显示更多信息对话框''' info_dialog = InfoDialog(infos) info_dialog.setWindowModality(Qt.ApplicationModal) # 窗口前置 info_dialog.exec() def call_change_dir(self, folder_id=-1): """按钮调用""" def callfunc(): self.list_refresher.set_values(folder_id) return callfunc def change_dir(self, dir_name): """双击切换工作目录""" dir_name = self.model_disk.item(dir_name.row(), 0).text() # 文件夹名 if dir_name == "..": # 返回上级路径 self.list_refresher.set_values(self._parent_id) elif dir_name in self._folder_list.keys(): folder_id = self._folder_list[dir_name][0] self.list_refresher.set_values(folder_id) def call_upload(self, infos): """上传文件(夹)""" self._old_work_id = self._work_id # 记录上传文件夹id self.upload_worker.set_values(self._disk, infos, self._old_work_id) self.upload_worker.start() def show_full_path(self): """路径框显示当前路径""" index = 1 for name in self._path_list_old.items(): self._locs[index].clicked.disconnect() self.disk_loc.removeWidget(self._locs[index]) self._locs[index].deleteLater() self._locs[index] = None del self._locs[index] index += 1 index = 1 for name, fid in self._path_list.items(): self._locs[index] = QPushButton(name, self.disk_tab) self._locs[index].setIcon(QIcon("./icon/folder.gif")) self._locs[index].setStyleSheet( "QPushButton {border: none; background:transparent;}") self.disk_loc.insertWidget(index, self._locs[index]) self._locs[index].clicked.connect(self.call_change_dir(fid)) index += 1 self._path_list_old = self._path_list def select_all_btn(self, action="reverse"): """默认反转按钮状态""" page = self.tabWidget.currentIndex() if page == 0: btn = self.btn_share_select_all table = self.table_share elif page == 1: btn = self.btn_disk_select_all table = self.table_disk elif page == 2: return else: return if btn.isEnabled(): if action == "reverse": if btn.text() == "全选": table.selectAll() btn.setText("取消") btn.setIcon(QIcon("./icon/select-none.ico")) elif btn.text() == "取消": table.clearSelection() btn.setText("全选") btn.setIcon(QIcon("./icon/select-all.ico")) elif action == "cancel": # 点击列表其中一个就表示放弃全选 btn.setText("全选") btn.setIcon(QIcon("./icon/select-all.ico")) else: table.selectAll() btn.setText("取消") btn.setIcon(QIcon("./icon/select-none.ico")) def finished_upload(self): """上传完成调用""" if self._old_work_id == self._work_id: self.list_refresher.set_values(self._work_id, True, True, False) else: self._old_work_id = self._work_id self.show_status("上传完成!", 7000) def init_disk_ui(self): self.model_disk = QStandardItemModel(1, 3) self.config_tableview("disk") self.btn_disk_delete.setIcon(QIcon("./icon/delete.ico")) self.btn_disk_dl.setIcon(QIcon("./icon/downloader.ico")) self.btn_disk_select_all.setIcon(QIcon("./icon/select-all.ico")) self.btn_disk_select_all.setToolTip("按下 Ctrl/Alt + A 全选或则取消全选") self.btn_disk_select_all.clicked.connect( lambda: self.select_all_btn("reverse")) self.table_disk.clicked.connect(lambda: self.select_all_btn("cancel")) self.btn_disk_dl.clicked.connect(self.call_downloader) self.btn_disk_mkdir.setIcon(QIcon("./icon/add-folder.ico")) self.btn_disk_mkdir.clicked.connect(self.call_mkdir) self.btn_disk_delete.clicked.connect(self.call_remove_files) self.table_disk.doubleClicked.connect(self.change_dir) # 上传器 self.upload_worker = UploadWorker() self.upload_worker.finished.connect(self.finished_upload) self.upload_worker.code.connect(self.show_status) def show_status(self, msg, duration=0): self._msg_label.setText(msg) # self.statusbar.showMessage(msg, duration) # QCoreApplication.processEvents() # 重绘界面,在弱网络情况导致程序闪退 if duration != 0: QTimer.singleShot(duration, lambda: self._msg_label.setText("")) # shared url def call_get_shared_info(self): if not self.get_shared_info_thread.isRunning(): # 防止快速多次调用 self.line_share_url.setEnabled(False) self.btn_extract.setEnabled(False) text = self.line_share_url.text().strip() self.get_shared_info_thread.set_values(text) def show_share_url_file_lists(self, infos): if infos["code"] == LanZouCloud.SUCCESS: file_count = len(infos["info"].keys()) self.model_share.setHorizontalHeaderLabels( ["文件{}个".format(file_count), "大小", "时间"]) for infos in infos["info"].values(): name = QStandardItem(self.set_file_icon(infos[1]), infos[1]) name.setData(infos) self.model_share.appendRow( [name, QStandardItem(infos[2]), QStandardItem(infos[3])]) for r in range(self.model_share.rowCount()): # 右对齐 self.model_share.item(r, 1).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.model_share.item(r, 2).setTextAlignment(Qt.AlignRight | Qt.AlignVCenter) self.table_share.setDisabled(False) self.btn_share_select_all.setDisabled(False) self.btn_share_select_all.setToolTip("按下 Ctrl/Alt + A 全选或则取消全选") self.btn_share_dl.setDisabled(False) else: self.btn_share_select_all.setDisabled(True) self.btn_share_select_all.setToolTip("") self.btn_share_dl.setDisabled(True) self.table_share.setDisabled(True) def set_download_path(self): """设置下载路径""" dl_path = QFileDialog.getExistingDirectory() dl_path = os.path.normpath(dl_path) # windows backslash if dl_path == self.settings["path"]: return if dl_path == "": dl_path = os.path.dirname( os.path.abspath(__file__)) + os.sep + "downloads" up_info = {"path": dl_path} else: up_info = {"path": dl_path} update_settings(self._config, up_info) self.load_settings() self.line_dl_path.setText(self.settings["path"]) def init_extract_share_ui(self): self.btn_share_select_all.setDisabled(True) self.btn_share_dl.setDisabled(True) self.table_share.setDisabled(True) self.model_share = QStandardItemModel(1, 3) self.config_tableview("share") # 获取分享链接信息线程 self.get_shared_info_thread = GetSharedInfo() self.get_shared_info_thread.update.connect( lambda: self.model_share.removeRows(0, self.model_share.rowCount() )) # 清理旧的信息 self.get_shared_info_thread.msg.connect(self.show_status) # 提示信息 self.get_shared_info_thread.infos.connect( self.show_share_url_file_lists) # 内容信息 self.get_shared_info_thread.finished.connect( lambda: self.btn_extract.setEnabled(True)) self.get_shared_info_thread.finished.connect( lambda: self.line_share_url.setEnabled(True)) # 控件设置 self.line_share_url.setPlaceholderText( "蓝奏云链接,如有提取码,放后面,空格或汉字等分割,回车键提取") self.line_share_url.returnPressed.connect(self.call_get_shared_info) self.btn_extract.clicked.connect(self.call_get_shared_info) self.btn_share_dl.clicked.connect(self.call_downloader) self.btn_share_dl.setIcon(QIcon("./icon/downloader.ico")) self.btn_share_select_all.setIcon(QIcon("./icon/select-all.ico")) self.btn_share_select_all.clicked.connect( lambda: self.select_all_btn("reverse")) self.table_share.clicked.connect( lambda: self.select_all_btn("cancel")) # 全选按钮 # 添加文件下载路径选择器 self.line_dl_path = MyLineEdit(self.share_tab) self.line_dl_path.setObjectName("line_dl_path") self.horizontalLayout_share_2.insertWidget(2, self.line_dl_path) self.line_dl_path.setText(self.settings["path"]) self.line_dl_path.clicked.connect(self.set_download_path) # QSS self.label_share_url.setStyleSheet( "#label_share_url {color: rgb(255,255,60);}") self.label_dl_path.setStyleSheet( "#label_dl_path {color: rgb(255,255,60);}") def keyPressEvent(self, e): if e.key() == Qt.Key_A: # Ctrl/Alt + A 全选 if e.modifiers() and Qt.ControlModifier: self.select_all_btn() def set_window_at_center(self): screen = QDesktopWidget().screenGeometry() size = self.geometry() new_left = int((screen.width() - size.width()) / 2) new_top = int((screen.height() - size.height()) / 2) self.move(new_left, new_top) def open_wiki_url(self): # 打开使用说明页面 url = QUrl('https://github.com/rachpt/lanzou-gui/wiki') if not QDesktopServices.openUrl(url): self.show_status('Could not open wiki page!', 5000)
class GetSharedInfo(QThread): '''提取界面获取分享链接信息''' infos = pyqtSignal(object) msg = pyqtSignal(str, int) update = pyqtSignal() def __init__(self, parent=None): super(GetSharedInfo, self).__init__(parent) self._disk = LanZouCloud() self.share_url = "" self.pwd = "" self.is_file = "" self.is_folder = "" self._mutex = QMutex() self._is_work = False self._pat = r"(https?://(www\.)?lanzous.com/[bi][a-z0-9]+)[^0-9a-z]*([a-z0-9]+)?" def set_values(self, text): '''获取分享链接信息''' if not text: return for share_url, _, pwd in re.findall(self._pat, text): if is_file_url(share_url): # 文件链接 is_file = True is_folder = False self.msg.emit("正在获取文件链接信息……", 0) elif is_folder_url(share_url): # 文件夹链接 is_folder = True is_file = False self.msg.emit("正在获取文件夹链接信息,可能需要几秒钟,请稍后……", 0) else: self.msg.emit(f"{share_url} 为非法链接!", 0) self.btn_extract.setEnabled(True) self.line_share_url.setEnabled(True) return self.update.emit() # 清理旧的显示信息 self.share_url = share_url self.pwd = pwd self.is_file = is_file self.is_folder = is_folder self.start() break def __del__(self): self.wait() def stop(self): # 用于手动停止 self._mutex.lock() self._is_work = False self._mutex.unlock() def emit_msg(self, infos): '''根据查询信息发送状态信号''' show_time = 7000 # 提示显示时间,单位 ms if infos["code"] == LanZouCloud.FILE_CANCELLED: self.msg.emit("<font color='red'>文件不存在,或已删除!</font>", show_time) elif infos["code"] == LanZouCloud.URL_INVALID: self.msg.emit("<font color='red'>链接非法!</font>", show_time) elif infos["code"] == LanZouCloud.PASSWORD_ERROR: self.msg.emit( "<font color='red'>提取码 [<b><font color='magenta'>{}</font></b>] 错误!</font>" .format(self.pwd), show_time) elif infos["code"] == LanZouCloud.LACK_PASSWORD: self.msg.emit("<font color='red'>请在链接后面跟上提取码,空格分割!</font>", show_time) elif infos["code"] == LanZouCloud.NETWORK_ERROR: self.msg.emit( "<font color='red'>网络错误!{}</font>".format(infos["info"]), show_time) elif infos["code"] == LanZouCloud.SUCCESS: self.msg.emit("<font color='#00CC00'>提取成功!</font>", show_time) def run(self): if not self._is_work: self._mutex.lock() self._is_work = True try: if self.is_file: # 链接为文件 _infos = self._disk.get_share_file_info( self.share_url, self.pwd) self.emit_msg(_infos) elif self.is_folder: # 链接为文件夹 _infos = self._disk.get_share_folder_info( self.share_url, self.pwd) self.emit_msg(_infos) self.infos.emit(_infos) except TimeoutError: self.msg.emit("font color='red'>网络超时!请稍后重试</font>", 8000) self._is_work = False self._mutex.unlock() else: self.msg.emit("<font color='blue'>后台正在运行,稍后重试!</font>", 3000)
import os import sys from lanzou.api import LanZouCloud if __name__ == '__main__': lzy = LanZouCloud() cookie = {'ylogin': os.getenv('LZ_YLOGIN'), 'phpdisk_info': os.getenv('LZ_PHPDISK_INFO')} lzy.login_by_cookie(cookie) bilibili_hd_dir_id = lzy.get_dir_list().find_by_name('BilibiliHD2').id bilibili_hd_v01_dir_id = lzy.get_dir_list(bilibili_hd_dir_id).find_by_name('0.1').id def handler(fid, is_file): lzy.set_passwd(fid, '233', is_file) code = lzy.upload_file(sys.argv[1], bilibili_hd_v01_dir_id, uploaded_handler=handler) if code == 0: print('success') else: print(f'::warning :: upload to lanzou fail with code ${code}') exit(code)
def __init__(self, cookie): self.lzy = LanZouCloud() self.login_ok = self.lzy.login_by_cookie(cookie) == LanZouCloud.SUCCESS if self.login_ok: self._folder_dnf_calc_code = self.lzy.get_folder_info_by_id(self._folder_id_dnf_calc_code).folder self._folder_dnf_calc_publish = self.lzy.get_folder_info_by_id(self._folder_id_dnf_calc_publish).folder
class Uploader: # 源码目录 _folder_id_dnf_calc_code = "1989903" # 版本目录 _folder_id_dnf_calc_publish = "1860455" def __init__(self, cookie): self.lzy = LanZouCloud() self.login_ok = self.lzy.login_by_cookie(cookie) == LanZouCloud.SUCCESS if self.login_ok: self._folder_dnf_calc_code = self.lzy.get_folder_info_by_id(self._folder_id_dnf_calc_code).folder self._folder_dnf_calc_publish = self.lzy.get_folder_info_by_id(self._folder_id_dnf_calc_publish).folder def upload_to_lanzouyun(self, filepath, target_folder): def on_uploaded(fid, is_file): if not is_file: return files = self.lzy.get_file_list(target_folder.id) self.lzy.move_file(fid, target_folder.id) # 上传到指定的文件夹中 retCode = self.lzy.upload_file(filepath, -1, callback=self.show_progress, uploaded_handler=on_uploaded) if retCode != LanZouCloud.SUCCESS: # logger.error("上传失败,retCode={}".format(retCode)) return False return True def show_progress(self, file_name, total_size, now_size): """显示进度的回调函数""" percent = now_size / total_size bar_len = 40 # 进度条长总度 bar_str = '>' * round(bar_len * percent) + '=' * round(bar_len * (1 - percent)) print('\r{:.2f}%\t[{}] {:.1f}/{:.1f}MB | {} '.format( percent * 100, bar_str, now_size / 1048576, total_size / 1048576, file_name), end='') if total_size == now_size: print('') # 下载完成换行 def clearOldVersion(self): # 清空版本目录 path = self.lzy.get_file_list(self._folder_id_dnf_calc_code) for item in path: if item.name.startswith("源码"): self.lzy.delete(item.id) path = self.lzy.get_file_list(self._folder_id_dnf_calc_publish) for item in path: if item.name.startswith("DNF计算器") or item.name.startswith("更新日志"): self.lzy.delete(item.id)
def __init__(self, cookie): self.lzy = LanZouCloud() self.login_ok = self.lzy.login_by_cookie(cookie) == LanZouCloud.SUCCESS
from lanzou.api import LanZouCloud import requests import sys lzy = LanZouCloud() def handler(fid, is_file): if is_file: lzy.set_desc(fid, '文件由GitHub Action修改', is_file=True) print("已上传") def show_progress(file_name, total_size, now_size): """显示进度的回调函数""" percent = now_size / total_size bar_len = 40 # 进度条长总度 bar_str = '>' * round(bar_len * percent) + '=' * round(bar_len * (1 - percent)) print('\r{:.2f}%\t[{}] {:.1f}/{:.1f}MB | {} '.format( percent * 100, bar_str, now_size / 1048576, total_size / 1048576, file_name), end='') if total_size == now_size: print('') # 下载完成换行 def upload(path, id, trynum=1, num=0): """ path:文件路径 id:文件夹id trynum:可执行次数 num:执行次数 """ print('开始上传文件') code = lzy.upload_file(path,