46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
|
from app import db
|
||
|
|
from io import BytesIO
|
||
|
|
from pandas import DataFrame, read_sql
|
||
|
|
from base64 import b64encode
|
||
|
|
|
||
|
|
|
||
|
|
def create_dataframe(query) -> DataFrame:
|
||
|
|
df = read_sql(sql=query, con=db.engine)
|
||
|
|
return df
|
||
|
|
|
||
|
|
|
||
|
|
def render_table(df) -> str:
|
||
|
|
df.fillna(value=0, inplace=True)
|
||
|
|
table = df.to_html(classes=["table-striped", "table-hover"])
|
||
|
|
return table
|
||
|
|
|
||
|
|
|
||
|
|
def render_plot(df):
|
||
|
|
df.fillna(value=0, inplace=True)
|
||
|
|
plot = df.plot("year", ["surface", "length", "elevation"], kind="bar")
|
||
|
|
plot_figure = plot.get_figure()
|
||
|
|
figure = BytesIO()
|
||
|
|
plot_figure.savefig(figure)
|
||
|
|
figure.seek(0)
|
||
|
|
return figure
|
||
|
|
|
||
|
|
|
||
|
|
def encode_plot(plot):
|
||
|
|
buffer = b"".join(plot)
|
||
|
|
buf = b64encode(buffer)
|
||
|
|
encoded_plot = buf.decode("utf-8")
|
||
|
|
return encoded_plot
|
||
|
|
|
||
|
|
|
||
|
|
def create_table(query) -> str:
|
||
|
|
df = create_dataframe(query)
|
||
|
|
html_table = render_table(df)
|
||
|
|
return html_table
|
||
|
|
|
||
|
|
|
||
|
|
def create_plot(query):
|
||
|
|
df = create_dataframe(query)
|
||
|
|
plot = render_plot(df)
|
||
|
|
encoded_plot = encode_plot(plot)
|
||
|
|
return encoded_plot
|