Add extra_json column to Query

superset/migrations/versions/0b1f1ab473c0_add_extra_column_to_query.py

@@ -0,0 +1,21 @@
+"""Add extra column to Query
+
+Revision ID: 0b1f1ab473c0
+Revises: 55e910a74826
+Create Date: 2018-11-05 08:42:56.181012
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+# revision identifiers, used by Alembic.
+revision = '0b1f1ab473c0'
+down_revision = '55e910a74826'
+
+
+def upgrade():
+    op.add_column('query', sa.Column('extra_json', sa.Text(), nullable=True))
+
+
+def downgrade():
+    op.drop_column('query', 'extra_json')

superset/models/helpers.py

@@ -305,3 +305,14 @@ def __init__(  # noqa
         self.duration = duration
         self.status = status
         self.error_message = error_message
+
+
+class ExtraJSONMixin:
+    extra_json = sa.Column(sa.Text)
+
+    @property
+    def extra(self):
+        return json.loads(self.extra_json)
+
+    def set_extra_json(self, d):
+        self.extra_json = json.dumps(d)

superset/models/sql_lab.py

@@ -1,6 +1,7 @@
 # pylint: disable=C,R,W
 """A collection of ORM sqlalchemy models for SQL Lab"""
 from datetime import datetime
+import json
 import re
 
 from flask import Markup
@@ -12,11 +13,11 @@
 from sqlalchemy.orm import backref, relationship
 
 from superset import security_manager
-from superset.models.helpers import AuditMixinNullable
+from superset.models.helpers import AuditMixinNullable, ExtraJSONMixin
 from superset.utils.core import QueryStatus, user_label
 
 
-class Query(Model):
+class Query(Model, ExtraJSONMixin):
     """ORM model for SQL query
 
     Now that SQL Lab support multi-statement execution, an entry in this