Difficult-Rocket/Difficult_Rocket/utils/tools.py

#  -------------------------------
#  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 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

from lib_not_dr import loggers

# logger
tools_logger = loggers.config.get_logger("tools")
"""
file config
"""

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("./config/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))


# 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