virtual-data-analyst / functions /chart_functions.py
nolanzandi's picture
Add link to edit/share graphs
4084d71 verified
raw
history blame
944 Bytes
from typing import List
from quickchart import QuickChart
def chart_generation_func(queries: List[str], session_hash):
print("CHART GENERATION")
query_dict = queries[0]
print(query_dict)
qc = QuickChart()
qc.width = 1000
qc.height = 500
# Config can be set as a string or as a nested dict
qc.config = query_dict
url_id = qc.get_short_url().rsplit('/', 1)[-1]
url_base = qc.get_url_base()
# You can get the chart URL...
interactive_url = url_base + '/chart-maker/view/' + url_id
edit_url = url_base + '/chart-maker/edit/' + url_id
iframe = '<div style=overflow:auto;><iframe\n scrolling="yes"\n width="1000px"\n height="500px"\n src="' + interactive_url + '"\n frameborder="0"\n allowfullscreen\n></iframe>\n <p>Edit, share, and download this graph <a target="_blank" href="' + edit_url + '">here</a></p></div>'
return {"reply": iframe}