ENH : move draw_all to Gcf class method

tacaswell · tacaswell · commit 28dcf3c15361 · 2015-05-25T00:49:40.000-04:00
diff --git a/lib/matplotlib/_pylab_helpers.py b/lib/matplotlib/_pylab_helpers.py
@@ -139,5 +139,14 @@ def set_active(cls, manager):
         cls._activeQue.append(manager)
         cls.figs[manager.num] = manager
 
+    @classmethod
+    def draw_all(cls):
+        """
+        Redraw all figures registered with the pyplot
+        state machine.
+        """
+        for f_mgr in cls.get_all_fig_managers():
+            # TODO add logic to check if figure is dirty
+            f_mgr.canvas.draw()
 
 atexit.register(Gcf.destroy_all)
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -139,15 +139,7 @@ def displayhook(*args):
 
         sys.displayhook = displayhook
 
-
-def draw_all():
-    """
-    Redraw all figures registered with the pyplot
-    state machine.
-    """
-    for f_mgr in _pylab_helpers.Gcf.get_all_fig_managers():
-        # TODO add logic to check if figure is dirty
-        f_mgr.canvas.draw()
+draw_all = _pylab_helpers.Gcf.draw_all
 
 
 @docstring.copy_dedent(Artist.findobj)
