Function make_baselines_async

test/image/make_baseline.py:85–153

()

async def make_baselines_async():
    kopts = dict(plotlyjs=plotlyjs)
    if mathjax is not None:
        kopts["mathjax"] = mathjax

    async with kaleido.Kaleido(n=1, **kopts) as k:
        for name in allNames:
            outName = name
            if mathjax_version == 3:
                outName = "mathjax3___" + name

            print(outName)

            created = False

            MAX_RETRY = 2  # 1 means retry once
            for attempt in range(0, MAX_RETRY + 1):
                with open(os.path.join(dirIn, name + ".json"), "r") as _in:
                    fig = json.load(_in)

                    width = 700
                    height = 500
                    if "layout" in fig:
                        layout = fig["layout"]
                        if "autosize" not in layout or not layout["autosize"]:
                            if "width" in layout:
                                width = layout["width"]
                            if "height" in layout:
                                height = layout["height"]

                    if (
                        "b64" in sys.argv
                        or "b64=" in sys.argv
                        or "b64-json" in sys.argv
                    ):
                        newFig = dict()
                        arraysToB64(fig, newFig)
                        fig = newFig
                        if "b64-json" in sys.argv and attempt == 0:
                            print(json.dumps(fig, indent=2))

                    try:
                        bytes = await k.calc_fig(
                            fig,
                            opts=dict(
                                format="png",
                                width=width,
                                height=height,
                            ),
                            topojson=topojson,
                        )
                        filename = os.path.join(dirOut, outName + ".png")
                        with open(filename, "wb") as f:
                            f.write(bytes)
                        created = True
                    except Exception as e:
                        print(e)
                        if attempt < MAX_RETRY: