def initUI(self):
self.setWindowTitle("移动文件(夹)")
self.setWindowIcon(QIcon(SRC_DIR + "move.ico"))
self.lb_name = QLabel()
self.lb_name.setText("文件(夹)名：")
self.lb_name.setAlignment(Qt.AlignRight | Qt.AlignTrailing
| Qt.AlignVCenter)
self.tx_name = AutoResizingTextEdit()
self.tx_name.setFocusPolicy(Qt.NoFocus) # 只读
self.tx_name.setReadOnly(True)
self.lb_new_path = QLabel()
self.lb_new_path.setText("目标文件夹：")
self.lb_new_path.setAlignment(Qt.AlignRight | Qt.AlignTrailing
| Qt.AlignVCenter)
self.tx_new_path = QComboBox()
self.buttonBox = QDialogButtonBox()
self.buttonBox.setOrientation(Qt.Horizontal)
self.buttonBox.setStandardButtons(QDialogButtonBox.Ok
| QDialogButtonBox.Cancel)
self.buttonBox.button(QDialogButtonBox.Ok).setText("确定")
self.buttonBox.button(QDialogButtonBox.Cancel).setText("取消")
self.grid = QGridLayout()
self.grid.setSpacing(10)
self.grid.addWidget(self.lb_name, 1, 0)
self.grid.addWidget(self.tx_name, 1, 1)
self.grid.addWidget(self.lb_new_path, 2, 0)
self.grid.addWidget(self.tx_new_path, 2, 1)
self.grid.addWidget(self.buttonBox, 3, 0, 1, 2)
self.setLayout(self.grid)
self.buttonBox.accepted.connect(self.btn_ok)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.setMinimumWidth(280)
def initUI(self):
self.setWindowTitle("设置")
logo = QLabel()
logo.setPixmap(QPixmap(SRC_DIR + "logo2.gif"))
logo.setStyleSheet("background-color:rgb(255,255,255);")
logo.setAlignment(Qt.AlignCenter)
self.download_threads_lb = QLabel("同时下载文件数")
self.download_threads_var = QLineEdit()
self.download_threads_var.setPlaceholderText("范围：1-9")
self.download_threads_var.setToolTip("范围：1-9")
self.download_threads_var.setInputMask("D")
self.max_size_lb = QLabel("分卷大小(MB)")
self.max_size_var = QLineEdit()
self.max_size_var.setPlaceholderText("普通用户最大100，vip用户根据具体情况设置")
self.max_size_var.setToolTip("普通用户最大100，vip用户根据具体情况设置")
self.max_size_var.setInputMask("D99")
self.timeout_lb = QLabel("请求超时(秒)")
self.timeout_var = QLineEdit()
self.timeout_var.setPlaceholderText("范围：1-99")
self.timeout_var.setToolTip("范围：1-99")
self.timeout_var.setInputMask("D9")
self.upload_delay_lb = QLabel("上传延时(秒)")
self.upload_delay_var = QLineEdit()
self.upload_delay_var.setPlaceholderText("范围：1-99")
self.upload_delay_var.setToolTip("范围：1-99")
self.upload_delay_var.setInputMask("D9")
self.dl_path_lb = QLabel("下载保存路径")
self.dl_path_var = MyLineEdit(self)
self.dl_path_var.clicked.connect(self.set_download_path)
self.time_fmt_box = QCheckBox("使用[年-月-日]时间格式")
self.time_fmt_box.setToolTip("文件上传日期显示格式")
self.to_tray_box = QCheckBox("关闭到系统托盘")
self.to_tray_box.setToolTip("点击关闭软件按钮是最小化软件至系统托盘")
self.watch_clipboard_box = QCheckBox("监听系统剪切板")
self.watch_clipboard_box.setToolTip("检测到系统剪切板中有符合规范的蓝奏链接时自动唤起软件，并提取")
self.debug_box = QCheckBox("开启调试日志")
self.debug_box.setToolTip("记录软件 debug 信息至 debug-lanzou-gui.log 文件")
self.set_pwd_box = QCheckBox("上传文件自动设置密码")
self.set_pwd_var = AutoResizingTextEdit()
self.set_pwd_var.setPlaceholderText(" 2-8 位数字或字母")
self.set_pwd_var.setToolTip("2-8 位数字或字母")
self.set_desc_box = QCheckBox("上传文件自动设置描述")
self.set_desc_var = AutoResizingTextEdit()
self.big_file_box = QCheckBox(f"允许上传超过 {self.max_size}MB 的大文件")
self.big_file_box.setToolTip("开启大文件上传支持 (功能下线)")
self.upgrade_box = QCheckBox("自动检测新版本")
self.upgrade_box.setToolTip("在软件打开时自动检测是否有新的版本发布，如有则弹出更新信息")
self.time_fmt_box.toggle()
self.time_fmt_box.stateChanged.connect(self.change_time_fmt)
self.to_tray_box.stateChanged.connect(self.change_to_tray)
self.watch_clipboard_box.stateChanged.connect(
self.change_watch_clipboard)
self.debug_box.stateChanged.connect(self.change_debug)
self.set_pwd_box.stateChanged.connect(self.change_set_pwd)
self.set_pwd_var.editingFinished.connect(self.check_pwd)
self.set_desc_box.stateChanged.connect(self.change_set_desc)
self.big_file_box.stateChanged.connect(self.change_big_file)
self.upgrade_box.stateChanged.connect(self.change_upgrade)
buttonBox = QDialogButtonBox()
buttonBox.setOrientation(Qt.Horizontal)
buttonBox.setStandardButtons(QDialogButtonBox.Reset
| QDialogButtonBox.Save
| QDialogButtonBox.Cancel)
buttonBox.button(QDialogButtonBox.Reset).setText("重置")
buttonBox.button(QDialogButtonBox.Save).setText("保存")
buttonBox.button(QDialogButtonBox.Cancel).setText("取消")
buttonBox.button(QDialogButtonBox.Reset).clicked.connect(
lambda: self.set_values(reset=True))
buttonBox.button(QDialogButtonBox.Save).clicked.connect(self.slot_save)
buttonBox.rejected.connect(self.reject)
form = QFormLayout()
form.setLabelAlignment(Qt.AlignRight)
form.setSpacing(10)
form.addRow(self.download_threads_lb, self.download_threads_var)
form.addRow(self.timeout_lb, self.timeout_var)
form.addRow(self.upload_delay_lb, self.upload_delay_var)
form.addRow(self.max_size_lb, self.max_size_var)
form.addRow(self.dl_path_lb, self.dl_path_var)
vbox = QVBoxLayout()
vbox.addWidget(logo)
vbox.addStretch(1)
vbox.addLayout(form)
vbox.addStretch(1)
hbox = QHBoxLayout()
hbox.addWidget(self.time_fmt_box)
hbox.addWidget(self.to_tray_box)
hbox.addWidget(self.watch_clipboard_box)
hbox.addWidget(self.debug_box)
vbox.addLayout(hbox)
vbox.addStretch(1)
hbox_2 = QHBoxLayout()
hbox_2.addWidget(self.set_pwd_box)
hbox_2.addWidget(self.set_pwd_var)
vbox.addLayout(hbox_2)
vbox.addStretch(1)
hbox_3 = QHBoxLayout()
hbox_3.addWidget(self.set_desc_box)
hbox_3.addWidget(self.set_desc_var)
vbox.addLayout(hbox_3)
hbox_4 = QHBoxLayout()
hbox_4.addWidget(self.big_file_box)
hbox_4.addWidget(self.upgrade_box)
vbox.addStretch(1)
vbox.addLayout(hbox_4)
vbox.addStretch(2)
vbox.addWidget(buttonBox)
self.setLayout(vbox)
self.setMinimumWidth(500)
class SettingDialog(QDialog):
saved = pyqtSignal()
def __init__(self, parent=None):
super(SettingDialog, self).__init__(parent)
self._config = object
self.download_threads = 3
self.max_size = 100
self.timeout = 5
self.dl_path = None
self.time_fmt = False
self.to_tray = False
self.watch_clipboard = False
self.debug = False
self.set_pwd = False
self.set_desc = False
self.upload_delay = 0
self.allow_big_file = False
self.upgrade = True
self.pwd = ""
self.desc = ""
self.initUI()
self.setStyleSheet(dialog_qss_style)
def open_dialog(self, config):
""""打开前先更新一下显示界面"""
self._config = config
if self._config.name:
self.setWindowTitle(f"设置 <{self._config.name}>")
else:
self.setWindowTitle("设置")
self.cwd = self._config.path
self.set_values()
self.exec()
def show_values(self):
"""控件显示值"""
self.download_threads_var.setText(str(self.download_threads))
self.max_size_var.setText(str(self.max_size))
self.timeout_var.setText(str(self.timeout))
self.dl_path_var.setText(str(self.dl_path))
self.time_fmt_box.setChecked(self.time_fmt)
self.to_tray_box.setChecked(self.to_tray)
self.watch_clipboard_box.setChecked(self.watch_clipboard)
self.debug_box.setChecked(self.debug)
self.set_pwd_box.setChecked(self.set_pwd)
self.set_pwd_var.setEnabled(self.set_pwd)
self.set_pwd_var.setText(self.pwd)
self.set_desc_box.setChecked(self.set_desc)
self.set_desc_var.setEnabled(self.set_desc)
self.set_desc_var.setText(self.desc)
self.upload_delay_var.setText(str(self.upload_delay))
self.big_file_box.setChecked(self.allow_big_file)
self.big_file_box.setText(f"允许上传超过 {self.max_size}MB 的大文件")
self.big_file_box.setDisabled(True) # 关闭允许上传大文件设置入口
self.upgrade_box.setChecked(self.upgrade)
def set_values(self, reset=False):
"""设置控件对应变量初始值"""
settings = self._config.default_settings if reset else self._config.settings
self.download_threads = settings["download_threads"]
self.max_size = settings["max_size"]
self.timeout = settings["timeout"]
self.dl_path = settings["dl_path"]
self.time_fmt = settings["time_fmt"]
self.to_tray = settings["to_tray"]
self.watch_clipboard = settings["watch_clipboard"]
self.debug = settings["debug"]
self.set_pwd = settings["set_pwd"]
self.pwd = settings["pwd"]
self.set_desc = settings["set_desc"]
self.desc = settings["desc"]
self.upload_delay = settings["upload_delay"]
if 'upgrade' in settings:
self.upgrade = settings["upgrade"]
self.show_values()
def get_values(self) -> dict:
"""读取输入控件的值"""
if self.download_threads_var.text():
self.download_threads = int(self.download_threads_var.text())
if self.max_size_var.text():
self.max_size = int(self.max_size_var.text())
if self.timeout_var.text():
self.timeout = int(self.timeout_var.text())
if self.upload_delay_var.text():
self.upload_delay = int(self.upload_delay_var.text())
self.dl_path = str(self.dl_path_var.text())
self.pwd = str(self.set_pwd_var.toPlainText())
self.desc = str(self.set_desc_var.toPlainText())
return {
"download_threads": self.download_threads,
"max_size": self.max_size,
"timeout": self.timeout,
"dl_path": self.dl_path,
"time_fmt": self.time_fmt,
"to_tray": self.to_tray,
"watch_clipboard": self.watch_clipboard,
"debug": self.debug,
"set_pwd": self.set_pwd,
"pwd": self.pwd,
"set_desc": self.set_desc,
"desc": self.desc,
"upload_delay": self.upload_delay,
"allow_big_file": self.allow_big_file,
"upgrade": self.upgrade
}
def initUI(self):
self.setWindowTitle("设置")
logo = QLabel()
logo.setPixmap(QPixmap(SRC_DIR + "logo2.gif"))
logo.setStyleSheet("background-color:rgb(255,255,255);")
logo.setAlignment(Qt.AlignCenter)
self.download_threads_lb = QLabel("同时下载文件数")
self.download_threads_var = QLineEdit()
self.download_threads_var.setPlaceholderText("范围：1-9")
self.download_threads_var.setToolTip("范围：1-9")
self.download_threads_var.setInputMask("D")
self.max_size_lb = QLabel("分卷大小(MB)")
self.max_size_var = QLineEdit()
self.max_size_var.setPlaceholderText("普通用户最大100，vip用户根据具体情况设置")
self.max_size_var.setToolTip("普通用户最大100，vip用户根据具体情况设置")
self.max_size_var.setInputMask("D99")
self.timeout_lb = QLabel("请求超时(秒)")
self.timeout_var = QLineEdit()
self.timeout_var.setPlaceholderText("范围：1-99")
self.timeout_var.setToolTip("范围：1-99")
self.timeout_var.setInputMask("D9")
self.upload_delay_lb = QLabel("上传延时(秒)")
self.upload_delay_var = QLineEdit()
self.upload_delay_var.setPlaceholderText("范围：1-99")
self.upload_delay_var.setToolTip("范围：1-99")
self.upload_delay_var.setInputMask("D9")
self.dl_path_lb = QLabel("下载保存路径")
self.dl_path_var = MyLineEdit(self)
self.dl_path_var.clicked.connect(self.set_download_path)
self.time_fmt_box = QCheckBox("使用[年-月-日]时间格式")
self.time_fmt_box.setToolTip("文件上传日期显示格式")
self.to_tray_box = QCheckBox("关闭到系统托盘")
self.to_tray_box.setToolTip("点击关闭软件按钮是最小化软件至系统托盘")
self.watch_clipboard_box = QCheckBox("监听系统剪切板")
self.watch_clipboard_box.setToolTip("检测到系统剪切板中有符合规范的蓝奏链接时自动唤起软件，并提取")
self.debug_box = QCheckBox("开启调试日志")
self.debug_box.setToolTip("记录软件 debug 信息至 debug-lanzou-gui.log 文件")
self.set_pwd_box = QCheckBox("上传文件自动设置密码")
self.set_pwd_var = AutoResizingTextEdit()
self.set_pwd_var.setPlaceholderText(" 2-8 位数字或字母")
self.set_pwd_var.setToolTip("2-8 位数字或字母")
self.set_desc_box = QCheckBox("上传文件自动设置描述")
self.set_desc_var = AutoResizingTextEdit()
self.big_file_box = QCheckBox(f"允许上传超过 {self.max_size}MB 的大文件")
self.big_file_box.setToolTip("开启大文件上传支持 (功能下线)")
self.upgrade_box = QCheckBox("自动检测新版本")
self.upgrade_box.setToolTip("在软件打开时自动检测是否有新的版本发布，如有则弹出更新信息")
self.time_fmt_box.toggle()
self.time_fmt_box.stateChanged.connect(self.change_time_fmt)
self.to_tray_box.stateChanged.connect(self.change_to_tray)
self.watch_clipboard_box.stateChanged.connect(
self.change_watch_clipboard)
self.debug_box.stateChanged.connect(self.change_debug)
self.set_pwd_box.stateChanged.connect(self.change_set_pwd)
self.set_pwd_var.editingFinished.connect(self.check_pwd)
self.set_desc_box.stateChanged.connect(self.change_set_desc)
self.big_file_box.stateChanged.connect(self.change_big_file)
self.upgrade_box.stateChanged.connect(self.change_upgrade)
buttonBox = QDialogButtonBox()
buttonBox.setOrientation(Qt.Horizontal)
buttonBox.setStandardButtons(QDialogButtonBox.Reset
| QDialogButtonBox.Save
| QDialogButtonBox.Cancel)
buttonBox.button(QDialogButtonBox.Reset).setText("重置")
buttonBox.button(QDialogButtonBox.Save).setText("保存")
buttonBox.button(QDialogButtonBox.Cancel).setText("取消")
buttonBox.button(QDialogButtonBox.Reset).clicked.connect(
lambda: self.set_values(reset=True))
buttonBox.button(QDialogButtonBox.Save).clicked.connect(self.slot_save)
buttonBox.rejected.connect(self.reject)
form = QFormLayout()
form.setLabelAlignment(Qt.AlignRight)
form.setSpacing(10)
form.addRow(self.download_threads_lb, self.download_threads_var)
form.addRow(self.timeout_lb, self.timeout_var)
form.addRow(self.upload_delay_lb, self.upload_delay_var)
form.addRow(self.max_size_lb, self.max_size_var)
form.addRow(self.dl_path_lb, self.dl_path_var)
vbox = QVBoxLayout()
vbox.addWidget(logo)
vbox.addStretch(1)
vbox.addLayout(form)
vbox.addStretch(1)
hbox = QHBoxLayout()
hbox.addWidget(self.time_fmt_box)
hbox.addWidget(self.to_tray_box)
hbox.addWidget(self.watch_clipboard_box)
hbox.addWidget(self.debug_box)
vbox.addLayout(hbox)
vbox.addStretch(1)
hbox_2 = QHBoxLayout()
hbox_2.addWidget(self.set_pwd_box)
hbox_2.addWidget(self.set_pwd_var)
vbox.addLayout(hbox_2)
vbox.addStretch(1)
hbox_3 = QHBoxLayout()
hbox_3.addWidget(self.set_desc_box)
hbox_3.addWidget(self.set_desc_var)
vbox.addLayout(hbox_3)
hbox_4 = QHBoxLayout()
hbox_4.addWidget(self.big_file_box)
hbox_4.addWidget(self.upgrade_box)
vbox.addStretch(1)
vbox.addLayout(hbox_4)
vbox.addStretch(2)
vbox.addWidget(buttonBox)
self.setLayout(vbox)
self.setMinimumWidth(500)
def change_time_fmt(self, state):
if state == Qt.Checked:
self.time_fmt = True
else:
self.time_fmt = False
def change_to_tray(self, state):
if state == Qt.Checked:
self.to_tray = True
else:
self.to_tray = False
def change_watch_clipboard(self, state):
if state == Qt.Checked:
self.watch_clipboard = True
else:
self.watch_clipboard = False
def change_debug(self, state):
if state == Qt.Checked:
self.debug = True
else:
self.debug = False
def change_big_file(self, state):
if state == Qt.Checked:
self.allow_big_file = True
else:
self.allow_big_file = False
def change_upgrade(self, state):
if state == Qt.Checked:
self.upgrade = True
else:
self.upgrade = False
def change_set_pwd(self, state):
if state == Qt.Checked:
self.set_pwd = True
self.set_pwd_var.setDisabled(False)
else:
self.set_pwd = False
self.set_pwd_var.setDisabled(True)
def change_set_desc(self, state):
if state == Qt.Checked:
self.set_desc = True
self.set_desc_var.setDisabled(False)
else:
self.set_desc = False
self.set_desc_var.setDisabled(True)
def check_pwd(self):
pwd = self.set_pwd_var.toPlainText()
pwd = ''.join(list(filter(str.isalnum, pwd)))
if len(pwd) < 2:
pwd = ""
self.set_pwd_var.setText(pwd[:8])
def set_download_path(self):
"""设置下载路径"""
dl_path = QFileDialog.getExistingDirectory(self, "选择文件下载保存文件夹",
self.cwd)
dl_path = os.path.normpath(dl_path) # windows backslash
if dl_path == self.dl_path or dl_path == ".":
return None
self.dl_path_var.setText(dl_path)
self.dl_path = dl_path
def slot_save(self):
"""保存槽函数"""
self._config.settings = self.get_values()
self.saved.emit()
self.close()
class MoveFileDialog(QDialog):
'''移动文件对话框'''
new_infos = pyqtSignal(object)
def __init__(self, parent=None):
super(MoveFileDialog, self).__init__(parent)
self.infos = None
self.dirs = {}
self.initUI()
self.setStyleSheet(dialog_qss_style)
def update_ui(self):
names = "\n".join([i.name for i in self.infos])
names_tip = "\n".join([i.name for i in self.infos])
self.tx_name.setText(names)
self.tx_name.setToolTip(names_tip)
self.tx_new_path.clear()
f_icon = QIcon(SRC_DIR + "folder.gif")
for f_name, fid in self.dirs.items():
if len(f_name) > 50: # 防止文件夹名字过长？
f_name = f_name[:47] + "..."
self.tx_new_path.addItem(f_icon,
"id：{:>8}，name：{}".format(fid, f_name))
def set_values(self, infos, all_dirs_dict):
self.infos = infos
self.dirs = all_dirs_dict
self.update_ui()
self.exec()
def initUI(self):
self.setWindowTitle("移动文件(夹)")
self.setWindowIcon(QIcon(SRC_DIR + "move.ico"))
self.lb_name = QLabel()
self.lb_name.setText("文件(夹)名：")
self.lb_name.setAlignment(Qt.AlignRight | Qt.AlignTrailing
| Qt.AlignVCenter)
self.tx_name = AutoResizingTextEdit()
self.tx_name.setFocusPolicy(Qt.NoFocus) # 只读
self.tx_name.setReadOnly(True)
self.lb_new_path = QLabel()
self.lb_new_path.setText("目标文件夹：")
self.lb_new_path.setAlignment(Qt.AlignRight | Qt.AlignTrailing
| Qt.AlignVCenter)
self.tx_new_path = QComboBox()
self.buttonBox = QDialogButtonBox()
self.buttonBox.setOrientation(Qt.Horizontal)
self.buttonBox.setStandardButtons(QDialogButtonBox.Ok
| QDialogButtonBox.Cancel)
self.buttonBox.button(QDialogButtonBox.Ok).setText("确定")
self.buttonBox.button(QDialogButtonBox.Cancel).setText("取消")
self.grid = QGridLayout()
self.grid.setSpacing(10)
self.grid.addWidget(self.lb_name, 1, 0)
self.grid.addWidget(self.tx_name, 1, 1)
self.grid.addWidget(self.lb_new_path, 2, 0)
self.grid.addWidget(self.tx_new_path, 2, 1)
self.grid.addWidget(self.buttonBox, 3, 0, 1, 2)
self.setLayout(self.grid)
self.buttonBox.accepted.connect(self.btn_ok)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.setMinimumWidth(280)
def btn_ok(self):
new_id = self.tx_new_path.currentText().split("，")[0].split("：")[1]
for index in range(len(self.infos)):
self.infos[index].new_id = int(new_id)
self.new_infos.emit(self.infos)
class InfoDialog(QDialog):
"""文件信息对话框"""
get_dl_link = pyqtSignal(str, str)
closed = pyqtSignal()
def __init__(self, parent=None):
super().__init__(parent)
self.infos = None
self.initUI()
self.setStyleSheet(dialog_qss_style)
def update_ui(self):
self.tx_dl_link.setPlaceholderText("单击获取")
self.tx_name.setText(self.infos.name)
if self.infos.is_file:
self.setWindowTitle("文件信息")
self.lb_name.setText("文件名：")
self.lb_desc.setText("文件描述：")
self.tx_dl_link.setText("") # 清空旧的信息
self.lb_dl_link.setVisible(True)
self.tx_dl_link.setVisible(True)
else:
self.setWindowTitle("文件夹信息")
self.lb_name.setText("文件夹名：")
self.lb_desc.setText("文件夹描述：")
self.lb_dl_link.setVisible(False)
self.tx_dl_link.setVisible(False)
if self.infos.size:
self.tx_size.setText(self.infos.size)
self.lb_size.setVisible(True)
self.tx_size.setVisible(True)
else:
self.tx_size.setVisible(False)
self.lb_size.setVisible(False)
if self.infos.time:
self.lb_time.setVisible(True)
self.tx_time.setVisible(True)
self.tx_time.setText(self.infos.time)
else:
self.lb_time.setVisible(False)
self.tx_time.setVisible(False)
if self.infos.downs:
self.lb_dl_count.setVisible(True)
self.tx_dl_count.setVisible(True)
self.tx_dl_count.setText(str(self.infos.downs))
else:
self.tx_dl_count.setVisible(False)
self.lb_dl_count.setVisible(False)
if self.infos.pwd:
self.tx_pwd.setText(self.infos.pwd)
self.tx_pwd.setPlaceholderText("")
else:
self.tx_pwd.setText("")
self.tx_pwd.setPlaceholderText("无")
if self.infos.desc:
self.tx_desc.setText(self.infos.desc)
self.tx_desc.setPlaceholderText("")
else:
self.tx_desc.setText("")
self.tx_desc.setPlaceholderText("无")
self.tx_share_url.setText(self.infos.url)
self.adjustSize()
def set_values(self, infos):
self.infos = infos
self.update_ui()
def set_dl_link_tx(self, text):
self.tx_dl_link.setText(text)
self.adjustSize()
def call_get_dl_link(self):
url = self.tx_share_url.text()
pwd = self.tx_pwd.text()
self.get_dl_link.emit(url, pwd)
self.tx_dl_link.setPlaceholderText("后台获取中，请稍后！")
def call_get_short_url(self):
self.tx_short.setPlaceholderText("后台获取中，请稍后！")
url = self.tx_share_url.text()
from lanzou.api.extra import get_short_url
short_url = get_short_url(url)
if short_url:
self.tx_short.setText(short_url)
self.tx_short.setPlaceholderText("")
else:
self.tx_short.setText("")
self.tx_short.setPlaceholderText("生成失败！")
def clean(self):
self.tx_short.setText("")
self.tx_short.setPlaceholderText("单击获取")
def initUI(self):
self.setWindowIcon(QIcon("./src/share.ico"))
self.setWindowTitle("文件信息")
self.buttonBox = QDialogButtonBox()
self.buttonBox.setOrientation(Qt.Horizontal)
self.buttonBox.setStandardButtons(QDialogButtonBox.Close)
self.buttonBox.button(QDialogButtonBox.Close).setText("关闭")
self.buttonBox.rejected.connect(self.reject)
self.buttonBox.rejected.connect(self.clean)
self.buttonBox.rejected.connect(self.closed.emit)
self.logo = QLabel()
self.logo.setPixmap(QPixmap("./src/q9.gif"))
self.logo.setAlignment(Qt.AlignCenter)
self.logo.setStyleSheet("background-color:rgb(255,204,51);")
self.lb_name = QLabel()
self.lb_name.setText("文件名：")
self.tx_name = AutoResizingTextEdit()
self.tx_name.setReadOnly(True)
self.tx_name.setMinimumLines(1)
self.lb_size = QLabel()
self.lb_size.setText("文件大小：")
self.tx_size = QLabel()
self.lb_time = QLabel()
self.lb_time.setText("上传时间：")
self.tx_time = QLabel()
self.lb_dl_count = QLabel()
self.lb_dl_count.setText("下载次数：")
self.tx_dl_count = QLabel()
self.lb_share_url = QLabel()
self.lb_share_url.setText("分享链接：")
self.tx_share_url = QLineEdit()
self.tx_share_url.setReadOnly(True)
self.lb_pwd = QLabel()
self.lb_pwd.setText("提取码：")
self.tx_pwd = QLineEdit()
self.tx_pwd.setReadOnly(True)
self.lb_short = QLabel()
self.lb_short.setText("短链接：")
self.tx_short = AutoResizingTextEdit(self)
self.tx_short.setPlaceholderText("单击获取")
self.tx_short.clicked.connect(self.call_get_short_url)
self.tx_short.setReadOnly(True)
self.tx_short.setMinimumLines(1)
self.lb_desc = QLabel()
self.lb_desc.setText("文件描述：")
self.tx_desc = AutoResizingTextEdit()
self.tx_desc.setReadOnly(True)
self.tx_desc.setMinimumLines(1)
self.lb_dl_link = QLabel()
self.lb_dl_link.setText("下载直链：")
self.tx_dl_link = AutoResizingTextEdit(self)
self.tx_dl_link.setPlaceholderText("单击获取")
self.tx_dl_link.clicked.connect(self.call_get_dl_link)
self.tx_dl_link.setReadOnly(True)
self.tx_dl_link.setMinimumLines(1)
vbox = QVBoxLayout()
vbox.addWidget(self.logo)
vbox.addStretch(1)
form = QFormLayout()
form.setLabelAlignment(Qt.AlignRight)
form.addRow(self.lb_name, self.tx_name)
form.addRow(self.lb_size, self.tx_size)
form.addRow(self.lb_time, self.tx_time)
form.addRow(self.lb_dl_count, self.tx_dl_count)
form.addRow(self.lb_share_url, self.tx_share_url)
form.addRow(self.lb_pwd, self.tx_pwd)
form.addRow(self.lb_short, self.tx_short)
form.addRow(self.lb_desc, self.tx_desc)
form.addRow(self.lb_dl_link, self.tx_dl_link)
vbox.addLayout(form)
vbox.addStretch(1)
vbox.addWidget(self.buttonBox)
self.setLayout(vbox)
self.setMinimumWidth(500)
def initUI(self):
self.setWindowIcon(QIcon("./src/share.ico"))
self.setWindowTitle("文件信息")
self.buttonBox = QDialogButtonBox()
self.buttonBox.setOrientation(Qt.Horizontal)
self.buttonBox.setStandardButtons(QDialogButtonBox.Close)
self.buttonBox.button(QDialogButtonBox.Close).setText("关闭")
self.buttonBox.rejected.connect(self.reject)
self.buttonBox.rejected.connect(self.clean)
self.buttonBox.rejected.connect(self.closed.emit)
self.logo = QLabel()
self.logo.setPixmap(QPixmap("./src/q9.gif"))
self.logo.setAlignment(Qt.AlignCenter)
self.logo.setStyleSheet("background-color:rgb(255,204,51);")
self.lb_name = QLabel()
self.lb_name.setText("文件名：")
self.tx_name = AutoResizingTextEdit()
self.tx_name.setReadOnly(True)
self.tx_name.setMinimumLines(1)
self.lb_size = QLabel()
self.lb_size.setText("文件大小：")
self.tx_size = QLabel()
self.lb_time = QLabel()
self.lb_time.setText("上传时间：")
self.tx_time = QLabel()
self.lb_dl_count = QLabel()
self.lb_dl_count.setText("下载次数：")
self.tx_dl_count = QLabel()
self.lb_share_url = QLabel()
self.lb_share_url.setText("分享链接：")
self.tx_share_url = QLineEdit()
self.tx_share_url.setReadOnly(True)
self.lb_pwd = QLabel()
self.lb_pwd.setText("提取码：")
self.tx_pwd = QLineEdit()
self.tx_pwd.setReadOnly(True)
self.lb_short = QLabel()
self.lb_short.setText("短链接：")
self.tx_short = AutoResizingTextEdit(self)
self.tx_short.setPlaceholderText("单击获取")
self.tx_short.clicked.connect(self.call_get_short_url)
self.tx_short.setReadOnly(True)
self.tx_short.setMinimumLines(1)
self.lb_desc = QLabel()
self.lb_desc.setText("文件描述：")
self.tx_desc = AutoResizingTextEdit()
self.tx_desc.setReadOnly(True)
self.tx_desc.setMinimumLines(1)
self.lb_dl_link = QLabel()
self.lb_dl_link.setText("下载直链：")
self.tx_dl_link = AutoResizingTextEdit(self)
self.tx_dl_link.setPlaceholderText("单击获取")
self.tx_dl_link.clicked.connect(self.call_get_dl_link)
self.tx_dl_link.setReadOnly(True)
self.tx_dl_link.setMinimumLines(1)
vbox = QVBoxLayout()
vbox.addWidget(self.logo)
vbox.addStretch(1)
form = QFormLayout()
form.setLabelAlignment(Qt.AlignRight)
form.addRow(self.lb_name, self.tx_name)
form.addRow(self.lb_size, self.tx_size)
form.addRow(self.lb_time, self.tx_time)
form.addRow(self.lb_dl_count, self.tx_dl_count)
form.addRow(self.lb_share_url, self.tx_share_url)
form.addRow(self.lb_pwd, self.tx_pwd)
form.addRow(self.lb_short, self.tx_short)
form.addRow(self.lb_desc, self.tx_desc)
form.addRow(self.lb_dl_link, self.tx_dl_link)
vbox.addLayout(form)
vbox.addStretch(1)
vbox.addWidget(self.buttonBox)
self.setLayout(vbox)
self.setMinimumWidth(500)
def initUI(self):
self.setWindowTitle("登录蓝奏云")
self.setWindowIcon(QIcon(SRC_DIR + "login.ico"))
logo = QLabel()
logo.setPixmap(QPixmap(SRC_DIR + "logo3.gif"))
logo.setStyleSheet("background-color:rgb(0,153,255);")
logo.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.tabs = QTabWidget()
self.auto_tab = QWidget()
self.hand_tab = QWidget()
# Add tabs
self.tabs.addTab(self.auto_tab,"自动获取Cookie")
self.tabs.addTab(self.hand_tab,"手动输入Cookie")
self.auto_get_cookie_ok = AutoResizingTextEdit("🔶点击👇自动获取浏览器登录信息👇")
self.auto_get_cookie_ok.setReadOnly(True)
self.auto_get_cookie_btn = QPushButton("自动读取浏览器登录信息")
auto_cookie_notice = '支持浏览器：Chrome, Chromium, Opera, Edge, Firefox'
self.auto_get_cookie_btn.setToolTip(auto_cookie_notice)
self.auto_get_cookie_btn.clicked.connect(self.call_auto_get_cookie)
self.auto_get_cookie_btn.setStyleSheet("QPushButton {min-width: 210px;max-width: 210px;}")
self.name_lb = QLabel("&U 用户")
self.name_lb.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.name_ed = QLineEdit()
self.name_lb.setBuddy(self.name_ed)
self.pwd_lb = QLabel("&P 密码")
self.pwd_lb.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.pwd_ed = QLineEdit()
self.pwd_ed.setEchoMode(QLineEdit.EchoMode.Password)
self.pwd_lb.setBuddy(self.pwd_ed)
self.cookie_lb = QLabel("&Cookie")
self.cookie_ed = QTextEdit()
notice = "由于滑动验证的存在，需要输入cookie，cookie请使用浏览器获取\n" + \
"cookie会保存在本地，下次使用。其格式如下：\n ylogin=value1; phpdisk_info=value2"
self.cookie_ed.setPlaceholderText(notice)
self.cookie_lb.setBuddy(self.cookie_ed)
self.show_input_cookie_btn = QPushButton("显示Cookie输入框")
self.show_input_cookie_btn.setToolTip(notice)
self.show_input_cookie_btn.setStyleSheet("QPushButton {min-width: 110px;max-width: 110px;}")
self.show_input_cookie_btn.clicked.connect(self.change_show_input_cookie)
self.ok_btn = QPushButton("登录")
self.ok_btn.clicked.connect(self.change_ok_btn)
self.cancel_btn = QPushButton("取消")
self.cancel_btn.clicked.connect(self.change_cancel_btn)
lb_line_1 = QLabel()
lb_line_1.setText('<html><hr />切换用户</html>')
lb_line_2 = QLabel()
lb_line_2.setText('<html><hr /></html>')
self.form = QFormLayout()
self.form.setLabelAlignment(Qt.AlignmentFlag.AlignRight)
self.form.setFieldGrowthPolicy(QFormLayout.FieldGrowthPolicy.AllNonFixedFieldsGrow) # 覆盖MacOS的默认样式
self.form.addRow(self.name_lb, self.name_ed)
self.form.addRow(self.pwd_lb, self.pwd_ed)
if is_windows:
def set_assister_path():
"""设置辅助登录程序路径"""
assister_path = QFileDialog.getOpenFileName(self, "选择辅助登录程序路径",
self._cwd, "EXE Files (*.exe)")
if not assister_path[0]:
return None
assister_path = os.path.normpath(assister_path[0]) # windows backslash
if assister_path == self._cookie_assister:
return None
self.assister_ed.setText(assister_path)
self._cookie_assister = assister_path
self.assister_lb = QLabel("登录辅助程序")
self.assister_lb.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.assister_ed = MyLineEdit(self)
self.assister_ed.setText(self._cookie_assister)
self.assister_ed.clicked.connect(set_assister_path)
self.assister_lb.setBuddy(self.assister_ed)
self.form.addRow(self.assister_lb, self.assister_ed)
hbox = QHBoxLayout()
hbox.addWidget(self.show_input_cookie_btn)
hbox.addStretch(1)
hbox.addWidget(self.ok_btn)
hbox.addWidget(self.cancel_btn)
user_box = QHBoxLayout()
self.user_num = 0
self.user_btns = {}
for user in self._config.users_name:
user = str(user) # TODO: 可能需要删掉
self.user_btns[user] = QDoublePushButton(user)
self.user_btns[user].setStyleSheet("QPushButton {border:none;}")
if user == self._config.name:
self.user_btns[user].setStyleSheet("QPushButton {background-color:rgb(0,153,2);}")
self.tabs.setCurrentIndex(1)
self.user_btns[user].setToolTip(f"点击选中，双击切换至用户：{user}")
self.user_btns[user].doubleClicked.connect(self.choose_user)
self.user_btns[user].clicked.connect(self.delete_chose_user)
user_box.addWidget(self.user_btns[user])
self.user_num += 1
user_box.addStretch(1)
self.layout = QVBoxLayout(self)
self.layout.addWidget(logo)
vbox = QVBoxLayout()
if self._config.name:
vbox.addWidget(lb_line_1)
user_box.setAlignment(Qt.AlignmentFlag.AlignCenter)
vbox.addLayout(user_box)
vbox.addWidget(lb_line_2)
if self.user_num > 1:
self.del_user_btn = QPushButton("删除账户")
self.del_user_btn.setIcon(QIcon(SRC_DIR + "delete.ico"))
self.del_user_btn.setStyleSheet("QPushButton {min-width: 180px;max-width: 180px;}")
self.del_user_btn.clicked.connect(self.call_del_chose_user)
vbox.addWidget(self.del_user_btn)
else:
self.del_user_btn = None
vbox.addStretch(1)
vbox.addLayout(self.form)
vbox.addStretch(1)
vbox.addLayout(hbox)
vbox.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.hand_tab.setLayout(vbox)
auto_cookie_vbox = QVBoxLayout()
auto_cookie_vbox.addWidget(self.auto_get_cookie_ok)
auto_cookie_vbox.addWidget(self.auto_get_cookie_btn)
auto_cookie_vbox.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.auto_tab.setLayout(auto_cookie_vbox)
self.layout.addWidget(self.tabs)
self.setLayout(self.layout)
self.update_selection(self._config.name)
class LoginDialog(QDialog):
"""登录对话框"""
clicked_ok = pyqtSignal()
def __init__(self, config):
super().__init__()
self._cwd = os.getcwd()
self._config = config
self._cookie_assister = 'login_assister.exe'
self._user = ""
self._pwd = ""
self._cookie = {}
self._del_user = ""
self.initUI()
self.setStyleSheet(dialog_qss_style)
self.setMinimumWidth(560)
self.name_ed.setFocus()
# 信号
self.name_ed.textChanged.connect(self.set_user)
self.pwd_ed.textChanged.connect(self.set_pwd)
self.cookie_ed.textChanged.connect(self.set_cookie)
def update_selection(self, user):
"""显示已经保存的登录用户信息"""
user_info = self._config.get_user_info(user)
if user_info:
self._user = user_info[0]
self._pwd = user_info[1]
self._cookie = user_info[2]
# 更新控件显示内容
self.name_ed.setText(self._user)
self.pwd_ed.setText(self._pwd)
try: text = ";".join([f'{k}={v}' for k, v in self._cookie.items()])
except: text = ''
self.cookie_ed.setPlainText(text)
def initUI(self):
self.setWindowTitle("登录蓝奏云")
self.setWindowIcon(QIcon(SRC_DIR + "login.ico"))
logo = QLabel()
logo.setPixmap(QPixmap(SRC_DIR + "logo3.gif"))
logo.setStyleSheet("background-color:rgb(0,153,255);")
logo.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.tabs = QTabWidget()
self.auto_tab = QWidget()
self.hand_tab = QWidget()
# Add tabs
self.tabs.addTab(self.auto_tab,"自动获取Cookie")
self.tabs.addTab(self.hand_tab,"手动输入Cookie")
self.auto_get_cookie_ok = AutoResizingTextEdit("🔶点击👇自动获取浏览器登录信息👇")
self.auto_get_cookie_ok.setReadOnly(True)
self.auto_get_cookie_btn = QPushButton("自动读取浏览器登录信息")
auto_cookie_notice = '支持浏览器：Chrome, Chromium, Opera, Edge, Firefox'
self.auto_get_cookie_btn.setToolTip(auto_cookie_notice)
self.auto_get_cookie_btn.clicked.connect(self.call_auto_get_cookie)
self.auto_get_cookie_btn.setStyleSheet("QPushButton {min-width: 210px;max-width: 210px;}")
self.name_lb = QLabel("&U 用户")
self.name_lb.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.name_ed = QLineEdit()
self.name_lb.setBuddy(self.name_ed)
self.pwd_lb = QLabel("&P 密码")
self.pwd_lb.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.pwd_ed = QLineEdit()
self.pwd_ed.setEchoMode(QLineEdit.EchoMode.Password)
self.pwd_lb.setBuddy(self.pwd_ed)
self.cookie_lb = QLabel("&Cookie")
self.cookie_ed = QTextEdit()
notice = "由于滑动验证的存在，需要输入cookie，cookie请使用浏览器获取\n" + \
"cookie会保存在本地，下次使用。其格式如下：\n ylogin=value1; phpdisk_info=value2"
self.cookie_ed.setPlaceholderText(notice)
self.cookie_lb.setBuddy(self.cookie_ed)
self.show_input_cookie_btn = QPushButton("显示Cookie输入框")
self.show_input_cookie_btn.setToolTip(notice)
self.show_input_cookie_btn.setStyleSheet("QPushButton {min-width: 110px;max-width: 110px;}")
self.show_input_cookie_btn.clicked.connect(self.change_show_input_cookie)
self.ok_btn = QPushButton("登录")
self.ok_btn.clicked.connect(self.change_ok_btn)
self.cancel_btn = QPushButton("取消")
self.cancel_btn.clicked.connect(self.change_cancel_btn)
lb_line_1 = QLabel()
lb_line_1.setText('<html><hr />切换用户</html>')
lb_line_2 = QLabel()
lb_line_2.setText('<html><hr /></html>')
self.form = QFormLayout()
self.form.setLabelAlignment(Qt.AlignmentFlag.AlignRight)
self.form.setFieldGrowthPolicy(QFormLayout.FieldGrowthPolicy.AllNonFixedFieldsGrow) # 覆盖MacOS的默认样式
self.form.addRow(self.name_lb, self.name_ed)
self.form.addRow(self.pwd_lb, self.pwd_ed)
if is_windows:
def set_assister_path():
"""设置辅助登录程序路径"""
assister_path = QFileDialog.getOpenFileName(self, "选择辅助登录程序路径",
self._cwd, "EXE Files (*.exe)")
if not assister_path[0]:
return None
assister_path = os.path.normpath(assister_path[0]) # windows backslash
if assister_path == self._cookie_assister:
return None
self.assister_ed.setText(assister_path)
self._cookie_assister = assister_path
self.assister_lb = QLabel("登录辅助程序")
self.assister_lb.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.assister_ed = MyLineEdit(self)
self.assister_ed.setText(self._cookie_assister)
self.assister_ed.clicked.connect(set_assister_path)
self.assister_lb.setBuddy(self.assister_ed)
self.form.addRow(self.assister_lb, self.assister_ed)
hbox = QHBoxLayout()
hbox.addWidget(self.show_input_cookie_btn)
hbox.addStretch(1)
hbox.addWidget(self.ok_btn)
hbox.addWidget(self.cancel_btn)
user_box = QHBoxLayout()
self.user_num = 0
self.user_btns = {}
for user in self._config.users_name:
user = str(user) # TODO: 可能需要删掉
self.user_btns[user] = QDoublePushButton(user)
self.user_btns[user].setStyleSheet("QPushButton {border:none;}")
if user == self._config.name:
self.user_btns[user].setStyleSheet("QPushButton {background-color:rgb(0,153,2);}")
self.tabs.setCurrentIndex(1)
self.user_btns[user].setToolTip(f"点击选中，双击切换至用户：{user}")
self.user_btns[user].doubleClicked.connect(self.choose_user)
self.user_btns[user].clicked.connect(self.delete_chose_user)
user_box.addWidget(self.user_btns[user])
self.user_num += 1
user_box.addStretch(1)
self.layout = QVBoxLayout(self)
self.layout.addWidget(logo)
vbox = QVBoxLayout()
if self._config.name:
vbox.addWidget(lb_line_1)
user_box.setAlignment(Qt.AlignmentFlag.AlignCenter)
vbox.addLayout(user_box)
vbox.addWidget(lb_line_2)
if self.user_num > 1:
self.del_user_btn = QPushButton("删除账户")
self.del_user_btn.setIcon(QIcon(SRC_DIR + "delete.ico"))
self.del_user_btn.setStyleSheet("QPushButton {min-width: 180px;max-width: 180px;}")
self.del_user_btn.clicked.connect(self.call_del_chose_user)
vbox.addWidget(self.del_user_btn)
else:
self.del_user_btn = None
vbox.addStretch(1)
vbox.addLayout(self.form)
vbox.addStretch(1)
vbox.addLayout(hbox)
vbox.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.hand_tab.setLayout(vbox)
auto_cookie_vbox = QVBoxLayout()
auto_cookie_vbox.addWidget(self.auto_get_cookie_ok)
auto_cookie_vbox.addWidget(self.auto_get_cookie_btn)
auto_cookie_vbox.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.auto_tab.setLayout(auto_cookie_vbox)
self.layout.addWidget(self.tabs)
self.setLayout(self.layout)
self.update_selection(self._config.name)
def call_del_chose_user(self):
if self._del_user:
if self._del_user != self._config.name:
self.user_num -= 1
self._config.del_user(self._del_user)
self.user_btns[self._del_user].close()
self._del_user = ""
if self.user_num <= 1:
self.del_user_btn.close()
self.del_user_btn = None
return
else:
title = '不能删除'
msg = '不能删除当前登录账户，请先切换用户！'
else:
title = '请选择账户'
msg = '请单击选择需要删除的账户\n\n注意不能删除当前账户(绿色)'
message_box = QMessageBox(self)
message_box.setIcon(QMessageBox.Icon.Critical)
message_box.setStyleSheet(btn_style)
message_box.setWindowTitle(title)
message_box.setText(msg)
message_box.setStandardButtons(QMessageBox.StandardButton.Close)
buttonC = message_box.button(QMessageBox.StandardButton.Close)
buttonC.setText('关闭')
message_box.exec()
def delete_chose_user(self):
"""更改单击选中需要删除的用户"""
user = str(self.sender().text())
self._del_user = user
if self.del_user_btn:
self.del_user_btn.setText(f"删除 <{user}>")
def choose_user(self):
"""切换用户"""
user = self.sender().text()
if user != self._config.name:
self.ok_btn.setText("切换用户")
else:
self.ok_btn.setText("登录")
self.update_selection(user)
def change_show_input_cookie(self):
row_c = 4 if is_windows else 3
if self.form.rowCount() < row_c:
self.org_height = self.height()
self.form.addRow(self.cookie_lb, self.cookie_ed)
self.show_input_cookie_btn.setText("隐藏Cookie输入框")
self.change_height = None
self.adjustSize()
else:
if not self.change_height:
self.change_height = self.height()
if self.cookie_ed.isVisible():
self.cookie_lb.setVisible(False)
self.cookie_ed.setVisible(False)
self.show_input_cookie_btn.setText("显示Cookie输入框")
start_height, end_height = self.change_height, self.org_height
else:
self.cookie_lb.setVisible(True)
self.cookie_ed.setVisible(True)
self.show_input_cookie_btn.setText("隐藏Cookie输入框")
start_height, end_height = self.org_height, self.change_height
gm = self.geometry()
x, y = gm.x(), gm.y()
wd = self.width()
self.animation = QPropertyAnimation(self, b'geometry')
self.animation.setDuration(400)
self.animation.setStartValue(QRect(x, y, wd, start_height))
self.animation.setEndValue(QRect(x, y, wd, end_height))
self.animation.start()
def set_user(self, user):
self._user = user
if not user:
return None
if user not in self._config.users_name:
self.ok_btn.setText("添加用户")
self.cookie_ed.setPlainText("")
elif user != self._config.name:
self.update_selection(user)
self.ok_btn.setText("切换用户")
else:
self.update_selection(user)
self.ok_btn.setText("登录")
def set_pwd(self, pwd):
if self._user in self._config.users_name:
user_info = self._config.get_user_info(self._user)
if pwd and pwd != user_info[1]: # 改变密码，cookie作废
self.cookie_ed.setPlainText("")
self._cookie = None
if not pwd: # 输入空密码，表示删除对pwd的存储，并使用以前的cookie
self._cookie = user_info[2]
try: text = ";".join([f'{k}={v}' for k, v in self._cookie.items()])
except: text = ''
self.cookie_ed.setPlainText(text)
self._pwd = pwd
def set_cookie(self):
cookies = self.cookie_ed.toPlainText()
if cookies:
try:
self._cookie = {kv.split("=")[0].strip(" "): kv.split("=")[1].strip(" ") for kv in cookies.split(";") if kv.strip(" ") }
except: self._cookie = None
def change_cancel_btn(self):
self.update_selection(self._config.name)
self.close()
def change_ok_btn(self):
if self._user and self._pwd:
if self._user not in self._config.users_name:
self._cookie = None
if self._cookie:
up_info = {"name": self._user, "pwd": self._pwd, "cookie": self._cookie, "work_id": -1}
if self.ok_btn.text() == "切换用户":
self._config.change_user(self._user)
else:
self._config.set_infos(up_info)
self.clicked_ok.emit()
self.close()
elif USE_WEB_ENG:
self.web = LoginWindow(self._user, self._pwd)
self.web.cookie.connect(self.get_cookie_by_web)
self.web.setWindowModality(Qt.WindowModality.ApplicationModal)
self.web.exec()
elif os.path.isfile(self._cookie_assister):
try:
result = os.popen(f'{self._cookie_assister} {self._user} {self._pwd}')
cookie = result.read()
try:
self._cookie = {kv.split("=")[0].strip(" "): kv.split("=")[1].strip(" ") for kv in cookie.split(";")}
except: self._cookie = None
if not self._cookie:
return None
up_info = {"name": self._user, "pwd": self._pwd, "cookie": self._cookie, "work_id": -1}
self._config.set_infos(up_info)
self.clicked_ok.emit()
self.close()
except: pass
else:
title = '请使用 Cookie 登录或是选择 登录辅助程序'
msg = '没有输入 Cookie，或者没有找到登录辅助程序！\n\n' + \
'推荐使用浏览器获取 cookie 填入 cookie 输入框\n\n' + \
'如果不嫌文件体积大，请下载登录辅助程序：\n' + \
'https://github.com/rachpt/lanzou-gui/releases'
message_box = QMessageBox(self)
message_box.setIcon(QMessageBox.Icon.Critical)
message_box.setStyleSheet(btn_style)
message_box.setWindowTitle(title)
message_box.setText(msg)
message_box.setStandardButtons(QMessageBox.StandardButton.Close)
buttonC = message_box.button(QMessageBox.StandardButton.Close)
buttonC.setText('关闭')
message_box.exec()
def get_cookie_by_web(self, cookie):
"""使用辅助登录程序槽函数"""
self._cookie = cookie
self._close_dialog()
def call_auto_get_cookie(self):
"""自动读取浏览器cookie槽函数"""
try:
self._cookie = get_cookie_from_browser()
except Exception as e:
logger.error(f"Browser_cookie3 Error: {e}")
self.auto_get_cookie_ok.setPlainText(f"❌获取失败，错误信息\n{e}")
else:
if self._cookie:
self._user = self._pwd = ''
self.auto_get_cookie_ok.setPlainText("✅获取成功即将登录……")
QTimer.singleShot(2000, self._close_dialog)
else:
self.auto_get_cookie_ok.setPlainText("❌获取失败\n请提前使用支持的浏览器登录蓝奏云，读取前完全退出浏览器！\n支持的浏览器与顺序：\nchrome, chromium, opera, edge, firefox")
def _close_dialog(self):
"""关闭对话框"""
up_info = {"name": self._user, "pwd": self._pwd, "cookie": self._cookie}
self._config.set_infos(up_info)
self.clicked_ok.emit()
self.close()