let get_diff_image return None and deal with it downstream

cknd · cknd · commit f38575e3ac01 · 2017-05-23T16:27:20.000+02:00
diff --git a/lib/matplotlib/backends/backend_webagg_core.py b/lib/matplotlib/backends/backend_webagg_core.py
@@ -236,8 +236,7 @@ def get_diff_image(self):
                 self._last_renderer, renderer)
             self._force_full = False
             self._png_is_old = False
-            self.buff = buff
-        return self.buff
+            return buff
 
     def get_renderer(self, cleared=None):
         # Mirrors super.get_renderer, but caches the old one
@@ -487,8 +486,9 @@ def handle_json(self, content):
     def refresh_all(self):
         if self.web_sockets:
             diff = self.canvas.get_diff_image()
-            for s in self.web_sockets:
-                s.send_binary(diff)
+            if diff is not None:
+                for s in self.web_sockets:
+                    s.send_binary(diff)
 
     @classmethod
     def get_javascript(cls, stream=None):
