Saving static images about dataplotly HOT 2 OPEN

import os
import urllib
from base64 import b64decode

import plotly
import plotly.graph_objs as go
from PyQt5.QtCore import QMimeDatabase, pyqtProperty
from PyQt5.QtWebKitWidgets import QWebView

vl = iface.activeLayer()
f1 = [i["PH"] for i in vl.getFeatures()]
f2 = [i["EC"] for i in vl.getFeatures()]
data = [go.Scatter(x=f1, y=f2)]
fig = go.Figure(data=data)

js_func_export = (
    # js function to export image at the webpage size
    # and store graph div id in the gdid variable
    "<script>"
    "function export_img(gd, name, w=null, h=null) {"
    "Plotly.downloadImage(gd, {format: pyObj.fileformat, width: w, height: h, filename: name});};"
    "</script>"
)
# custom buttons in the modeBar to export plot
export_button = (
    "{"
    'name: "Export image",'
    'icon: Plotly.Icons["camera-retro"],'
    "click: function(gd) {export_img(gd)}"
    "}"
)
# create the html DIV
html_plot = js_func_export
html_plot += plotly.offline.plot(
    fig,
    show_link=False,
    validate=True,
    output_type="div",
    image_filename="plot_image",
    image_width=800,
    image_height=600,
    include_mathjax=False,
    config={
        "displaylogo": False,
        "modeBarButtonsToAdd": ["exportButton"],
        "modeBarButtonsToRemove": ["toImage"],
    },
)

html_plot = html_plot.replace('"exportButton"', export_button)


def download_picture(webview, request):
    # avoid "Frame load interrupted by policy change" error
    webview.stop()
    # format default image file name
    image_filename = "my_dataplotly_plot"
    # save image
    request_content = request.url().toString()
    data = request_content.split(",")
    # verify mime type
    mime_type = QMimeDatabase().mimeTypeForFile(request_content).name()
    if mime_type == "application/octet-stream":
        # unquote the mime data
        image = urllib.parse.unquote(data[1])
        svg = image[:4] == "<svg"
        file_format = ["png", "svg"][svg]
        path = QFileDialog.getSaveFileName(
            webview,
            "Save as...",
            f"{os.path.expanduser('~')}/{image_filename}.{file_format}",
            f"Pictures (*.{file_format}",
        )
        # if a save image path is provided
        if path[0] != "":
            if svg:
                # write svg file
                with open(path[0], "w") as file:
                    file.write(image)
            else:
                # decode base 64 png to get the binary data
                png = b64decode(image)
                # write png file
                with open(path[0], "wb") as file:
                    file.write(png)


class PlotImageFormat(QObject):
    def __init__(self, format: str):
        super(PlotImageFormat, self).__init__()
        self._format = format

    def _fileformat(self):
        return self._format

    fileformat = pyqtProperty(str, fget=_fileformat)


wv = QWebView()
wv.setHtml(html_plot)
wv.page().downloadRequested.connect(lambda r: download_picture(wv, r))

pif = PlotImageFormat("svg")
wv.page().mainFrame().addToJavaScriptWindowObject("pyObj", pif)

wv.show()