# ------------------------------- # Difficult Rocket # Copyright © 2021-2022 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 logging import configparser from xml.dom.minidom import parse from libs import toml from Difficult_Rocket.exception.unsupport import NoMoreJson5 # logger tools_logger = logging.getLogger('part-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}', 'Error': 'get some unknown error when read {filetype} file {filename}! \n file type: {} \n file name: {} \n stack: {stack}'} def load_file(file_name: str, stack=None): f_type = file_name[file_name.rfind('.') + 1:] # 从最后一个.到末尾 (截取文件格式) try: get_file = NotImplementedError('解析失败,请检查文件类型/文件内容/文件是否存在!') if f_type == 'xml': xml_load = parse(file_name) if stack is not None: get_file = xml_load.getElementsByTagName(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': get_file = toml.load(file_name) elif f_type == 'json5': raise NoMoreJson5("我说什么也不用json5了!喵的") except Exception as exp: error_type = type(exp).__name__ 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['Error'].format(filetype=f_type, filename=file_name, stack=stack)) raise return get_file # main config main_config_file = load_file('./configs/main.config') 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): At = 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(At) 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