import re
from tinydb import TinyDB, Query
from src.gr_func import _get_medicine_elixir_config,material_table


def get_need_material(medicine_select, medicine_level_select="ALL",material_max_num=16) ->list:
    material = Query()
    m = _get_medicine_elixir_config(medicine_select)
    func1_type = m["func1_type"]
    func1_power = m["func1_power"]
    func2_type = m["func2_type"]
    func2_power = m["func2_power"]
    if medicine_level_select == "ALL":
        a = material_table.search((material.main_func_t == func1_type) | (material.auxi_func_t == func1_type) | (
                    material.main_func_t == func2_type) | (material.auxi_func_t == func2_type))
    else:
        a = material_table.search((material.level == medicine_level_select) & (
                    (material.main_func_t == func1_type) | (material.auxi_func_t == func1_type) | (
                        material.main_func_t == func2_type) | (material.auxi_func_t == func2_type)))

    def get_num(material0):
        global material_second_f
        name = material0["name"]
        if material0["main_func_t"] == func1_type:
            material_second_f = (func2_type,False)
            num = func1_power / material0["main_func_p"]
        elif material0["auxi_func_t"] == func1_type:
            material_second_f = (func2_type,True)
            num = func1_power / material0["auxi_func_p"]
        elif material0["main_func_t"] == func2_type:
            material_second_f = (func1_type,False)
            num = func2_power / material0["main_func_p"]
        elif material0["auxi_func_t"] == func2_type:
            material_second_f = (func1_type,True)
            num = func2_power / material0["auxi_func_p"]
        num = int(num) + 1 if num > int(num) else int(num)
        return (name,num,material_second_f)
    rtn = list(map(get_num, a))
    rtn = list(filter(lambda x:x[1]<=material_max_num, rtn))

    def check_material(material0):
        if material0[1] > material_max_num:
            return False
        material_t = material.main_func_t if material0[2][1] else material.auxi_func_t
        a = material_table.search(material_t == material0[2][0])
        if a == []:
            return False
        return True

    rtn = list(filter(check_material, rtn))
    rtn = list(map(lambda x: (x[0],x[1]), rtn))
    return rtn

grade_str = "一二三四五六七八九"

def sort_yaocai(text,medicine_select,material_num):
    material_need_dict = {}
    if medicine_select != "无":
        material_need_list = get_need_material(medicine_select,material_max_num=material_num)
        for name,num in material_need_list:
            material_need_dict[name[:-4]] = num
        print(material_need_dict)
    regex = re.compile("名字：.+\n品级：.+\n.+\n.+\n拥有数量:\d+")
    yaocai_l = regex.findall(text)
    rtn = []
    for yaocai in yaocai_l:
        yaocai = yaocai.split("\n")
        name = yaocai[0][3:]
        num = int(yaocai[-1][5:])
        grade = grade_str.index(yaocai[1][3])+1

        flag = material_need_dict.get(name)
        if flag is not None:
            if num >= flag:
                flag = "+"
            else:
                num = f"{num}({flag})"
                flag = "-"
        rtn.append((name,grade,num,flag))
    return rtn