Move check for ImageMagick Windows path to bin_path().

anntzer · anntzer · commit bb7a182a1e78 · 2018-10-02T17:56:53.000+02:00
That's a perfectly reasonable place to put the registry check and avoids
introducing an additional private helper method.
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -704,7 +704,7 @@ class AVConvFileWriter(AVConvBase, FFMpegFileWriter):
     '''
 
 
-# Base class for animated GIFs with convert utility
+# Base class for animated GIFs with ImageMagick
 class ImageMagickBase(object):
     '''Mixin class for ImageMagick output.
 
@@ -724,48 +724,33 @@ def output_args(self):
         return [self.outfile]
 
     @classmethod
-    def _init_from_registry(cls):
-        if sys.platform != 'win32' or rcParams[cls.exec_key] != 'convert':
-            return
-        import winreg
-        for flag in (0, winreg.KEY_WOW64_32KEY, winreg.KEY_WOW64_64KEY):
-            try:
-                hkey = winreg.OpenKeyEx(winreg.HKEY_LOCAL_MACHINE,
-                                        r'Software\Imagemagick\Current',
-                                        0, winreg.KEY_QUERY_VALUE | flag)
-                binpath = winreg.QueryValueEx(hkey, 'BinPath')[0]
-                winreg.CloseKey(hkey)
-                break
-            except Exception:
-                binpath = ''
-        if binpath:
-            for exe in ('convert.exe', 'magick.exe'):
-                path = os.path.join(binpath, exe)
-                if os.path.exists(path):
-                    binpath = path
-                    break
-            else:
-                binpath = ''
-        rcParams[cls.exec_key] = rcParamsDefault[cls.exec_key] = binpath
-
-    @classmethod
-    def isAvailable(cls):
-        '''
-        Check to see if a ImageMagickWriter is actually available.
-
-        Done by first checking the windows registry (if applicable) and then
-        running the commandline tool.
-        '''
-        bin_path = cls.bin_path()
-        if bin_path == "convert":
-            cls._init_from_registry()
-        return super().isAvailable()
-
-
-# Note: the base classes need to be in that order to get
-# isAvailable() from ImageMagickBase called and not the
-# one from MovieWriter. The latter is then called by the
-# former.
+    def bin_path(cls):
+        binpath = super().bin_path()
+        if sys.platform == 'win32' and binpath == 'convert':
+            # Check the registry to avoid confusing ImageMagick's convert with
+            # Windows's builtin convert.exe.
+            import winreg
+            binpath = ''
+            for flag in (0, winreg.KEY_WOW64_32KEY, winreg.KEY_WOW64_64KEY):
+                try:
+                    with winreg.OpenKeyEx(
+                            winreg.HKEY_LOCAL_MACHINE,
+                            r'Software\Imagemagick\Current',
+                            0, winreg.KEY_QUERY_VALUE | flag) as hkey:
+                        parent = winreg.QueryValueEx(hkey, 'BinPath')[0]
+                except OSError:
+                    pass
+            if binpath:
+                for exe in ('convert.exe', 'magick.exe'):
+                    candidate = os.path.join(parent, exe)
+                    if os.path.exists(candidate):
+                        binpath = candidate
+                        break
+            rcParams[cls.exec_key] = rcParamsDefault[cls.exec_key] = binpath
+        return binpath
+
+
+# Combine ImageMagick options with pipe-based writing
 @writers.register('imagemagick')
 class ImageMagickWriter(ImageMagickBase, MovieWriter):
     '''Pipe-based animated gif.
@@ -782,10 +767,7 @@ def _args(self):
                 + self.output_args)
 
 
-# Note: the base classes need to be in that order to get
-# isAvailable() from ImageMagickBase called and not the
-# one from MovieWriter. The latter is then called by the
-# former.
+# Combine ImageMagick options with temp file-based writing
 @writers.register('imagemagick_file')
 class ImageMagickFileWriter(ImageMagickBase, FileMovieWriter):
     '''File-based animated gif writer.
