424 lines
14 KiB
Python
424 lines
14 KiB
Python
# <AUTO_MAA:A MAA Multi Account Management and Automation Tool>
|
||
# Copyright © <2024> <DLmaster361>
|
||
|
||
# This file is part of AUTO_MAA.
|
||
|
||
# AUTO_MAA is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published
|
||
# by the Free Software Foundation, either version 3 of the License,
|
||
# or (at your option) any later version.
|
||
|
||
# AUTO_MAA is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty
|
||
# of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
|
||
# the GNU General Public License for more details.
|
||
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with AUTO_MAA. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
# DLmaster_361@163.com
|
||
|
||
"""
|
||
AUTO_MAA
|
||
AUTO_MAA主界面
|
||
v4.2
|
||
作者：DLmaster_361
|
||
"""
|
||
|
||
from loguru import logger
|
||
from PySide6.QtWidgets import (
|
||
QWidget,
|
||
QVBoxLayout,
|
||
QHBoxLayout,
|
||
QSpacerItem,
|
||
QSizePolicy,
|
||
QFileDialog,
|
||
)
|
||
from PySide6.QtCore import Qt, QSize, QUrl
|
||
from PySide6.QtGui import QDesktopServices, QColor
|
||
from qfluentwidgets import (
|
||
FluentIcon,
|
||
ScrollArea,
|
||
SimpleCardWidget,
|
||
PrimaryToolButton,
|
||
TextBrowser,
|
||
)
|
||
import re
|
||
import shutil
|
||
import requests
|
||
import json
|
||
import time
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
from app.core import Config, MainInfoBar
|
||
from .Widget import Banner, IconButton
|
||
|
||
|
||
class Home(QWidget):
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent)
|
||
self.setObjectName("主页")
|
||
|
||
self.banner = Banner()
|
||
self.banner_text = TextBrowser()
|
||
|
||
widget = QWidget()
|
||
Layout = QVBoxLayout(widget)
|
||
|
||
Layout.addWidget(self.banner)
|
||
Layout.addWidget(self.banner_text)
|
||
Layout.setStretch(0, 2)
|
||
Layout.setStretch(1, 3)
|
||
|
||
v_layout = QVBoxLayout(self.banner)
|
||
v_layout.setContentsMargins(0, 0, 0, 15)
|
||
v_layout.setSpacing(5)
|
||
v_layout.setAlignment(Qt.AlignmentFlag.AlignTop)
|
||
|
||
# 空白占位符
|
||
v_layout.addItem(
|
||
QSpacerItem(10, 20, QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Minimum)
|
||
)
|
||
|
||
# 顶部部分 (按钮组)
|
||
h1_layout = QHBoxLayout()
|
||
h1_layout.setAlignment(Qt.AlignmentFlag.AlignTop)
|
||
|
||
# 左边留白区域
|
||
h1_layout.addStretch()
|
||
|
||
# 按钮组
|
||
buttonGroup = ButtonGroup()
|
||
buttonGroup.setMaximumHeight(320)
|
||
h1_layout.addWidget(buttonGroup)
|
||
|
||
# 空白占位符
|
||
h1_layout.addItem(
|
||
QSpacerItem(20, 10, QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Minimum)
|
||
)
|
||
|
||
# 将顶部水平布局添加到垂直布局
|
||
v_layout.addLayout(h1_layout)
|
||
|
||
# 中间留白区域
|
||
v_layout.addItem(
|
||
QSpacerItem(10, 10, QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Minimum)
|
||
)
|
||
v_layout.addStretch()
|
||
|
||
# 中间留白区域
|
||
v_layout.addItem(
|
||
QSpacerItem(10, 10, QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Minimum)
|
||
)
|
||
v_layout.addStretch()
|
||
|
||
# 底部部分 (图片切换按钮)
|
||
h2_layout = QHBoxLayout()
|
||
h2_layout.setAlignment(Qt.AlignmentFlag.AlignTop)
|
||
|
||
# 左边留白区域
|
||
h2_layout.addItem(
|
||
QSpacerItem(20, 10, QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Minimum)
|
||
)
|
||
|
||
# # 公告卡片
|
||
# noticeCard = NoticeCard()
|
||
# h2_layout.addWidget(noticeCard)
|
||
|
||
h2_layout.addStretch()
|
||
|
||
# 自定义图像按钮布局
|
||
self.imageButton = PrimaryToolButton(FluentIcon.IMAGE_EXPORT)
|
||
self.imageButton.setFixedSize(56, 56)
|
||
self.imageButton.setIconSize(QSize(32, 32))
|
||
self.imageButton.clicked.connect(self.get_home_image)
|
||
|
||
v1_layout = QVBoxLayout()
|
||
v1_layout.addWidget(self.imageButton, alignment=Qt.AlignmentFlag.AlignBottom)
|
||
|
||
h2_layout.addLayout(v1_layout)
|
||
|
||
# 空白占位符
|
||
h2_layout.addItem(
|
||
QSpacerItem(25, 10, QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Minimum)
|
||
)
|
||
|
||
# 将底部水平布局添加到垂直布局
|
||
v_layout.addLayout(h2_layout)
|
||
|
||
layout = QVBoxLayout()
|
||
scrollArea = ScrollArea()
|
||
scrollArea.setWidgetResizable(True)
|
||
scrollArea.setWidget(widget)
|
||
layout.addWidget(scrollArea)
|
||
self.setLayout(layout)
|
||
|
||
self.set_banner()
|
||
|
||
def get_home_image(self) -> None:
|
||
"""获取主页图片"""
|
||
|
||
if (
|
||
Config.global_config.get(Config.global_config.function_HomeImageMode)
|
||
== "默认"
|
||
):
|
||
pass
|
||
elif (
|
||
Config.global_config.get(Config.global_config.function_HomeImageMode)
|
||
== "自定义"
|
||
):
|
||
|
||
file_path, _ = QFileDialog.getOpenFileName(
|
||
self, "打开自定义主页图片", "", "图片文件 (*.png *.jpg *.bmp)"
|
||
)
|
||
if file_path:
|
||
|
||
for file in Config.app_path.glob(
|
||
"resources/images/Home/BannerCustomize.*"
|
||
):
|
||
file.unlink()
|
||
|
||
shutil.copy(
|
||
file_path,
|
||
Config.app_path
|
||
/ f"resources/images/Home/BannerCustomize{Path(file_path).suffix}",
|
||
)
|
||
|
||
logger.info(f"自定义主页图片更换成功：{file_path}")
|
||
MainInfoBar.push_info_bar(
|
||
"success",
|
||
"主页图片更换成功",
|
||
"自定义主页图片更换成功！",
|
||
3000,
|
||
)
|
||
|
||
else:
|
||
logger.warning("自定义主页图片更换失败：未选择图片文件")
|
||
MainInfoBar.push_info_bar(
|
||
"warning",
|
||
"主页图片更换失败",
|
||
"未选择图片文件！",
|
||
5000,
|
||
)
|
||
elif (
|
||
Config.global_config.get(Config.global_config.function_HomeImageMode)
|
||
== "主题图像"
|
||
):
|
||
|
||
# 从远程服务器获取最新主题图像
|
||
for _ in range(3):
|
||
try:
|
||
response = requests.get(
|
||
"https://gitee.com/DLmaster_361/AUTO_MAA/raw/server/theme_image.json"
|
||
)
|
||
theme_image = response.json()
|
||
break
|
||
except Exception as e:
|
||
err = e
|
||
time.sleep(0.1)
|
||
else:
|
||
logger.error(f"获取最新主题图像时出错：\n{err}")
|
||
MainInfoBar.push_info_bar(
|
||
"error",
|
||
"主题图像获取失败",
|
||
f"获取最新主题图像信息时出错：\n{err}",
|
||
-1,
|
||
)
|
||
return None
|
||
|
||
if (Config.app_path / "resources/theme_image.json").exists():
|
||
with (Config.app_path / "resources/theme_image.json").open(
|
||
mode="r", encoding="utf-8"
|
||
) as f:
|
||
theme_image_local = json.load(f)
|
||
time_local = datetime.strptime(
|
||
theme_image_local["time"], "%Y-%m-%d %H:%M"
|
||
)
|
||
else:
|
||
time_local = datetime.strptime("2000-01-01 00:00", "%Y-%m-%d %H:%M")
|
||
|
||
if not (
|
||
Config.app_path / "resources/images/Home/BannerTheme.jpg"
|
||
).exists() or (
|
||
datetime.now()
|
||
> datetime.strptime(theme_image["time"], "%Y-%m-%d %H:%M")
|
||
and datetime.strptime(theme_image["time"], "%Y-%m-%d %H:%M")
|
||
> time_local
|
||
):
|
||
|
||
response = requests.get(theme_image["url"])
|
||
if response.status_code == 200:
|
||
|
||
with open(
|
||
Config.app_path / "resources/images/Home/BannerTheme.jpg", "wb"
|
||
) as file:
|
||
file.write(response.content)
|
||
|
||
logger.info(f"主题图像「{theme_image["name"]}」下载成功")
|
||
MainInfoBar.push_info_bar(
|
||
"success",
|
||
"主题图像下载成功",
|
||
f"「{theme_image["name"]}」下载成功！",
|
||
3000,
|
||
)
|
||
|
||
else:
|
||
|
||
logger.error("主题图像下载失败")
|
||
MainInfoBar.push_info_bar(
|
||
"error",
|
||
"主题图像下载失败",
|
||
f"主题图像下载失败：{response.status_code}",
|
||
-1,
|
||
)
|
||
|
||
with (Config.app_path / "resources/theme_image.json").open(
|
||
mode="w", encoding="utf-8"
|
||
) as f:
|
||
json.dump(theme_image, f, ensure_ascii=False, indent=4)
|
||
|
||
else:
|
||
|
||
logger.info("主题图像已是最新")
|
||
MainInfoBar.push_info_bar(
|
||
"info",
|
||
"主题图像已是最新",
|
||
"主题图像已是最新！",
|
||
3000,
|
||
)
|
||
|
||
self.set_banner()
|
||
|
||
def set_banner(self):
|
||
"""设置主页图像"""
|
||
if (
|
||
Config.global_config.get(Config.global_config.function_HomeImageMode)
|
||
== "默认"
|
||
):
|
||
self.banner.set_banner_image(
|
||
str(Config.app_path / "resources/images/Home/BannerDefault.png")
|
||
)
|
||
self.imageButton.hide()
|
||
self.banner_text.setVisible(False)
|
||
elif (
|
||
Config.global_config.get(Config.global_config.function_HomeImageMode)
|
||
== "自定义"
|
||
):
|
||
for file in Config.app_path.glob("resources/images/Home/BannerCustomize.*"):
|
||
self.banner.set_banner_image(str(file))
|
||
break
|
||
self.imageButton.show()
|
||
self.banner_text.setVisible(False)
|
||
elif (
|
||
Config.global_config.get(Config.global_config.function_HomeImageMode)
|
||
== "主题图像"
|
||
):
|
||
self.banner.set_banner_image(
|
||
str(Config.app_path / "resources/images/Home/BannerTheme.jpg")
|
||
)
|
||
self.imageButton.show()
|
||
self.banner_text.setVisible(True)
|
||
|
||
if (Config.app_path / "resources/theme_image.json").exists():
|
||
with (Config.app_path / "resources/theme_image.json").open(
|
||
mode="r", encoding="utf-8"
|
||
) as f:
|
||
theme_image = json.load(f)
|
||
html_content = theme_image["html"]
|
||
else:
|
||
html_content = "<h1>主题图像</h1><p>主题图像信息未知</p>"
|
||
|
||
self.banner_text.setHtml(re.sub(r"<img[^>]*>", "", html_content))
|
||
|
||
|
||
class ButtonGroup(SimpleCardWidget):
|
||
"""显示主页和 GitHub 按钮的竖直按钮组"""
|
||
|
||
def __init__(self, parent=None):
|
||
super().__init__(parent=parent)
|
||
|
||
self.setFixedSize(56, 180)
|
||
|
||
layout = QVBoxLayout(self)
|
||
layout.setAlignment(Qt.AlignmentFlag.AlignTop)
|
||
|
||
# 创建主页按钮
|
||
home_button = IconButton(
|
||
FluentIcon.HOME.icon(color=QColor("#fff")),
|
||
tip_title="AUTO_MAA官网",
|
||
tip_content="AUTO_MAA官方文档站",
|
||
isTooltip=True,
|
||
)
|
||
home_button.setIconSize(QSize(32, 32))
|
||
home_button.clicked.connect(self.open_home)
|
||
layout.addWidget(home_button)
|
||
|
||
# 创建 GitHub 按钮
|
||
github_button = IconButton(
|
||
FluentIcon.GITHUB.icon(color=QColor("#fff")),
|
||
tip_title="Github仓库",
|
||
tip_content="如果本项目有帮助到您~\n不妨给项目点一个Star⭐",
|
||
isTooltip=True,
|
||
)
|
||
github_button.setIconSize(QSize(32, 32))
|
||
github_button.clicked.connect(self.open_github)
|
||
layout.addWidget(github_button)
|
||
|
||
# # 创建 文档 按钮
|
||
# doc_button = IconButton(
|
||
# FluentIcon.DICTIONARY.icon(color=QColor("#fff")),
|
||
# tip_title="自助排障文档",
|
||
# tip_content="点击打开自助排障文档,好孩子都能看懂",
|
||
# isTooltip=True,
|
||
# )
|
||
# doc_button.setIconSize(QSize(32, 32))
|
||
# doc_button.clicked.connect(self.open_doc)
|
||
# layout.addWidget(doc_button)
|
||
|
||
# 创建 Q群 按钮
|
||
doc_button = IconButton(
|
||
FluentIcon.CHAT.icon(color=QColor("#fff")),
|
||
tip_title="官方社群",
|
||
tip_content="加入官方群聊【AUTO_MAA绝赞DeBug中！】",
|
||
isTooltip=True,
|
||
)
|
||
doc_button.setIconSize(QSize(32, 32))
|
||
doc_button.clicked.connect(self.open_chat)
|
||
layout.addWidget(doc_button)
|
||
|
||
# 创建 官方店铺 按钮 (当然没有)
|
||
doc_button = IconButton(
|
||
FluentIcon.SHOPPING_CART.icon(color=QColor("#fff")),
|
||
tip_title="官方店铺",
|
||
tip_content="暂时没有官方店铺，但是可以加入官方群聊哦~",
|
||
isTooltip=True,
|
||
)
|
||
doc_button.setIconSize(QSize(32, 32))
|
||
doc_button.clicked.connect(self.open_sales)
|
||
layout.addWidget(doc_button)
|
||
|
||
def _normalBackgroundColor(self):
|
||
return QColor(0, 0, 0, 96)
|
||
|
||
def open_home(self):
|
||
"""打开主页链接"""
|
||
QDesktopServices.openUrl(QUrl("https://clozya.github.io/AUTOMAA_docs"))
|
||
|
||
def open_github(self):
|
||
"""打开 GitHub 链接"""
|
||
QDesktopServices.openUrl(QUrl("https://github.com/DLmaster361/AUTO_MAA"))
|
||
|
||
def open_chat(self):
|
||
"""打开 Q群 链接"""
|
||
QDesktopServices.openUrl(QUrl("https://qm.qq.com/q/bd9fISNoME"))
|
||
|
||
def open_doc(self):
|
||
"""打开 文档 链接"""
|
||
QDesktopServices.openUrl(QUrl("https://clozya.github.io/AUTOMAA_docs"))
|
||
|
||
def open_sales(self):
|
||
"""其实还是打开 Q群 链接"""
|
||
QDesktopServices.openUrl(QUrl("https://qm.qq.com/q/bd9fISNoME"))
|