1072 lines
39 KiB
Python
1072 lines
39 KiB
Python
# -*- coding: UTF-8 -*-
|
||
'''
|
||
@Project :domainScanDemo
|
||
@File :system_settings.py
|
||
@IDE :PyCharm
|
||
@Author :梦伴
|
||
@Date :2026/4/10 15:00
|
||
@explain : 系统设置页面
|
||
'''
|
||
|
||
from PySide6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel, QLineEdit, QGroupBox, QCheckBox
|
||
from PySide6.QtCore import QTimer, QThreadPool, QRunnable, Slot
|
||
from PySide6.QtCore import Qt, QThread, Signal
|
||
from loguru import logger
|
||
import os
|
||
import json
|
||
import redis
|
||
|
||
from detect.juming import JM
|
||
from detect.juziseo import Juziseo
|
||
from detect.jucha import JC
|
||
from app.config import config
|
||
|
||
|
||
class RedisTask(QRunnable):
|
||
"""
|
||
Redis操作后台任务
|
||
"""
|
||
def __init__(self, func, *args, **kwargs):
|
||
"""
|
||
初始化Redis任务
|
||
|
||
:param func: 要执行的函数
|
||
:param args: 函数参数
|
||
:param kwargs: 函数关键字参数
|
||
"""
|
||
super().__init__()
|
||
self.func = func
|
||
self.args = args
|
||
self.kwargs = kwargs
|
||
|
||
@Slot()
|
||
def run(self):
|
||
"""
|
||
执行Redis操作
|
||
"""
|
||
try:
|
||
self.func(*self.args, **self.kwargs)
|
||
except Exception as e:
|
||
logger.error(f"Redis操作失败: {e}")
|
||
|
||
|
||
class LoginThread(QThread):
|
||
"""
|
||
登录线程
|
||
"""
|
||
finished = Signal(bool, str)
|
||
|
||
def __init__(self, email, password, platform):
|
||
"""
|
||
初始化登录线程
|
||
|
||
:param email: 账号
|
||
:param password: 密码
|
||
:param platform: 平台 (juming: 聚名, juziseo: 桔子SEO)
|
||
"""
|
||
super().__init__()
|
||
self.email = email
|
||
self.password = password
|
||
self.platform = platform
|
||
|
||
def run(self):
|
||
"""
|
||
运行登录线程
|
||
"""
|
||
try:
|
||
if self.platform == "juming":
|
||
# 初始化聚名客户端
|
||
jm = JM()
|
||
|
||
# 加载 Cookie
|
||
jm.load_cookies()
|
||
logger.info("已加载聚名 Cookie")
|
||
|
||
# 登录
|
||
logger.info(f"开始登录聚名网: {self.email}")
|
||
login_result = jm.user_zh_p_login(self.email, self.password)
|
||
if not login_result[0]:
|
||
self.finished.emit(False, f"登录失败: {login_result[1]}")
|
||
return
|
||
logger.info("登录成功")
|
||
|
||
# 保存cookies
|
||
jm.save_cookies()
|
||
logger.info("已保存聚名 Cookie")
|
||
|
||
self.finished.emit(True, "登录成功")
|
||
elif self.platform == "juziseo":
|
||
# 初始化桔子SEO客户端
|
||
juziseo = Juziseo()
|
||
|
||
# 加载 Cookie
|
||
juziseo.load_cookies()
|
||
logger.info("已加载桔子SEO Cookie")
|
||
|
||
# 登录
|
||
logger.info(f"开始登录桔子SEO: {self.email}")
|
||
login_result = juziseo.login(self.email, self.password)
|
||
if not login_result[0]:
|
||
self.finished.emit(False, f"登录失败: {login_result[1]}")
|
||
return
|
||
logger.info("登录成功")
|
||
|
||
# 保存cookies
|
||
juziseo.save_cookies()
|
||
logger.info("已保存桔子SEO Cookie")
|
||
|
||
self.finished.emit(True, "登录成功")
|
||
elif self.platform == "jucha":
|
||
# 初始化聚查客户端
|
||
jc = JC()
|
||
|
||
# 加载聚名 Cookie
|
||
jc.load_juming_cookies()
|
||
logger.info("已加载聚名 Cookie")
|
||
|
||
# 登录
|
||
logger.info("开始登录聚查网")
|
||
login_result = jc.auth_login()
|
||
if not login_result[0]:
|
||
self.finished.emit(False, f"登录失败: {login_result[1]}")
|
||
return
|
||
logger.info("登录成功")
|
||
|
||
# 保存cookies
|
||
jc.save_cookies()
|
||
logger.info("已保存聚查 Cookie")
|
||
|
||
self.finished.emit(True, "登录成功")
|
||
except Exception as e:
|
||
logger.error(f"登录失败: {e}")
|
||
self.finished.emit(False, f"登录失败: {str(e)}")
|
||
|
||
|
||
class SystemSettingsWidget(QWidget):
|
||
"""
|
||
系统设置页面
|
||
"""
|
||
|
||
def __init__(self):
|
||
"""
|
||
初始化系统设置页面
|
||
"""
|
||
super().__init__()
|
||
|
||
# 初始化Redis客户端
|
||
self.redis_client = None
|
||
self.use_redis = False
|
||
self.thread_pool = QThreadPool()
|
||
self.thread_pool.setMaxThreadCount(5) # 最大线程数
|
||
self.init_redis()
|
||
|
||
# 创建布局
|
||
layout = QVBoxLayout(self)
|
||
|
||
# 创建登录区域容器
|
||
login_container = QGroupBox("登录设置")
|
||
login_container.setStyleSheet("""
|
||
QGroupBox {
|
||
font-size: 14px;
|
||
font-weight: bold;
|
||
color: #333;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
margin-bottom: 20px;
|
||
padding: 15px;
|
||
}
|
||
""")
|
||
login_layout = QVBoxLayout(login_container)
|
||
|
||
# 聚名登录区域
|
||
juming_subgroup = QGroupBox("聚名网登录")
|
||
juming_subgroup.setStyleSheet("""
|
||
QGroupBox {
|
||
font-size: 13px;
|
||
font-weight: bold;
|
||
color: #333;
|
||
border: 1px solid #eee;
|
||
border-radius: 4px;
|
||
margin-bottom: 15px;
|
||
padding: 10px;
|
||
}
|
||
""")
|
||
juming_layout = QVBoxLayout(juming_subgroup)
|
||
|
||
# 聚名输入区域（两列布局）
|
||
juming_input_layout = QHBoxLayout()
|
||
juming_input_layout.setSpacing(20)
|
||
|
||
# 邮箱输入
|
||
email_layout = QVBoxLayout()
|
||
email_label = QLabel("邮箱:")
|
||
email_label.setStyleSheet("font-size: 14px; color: #666; margin-bottom: 8px;")
|
||
self.juming_email_input = QLineEdit()
|
||
self.juming_email_input.setPlaceholderText("请输入聚名网账号")
|
||
self.juming_email_input.setStyleSheet("""
|
||
QLineEdit {
|
||
font-size: 14px;
|
||
padding: 8px;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
min-width: 250px;
|
||
min-height: 22px;
|
||
}
|
||
QLineEdit:focus {
|
||
border-color: #4CAF50;
|
||
outline: none;
|
||
}
|
||
""")
|
||
email_layout.addWidget(email_label)
|
||
email_layout.addWidget(self.juming_email_input)
|
||
juming_input_layout.addLayout(email_layout)
|
||
|
||
# 密码输入
|
||
password_layout = QVBoxLayout()
|
||
password_label = QLabel("密码:")
|
||
password_label.setStyleSheet("font-size: 14px; color: #666; margin-bottom: 8px;")
|
||
self.juming_password_input = QLineEdit()
|
||
self.juming_password_input.setPlaceholderText("请输入聚名网密码")
|
||
self.juming_password_input.setEchoMode(QLineEdit.Password)
|
||
self.juming_password_input.setStyleSheet("""
|
||
QLineEdit {
|
||
font-size: 14px;
|
||
padding: 8px;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
min-width: 250px;
|
||
min-height: 22px;
|
||
}
|
||
QLineEdit:focus {
|
||
border-color: #4CAF50;
|
||
outline: none;
|
||
}
|
||
""")
|
||
password_layout.addWidget(password_label)
|
||
password_layout.addWidget(self.juming_password_input)
|
||
juming_input_layout.addLayout(password_layout)
|
||
juming_layout.addLayout(juming_input_layout)
|
||
|
||
# 登录说明
|
||
juming_note = QLabel("聚名聚查联名登录：登录聚名后会自动登录聚查")
|
||
juming_note.setStyleSheet("font-size: 12px; color: #666; margin-top: 12px; margin-bottom: 12px;")
|
||
juming_layout.addWidget(juming_note)
|
||
|
||
# 登录按钮布局
|
||
juming_btn_layout = QHBoxLayout()
|
||
juming_btn_layout.addStretch() # 添加弹性空间，将按钮推到右侧
|
||
juming_login_btn = QPushButton("登录聚名网")
|
||
juming_login_btn.clicked.connect(self.login_juming)
|
||
juming_login_btn.setStyleSheet("""
|
||
QPushButton {
|
||
font-size: 14px;
|
||
padding: 8px 16px;
|
||
background-color: #4CAF50;
|
||
color: white;
|
||
border: none;
|
||
border-radius: 4px;
|
||
}
|
||
QPushButton:hover {
|
||
background-color: #45a049;
|
||
}
|
||
QPushButton:disabled {
|
||
background-color: #cccccc;
|
||
}
|
||
""")
|
||
juming_btn_layout.addWidget(juming_login_btn)
|
||
juming_layout.addLayout(juming_btn_layout)
|
||
|
||
login_layout.addWidget(juming_subgroup)
|
||
|
||
# 桔子SEO登录区域
|
||
juziseo_subgroup = QGroupBox("桔子SEO登录")
|
||
juziseo_subgroup.setStyleSheet("""
|
||
QGroupBox {
|
||
font-size: 13px;
|
||
font-weight: bold;
|
||
color: #333;
|
||
border: 1px solid #eee;
|
||
border-radius: 4px;
|
||
margin-bottom: 15px;
|
||
padding: 10px;
|
||
}
|
||
""")
|
||
juziseo_layout = QVBoxLayout(juziseo_subgroup)
|
||
|
||
# 桔子SEO输入区域（两列布局）
|
||
juziseo_input_layout = QHBoxLayout()
|
||
juziseo_input_layout.setSpacing(20)
|
||
|
||
# 账号输入
|
||
juziseo_email_layout = QVBoxLayout()
|
||
juziseo_email_label = QLabel("账号:")
|
||
juziseo_email_label.setStyleSheet("font-size: 14px; color: #666; margin-bottom: 8px;")
|
||
self.juziseo_email_input = QLineEdit()
|
||
self.juziseo_email_input.setPlaceholderText("请输入桔子SEO账号")
|
||
self.juziseo_email_input.setStyleSheet("""
|
||
QLineEdit {
|
||
font-size: 14px;
|
||
padding: 8px;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
min-width: 250px;
|
||
min-height: 22px;
|
||
}
|
||
QLineEdit:focus {
|
||
border-color: #2196F3;
|
||
outline: none;
|
||
}
|
||
""")
|
||
juziseo_email_layout.addWidget(juziseo_email_label)
|
||
juziseo_email_layout.addWidget(self.juziseo_email_input)
|
||
juziseo_input_layout.addLayout(juziseo_email_layout)
|
||
|
||
# 密码输入
|
||
juziseo_password_layout = QVBoxLayout()
|
||
juziseo_password_label = QLabel("密码:")
|
||
juziseo_password_label.setStyleSheet("font-size: 14px; color: #666; margin-bottom: 8px;")
|
||
self.juziseo_password_input = QLineEdit()
|
||
self.juziseo_password_input.setPlaceholderText("请输入桔子SEO密码")
|
||
self.juziseo_password_input.setEchoMode(QLineEdit.Password)
|
||
self.juziseo_password_input.setStyleSheet("""
|
||
QLineEdit {
|
||
font-size: 14px;
|
||
padding: 8px;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
min-width: 250px;
|
||
min-height: 22px;
|
||
}
|
||
QLineEdit:focus {
|
||
border-color: #2196F3;
|
||
outline: none;
|
||
}
|
||
""")
|
||
juziseo_password_layout.addWidget(juziseo_password_label)
|
||
juziseo_password_layout.addWidget(self.juziseo_password_input)
|
||
juziseo_input_layout.addLayout(juziseo_password_layout)
|
||
juziseo_layout.addLayout(juziseo_input_layout)
|
||
|
||
# 登录按钮布局
|
||
juziseo_btn_layout = QHBoxLayout()
|
||
juziseo_btn_layout.addStretch() # 添加弹性空间，将按钮推到右侧
|
||
juziseo_login_btn = QPushButton("登录桔子SEO")
|
||
juziseo_login_btn.clicked.connect(self.login_juziseo)
|
||
juziseo_login_btn.setStyleSheet("""
|
||
QPushButton {
|
||
font-size: 14px;
|
||
padding: 8px 16px;
|
||
background-color: #2196F3;
|
||
color: white;
|
||
border: none;
|
||
border-radius: 4px;
|
||
}
|
||
QPushButton:hover {
|
||
background-color: #0b7dda;
|
||
}
|
||
QPushButton:disabled {
|
||
background-color: #cccccc;
|
||
}
|
||
""")
|
||
juziseo_btn_layout.addWidget(juziseo_login_btn)
|
||
juziseo_layout.addLayout(juziseo_btn_layout)
|
||
|
||
login_layout.addWidget(juziseo_subgroup)
|
||
|
||
layout.addWidget(login_container)
|
||
|
||
# 创建代理IP配置区域
|
||
proxy_group = QGroupBox("代理IP配置")
|
||
proxy_group.setStyleSheet("""
|
||
QGroupBox {
|
||
font-size: 14px;
|
||
font-weight: bold;
|
||
color: #333;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
margin-bottom: 20px;
|
||
padding: 15px;
|
||
}
|
||
""")
|
||
proxy_layout = QVBoxLayout(proxy_group)
|
||
|
||
# 启用代理IP复选框
|
||
proxy_enable_layout = QHBoxLayout()
|
||
proxy_enable_label = QLabel("启用代理IP:")
|
||
proxy_enable_label.setStyleSheet("font-size: 14px; color: #333; min-width: 100px; font-weight: bold;")
|
||
self.proxy_enable_checkbox = QCheckBox()
|
||
self.proxy_enable_checkbox.setStyleSheet("""
|
||
QCheckBox {
|
||
font-size: 14px;
|
||
spacing: 8px;
|
||
}
|
||
QCheckBox::indicator {
|
||
width: 20px;
|
||
height: 20px;
|
||
}
|
||
QCheckBox::indicator:unchecked {
|
||
border: 2px solid #999;
|
||
background-color: white;
|
||
border-radius: 4px;
|
||
}
|
||
QCheckBox::indicator:checked {
|
||
border: 2px solid #9c27b0;
|
||
background-color: #9c27b0;
|
||
border-radius: 4px;
|
||
}
|
||
QCheckBox::indicator:checked::after {
|
||
content: '✓';
|
||
color: white;
|
||
font-size: 14px;
|
||
font-weight: bold;
|
||
position: absolute;
|
||
top: 0px;
|
||
left: 3px;
|
||
}
|
||
""")
|
||
self.proxy_enable_checkbox.stateChanged.connect(self.save_proxy_config)
|
||
proxy_enable_layout.addWidget(proxy_enable_label)
|
||
proxy_enable_layout.addWidget(self.proxy_enable_checkbox)
|
||
proxy_enable_layout.setContentsMargins(0, 0, 0, 10)
|
||
proxy_layout.addLayout(proxy_enable_layout)
|
||
|
||
# 代理IP链接输入
|
||
proxy_url_layout = QHBoxLayout()
|
||
proxy_url_label = QLabel("代理IP链接:")
|
||
proxy_url_label.setStyleSheet("font-size: 14px; color: #666; min-width: 100px;")
|
||
self.proxy_url_input = QLineEdit()
|
||
self.proxy_url_input.setPlaceholderText("请输入获取代理IP的链接")
|
||
self.proxy_url_input.setStyleSheet("""
|
||
QLineEdit {
|
||
font-size: 14px;
|
||
padding: 8px;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
}
|
||
QLineEdit:focus {
|
||
border-color: #9c27b0;
|
||
outline: none;
|
||
}
|
||
""")
|
||
# 输入时自动保存
|
||
self.proxy_url_input.textChanged.connect(self.save_proxy_config)
|
||
proxy_url_layout.addWidget(proxy_url_label)
|
||
proxy_url_layout.addWidget(self.proxy_url_input)
|
||
proxy_url_layout.setContentsMargins(0, 0, 0, 15)
|
||
proxy_layout.addLayout(proxy_url_layout)
|
||
|
||
layout.addWidget(proxy_group)
|
||
|
||
# 创建域名后缀配置区域
|
||
domain_suffix_group = QGroupBox("域名后缀配置")
|
||
domain_suffix_group.setStyleSheet("""
|
||
QGroupBox {
|
||
font-size: 14px;
|
||
font-weight: bold;
|
||
color: #333;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
margin-bottom: 20px;
|
||
padding: 15px;
|
||
}
|
||
""")
|
||
domain_suffix_layout = QVBoxLayout(domain_suffix_group)
|
||
|
||
# 域名后缀输入
|
||
suffix_layout = QHBoxLayout()
|
||
suffix_label = QLabel("保留后缀:")
|
||
suffix_label.setStyleSheet("font-size: 14px; color: #666; min-width: 80px;")
|
||
self.suffix_input = QLineEdit()
|
||
self.suffix_input.setPlaceholderText("请输入需要保留的域名后缀，多个用逗号分隔，如：.com,.net")
|
||
self.suffix_input.setStyleSheet("""
|
||
QLineEdit {
|
||
font-size: 14px;
|
||
padding: 8px;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
}
|
||
QLineEdit:focus {
|
||
border-color: #ff9800;
|
||
outline: none;
|
||
}
|
||
""")
|
||
# 输入时自动保存
|
||
self.suffix_input.textChanged.connect(self.save_suffixes)
|
||
suffix_layout.addWidget(suffix_label)
|
||
suffix_layout.addWidget(self.suffix_input)
|
||
suffix_layout.setContentsMargins(0, 0, 0, 5)
|
||
domain_suffix_layout.addLayout(suffix_layout)
|
||
|
||
# 添加分隔符提示
|
||
suffix_note = QLabel("提示：多个域名后缀请用英文逗号分割")
|
||
suffix_note.setStyleSheet("font-size: 12px; color: #999; margin-bottom: 10px;")
|
||
domain_suffix_layout.addWidget(suffix_note)
|
||
|
||
layout.addWidget(domain_suffix_group)
|
||
|
||
# 创建检测选项配置区域
|
||
detect_group = QGroupBox("检测选项配置")
|
||
detect_group.setStyleSheet("""
|
||
QGroupBox {
|
||
font-size: 14px;
|
||
font-weight: bold;
|
||
color: #333;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
margin-bottom: 20px;
|
||
padding: 15px;
|
||
}
|
||
""")
|
||
detect_layout = QVBoxLayout(detect_group)
|
||
|
||
# 检测选项列表
|
||
self.detect_checkboxes = {}
|
||
detect_options = [
|
||
("detect_register", "1. 检测注册"),
|
||
("detect_chinaz", "2. 站长之家查询"),
|
||
("detect_aizhan", "3. 爱站网查询"),
|
||
("detect_baidu_site", "4. 百度site查询"),
|
||
("detect_360_site", "5. 360的site查询"),
|
||
("detect_baidu_security", "6. 百度网址安全中心查询"),
|
||
("detect_whois", "7. 聚查中的WHOIS 查询"),
|
||
("detect_beian", "8. 聚查中的备案相关查询"),
|
||
("detect_intercept", "9. 聚查中的拦截检测相关查询"),
|
||
("detect_juziseo", "10. 桔子历史"),
|
||
("detect_juziseo_outlink", "11. 桔子外链")
|
||
]
|
||
|
||
# 创建横向布局容器
|
||
detect_checkbox_container = QHBoxLayout()
|
||
detect_checkbox_container.setSpacing(20)
|
||
detect_checkbox_container.setContentsMargins(0, 0, 0, 0)
|
||
|
||
# 创建两列布局
|
||
left_column = QVBoxLayout()
|
||
right_column = QVBoxLayout()
|
||
|
||
# 将检测选项分为两列
|
||
for i, (key, label_text) in enumerate(detect_options):
|
||
checkbox = QCheckBox(label_text)
|
||
checkbox.setStyleSheet("font-size: 14px;")
|
||
checkbox.stateChanged.connect(self.save_detect_options)
|
||
self.detect_checkboxes[key] = checkbox
|
||
|
||
# 前5个选项放在左列，后5个放在右列
|
||
if i < 5:
|
||
left_column.addWidget(checkbox)
|
||
else:
|
||
right_column.addWidget(checkbox)
|
||
|
||
# 添加列到容器
|
||
detect_checkbox_container.addLayout(left_column)
|
||
detect_checkbox_container.addLayout(right_column)
|
||
|
||
# 添加容器到检测布局
|
||
detect_layout.addLayout(detect_checkbox_container)
|
||
|
||
# 创建检测配置容器
|
||
detect_config_container = QHBoxLayout()
|
||
detect_config_container.setSpacing(20)
|
||
detect_config_container.setContentsMargins(0, 0, 0, 20)
|
||
|
||
# 添加检测选项配置
|
||
detect_config_container.addWidget(detect_group, 2) # 占2份空间
|
||
|
||
# 创建检测线程配置区域
|
||
thread_group = QGroupBox("检测线程配置")
|
||
thread_group.setStyleSheet("""
|
||
QGroupBox {
|
||
font-size: 14px;
|
||
font-weight: bold;
|
||
color: #333;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
margin-bottom: 20px;
|
||
padding: 15px;
|
||
}
|
||
""")
|
||
thread_layout = QVBoxLayout(thread_group)
|
||
|
||
# 检测线程数输入
|
||
thread_count_layout = QHBoxLayout()
|
||
thread_count_label = QLabel("线程数量:")
|
||
thread_count_label.setStyleSheet("font-size: 14px; color: #666; min-width: 80px;")
|
||
self.thread_count_input = QLineEdit()
|
||
self.thread_count_input.setPlaceholderText("请输入检测线程数量，默认为100")
|
||
self.thread_count_input.setStyleSheet("""
|
||
QLineEdit {
|
||
font-size: 14px;
|
||
padding: 8px;
|
||
border: 1px solid #ddd;
|
||
border-radius: 4px;
|
||
}
|
||
QLineEdit:focus {
|
||
border-color: #2196F3;
|
||
outline: none;
|
||
}
|
||
""")
|
||
# 输入时自动保存
|
||
self.thread_count_input.textChanged.connect(self.save_thread_count)
|
||
thread_count_layout.addWidget(thread_count_label)
|
||
thread_count_layout.addWidget(self.thread_count_input)
|
||
thread_count_layout.setContentsMargins(0, 0, 0, 5)
|
||
thread_layout.addLayout(thread_count_layout)
|
||
|
||
# 添加线程数提示
|
||
thread_note = QLabel("提示：线程数量不宜设置过大，建议根据服务器性能调整")
|
||
thread_note.setStyleSheet("font-size: 12px; color: #999; margin-bottom: 10px;")
|
||
thread_layout.addWidget(thread_note)
|
||
|
||
# 添加检测线程配置到容器
|
||
detect_config_container.addWidget(thread_group, 1) # 占1份空间
|
||
|
||
# 将容器添加到主布局
|
||
layout.addLayout(detect_config_container)
|
||
|
||
# 创建状态标签
|
||
self.status_label = QLabel("")
|
||
self.status_label.setAlignment(Qt.AlignCenter)
|
||
self.status_label.setStyleSheet("font-size: 14px; color: #333; margin-top: 10px; padding: 10px; background-color: #f0f8ff; border-radius: 4px;")
|
||
layout.addWidget(self.status_label)
|
||
|
||
# 加载保存的账号密码
|
||
self.load_credentials()
|
||
|
||
# 加载域名后缀
|
||
self.load_suffixes()
|
||
|
||
# 加载检测线程数
|
||
self.load_thread_count()
|
||
|
||
# 加载cookie
|
||
self.load_cookies()
|
||
|
||
# 加载检测选项
|
||
self.load_detect_options()
|
||
|
||
# 加载代理IP配置
|
||
self.load_proxy_config()
|
||
|
||
logger.info("系统设置页面创建完成")
|
||
|
||
def init_redis(self):
|
||
"""
|
||
初始化Redis客户端连接
|
||
"""
|
||
try:
|
||
# 使用连接池
|
||
pool = redis.ConnectionPool(
|
||
host=config.REDIS_HOST,
|
||
port=config.REDIS_PORT,
|
||
password=config.REDIS_PASSWORD,
|
||
db=config.REDIS_DB,
|
||
decode_responses=True,
|
||
socket_connect_timeout=3, # 减少连接超时时间
|
||
socket_timeout=3, # 减少操作超时时间
|
||
max_connections=10 # 连接池大小
|
||
)
|
||
self.redis_client = redis.Redis(connection_pool=pool)
|
||
# 测试连接
|
||
self.redis_client.ping()
|
||
logger.info(f"Redis 连接成功: {config.REDIS_HOST}:{config.REDIS_PORT}")
|
||
self.use_redis = True
|
||
except Exception as e:
|
||
logger.warning(f"Redis 连接失败: {e}，将只保存到本地文件")
|
||
self.redis_client = None
|
||
self.use_redis = False
|
||
|
||
def check_redis_connection(self):
|
||
"""
|
||
检查Redis连接状态，如果连接已断开则重新连接
|
||
"""
|
||
if not self.use_redis or self.redis_client is None:
|
||
return False
|
||
|
||
try:
|
||
# 测试连接
|
||
self.redis_client.ping()
|
||
return True
|
||
except Exception as e:
|
||
logger.warning(f"Redis 连接已断开，尝试重新连接: {e}")
|
||
# 重新连接
|
||
self.init_redis()
|
||
return self.use_redis
|
||
|
||
def save_credentials(self):
|
||
"""
|
||
保存账号密码到本地文件和Redis
|
||
"""
|
||
credentials = {
|
||
"juming": {
|
||
"email": self.juming_email_input.text().strip(),
|
||
"password": self.juming_password_input.text().strip()
|
||
},
|
||
"juziseo": {
|
||
"email": self.juziseo_email_input.text().strip(),
|
||
"password": self.juziseo_password_input.text().strip()
|
||
}
|
||
}
|
||
|
||
try:
|
||
# 保存到本地文件（UI线程执行）
|
||
with open('credentials.json', 'w', encoding='utf-8') as f:
|
||
json.dump(credentials, f, indent=2, ensure_ascii=False)
|
||
logger.info("账号密码保存成功")
|
||
|
||
# 保存到Redis（后台线程执行）
|
||
def save_to_redis():
|
||
if self.check_redis_connection():
|
||
self.redis_client.set('domain_tool:credentials', json.dumps(credentials))
|
||
logger.info("账号密码已同步到Redis")
|
||
# 发布配置更新消息
|
||
self.redis_client.publish('domain_tool:config_update', 'credentials')
|
||
|
||
# 提交到线程池执行
|
||
task = RedisTask(save_to_redis)
|
||
self.thread_pool.start(task)
|
||
except Exception as e:
|
||
logger.error(f"保存账号密码失败: {e}")
|
||
|
||
def save_suffixes(self):
|
||
"""
|
||
保存域名后缀到本地文件和Redis
|
||
"""
|
||
suffixes = self.suffix_input.text().strip()
|
||
|
||
try:
|
||
# 保存到本地文件（UI线程执行）
|
||
with open('domain_suffixes.json', 'w', encoding='utf-8') as f:
|
||
json.dump({"suffixes": suffixes}, f, indent=2, ensure_ascii=False)
|
||
logger.info("域名后缀保存成功")
|
||
|
||
# 保存到Redis（后台线程执行）
|
||
def save_to_redis():
|
||
if self.check_redis_connection():
|
||
self.redis_client.set('domain_tool:domain_suffixes', suffixes)
|
||
logger.info("域名后缀已同步到Redis")
|
||
# 发布配置更新消息
|
||
self.redis_client.publish('domain_tool:config_update', 'domain_suffixes')
|
||
|
||
# 提交到线程池执行
|
||
task = RedisTask(save_to_redis)
|
||
self.thread_pool.start(task)
|
||
except Exception as e:
|
||
logger.error(f"保存域名后缀失败: {e}")
|
||
|
||
def load_suffixes(self):
|
||
"""
|
||
从本地文件加载域名后缀
|
||
"""
|
||
try:
|
||
if os.path.exists('domain_suffixes.json'):
|
||
with open('domain_suffixes.json', 'r', encoding='utf-8') as f:
|
||
data = json.load(f)
|
||
suffixes = data.get('suffixes', '.com,.net')
|
||
self.suffix_input.setText(suffixes)
|
||
logger.info("域名后缀加载成功")
|
||
else:
|
||
# 默认值
|
||
self.suffix_input.setText('.com,.net')
|
||
self.save_suffixes()
|
||
logger.info("使用默认域名后缀")
|
||
except Exception as e:
|
||
logger.error(f"加载域名后缀失败: {e}")
|
||
# 使用默认值
|
||
self.suffix_input.setText('.com,.net')
|
||
|
||
def load_credentials(self):
|
||
"""
|
||
从本地文件加载账号密码
|
||
"""
|
||
try:
|
||
if os.path.exists('credentials.json'):
|
||
with open('credentials.json', 'r', encoding='utf-8') as f:
|
||
credentials = json.load(f)
|
||
|
||
# 加载聚名账号密码
|
||
if 'juming' in credentials:
|
||
self.juming_email_input.setText(credentials['juming'].get('email', ''))
|
||
self.juming_password_input.setText(credentials['juming'].get('password', ''))
|
||
|
||
# 加载桔子SEO账号密码
|
||
if 'juziseo' in credentials:
|
||
self.juziseo_email_input.setText(credentials['juziseo'].get('email', ''))
|
||
self.juziseo_password_input.setText(credentials['juziseo'].get('password', ''))
|
||
|
||
logger.info("账号密码加载成功")
|
||
except Exception as e:
|
||
logger.error(f"加载账号密码失败: {e}")
|
||
|
||
def load_cookies(self):
|
||
"""
|
||
加载cookie
|
||
"""
|
||
try:
|
||
# 加载聚名cookie
|
||
jm = JM()
|
||
jm.load_cookies()
|
||
logger.info("聚名cookie加载成功")
|
||
|
||
# 加载桔子SEO cookie
|
||
juziseo = Juziseo()
|
||
juziseo.load_cookies()
|
||
logger.info("桔子SEO cookie加载成功")
|
||
|
||
# 加载聚查cookie
|
||
jc = JC()
|
||
jc.load_cookies()
|
||
logger.info("聚查cookie加载成功")
|
||
except Exception as e:
|
||
logger.error(f"加载cookie失败: {e}")
|
||
|
||
def login_juming(self):
|
||
"""
|
||
登录聚名网
|
||
"""
|
||
email = self.juming_email_input.text().strip()
|
||
password = self.juming_password_input.text().strip()
|
||
|
||
if not email or not password:
|
||
self.status_label.setText("请输入账号和密码")
|
||
return
|
||
|
||
self.status_label.setText("正在登录聚名网...")
|
||
|
||
# 创建并启动登录线程
|
||
self.login_thread = LoginThread(email, password, "juming")
|
||
self.login_thread.finished.connect(self.login_finished)
|
||
self.login_thread.start()
|
||
|
||
logger.info(f"开始登录聚名网: {email}")
|
||
|
||
def login_juziseo(self):
|
||
"""
|
||
登录桔子SEO
|
||
"""
|
||
email = self.juziseo_email_input.text().strip()
|
||
password = self.juziseo_password_input.text().strip()
|
||
|
||
if not email or not password:
|
||
self.status_label.setText("请输入账号和密码")
|
||
return
|
||
|
||
self.status_label.setText("正在登录桔子SEO...")
|
||
|
||
# 创建并启动登录线程
|
||
self.login_thread = LoginThread(email, password, "juziseo")
|
||
self.login_thread.finished.connect(self.login_finished)
|
||
self.login_thread.start()
|
||
|
||
logger.info(f"开始登录桔子SEO: {email}")
|
||
|
||
def login_jucha(self):
|
||
"""
|
||
登录聚查网
|
||
"""
|
||
# 检查聚名Cookie是否存在
|
||
if not os.path.exists('juming_cookies.pkl'):
|
||
self.status_label.setText("请先登录聚名网")
|
||
return
|
||
|
||
self.status_label.setText("正在登录聚查网...")
|
||
|
||
# 创建并启动登录线程
|
||
self.login_thread = LoginThread("", "", "jucha")
|
||
self.login_thread.finished.connect(self.login_finished)
|
||
self.login_thread.start()
|
||
|
||
logger.info("开始登录聚查网")
|
||
|
||
def save_detect_options(self):
|
||
"""
|
||
保存检测选项到本地文件和Redis
|
||
"""
|
||
detect_options = {}
|
||
for key, checkbox in self.detect_checkboxes.items():
|
||
detect_options[key] = checkbox.isChecked()
|
||
|
||
try:
|
||
# 保存到本地文件（UI线程执行）
|
||
with open('detect_options.json', 'w', encoding='utf-8') as f:
|
||
json.dump(detect_options, f, indent=2, ensure_ascii=False)
|
||
logger.info("检测选项保存成功")
|
||
|
||
# 保存到Redis（后台线程执行）
|
||
def save_to_redis():
|
||
if self.check_redis_connection():
|
||
self.redis_client.set('domain_tool:detect_options', json.dumps(detect_options))
|
||
logger.info("检测选项已同步到Redis")
|
||
# 发布配置更新消息
|
||
self.redis_client.publish('domain_tool:config_update', 'detect_options')
|
||
|
||
# 提交到线程池执行
|
||
task = RedisTask(save_to_redis)
|
||
self.thread_pool.start(task)
|
||
except Exception as e:
|
||
logger.error(f"保存检测选项失败: {e}")
|
||
|
||
def load_detect_options(self):
|
||
"""
|
||
从本地文件加载检测选项
|
||
"""
|
||
try:
|
||
if os.path.exists('detect_options.json'):
|
||
with open('detect_options.json', 'r', encoding='utf-8') as f:
|
||
detect_options = json.load(f)
|
||
|
||
for key, checked in detect_options.items():
|
||
if key in self.detect_checkboxes:
|
||
self.detect_checkboxes[key].setChecked(checked)
|
||
|
||
logger.info("检测选项加载成功")
|
||
else:
|
||
# 默认值：全部勾选
|
||
for checkbox in self.detect_checkboxes.values():
|
||
checkbox.setChecked(True)
|
||
self.save_detect_options()
|
||
logger.info("使用默认检测选项")
|
||
except Exception as e:
|
||
logger.error(f"加载检测选项失败: {e}")
|
||
# 使用默认值
|
||
for checkbox in self.detect_checkboxes.values():
|
||
checkbox.setChecked(True)
|
||
|
||
def save_proxy_config(self):
|
||
"""
|
||
保存代理IP配置到本地文件和Redis
|
||
"""
|
||
proxy_enable = self.proxy_enable_checkbox.isChecked()
|
||
proxy_url = self.proxy_url_input.text().strip()
|
||
|
||
try:
|
||
# 保存到本地文件（UI线程执行）
|
||
with open('proxy_config.json', 'w', encoding='utf-8') as f:
|
||
json.dump({"proxy_enable": proxy_enable, "proxy_url": proxy_url}, f, indent=2, ensure_ascii=False)
|
||
logger.info("代理IP配置保存成功")
|
||
|
||
# 保存到Redis（后台线程执行）
|
||
def save_to_redis():
|
||
if self.check_redis_connection():
|
||
self.redis_client.set('domain_tool:proxy_config', json.dumps({"proxy_enable": proxy_enable, "proxy_url": proxy_url}))
|
||
logger.info("代理IP配置已同步到Redis")
|
||
# 发布配置更新消息
|
||
self.redis_client.publish('domain_tool:config_update', 'proxy_config')
|
||
|
||
# 提交到线程池执行
|
||
task = RedisTask(save_to_redis)
|
||
self.thread_pool.start(task)
|
||
except Exception as e:
|
||
logger.error(f"保存代理IP配置失败: {e}")
|
||
|
||
def load_proxy_config(self):
|
||
"""
|
||
从本地文件加载代理IP配置
|
||
"""
|
||
try:
|
||
if os.path.exists('proxy_config.json'):
|
||
with open('proxy_config.json', 'r', encoding='utf-8') as f:
|
||
proxy_config = json.load(f)
|
||
proxy_enable = proxy_config.get('proxy_enable', False)
|
||
proxy_url = proxy_config.get('proxy_url', '')
|
||
self.proxy_enable_checkbox.setChecked(proxy_enable)
|
||
self.proxy_url_input.setText(proxy_url)
|
||
logger.info("代理IP配置加载成功")
|
||
else:
|
||
# 默认值：未启用，空链接
|
||
self.proxy_enable_checkbox.setChecked(False)
|
||
self.proxy_url_input.setText('')
|
||
self.save_proxy_config()
|
||
logger.info("使用默认代理IP配置")
|
||
except Exception as e:
|
||
logger.error(f"加载代理IP配置失败: {e}")
|
||
# 使用默认值
|
||
self.proxy_enable_checkbox.setChecked(False)
|
||
self.proxy_url_input.setText('')
|
||
|
||
def toggle_password_visibility(self, password_input):
|
||
"""
|
||
切换密码输入框的可见性
|
||
|
||
:param password_input: 密码输入框
|
||
"""
|
||
if password_input.echoMode() == QLineEdit.Password:
|
||
password_input.setEchoMode(QLineEdit.Normal)
|
||
# 找到对应的按钮并修改图标
|
||
if password_input == self.juming_password_input:
|
||
self.juming_password_visibility_btn.setText("🙈")
|
||
elif password_input == self.juziseo_password_input:
|
||
self.juziseo_password_visibility_btn.setText("🙈")
|
||
else:
|
||
password_input.setEchoMode(QLineEdit.Password)
|
||
# 找到对应的按钮并修改图标
|
||
if password_input == self.juming_password_input:
|
||
self.juming_password_visibility_btn.setText("👁")
|
||
elif password_input == self.juziseo_password_input:
|
||
self.juziseo_password_visibility_btn.setText("👁")
|
||
|
||
def login_finished(self, success, message):
|
||
"""
|
||
登录完成
|
||
|
||
:param success: 是否成功
|
||
:param message: 消息
|
||
"""
|
||
self.status_label.setText(message)
|
||
|
||
# 登录成功后保存账号密码
|
||
if success:
|
||
self.save_credentials()
|
||
# 聚名登录成功后自动登录聚查
|
||
if self.login_thread.platform == "juming":
|
||
self.status_label.setText("聚名登录成功，正在自动登录聚查...")
|
||
# 延迟一秒后登录聚查，确保聚名cookie已保存
|
||
QTimer.singleShot(1000, self.login_jucha)
|
||
|
||
logger.info(f"登录完成: {message}")
|
||
|
||
def save_thread_count(self):
|
||
"""
|
||
保存检测线程数到本地文件和Redis
|
||
"""
|
||
thread_count = self.thread_count_input.text().strip()
|
||
|
||
try:
|
||
# 保存到本地文件（UI线程执行）
|
||
with open('thread_count.json', 'w', encoding='utf-8') as f:
|
||
json.dump({"thread_count": thread_count}, f, indent=2, ensure_ascii=False)
|
||
logger.info("检测线程数配置保存成功")
|
||
|
||
# 保存到Redis（后台线程执行）
|
||
def save_to_redis():
|
||
if self.check_redis_connection():
|
||
self.redis_client.set('domain_tool:thread_count', thread_count)
|
||
logger.info("检测线程数配置已同步到Redis")
|
||
# 发布配置更新消息
|
||
self.redis_client.publish('domain_tool:config_update', 'thread_count')
|
||
|
||
# 提交到线程池执行
|
||
task = RedisTask(save_to_redis)
|
||
self.thread_pool.start(task)
|
||
except Exception as e:
|
||
logger.error(f"保存检测线程数配置失败: {e}")
|
||
|
||
def load_thread_count(self):
|
||
"""
|
||
从本地文件加载检测线程数
|
||
"""
|
||
try:
|
||
if os.path.exists('thread_count.json'):
|
||
with open('thread_count.json', 'r', encoding='utf-8') as f:
|
||
thread_config = json.load(f)
|
||
thread_count = thread_config.get('thread_count', '100')
|
||
self.thread_count_input.setText(thread_count)
|
||
logger.info("检测线程数配置加载成功")
|
||
else:
|
||
# 默认值：100
|
||
self.thread_count_input.setText('100')
|
||
self.save_thread_count()
|
||
logger.info("使用默认检测线程数配置")
|
||
except Exception as e:
|
||
logger.error(f"加载检测线程数配置失败: {e}")
|
||
# 使用默认值
|
||
self.thread_count_input.setText('100') |