imaging.py 文件源码

def make_heat_img(path_to, filename):
    try:
        info = name_deconstructor(filename=filename, t="s")
        info["broker"] = str(slugify(info["broker"])).replace("-", "_")

        file_name = join(path_to, info["filename"])
        file_name = ext_drop(filename=file_name)
        df = await df_multi_reader(filename=file_name)

        if len(df.index) > settings.MIN_TRADES:
            info["direction"] = 1
            longs = await convert_to_perc(data=df.LONG_PL, info=info)
            info["direction"] = 2
            shorts = await convert_to_perc(data=df.SHORT_PL, info=info)
            info["direction"] = 0
            long_short = await convert_to_perc(data=(df.LONG_PL + df.SHORT_PL), info=info)

            if not longs is None:
                info["direction"] = "longs"
                await save_heatmap(data=longs, info=info)
                await make_yearly_returns(returns=longs, info=info)
            if not shorts is None:
                info["direction"] = "shorts"
                await save_heatmap(data=longs, info=info)
                await make_yearly_returns(returns=longs, info=info)
            if not long_short is None:
                info["direction"] = "longs_shorts"
                await save_heatmap(data=longs, info=info)
                await make_yearly_returns(returns=longs, info=info)

    except Exception as err:
        print(colored.red("At make_heat_img {}".format(err)))