diff --git a/PIL/EpsImagePlugin.py b/PIL/EpsImagePlugin.py index 08110abf365..e6842e91ec8 100644 --- a/PIL/EpsImagePlugin.py +++ b/PIL/EpsImagePlugin.py @@ -419,7 +419,6 @@ def close(self): Image.register_save(EpsImageFile.format, _save) -Image.register_extension(EpsImageFile.format, ".ps") -Image.register_extension(EpsImageFile.format, ".eps") +Image.register_extensions(EpsImageFile.format, [".ps", ".eps"]) Image.register_mime(EpsImageFile.format, "application/postscript") diff --git a/PIL/FitsStubImagePlugin.py b/PIL/FitsStubImagePlugin.py index e3a7eb4a6ce..be926cadb84 100644 --- a/PIL/FitsStubImagePlugin.py +++ b/PIL/FitsStubImagePlugin.py @@ -72,5 +72,4 @@ def _save(im, fp, filename): Image.register_open(FITSStubImageFile.format, FITSStubImageFile, _accept) Image.register_save(FITSStubImageFile.format, _save) -Image.register_extension(FITSStubImageFile.format, ".fit") -Image.register_extension(FITSStubImageFile.format, ".fits") +Image.register_extensions(FITSStubImageFile.format, [".fit", ".fits"]) diff --git a/PIL/FliImagePlugin.py b/PIL/FliImagePlugin.py index 52bebe80b14..733d2d8f193 100644 --- a/PIL/FliImagePlugin.py +++ b/PIL/FliImagePlugin.py @@ -162,5 +162,4 @@ def tell(self): Image.register_open(FliImageFile.format, FliImageFile, _accept) -Image.register_extension(FliImageFile.format, ".fli") -Image.register_extension(FliImageFile.format, ".flc") +Image.register_extensions(FliImageFile.format, [".fli", ".flc"]) diff --git a/PIL/FtexImagePlugin.py b/PIL/FtexImagePlugin.py index 0d08f4cc4da..9b9809062d2 100644 --- a/PIL/FtexImagePlugin.py +++ b/PIL/FtexImagePlugin.py @@ -91,5 +91,4 @@ def _validate(prefix): Image.register_open(FtexImageFile.format, FtexImageFile, _validate) -Image.register_extension(FtexImageFile.format, ".ftc") -Image.register_extension(FtexImageFile.format, ".ftu") +Image.register_extensions(FtexImageFile.format, [".ftc", ".ftu"]) diff --git a/PIL/Hdf5StubImagePlugin.py b/PIL/Hdf5StubImagePlugin.py index a5d6b1bc185..de4d5bb0c97 100644 --- a/PIL/Hdf5StubImagePlugin.py +++ b/PIL/Hdf5StubImagePlugin.py @@ -69,5 +69,4 @@ def _save(im, fp, filename): Image.register_open(HDF5StubImageFile.format, HDF5StubImageFile, _accept) Image.register_save(HDF5StubImageFile.format, _save) -Image.register_extension(HDF5StubImageFile.format, ".h5") -Image.register_extension(HDF5StubImageFile.format, ".hdf") +Image.register_extensions(HDF5StubImageFile.format, [".h5", ".hdf"]) diff --git a/PIL/Image.py b/PIL/Image.py index 2904e4e0743..1c7518df0de 100644 --- a/PIL/Image.py +++ b/PIL/Image.py @@ -2733,6 +2733,16 @@ def register_extension(id, extension): """ EXTENSION[extension.lower()] = id.upper() +def register_extensions(id, extensions): + """ + Registers image extensions. This function should not be + used in application code. + + :param id: An image format identifier. + :param extensions: A list of extensions used for this format. + """ + for extension in extensions: + register_extension(id, extension) def registered_extensions(): """ diff --git a/PIL/Jpeg2KImagePlugin.py b/PIL/Jpeg2KImagePlugin.py index 101f55f7673..c67cc14bfdc 100644 --- a/PIL/Jpeg2KImagePlugin.py +++ b/PIL/Jpeg2KImagePlugin.py @@ -269,12 +269,7 @@ def _save(im, fp, filename): Image.register_open(Jpeg2KImageFile.format, Jpeg2KImageFile, _accept) Image.register_save(Jpeg2KImageFile.format, _save) -Image.register_extension(Jpeg2KImageFile.format, '.jp2') -Image.register_extension(Jpeg2KImageFile.format, '.j2k') -Image.register_extension(Jpeg2KImageFile.format, '.jpc') -Image.register_extension(Jpeg2KImageFile.format, '.jpf') -Image.register_extension(Jpeg2KImageFile.format, '.jpx') -Image.register_extension(Jpeg2KImageFile.format, '.j2c') +Image.register_extensions(Jpeg2KImageFile.format, [".jp2", ".j2k", ".jpc", ".jpf", ".jpx", ".j2c"]) Image.register_mime(Jpeg2KImageFile.format, 'image/jp2') Image.register_mime(Jpeg2KImageFile.format, 'image/jpx') diff --git a/PIL/JpegImagePlugin.py b/PIL/JpegImagePlugin.py index d28c5e506ba..d286f793f78 100644 --- a/PIL/JpegImagePlugin.py +++ b/PIL/JpegImagePlugin.py @@ -783,9 +783,6 @@ def jpeg_factory(fp=None, filename=None): Image.register_open(JpegImageFile.format, jpeg_factory, _accept) Image.register_save(JpegImageFile.format, _save) -Image.register_extension(JpegImageFile.format, ".jfif") -Image.register_extension(JpegImageFile.format, ".jpe") -Image.register_extension(JpegImageFile.format, ".jpg") -Image.register_extension(JpegImageFile.format, ".jpeg") +Image.register_extensions(JpegImageFile.format, [".jfif", ".jpe", ".jpg", ".jpeg"]) Image.register_mime(JpegImageFile.format, "image/jpeg") diff --git a/PIL/MpegImagePlugin.py b/PIL/MpegImagePlugin.py index bdc5e3689d9..fca7f9d9fe4 100644 --- a/PIL/MpegImagePlugin.py +++ b/PIL/MpegImagePlugin.py @@ -80,7 +80,6 @@ def _open(self): Image.register_open(MpegImageFile.format, MpegImageFile) -Image.register_extension(MpegImageFile.format, ".mpg") -Image.register_extension(MpegImageFile.format, ".mpeg") +Image.register_extensions(MpegImageFile.format, [".mpg", ".mpeg"]) Image.register_mime(MpegImageFile.format, "video/mpeg") diff --git a/PIL/PpmImagePlugin.py b/PIL/PpmImagePlugin.py index b91f9912bdc..bd1b7a36467 100644 --- a/PIL/PpmImagePlugin.py +++ b/PIL/PpmImagePlugin.py @@ -164,6 +164,4 @@ def _save(im, fp, filename): Image.register_open(PpmImageFile.format, PpmImageFile, _accept) Image.register_save(PpmImageFile.format, _save) -Image.register_extension(PpmImageFile.format, ".pbm") -Image.register_extension(PpmImageFile.format, ".pgm") -Image.register_extension(PpmImageFile.format, ".ppm") +Image.register_extensions(PpmImageFile.format, [".pbm", ".pgm", ".ppm"]) diff --git a/PIL/SgiImagePlugin.py b/PIL/SgiImagePlugin.py index d5db91c2fc6..c96a8d821ab 100644 --- a/PIL/SgiImagePlugin.py +++ b/PIL/SgiImagePlugin.py @@ -149,9 +149,7 @@ def _save(im, fp, filename): Image.register_save(SgiImageFile.format, _save) Image.register_mime(SgiImageFile.format, "image/sgi") Image.register_mime(SgiImageFile.format, "image/rgb") -Image.register_extension(SgiImageFile.format, ".bw") -Image.register_extension(SgiImageFile.format, ".rgb") -Image.register_extension(SgiImageFile.format, ".rgba") -Image.register_extension(SgiImageFile.format, ".sgi") + +Image.register_extensions(SgiImageFile.format, [".bw", ".rgb", ".rgba", ".sgi"]) # End of file diff --git a/PIL/TiffImagePlugin.py b/PIL/TiffImagePlugin.py index d6fc9ebc5d7..6a01feb35e4 100644 --- a/PIL/TiffImagePlugin.py +++ b/PIL/TiffImagePlugin.py @@ -1811,7 +1811,6 @@ def _save_all(im, fp, filename): Image.register_save(TiffImageFile.format, _save) Image.register_save_all(TiffImageFile.format, _save_all) -Image.register_extension(TiffImageFile.format, ".tif") -Image.register_extension(TiffImageFile.format, ".tiff") +Image.register_extensions(TiffImageFile.format, [".tif", ".tiff"]) Image.register_mime(TiffImageFile.format, "image/tiff") diff --git a/PIL/WmfImagePlugin.py b/PIL/WmfImagePlugin.py index c6c002ffd72..fea436b3a63 100644 --- a/PIL/WmfImagePlugin.py +++ b/PIL/WmfImagePlugin.py @@ -164,5 +164,4 @@ def _save(im, fp, filename): Image.register_open(WmfStubImageFile.format, WmfStubImageFile, _accept) Image.register_save(WmfStubImageFile.format, _save) -Image.register_extension(WmfStubImageFile.format, ".wmf") -Image.register_extension(WmfStubImageFile.format, ".emf") +Image.register_extensions(WmfStubImageFile.format, [".wmf", ".emf"]) diff --git a/Tests/test_image.py b/Tests/test_image.py index e65e0e4c791..c960e84de04 100644 --- a/Tests/test_image.py +++ b/Tests/test_image.py @@ -457,6 +457,22 @@ def test_radial_gradient(self): target = Image.open(target_file).convert(mode) self.assert_image_equal(im, target) + def test_register_extensions(self): + test_format = "a" + exts = ["b", "c"] + for ext in exts: + Image.register_extension(test_format, ext) + ext_individual = Image.EXTENSION.copy() + for ext in exts: + del Image.EXTENSION[ext] + + Image.register_extensions(test_format, exts) + ext_multiple = Image.EXTENSION.copy() + for ext in exts: + del Image.EXTENSION[ext] + + self.assertEqual(ext_individual, ext_multiple) + def test_remap_palette(self): # Test illegal image mode im = hopper() @@ -465,7 +481,7 @@ def test_remap_palette(self): def test__new(self): from PIL import ImagePalette - + im = hopper('RGB') im_p = hopper('P') @@ -473,7 +489,7 @@ def test__new(self): blank_pa = Image.new('PA', (10,10)) blank_p.palette = None blank_pa.palette = None - + def _make_new(base_image, im, palette_result=None): new_im = base_image._new(im) self.assertEqual(new_im.mode, im.mode) @@ -483,12 +499,12 @@ def _make_new(base_image, im, palette_result=None): self.assertEqual(new_im.palette.tobytes(), palette_result.tobytes()) else: self.assertEqual(new_im.palette, None) - + _make_new(im, im_p, im_p.palette) _make_new(im_p, im, None) _make_new(im, blank_p, ImagePalette.ImagePalette()) _make_new(im, blank_pa, ImagePalette.ImagePalette()) - + class MockEncoder(object): pass