This commit is contained in:
Maxime Beauchemin 2014-12-04 07:16:37 +00:00
Родитель fd01a768e6
Коммит 4fd239f601
2 изменённых файлов: 10 добавлений и 5 удалений

Просмотреть файл

@ -127,6 +127,7 @@ class Airflow(BaseView):
session.close()
return self.render(
'airflow/query.html', form=form,
title="Query",
results=results, has_data=has_data)
@expose('/chart')
@ -159,15 +160,20 @@ class Airflow(BaseView):
has_data = False
table = None
height = "{0}px".format(chart.height)
failed = False
try:
df = hook.get_pandas_df(sql)
has_data = len(df)
except Exception as e:
flash(str(e), 'error')
has_data = False
failed = True
show_chart = True
if not has_data:
if failed:
show_sql = True
show_chart = False
elif not has_data:
show_sql = True
show_chart = False
flash('No data was returned', 'error')
@ -178,8 +184,6 @@ class Airflow(BaseView):
"SQL needs to return at least 3 columns (series, x, y)",
'error')
else:
# Preparing the data in a format that chartkick likes
for i, (series, x, y) in df.iterrows():
series = str(series)
@ -194,7 +198,8 @@ class Airflow(BaseView):
'name': series,
'data': sorted(all_data[series], key=lambda r: r[0])
}
for series in sorted(all_data)
for series in sorted(
all_data, key=lambda s: all_data[s][0][1], reverse=True)
]
if chart.show_datatable:

Просмотреть файл

@ -1,6 +1,6 @@
{% extends "airflow/master.html" %}
{% block title %}Airflow - DAGs{% endblock %}
{% block title %}{{ title }}{% endblock %}
{% block head_css %}
{{ super() }}