Merge pull request #12 from mistercrunch/reload_Fix
health check + reload dags
This commit is contained in:
Коммит
74018b52bf
|
@ -57,6 +57,11 @@ class GraphForm(Form):
|
||||||
def index():
|
def index():
|
||||||
return redirect(url_for('admin.index'))
|
return redirect(url_for('admin.index'))
|
||||||
|
|
||||||
|
@app.route('/health')
|
||||||
|
def health():
|
||||||
|
""" We can add an array of tests here to check the server's health """
|
||||||
|
content = Markup(markdown.markdown("The server is healthy!"))
|
||||||
|
return content;
|
||||||
|
|
||||||
class HomeView(AdminIndexView):
|
class HomeView(AdminIndexView):
|
||||||
"""
|
"""
|
||||||
|
@ -504,7 +509,7 @@ class DagModelView(ModelViewOnly):
|
||||||
'filepath': filepath_formatter,
|
'filepath': filepath_formatter,
|
||||||
}
|
}
|
||||||
column_list = ('dag_id', 'task_count', 'filepath')
|
column_list = ('dag_id', 'task_count', 'filepath')
|
||||||
mv = DagModelView(models.DAG, session, name="DAGs")
|
mv = DagModelView(models.DAG, session, name="DAGs", endpoint="dags")
|
||||||
admin.add_view(mv)
|
admin.add_view(mv)
|
||||||
|
|
||||||
|
|
||||||
|
@ -571,6 +576,14 @@ class LogModelView(ModelViewOnly):
|
||||||
mv = LogModelView(models.Log, session, name="Logs", category="Admin")
|
mv = LogModelView(models.Log, session, name="Logs", category="Admin")
|
||||||
admin.add_view(mv)
|
admin.add_view(mv)
|
||||||
|
|
||||||
|
class ReloadTaskView(BaseView):
|
||||||
|
@expose('/')
|
||||||
|
def index(self):
|
||||||
|
logging.info("Reloading the dags")
|
||||||
|
bag = models.DagBag();
|
||||||
|
return redirect(url_for('dags.index_view'))
|
||||||
|
admin.add_view(ReloadTaskView(name='Reload DAGs', category="Admin"))
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
logging.info("Starting the web server.")
|
logging.info("Starting the web server.")
|
||||||
app.run(debug=True)
|
app.run(debug=True)
|
||||||
|
|
Загрузка…
Ссылка в новой задаче