From 85a6da19ee598605f63b557c69620052d3e04ad0 Mon Sep 17 00:00:00 2001 From: "Hugh A. Miles II" Date: Tue, 14 Aug 2018 11:40:14 -0700 Subject: [PATCH] Fix annotation_json endpoint (#5621) * fix annotation_json endpoint * add time back * set db to None * change if condition * remove prop * add guard for to_dttm * linting --- superset/connectors/sqla/models.py | 6 ++++-- superset/viz.py | 3 ++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/superset/connectors/sqla/models.py b/superset/connectors/sqla/models.py index 61483cc349..648bff44e8 100644 --- a/superset/connectors/sqla/models.py +++ b/superset/connectors/sqla/models.py @@ -47,8 +47,10 @@ class AnnotationDatasource(BaseDatasource): error_message = None qry = db.session.query(Annotation) qry = qry.filter(Annotation.layer_id == query_obj['filter'][0]['val']) - qry = qry.filter(Annotation.start_dttm >= query_obj['from_dttm']) - qry = qry.filter(Annotation.end_dttm <= query_obj['to_dttm']) + if query_obj['from_dttm']: + qry = qry.filter(Annotation.start_dttm >= query_obj['from_dttm']) + if query_obj['to_dttm']: + qry = qry.filter(Annotation.end_dttm <= query_obj['to_dttm']) status = QueryStatus.SUCCESS try: df = pd.read_sql_query(qry.statement, db.engine) diff --git a/superset/viz.py b/superset/viz.py index c295e5a88a..e50ee1b77a 100644 --- a/superset/viz.py +++ b/superset/viz.py @@ -379,7 +379,8 @@ class BaseViz(object): if query_obj and not is_loaded: try: df = self.get_df(query_obj) - if hasattr(self.datasource.database, 'db_engine_spec'): + if hasattr(self.datasource, 'database') and \ + hasattr(self.datasource.database, 'db_engine_spec'): db_engine_spec = self.datasource.database.db_engine_spec df = db_engine_spec.adjust_df_column_names(df, self.form_data) if self.status != utils.QueryStatus.FAILED: