From 28611108d7c60d355dada9eb93d5beaa230d6b5f Mon Sep 17 00:00:00 2001
From: Maxime Beauchemin <maximebeauchemin@gmail.com>
Date: Thu, 31 May 2018 16:01:49 -0700
Subject: [PATCH] Refactor NULL handling into method, disable for DECK.gl vizes
 (#5106)

---
 superset/viz.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/superset/viz.py b/superset/viz.py
index a32456d77..dcd924ba8 100644
--- a/superset/viz.py
+++ b/superset/viz.py
@@ -145,6 +145,10 @@ class BaseViz(object):
         """
         pass
 
+    def handle_nulls(self, df):
+        fillna = self.get_fillna_for_columns(df.columns)
+        df = df.fillna(fillna)
+
     def get_fillna_for_col(self, col):
         """Returns the value for use as filler for a specific Column.type"""
         if col:
@@ -208,8 +212,7 @@ class BaseViz(object):
             self.df_metrics_to_num(df, query_obj.get('metrics') or [])
 
             df.replace([np.inf, -np.inf], np.nan)
-            fillna = self.get_fillna_for_columns(df.columns)
-            df = df.fillna(fillna)
+            self.handle_nulls(df)
         return df
 
     @staticmethod
@@ -2041,6 +2044,9 @@ class BaseDeckGLViz(BaseViz):
     credits = '<a href="https://uber.github.io/deck.gl/">deck.gl</a>'
     spatial_control_keys = []
 
+    def handle_nulls(self, df):
+        pass
+
     def get_metrics(self):
         self.metric = self.form_data.get('size')
         return [self.metric] if self.metric else []