220 lines
7.4 KiB
Python
220 lines
7.4 KiB
Python
# -------------------------------
|
||
# Difficult Rocket
|
||
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
|
||
# All rights reserved
|
||
# -------------------------------
|
||
|
||
"""
|
||
writen by shenjackyuanjie
|
||
mail: 3695888@qq.com
|
||
github: @shenjackyuanjie
|
||
gitee: @shenjackyuanjie
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import time
|
||
import math
|
||
import json
|
||
import rtoml
|
||
import logging
|
||
import configparser
|
||
|
||
from pathlib import Path
|
||
from typing import Union, Optional
|
||
from xml.etree import ElementTree
|
||
from defusedxml.ElementTree import parse
|
||
|
||
from Difficult_Rocket.exception.unsupport import NoMoreJson5
|
||
|
||
# logger
|
||
tools_logger = logging.getLogger('tools')
|
||
"""
|
||
file configs
|
||
"""
|
||
|
||
file_error = {FileNotFoundError: 'no {filetype} file was founded!:\n file name: {filename}\n file_type: {filetype}\n stack: {stack}',
|
||
KeyError: 'no stack in {filetype} file {filename} was found! \n file type: {} \n file name: {} \n stack: {stack}',
|
||
Exception: 'get some {error_type} when read {filetype} file {filename}! \n file type: {} \n file name: {} \n stack: {stack}'}
|
||
|
||
|
||
def load_file(file_name: Union[str, Path],
|
||
stack: Optional[Union[str, list, dict]] = None,
|
||
raise_error: Optional[bool] = True,
|
||
encoding: Optional[str] = 'utf-8') -> Union[dict, ElementTree.ElementTree]:
|
||
if isinstance(file_name, Path):
|
||
file_name = str(file_name)
|
||
f_type = file_name[file_name.rfind('.') + 1:] # 从最后一个.到末尾 (截取文件格式)
|
||
get_file = NotImplementedError('解析失败，请检查文件类型/文件内容/文件是否存在！')
|
||
try:
|
||
if f_type == 'xml':
|
||
xml_load: ElementTree.ElementTree = parse(file_name)
|
||
if stack is not None:
|
||
get_file = xml_load.findall(stack)
|
||
elif (f_type == 'config') or (f_type == 'conf') or (f_type == 'ini'):
|
||
get_file = configparser.ConfigParser()
|
||
get_file.read(file_name)
|
||
if stack:
|
||
get_file = get_file[stack]
|
||
elif f_type == 'toml':
|
||
with open(file_name, mode='r', encoding=encoding) as file:
|
||
get_file = rtoml.load(file)
|
||
if stack is not None:
|
||
get_file = get_file[stack]
|
||
elif f_type == 'json':
|
||
with open(file_name, mode='r', encoding=encoding) as file:
|
||
get_file = json.load(file)
|
||
if stack is not None:
|
||
get_file = get_file[stack]
|
||
elif f_type == 'json5':
|
||
raise NoMoreJson5("我说什么也不用json5了！喵的")
|
||
except Exception as exp:
|
||
error_type = type(exp)
|
||
if error_type in file_error:
|
||
tools_logger.error(file_error[error_type].format(filetype=f_type, filename=file_name, stack=stack))
|
||
else:
|
||
tools_logger.error(file_error[Exception].format(error_type=error_type, filetype=f_type, filename=file_name, stack=stack))
|
||
if raise_error:
|
||
raise exp from None
|
||
return get_file
|
||
|
||
|
||
def save_dict_file(file_name: str,
|
||
data: dict,
|
||
encoding: str = 'utf-8') -> bool:
|
||
f_type = file_name[file_name.rfind('.') + 1:] # 从最后一个.到末尾 (截取文件格式)
|
||
try:
|
||
if (f_type == 'config') or (f_type == 'conf') or (f_type == 'ini'):
|
||
return False
|
||
elif f_type == 'toml':
|
||
with open(file_name, mode='w', encoding=encoding) as file:
|
||
rtoml.dump(data, file)
|
||
elif f_type == 'json':
|
||
with open(file_name, mode='w', encoding=encoding) as file:
|
||
json.dump(data, file)
|
||
elif f_type == 'json5':
|
||
raise NoMoreJson5("我说什么也不用json5了！喵的")
|
||
except Exception as exp:
|
||
raise exp
|
||
|
||
|
||
# main config
|
||
main_config_file = load_file('./configs/main.toml')
|
||
|
||
|
||
def get_At(name, in_xml, need_type=str):
|
||
"""
|
||
get Attribute from a XML tree
|
||
will raise TypeError if input is not str or list
|
||
XML json5 no! toml yes!
|
||
"""
|
||
name_type = type(name)
|
||
if name_type == list:
|
||
At_list = []
|
||
for need_name in name:
|
||
if in_xml.hasAttribute(need_name):
|
||
get = in_xml.getAttribute(need_name)
|
||
At_list.append(need_type(get))
|
||
else:
|
||
continue
|
||
return At_list
|
||
elif name_type == str:
|
||
if in_xml.hasAttribute(name):
|
||
attr = in_xml.getAttribute(name)
|
||
else:
|
||
return None
|
||
else:
|
||
raise TypeError('only str and list type is ok but you give me a' + name_type + 'type')
|
||
return need_type(attr)
|
||
|
||
|
||
def default_name_handler(name_: str) -> str:
|
||
"""
|
||
won't change the string
|
||
just return one
|
||
"""
|
||
name = name_
|
||
name = name.replace('{time.time}', str(time.time()))
|
||
name = name.replace('{dir}', str(os.getcwd()))
|
||
name = name.replace('{py_v}', str(sys.version.split(' ')[0]))
|
||
name = name.replace('{version}', str(main_config_file['runtime']['version']))
|
||
name = name.replace('{write_v}', str(main_config_file['runtime']['write_py_v']))
|
||
return name
|
||
|
||
|
||
def name_handler(name: str, formats: dict = None) -> str:
|
||
if formats is None:
|
||
return default_name_handler(name)
|
||
name = default_name_handler(name)
|
||
for need_replace in formats:
|
||
replace = formats[need_replace]
|
||
if need_replace == '{date}':
|
||
if '{date}' in formats:
|
||
replace = time.strftime(formats['{date}'], time.gmtime(time.time()))
|
||
else:
|
||
replace = time.strftime(main_config_file['runtime']['date_fmt'], time.gmtime(time.time()))
|
||
name = name.replace(need_replace, replace)
|
||
return name
|
||
|
||
|
||
"""
|
||
some tools
|
||
"""
|
||
|
||
yes = ['true', '1', 1, 1.0, True]
|
||
no = ['false', '0', 0, 0.0, False, None]
|
||
|
||
|
||
def format_bool(thing) -> bool:
|
||
"""
|
||
:param thing 啥都行，只要是能传进来的都可以
|
||
|
||
如果看起来像"True" 比如 'true','1',1
|
||
就返回 True
|
||
如果看起来像"False" 比如 'false','0',0
|
||
就返回 False
|
||
都不像就 raise TypeError
|
||
感谢来自MCDReforged的各位同志《你能在MCDR群里聊除了MCDR的任何东西》
|
||
"""
|
||
if (thing in yes) or (isinstance(thing, str) and thing.lower() in yes):
|
||
return True
|
||
elif (thing in no) or (isinstance(thing, str) and thing.lower() in no):
|
||
return False
|
||
else:
|
||
raise TypeError("Need a 'like bool' not a {}".format(thing))
|
||
|
||
|
||
level_ = ['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL',
|
||
logging.DEBUG, logging.INFO, logging.WARNING, logging.ERROR, logging.CRITICAL]
|
||
|
||
|
||
def log_level(level):
|
||
if level in level_:
|
||
if (level == 'DEBUG') or (level == logging.DEBUG):
|
||
return logging.DEBUG
|
||
if (level == 'INFO') or (level == logging.INFO):
|
||
return logging.INFO
|
||
if (level == 'WARNING') or (level == logging.WARNING):
|
||
return logging.WARNING
|
||
if (level == 'ERROR') or (level == logging.ERROR):
|
||
return logging.ERROR
|
||
if (level == 'CRITICAL') or (level == logging.CRITICAL):
|
||
return logging.CRITICAL
|
||
else:
|
||
raise ValueError('Need a like logging.level thing not anything else')
|
||
|
||
|
||
# linear_algebra
|
||
|
||
def C_R_P(position, degrees): # stand for calculation
|
||
"""
|
||
very thanks for lenny from pyglet developer
|
||
https://github.com/LennyPhoenix
|
||
this part of code is write by him
|
||
"""
|
||
radians = degrees * (math.pi / 180)
|
||
cos = math.cos(radians)
|
||
sin = math.sin(radians)
|
||
rotated_pos = (position[0] * cos - position[1] * sin, position[0] * sin + position[1] * cos)
|
||
return rotated_pos
|