1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
import io
import csv
from typing import Literal, TypedDict, Any, Optional
def csv2dict(csvdata: str) -> dict:
with io.StringIO(csvdata) as f:
reader = csv.DictReader(f)
result = dict(header=reader.fieldnames, rows=[row for row in reader], raw=csvdata)
return result
def dict2csv(csvdict: dict) -> str:
# No longer needed since raw csv data are added to the table data
# dictionary in csv2dict
with io.StringIO() as f:
writer = csv.DictWriter(f, fieldnames=csvdict["header"])
writer.writerows(csvdict["rows"])
result = f.getvalue()
return result
class CustomWebviewPlot(TypedDict):
fig_type: Literal["plotly", "matplotlib", "table", "error"] = "plotly"
plotdata: Any
exportdata: Optional[str] = None
def process_custom_plot(plot_item: CustomWebviewPlot) -> CustomWebviewPlot:
figtype = plot_item["fig_type"]
plot_data = plot_item["plotdata"]
exportdata = plot_item.get("exportdata", None)
if figtype == "plotly":
figdict = plot_data.to_dict()
elif figtype == "matplotlib":
import mpld3
figdict = mpld3.fig_to_dict(plot_data)
elif figtype == "table":
figdict = csv2dict(plot_data)
if exportdata is None:
exportdata = figdict["raw"]
elif figtype == "error":
figdict = plot_data
else:
figdict = dict(error=f"unrecognized plot type {figtype}")
del plot_data # is this necessary? Does plot_item['plot_data'] also need to be deleted?
return CustomWebviewPlot(fig_type=figtype, plotdata=figdict, exportdata=exportdata)
|
