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.AlignmentFlag.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.Orientation.Horizontal)
buttonBox.setStandardButtons(QDialogButtonBox.StandardButton.Reset | QDialogButtonBox.StandardButton.Save | QDialogButtonBox.StandardButton.Cancel)
buttonBox.button(QDialogButtonBox.StandardButton.Reset).setText("重置")
buttonBox.button(QDialogButtonBox.StandardButton.Save).setText("保存")
buttonBox.button(QDialogButtonBox.StandardButton.Cancel).setText("取消")
buttonBox.button(QDialogButtonBox.StandardButton.Reset).clicked.connect(lambda: self.set_values(reset=True))
buttonBox.button(QDialogButtonBox.StandardButton.Save).clicked.connect(self.slot_save)
buttonBox.rejected.connect(self.reject)
form = QFormLayout()
form.setLabelAlignment(Qt.AlignmentFlag.AlignRight)
form.setFieldGrowthPolicy(QFormLayout.FieldGrowthPolicy.AllNonFixedFieldsGrow) # 覆盖MacOS的默认样式
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 AboutDialog(QDialog):
check_update = pyqtSignal(str, bool)
def __init__(self, parent=None):
super(AboutDialog, self).__init__(parent)
self._ver = ''
self._github = 'https://github.com/rachpt/lanzou-gui'
self._api_url = 'https://github.com/zaxtyson/LanZouCloud-API'
self._gitee = 'https://gitee.com/rachpt/lanzou-gui'
self._home_page = 'https://rachpt.cn/lanzou-gui/'
self.initUI()
self.setStyleSheet(others_style)
def set_values(self, version):
self._ver = version
self.lb_name_text.setText(f"v{version} (点击检查更新)") # 更新版本
def show_update(self, ver, msg):
self.lb_new_ver = QLabel("新版") # 检测新版
self.lb_new_ver_msg = QLabel()
self.lb_new_ver_msg.setOpenExternalLinks(True)
self.lb_new_ver_msg.setWordWrap(True)
if ver != '0':
self.lb_name_text.setText(f"{self._ver} ➡ {ver}")
self.lb_new_ver_msg.setText(msg)
self.lb_new_ver_msg.setMinimumWidth(700)
if self.form.rowCount() < 5:
self.form.insertRow(1, self.lb_new_ver, self.lb_new_ver_msg)
def initUI(self):
self.setWindowTitle("关于 lanzou-gui")
about = f'本项目使用PyQt6实现图形界面，可以完成蓝奏云的大部分功能<br/> \
得益于 <a href="{self._api_url}">API</a> 的功能，可以间接突破单文件最大 100MB 的限制，同时增加了批量上传/下载的功能<br/> \
Python 依赖见<a href="{self._github }/blob/master/requirements.txt">requirements.txt</a>，\
<a href="{self._github}/releases">releases</a> 有打包好了的 Windows 可执行程序，但可能不是最新的'
project_url = f'<a href="{self._home_page}">主页</a> | <a href="{self._github}">repo</a> | \
<a href="{self._gitee}">mirror repo</a>'
self.logo = QLabel() # logo
self.logo.setPixmap(QPixmap(SRC_DIR + "logo2.gif"))
self.logo.setStyleSheet("background-color:rgb(255,255,255);")
self.logo.setAlignment(Qt.AlignmentFlag.AlignCenter)
self.lb_qt_ver = QLabel("依赖") # QT 版本
self.lb_qt_text = QLabel(
f"QT: {QT_VERSION_STR}, PyQt: {PYQT_VERSION_STR}") # QT 版本
self.lb_name = QLabel("版本") # 版本
self.lb_name_text = QPushButton("") # 版本
self.lb_name_text.setToolTip("点击检查更新")
ver_style = "QPushButton {border:none; background:transparent;font-weight:bold;color:blue;}"
self.lb_name_text.setStyleSheet(ver_style)
self.lb_name_text.clicked.connect(
lambda: self.check_update.emit(self._ver, True))
self.lb_about = QLabel("关于") # about
self.lb_about_text = QLabel()
self.lb_about_text.setText(about)
self.lb_about_text.setOpenExternalLinks(True)
self.lb_author = QLabel("作者") # author
self.lb_author_mail = QLabel(
"<a href='mailto:[email protected]'>rachpt</a>")
self.lb_author_mail.setOpenExternalLinks(True)
self.lb_update = QLabel("项目") # 更新
self.lb_update_url = QLabel(project_url)
self.lb_update_url.setOpenExternalLinks(True)
self.buttonBox = QDialogButtonBox()
self.buttonBox.setOrientation(Qt.Orientation.Horizontal)
self.buttonBox.setStandardButtons(
QDialogButtonBox.StandardButton.Close)
self.buttonBox.button(
QDialogButtonBox.StandardButton.Close).setText("关闭")
self.buttonBox.rejected.connect(self.reject)
self.buttonBox.setStyleSheet(btn_style)
self.recommend = QLabel(
"<br />大文件推荐使用 <a href='https://github.com/Aruelius/cloud189'>cloud189-cli</a>"
)
self.recommend.setOpenExternalLinks(True)
self.line = QLine(QPoint(), QPoint(550, 0))
self.lb_line = QLabel('<html><hr /></html>')
vbox = QVBoxLayout()
vbox.addWidget(self.logo)
vbox.addStretch(1)
self.form = QFormLayout()
self.form.setLabelAlignment(Qt.AlignmentFlag.AlignRight)
self.form.setFormAlignment(Qt.AlignmentFlag.AlignLeft)
self.form.setHorizontalSpacing(40)
self.form.setVerticalSpacing(15)
self.form.addRow(self.lb_qt_ver, self.lb_qt_text)
self.form.addRow(self.lb_name, self.lb_name_text)
self.form.addRow(self.lb_update, self.lb_update_url)
self.form.addRow(self.lb_author, self.lb_author_mail)
self.form.addRow(self.lb_about, self.lb_about_text)
self.form.setFieldGrowthPolicy(QFormLayout.FieldGrowthPolicy.
AllNonFixedFieldsGrow) # 覆盖MacOS的默认样式
vbox.addLayout(self.form)
vbox.addStretch(1)
vbox.addWidget(self.recommend)
vbox.addWidget(self.lb_line)
donate = QLabel()
donate.setText("<b>捐助我</b> 如果你愿意")
donate.setAlignment(Qt.AlignmentFlag.AlignCenter)
hbox = QHBoxLayout()
hbox.addStretch(2)
for it in ["wechat", "alipay", "qqpay"]:
lb = QLabel()
lb.setPixmap(QPixmap(SRC_DIR + f"{it}.jpg"))
hbox.addWidget(lb)
hbox.addStretch(1)
hbox.addWidget(self.buttonBox)
vbox.addWidget(donate)
vbox.addLayout(hbox)
self.setLayout(vbox)
self.setMinimumWidth(720)
def paintEvent(self, event):
QDialog.paintEvent(self, event)
if not self.line.isNull():
painter = QPainter(self)
pen = QPen(Qt.GlobalColor.red, 3)
painter.setPen(pen)
painter.drawLine(self.line)
def initUI(self):
self.setWindowIcon(QIcon(SRC_DIR + "share.ico"))
self.setWindowTitle("文件信息")
self.buttonBox = QDialogButtonBox()
self.buttonBox.setOrientation(Qt.Orientation.Horizontal)
self.buttonBox.setStandardButtons(QDialogButtonBox.StandardButton.Close)
self.buttonBox.button(QDialogButtonBox.StandardButton.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_DIR + "q9.gif"))
self.logo.setAlignment(Qt.AlignmentFlag.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.AlignmentFlag.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)
form.setFieldGrowthPolicy(QFormLayout.FieldGrowthPolicy.AllNonFixedFieldsGrow) # 覆盖MacOS的默认样式
vbox.addLayout(form)
vbox.addStretch(1)
vbox.addWidget(self.buttonBox)
self.setLayout(vbox)
self.setMinimumWidth(500)
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()
