202 lines
8.3 KiB
Python
Executable File
202 lines
8.3 KiB
Python
Executable File
import json
|
||
import os
|
||
import sys
|
||
from pathlib import Path
|
||
from shutil import copy
|
||
|
||
from loguru import logger
|
||
import configparser as config
|
||
|
||
base_directory = os.path.dirname(os.path.abspath(__file__))
|
||
'''
|
||
if base_directory.endswith('MacOS'):
|
||
base_directory = os.path.join(os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir)), 'Resources')
|
||
'''
|
||
path = f'{base_directory}/config.ini'
|
||
|
||
|
||
class ConfigCenter:
|
||
"""
|
||
Config中心
|
||
"""
|
||
def __init__(self):
|
||
self.config = config.ConfigParser()
|
||
self.config.read(path, encoding='utf-8')
|
||
with open(f'{base_directory}/config/default_config.json', encoding="utf-8") as default:
|
||
self.default_data = json.load(default)
|
||
|
||
self.check_config()
|
||
self.schedule_name = self.read_conf('General', 'schedule')
|
||
self.old_schedule_name = self.schedule_name
|
||
|
||
def update_conf(self):
|
||
try:
|
||
self.config.read_file(open(path, 'r', encoding='utf-8'))
|
||
|
||
self.schedule_name = self.read_conf('General', 'schedule')
|
||
if self.schedule_name != self.old_schedule_name:
|
||
logger.info(f'已切换课程表: {self.schedule_name}')
|
||
schedule_center.update_schedule()
|
||
self.old_schedule_name = self.schedule_name
|
||
except Exception as e:
|
||
logger.error(f'更新配置文件时出错: {e}')
|
||
|
||
def read_conf(self, section='General', key=''):
|
||
if section in self.config and key in self.config[section]:
|
||
return self.config[section][key]
|
||
elif section in self.config and key == '':
|
||
return dict(self.config[section])
|
||
elif section in self.default_data and key in self.default_data[section]:
|
||
logger.info('配置文件出现问题，已尝试修复')
|
||
self.write_conf(section, key, self.default_data[section][key])
|
||
return self.default_data[section][key]
|
||
elif section in self.default_data and key == '':
|
||
logger.info('配置文件出现问题，已尝试修复')
|
||
self.write_conf(section, '', self.default_data[section])
|
||
return dict(self.default_data[section])
|
||
else:
|
||
return None
|
||
|
||
def write_conf(self, section, key, value):
|
||
if section not in self.config:
|
||
self.config.add_section(section)
|
||
|
||
self.config.set(section, key, str(value))
|
||
|
||
with open(path, 'w', encoding='utf-8') as configfile:
|
||
self.config.write(configfile)
|
||
|
||
def check_config(self):
|
||
if not os.path.exists(path): # 如果配置文件不存在，则copy默认配置文件
|
||
self.config.read_dict(self.default_data)
|
||
with open(path, 'w', encoding='utf-8') as configfile:
|
||
self.config.write(configfile)
|
||
if sys.platform != 'win32':
|
||
self.config.set('General', 'hide_method', '2')
|
||
with open(path, 'w', encoding='utf-8') as configfile:
|
||
self.config.write(configfile)
|
||
logger.info("配置文件不存在，已创建并写入默认配置。")
|
||
copy(f'{base_directory}/config/default.json', f'{base_directory}/config/schedule/新课表 - 1.json')
|
||
else:
|
||
with open(path, 'r', encoding='utf-8') as configfile:
|
||
self.config.read_file(configfile)
|
||
|
||
if self.config['Other']['version'] != self.default_data['Other']['version']: # 如果配置文件版本不同，则更新配置文件
|
||
logger.info(f"配置文件版本不同，将重新适配")
|
||
try:
|
||
for section, options in self.default_data.items():
|
||
if section not in self.config:
|
||
self.config[section] = options
|
||
else:
|
||
for key, value in options.items():
|
||
if key not in self.config[section]:
|
||
self.config[section][key] = str(value)
|
||
self.config.set('Other', 'version', self.default_data['Other']['version'])
|
||
with open(path, 'w', encoding='utf-8') as configfile:
|
||
self.config.write(configfile)
|
||
logger.info(f"配置文件已更新")
|
||
except Exception as e:
|
||
logger.error(f"配置文件更新失败: {e}")
|
||
|
||
if not os.path.exists(f"{base_directory}/config/schedule/{self.read_conf('General', 'schedule')}"):
|
||
# 如果config.ini课程表不存在，则创建
|
||
|
||
schedule_config = []
|
||
# 遍历目标目录下的所有文件
|
||
for file_name in os.listdir(f'{base_directory}/config/schedule'):
|
||
# 找json
|
||
if file_name.endswith('.json') and file_name != 'backup.json':
|
||
# 将文件路径添加到列表
|
||
schedule_config.append(file_name)
|
||
if not schedule_config:
|
||
copy(f'{base_directory}/config/default.json',
|
||
f'{base_directory}/config/schedule/{self.read_conf("General", "schedule")}')
|
||
logger.info(f"课程表不存在，已创建默认课程表")
|
||
else:
|
||
config_center.write_conf('General', 'schedule', schedule_config[0])
|
||
print(os.path.join(os.getcwd(), 'config', 'schedule'))
|
||
|
||
# 判断是否存在 Plugins 文件夹
|
||
plugins_dir = Path(base_directory) / 'plugins'
|
||
if not plugins_dir.exists():
|
||
plugins_dir.mkdir()
|
||
logger.info("Plugins 文件夹不存在，已创建。")
|
||
|
||
# 判断 Plugins 文件夹内是否存在 plugins_from_pp.json 文件
|
||
plugins_file = plugins_dir / 'plugins_from_pp.json'
|
||
if not plugins_file.exists():
|
||
with open(plugins_file, 'w', encoding='utf-8') as file:
|
||
# 使用 indent=4 来缩进，并确保数组元素在多行显示
|
||
json.dump({"plugins": []}, file, ensure_ascii=False, indent=4)
|
||
logger.info("plugins_from_pp.json 文件不存在，已创建。")
|
||
|
||
|
||
class ScheduleCenter:
|
||
"""
|
||
课程表中心
|
||
"""
|
||
def __init__(self):
|
||
self.schedule_data = None
|
||
self.update_schedule()
|
||
|
||
def update_schedule(self):
|
||
"""
|
||
更新课程表
|
||
"""
|
||
self.schedule_data = load_from_json(config_center.schedule_name)
|
||
|
||
def save_data(self, new_data, filename):
|
||
# 更新，添加或覆盖新的数据
|
||
self.schedule_data.update(new_data)
|
||
|
||
# 将更新后的数据保存回文件
|
||
try:
|
||
with open(f'{base_directory}/config/schedule/{filename}', 'w', encoding='utf-8') as file:
|
||
json.dump(self.schedule_data, file, ensure_ascii=False, indent=4)
|
||
return f"数据已成功保存到 config/schedule/{filename}"
|
||
except Exception as e:
|
||
logger.error(f"保存数据时出错: {e}")
|
||
|
||
|
||
def load_from_json(filename):
|
||
"""
|
||
从 JSON 文件中加载数据。
|
||
:param filename: 要加载的文件
|
||
:return: 返回从文件中加载的数据字典
|
||
"""
|
||
try:
|
||
with open(f'{base_directory}/config/schedule/{filename}', 'r', encoding='utf-8') as file:
|
||
data = json.load(file)
|
||
return data
|
||
except Exception as e:
|
||
logger.error(f"加载数据时出错: {e}")
|
||
return None
|
||
|
||
|
||
def save_data_to_json(new_data, filename):
|
||
# 初始化 data_dict 为一个空字典
|
||
data_dict = {}
|
||
|
||
# 如果文件存在，先读取文件中的现有数据
|
||
if os.path.exists(f'{base_directory}/config/schedule/{filename}'):
|
||
try:
|
||
with open(f'{base_directory}/config/schedule/{filename}', 'r', encoding='utf-8') as file:
|
||
data_dict = json.load(file)
|
||
except Exception as e:
|
||
logger.error(f"读取现有数据时出错: {e}")
|
||
|
||
# 更新 data_dict，添加或覆盖新的数据
|
||
data_dict.update(new_data)
|
||
|
||
# 将更新后的数据保存回文件
|
||
try:
|
||
with open(f'{base_directory}/config/schedule/{filename}', 'w', encoding='utf-8') as file:
|
||
json.dump(data_dict, file, ensure_ascii=False, indent=4)
|
||
return f"数据已成功保存到 config/schedule/{filename}"
|
||
except Exception as e:
|
||
logger.error(f"保存数据时出错: {e}")
|
||
|
||
|
||
config_center = ConfigCenter()
|
||
schedule_center = ScheduleCenter()
|