This commit is contained in:
Maxime Beauchemin 2014-11-07 08:37:30 -08:00
Родитель 9988e4d68c
Коммит fce8ecd38e
3 изменённых файлов: 42 добавлений и 10 удалений

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

@ -144,20 +144,18 @@ class Airflow(BaseView):
dag = dagbag.dags[dag_id]
task = dag.get_task(task_id)
s = ''
sql_attrs = ['hql', 'sql']
attributes = []
for attr_name in dir(task):
if not attr_name.startswith('_'):
attr = getattr(task, attr_name)
if type(attr) != type(self.task) and attr_name not in sql_attrs:
s += attr_name + ': ' + str(attr) + '\n'
s = s.replace('<', '&lt')
s = s.replace('>', '&gt')
html_code = "<h3>Attributes:</h3><pre><code>" + s + "</code></pre>"
attributes.append((attr_name, str(attr)))
title = "Task Details for {task_id}".format(**locals())
# Color coding the special attributes that are code
special_attrs = {}
for attr_name in sql_attrs:
if hasattr(task, attr_name):
source = getattr(task, attr_name)
@ -165,13 +163,17 @@ class Airflow(BaseView):
f = open(dag.folder + '/' + source, 'r')
source = f.read()
f.close()
html_code = "<h3>"+attr_name+"</h3>" + highlight(
source,
special_attrs[attr_name] = highlight(
getattr(task, attr_name),
SqlLexer(),
HtmlFormatter(noclasses=True)) + html_code
HtmlFormatter(noclasses=True)
)
return self.render(
'admin/code.html', html_code=html_code, dag=dag, title=title)
'admin/task.html',
attributes=attributes,
special_attrs=special_attrs,
dag=dag, title=title)
@expose('/tree')
def tree(self):

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

@ -46,3 +46,6 @@ textarea {
input, select {
margin: 0px;
}
.code {
font-family: monospace;
}

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

@ -0,0 +1,27 @@
{% extends "admin/dag.html" %}
{% block title %}Airflow - DAGs{% endblock %}
{% block body %}
{{ super() }}
<h2>{{ title }}</h2>
<div class="panel">
{% for attr, value in special_attrs.items() %}
<h4>Attribute: {{ attr }}</h4>
{{ value|safe }}
{% endfor %}
<h4>Attributes</h4>
<table class="table table-striped table-bordered">
<tr>
<th>Attribute</th>
<th>Value</th>
</tr>
{% for attr, value in attributes %}
<tr>
<td>{{ attr }}</td>
<td class='code'>{{ value }}</td>
</tr>
{% endfor %}
</table>
{{ html_code|safe }}
</div>
{% endblock %}