345 lines
13 KiB
Python
345 lines
13 KiB
Python
import json
|
|
import os
|
|
import re
|
|
import configparser as config
|
|
from pathlib import Path
|
|
|
|
from datetime import datetime
|
|
import time
|
|
from dateutil import parser
|
|
from loguru import logger
|
|
from file import base_directory, config_center
|
|
|
|
import list_
|
|
|
|
if os.name == 'nt':
|
|
from win32com.client import Dispatch
|
|
|
|
conf = config.ConfigParser()
|
|
name = 'Class Widgets'
|
|
|
|
PLUGINS_DIR = Path(base_directory) / 'plugins'
|
|
|
|
# app 图标
|
|
if os.name == 'nt':
|
|
app_icon = os.path.join(base_directory, 'img', 'favicon.ico')
|
|
elif os.name == 'darwin':
|
|
app_icon = os.path.join(base_directory, 'img', 'favicon.icns')
|
|
else:
|
|
app_icon = os.path.join(base_directory, 'img', 'favicon.png')
|
|
|
|
update_countdown_custom_last = 0
|
|
countdown_cnt = 0
|
|
|
|
def load_theme_config(theme):
|
|
try:
|
|
with open(f'{base_directory}/ui/{theme}/theme.json', 'r', encoding='utf-8') as file:
|
|
data = json.load(file)
|
|
return data
|
|
except FileNotFoundError:
|
|
logger.warning(f"主题配置文件 {theme} 不存在，返回默认配置")
|
|
return f'{base_directory}/ui/default/theme.json'
|
|
except Exception as e:
|
|
logger.error(f"加载主题数据时出错: {e}")
|
|
return None
|
|
|
|
|
|
def load_plugin_config():
|
|
try:
|
|
if os.path.exists(f'{base_directory}/config/plugin.json'): # 如果配置文件存在
|
|
with open(f'{base_directory}/config/plugin.json', 'r', encoding='utf-8') as file:
|
|
data = json.load(file)
|
|
else:
|
|
with open(f'{base_directory}/config/plugin.json', 'w', encoding='utf-8') as file:
|
|
data = {"enabled_plugins": []}
|
|
json.dump(data, file, ensure_ascii=False, indent=4)
|
|
return data
|
|
except Exception as e:
|
|
logger.error(f"加载启用插件数据时出错: {e}")
|
|
return None
|
|
|
|
|
|
def save_plugin_config(data):
|
|
data_dict = load_plugin_config()
|
|
data_dict.update(data)
|
|
try:
|
|
with open(f'{base_directory}/config/plugin.json', 'w', encoding='utf-8') as file:
|
|
json.dump(data_dict, file, ensure_ascii=False, indent=4)
|
|
return True
|
|
except Exception as e:
|
|
logger.error(f"保存启用插件数据时出错: {e}")
|
|
return False
|
|
|
|
|
|
def save_installed_plugin(data):
|
|
data = {"plugins": data}
|
|
try:
|
|
with open(f'{base_directory}/plugins/plugins_from_pp.json', 'w', encoding='utf-8') as file:
|
|
json.dump(data, file, ensure_ascii=False, indent=4)
|
|
return True
|
|
except Exception as e:
|
|
logger.error(f"保存已安装插件数据时出错: {e}")
|
|
return False
|
|
|
|
|
|
def load_theme_width(theme):
|
|
try:
|
|
with open(f'{base_directory}/ui/{theme}/theme.json', 'r', encoding='utf-8') as file:
|
|
data = json.load(file)
|
|
return data['widget_width']
|
|
except Exception as e:
|
|
logger.error(f"加载主题宽度时出错: {e}")
|
|
return list_.widget_width
|
|
|
|
|
|
def is_temp_week():
|
|
if config_center.read_conf('Temp', 'set_week') is None or config_center.read_conf('Temp', 'set_week') == '':
|
|
return False
|
|
else:
|
|
return config_center.read_conf('Temp', 'set_week')
|
|
|
|
|
|
def is_temp_schedule():
|
|
if (
|
|
config_center.read_conf('Temp', 'temp_schedule') is None
|
|
or config_center.read_conf('Temp', 'temp_schedule') == ''
|
|
):
|
|
return False
|
|
else:
|
|
return config_center.read_conf('Temp', 'temp_schedule')
|
|
|
|
|
|
def add_shortcut_to_startmenu(file='', icon=''):
|
|
if os.name != 'nt':
|
|
return
|
|
try:
|
|
if file == "":
|
|
file_path = os.path.realpath(__file__)
|
|
else:
|
|
file_path = os.path.abspath(file) # 将相对路径转换为绝对路径
|
|
|
|
if icon == "":
|
|
icon_path = file_path # 如果未指定图标路径，则使用程序路径
|
|
else:
|
|
icon_path = os.path.abspath(icon) # 将相对路径转换为绝对路径
|
|
|
|
# 获取开始菜单文件夹路径
|
|
menu_folder = os.path.join(os.getenv('APPDATA'), 'Microsoft', 'Windows', 'Start Menu', 'Programs')
|
|
|
|
# 快捷方式文件名（使用文件名或自定义名称）
|
|
name = os.path.splitext(os.path.basename(file_path))[0] # 使用文件名作为快捷方式名称
|
|
shortcut_path = os.path.join(menu_folder, f'{name}.lnk')
|
|
|
|
# 创建快捷方式
|
|
shell = Dispatch('WScript.Shell')
|
|
shortcut = shell.CreateShortCut(shortcut_path)
|
|
shortcut.Targetpath = file_path
|
|
shortcut.WorkingDirectory = os.path.dirname(file_path)
|
|
shortcut.IconLocation = icon_path # 设置图标路径
|
|
shortcut.save()
|
|
except Exception as e:
|
|
logger.error(f"创建开始菜单快捷方式时出错: {e}")
|
|
|
|
|
|
def add_shortcut(file='', icon=''):
|
|
if os.name != 'nt':
|
|
return
|
|
try:
|
|
if file == "":
|
|
file_path = os.path.realpath(__file__)
|
|
else:
|
|
file_path = os.path.abspath(file)
|
|
|
|
if icon == "":
|
|
icon_path = file_path
|
|
else:
|
|
icon_path = os.path.abspath(icon)
|
|
|
|
# 获取桌面文件夹路径
|
|
desktop_folder = os.path.join(os.environ['USERPROFILE'], 'Desktop')
|
|
|
|
# 快捷方式文件名（使用文件名或自定义名称）
|
|
name = os.path.splitext(os.path.basename(file_path))[0] # 使用文件名作为快捷方式名称
|
|
shortcut_path = os.path.join(desktop_folder, f'{name}.lnk')
|
|
|
|
# 创建快捷方式
|
|
shell = Dispatch('WScript.Shell')
|
|
shortcut = shell.CreateShortCut(shortcut_path)
|
|
shortcut.Targetpath = file_path
|
|
shortcut.WorkingDirectory = os.path.dirname(file_path)
|
|
shortcut.IconLocation = icon_path # 设置图标路径
|
|
shortcut.save()
|
|
except Exception as e:
|
|
logger.error(f"创建桌面快捷方式时出错: {e}")
|
|
|
|
|
|
def add_to_startup(file_path=f'{base_directory}/ClassWidgets.exe', icon_path=''): # 注册到开机启动
|
|
if os.name != 'nt':
|
|
return
|
|
if file_path == "":
|
|
file_path = os.path.realpath(__file__)
|
|
else:
|
|
file_path = os.path.abspath(file_path) # 将相对路径转换为绝对路径
|
|
|
|
if icon_path == "":
|
|
icon_path = file_path # 如果未指定图标路径，则使用程序路径
|
|
else:
|
|
icon_path = os.path.abspath(icon_path) # 将相对路径转换为绝对路径
|
|
|
|
# 获取启动文件夹路径
|
|
startup_folder = os.path.join(os.getenv('APPDATA'), 'Microsoft', 'Windows', 'Start Menu', 'Programs', 'Startup')
|
|
|
|
# 快捷方式文件名（使用文件名或自定义名称）
|
|
name = os.path.splitext(os.path.basename(file_path))[0] # 使用文件名作为快捷方式名称
|
|
shortcut_path = os.path.join(startup_folder, f'{name}.lnk')
|
|
|
|
# 创建快捷方式
|
|
shell = Dispatch('WScript.Shell')
|
|
shortcut = shell.CreateShortCut(shortcut_path)
|
|
shortcut.Targetpath = file_path
|
|
shortcut.WorkingDirectory = os.path.dirname(file_path)
|
|
shortcut.IconLocation = icon_path # 设置图标路径
|
|
shortcut.save()
|
|
|
|
|
|
def remove_from_startup():
|
|
startup_folder = os.path.join(os.getenv('APPDATA'), 'Microsoft', 'Windows', 'Start Menu', 'Programs', 'Startup')
|
|
shortcut_path = os.path.join(startup_folder, f'{name}.lnk')
|
|
if os.path.exists(shortcut_path):
|
|
os.remove(shortcut_path)
|
|
|
|
|
|
def get_time_offset(): # 获取时差偏移
|
|
time_offset = config_center.read_conf('General', 'time_offset')
|
|
if time_offset is None or time_offset == '' or time_offset == '0':
|
|
return 0
|
|
else:
|
|
return int(time_offset)
|
|
|
|
def update_countdown(cnt):
|
|
global update_countdown_custom_last
|
|
global countdown_cnt
|
|
if (length:=len(config_center.read_conf('Date', 'cd_text_custom').split(','))) == 0:
|
|
countdown_cnt = -1
|
|
elif config_center.read_conf('Date', 'countdown_custom_mode') == '1':
|
|
countdown_cnt = cnt
|
|
elif (nowtime:=time.time()) - update_countdown_custom_last > int(config_center.read_conf('Date', 'countdown_upd_cd')):
|
|
update_countdown_custom_last = nowtime
|
|
countdown_cnt += 1
|
|
if countdown_cnt >= length:
|
|
countdown_cnt = 0 if length != 0 else -1
|
|
|
|
def get_cd_text_custom():
|
|
global countdown_cnt
|
|
if countdown_cnt == -1:
|
|
return '未设置'
|
|
if countdown_cnt >= len(li:=config_center.read_conf('Date', 'cd_text_custom').split(',')):
|
|
return '未设置'
|
|
return li[countdown_cnt] if countdown_cnt >= 0 else ''
|
|
|
|
|
|
def get_custom_countdown():
|
|
global countdown_cnt
|
|
if countdown_cnt == -1:
|
|
return '未设置'
|
|
li = config_center.read_conf('Date', 'countdown_date').split(',')
|
|
if countdown_cnt == -1 or countdown_cnt >= len(li):
|
|
return '未设置' # 获取自定义倒计时
|
|
else:
|
|
custom_countdown = li[countdown_cnt]
|
|
if custom_countdown == '':
|
|
return '未设置'
|
|
try:
|
|
custom_countdown = parser.parse(custom_countdown)
|
|
except Exception as e:
|
|
logger.error(f"解析日期时出错: {custom_countdown}, 错误: {e}")
|
|
return '解析失败'
|
|
if custom_countdown < datetime.now():
|
|
return '0 天'
|
|
else:
|
|
cd_text = custom_countdown - datetime.now()
|
|
return f'{cd_text.days + 1} 天'
|
|
# return (
|
|
# f"{cd_text.days} 天 {cd_text.seconds // 3600} 小时 {cd_text.seconds // 60 % 60} 分"
|
|
# )
|
|
|
|
|
|
def get_week_type():
|
|
if (temp_schedule := config_center.read_conf('Temp', 'set_schedule')) not in ('', None): # 获取单双周
|
|
return int(temp_schedule)
|
|
start_date_str = config_center.read_conf('Date', 'start_date')
|
|
if start_date_str not in ('', None):
|
|
try:
|
|
start_date = parser.parse(start_date_str)
|
|
except (ValueError, TypeError):
|
|
logger.error(f"解析日期时出错: {start_date_str}")
|
|
return 0 # 解析失败默认单周
|
|
today = datetime.now()
|
|
week_num = (today - start_date).days // 7 + 1
|
|
if week_num % 2 == 0:
|
|
return 1 # 双周
|
|
else:
|
|
return 0 # 单周
|
|
else:
|
|
return 0 # 默认单周
|
|
|
|
|
|
def get_is_widget_in(widget='example.ui'):
|
|
widgets_list = list_.get_widget_config()
|
|
if widget in widgets_list:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def save_widget_conf_to_json(new_data):
|
|
# 初始化 data_dict 为一个空字典
|
|
data_dict = {}
|
|
if os.path.exists(f'{base_directory}/config/widget.json'):
|
|
try:
|
|
with open(f'{base_directory}/config/widget.json', 'r', encoding='utf-8') as file:
|
|
data_dict = json.load(file)
|
|
except Exception as e:
|
|
print(f"读取现有数据时出错: {e}")
|
|
return e
|
|
data_dict.update(new_data)
|
|
try:
|
|
with open(f'{base_directory}/config/widget.json', 'w', encoding='utf-8') as file:
|
|
json.dump(data_dict, file, ensure_ascii=False, indent=4)
|
|
return True
|
|
except Exception as e:
|
|
print(f"保存数据时出错: {e}")
|
|
return e
|
|
|
|
|
|
def load_plugins(): # 加载插件配置文件
|
|
plugin_dict = {}
|
|
for folder in Path(PLUGINS_DIR).iterdir():
|
|
if folder.is_dir() and (folder / 'plugin.json').exists():
|
|
try:
|
|
with open(f'{base_directory}/plugins/{folder.name}/plugin.json', 'r', encoding='utf-8') as file:
|
|
data = json.load(file)
|
|
except Exception as e:
|
|
logger.error(f"加载插件配置文件数据时出错，将跳过: {e}") # 跳过奇怪的文件夹
|
|
plugin_dict[str(folder.name)] = {}
|
|
plugin_dict[str(folder.name)]['name'] = data['name'] # 名称
|
|
plugin_dict[str(folder.name)]['version'] = data['version'] # 插件版本号
|
|
plugin_dict[str(folder.name)]['author'] = data['author'] # 作者
|
|
plugin_dict[str(folder.name)]['description'] = data['description'] # 描述
|
|
plugin_dict[str(folder.name)]['plugin_ver'] = data['plugin_ver'] # 插件架构版本
|
|
plugin_dict[str(folder.name)]['settings'] = data['settings'] # 设置
|
|
return plugin_dict
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print('AL_1S')
|
|
print(get_week_type())
|
|
print(load_plugins())
|
|
# save_data_to_json(test_data_dict, 'schedule-1.json')
|
|
# loaded_data = load_from_json('schedule-1.json')
|
|
# print(loaded_data)
|
|
# schedule = loaded_data.get('schedule')
|
|
|
|
# print(schedule['0'])
|
|
# add_shortcut_to_startmenu('Settings.exe', 'img/favicon.ico')
|