167 lines
7.0 KiB
Python
167 lines
7.0 KiB
Python
|
import base64
|
|||
|
|
|
|||
|
|
from loguru import logger
|
|||
|
|
from PyQt6.QtCore import Qt, QThread, pyqtSignal
|
|||
|
|
from PyQt6.QtGui import (QColor, QPainter, QPainterPath, QPen, QPixmap)
|
|||
|
|
|
|||
|
|
from ..api import miaoStarsBasicApi
|
|||
|
|
from ...core import cfg, qconfig, userConfig
|
|||
|
|
|
|||
|
|
|
|||
|
|
class CaptchaThread(QThread):
|
|||
|
|
captchaReady = pyqtSignal(QPixmap)
|
|||
|
|
captchaFailed = pyqtSignal(str)
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
super().__init__()
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _createRoundedPixmap(pixmap, radius=10):
|
|||
|
|
"""创建圆角图片"""
|
|||
|
|
try:
|
|||
|
|
# 获取原始图片尺寸
|
|||
|
|
if pixmap.isNull():
|
|||
|
|
logger.error("原始图片为空，无法创建圆角图片")
|
|||
|
|
return pixmap
|
|||
|
|
size = pixmap.size()
|
|||
|
|
# 创建透明背景的图片
|
|||
|
|
rounded_pixmap = QPixmap(size)
|
|||
|
|
rounded_pixmap.fill(Qt.GlobalColor.transparent)
|
|||
|
|
# 创建对象
|
|||
|
|
painter = QPainter(rounded_pixmap)
|
|||
|
|
painter.setRenderHints(
|
|||
|
|
QPainter.RenderHint.Antialiasing
|
|||
|
|
| QPainter.RenderHint.SmoothPixmapTransform
|
|||
|
|
)
|
|||
|
|
# 创建圆角矩形路径
|
|||
|
|
path = QPainterPath()
|
|||
|
|
path.addRoundedRect(0, 0, size.width(), size.height(), radius, radius)
|
|||
|
|
# 设置裁剪区域
|
|||
|
|
painter.setClipPath(path)
|
|||
|
|
# 绘制原始图片
|
|||
|
|
painter.drawPixmap(0, 0, pixmap)
|
|||
|
|
# 绘制边框
|
|||
|
|
pen = QPen(QColor(200, 200, 200)) # 浅灰色边框
|
|||
|
|
pen.setWidth(1)
|
|||
|
|
painter.setPen(pen)
|
|||
|
|
painter.drawRoundedRect(
|
|||
|
|
0, 0, size.width() - 1, size.height() - 1, radius, radius
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
painter.end()
|
|||
|
|
return rounded_pixmap
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"创建圆角图片失败：{e}")
|
|||
|
|
return pixmap # 如果出错，返回原始图片
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
try:
|
|||
|
|
logger.debug("开始获取验证码")
|
|||
|
|
response = miaoStarsBasicApi.getCaptcha()
|
|||
|
|
logger.debug(f"验证码API返回响应: {response}")
|
|||
|
|
|
|||
|
|
if response["code"] == 0:
|
|||
|
|
# 确保data字段存在且为字符串
|
|||
|
|
if "data" in response and isinstance(response["data"], str):
|
|||
|
|
# 分割base64前缀和实际数据
|
|||
|
|
try:
|
|||
|
|
captchaImageData = response["data"].split(",")[1]
|
|||
|
|
logger.debug(f"成功提取base64数据，长度: {len(captchaImageData)}")
|
|||
|
|
|
|||
|
|
# 解码base64数据
|
|||
|
|
captchaImage = base64.b64decode(captchaImageData)
|
|||
|
|
logger.debug(f"成功解码base64数据，长度: {len(captchaImage)} bytes")
|
|||
|
|
|
|||
|
|
# 加载图片
|
|||
|
|
pixmap = QPixmap()
|
|||
|
|
load_success = pixmap.loadFromData(captchaImage)
|
|||
|
|
|
|||
|
|
if load_success:
|
|||
|
|
logger.debug(f"成功加载图片，尺寸: {pixmap.width()}x{pixmap.height()}")
|
|||
|
|
# 创建圆角图片
|
|||
|
|
pixmap = self._createRoundedPixmap(pixmap, radius=10)
|
|||
|
|
self.captchaReady.emit(pixmap)
|
|||
|
|
else:
|
|||
|
|
logger.error("图片加载失败")
|
|||
|
|
self.captchaFailed.emit("验证码图片加载失败")
|
|||
|
|
self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
|
|||
|
|
except (IndexError, ValueError, TypeError) as e:
|
|||
|
|
logger.error(f"验证码数据格式错误: {e}")
|
|||
|
|
self.captchaFailed.emit(f"验证码数据格式错误: {str(e)}")
|
|||
|
|
self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
|
|||
|
|
else:
|
|||
|
|
logger.error("验证码响应中缺少有效的data字段")
|
|||
|
|
self.captchaFailed.emit("验证码数据无效")
|
|||
|
|
self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
|
|||
|
|
else:
|
|||
|
|
error_msg = response.get("msg", "获取验证码失败")
|
|||
|
|
logger.error(f"获取验证码失败: {error_msg}")
|
|||
|
|
self.captchaFailed.emit(error_msg)
|
|||
|
|
self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.exception(f"获取验证码过程中发生异常: {e}")
|
|||
|
|
self.captchaFailed.emit(str(e))
|
|||
|
|
self.captchaReady.emit(QPixmap(":app/images/loadFailure.png"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LoginThread(QThread):
|
|||
|
|
successLogin = pyqtSignal()
|
|||
|
|
errorLogin = pyqtSignal(str)
|
|||
|
|
|
|||
|
|
def __init__(self, email: str, password: str, captchaCode: str):
|
|||
|
|
super().__init__()
|
|||
|
|
logger.debug(f"初始化许可证服务线程 - 邮箱: {email}")
|
|||
|
|
self.email = email
|
|||
|
|
self.password = password
|
|||
|
|
self.captchaCode = captchaCode
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
logger.info(f"开始验证用户登录 - 邮箱: {self.email}")
|
|||
|
|
try:
|
|||
|
|
loginResponse = miaoStarsBasicApi.login(
|
|||
|
|
self.email, self.password, self.captchaCode
|
|||
|
|
)
|
|||
|
|
if loginResponse["code"] == 0:
|
|||
|
|
self.successLogin.emit()
|
|||
|
|
qconfig.set(cfg.email, self.email)
|
|||
|
|
qconfig.set(cfg.activationCode, self.password)
|
|||
|
|
userConfig.userData = loginResponse
|
|||
|
|
# 从登录响应中提取token并保存
|
|||
|
|
# 在basicApi的login方法中已经处理了token的设置，这里保存到userConfig中以便程序启动时恢复
|
|||
|
|
token = miaoStarsBasicApi.token
|
|||
|
|
if token:
|
|||
|
|
userConfig.setToken(token)
|
|||
|
|
else:
|
|||
|
|
self.errorLogin.emit(loginResponse["msg"])
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"登录验证过程中发生异常: {e}")
|
|||
|
|
self.errorLogin.emit("系统错误，请稍后重试")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RegisterThread(QThread):
|
|||
|
|
successRegister = pyqtSignal()
|
|||
|
|
errorRegister = pyqtSignal(str)
|
|||
|
|
|
|||
|
|
def __init__(self, email: str, password: str, captchaCode: str):
|
|||
|
|
super().__init__()
|
|||
|
|
logger.debug(f"初始化许可证服务线程 - 邮箱: {email}")
|
|||
|
|
self.email = email
|
|||
|
|
self.password = password
|
|||
|
|
self.captchaCode = captchaCode
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
logger.info(f"开始验证用户注册 - 邮箱: {self.email}")
|
|||
|
|
try:
|
|||
|
|
registerRespond = miaoStarsBasicApi.register(
|
|||
|
|
self.email, self.password, self.captchaCode
|
|||
|
|
)
|
|||
|
|
if registerRespond["code"] == 203:
|
|||
|
|
self.successRegister.emit()
|
|||
|
|
else:
|
|||
|
|
logger.error(f"注册失败: {registerRespond['msg']}")
|
|||
|
|
self.errorRegister.emit(registerRespond["msg"])
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"登录验证过程中发生异常: {e}")
|
|||
|
|
self.errorRegister.emit("系统错误，请稍后重试")
|