Fix crash when count_value_null is None

wlanslovenija · Jul 22, 2017 · a4ca3d5 · a4ca3d5
1 parent cce29e1
commit a4ca3d5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/datastream/backends/influxdb.py b/datastream/backends/influxdb.py
@@ -579,7 +579,9 @@ def count(self):
         if isinstance(results, dict):
             result_keys = results.keys()
             if 'count_value' in result_keys:
-                return results.get('count_value', 0) + results.get('count_value_null', 0)
+                count_value = results.get('count_value', 0) or 0
+                count_value_null = results.get('count_value_null', 0) or 0
+                return count_value + count_value_null
             else:
                 return max([results[key] for key in results.keys() if key.startswith('count')])
         elif isinstance(results, tuple):