python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2024-09-21 03:18:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5380 from radarhere/accept

Browse Source
This commit is contained in:
Hugo van Kemenade 2021-04-03 16:23:39 +03:00 committed by GitHub
commit e2ac1d1c34
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Tests/test_file_dds.py
View File

@ -136,25 +136,25 @@ def test_uncompressed_rgb():
)
def test__validate_true():
def test__accept_true():
"""Check valid prefix"""
# Arrange
prefix = b"DDS etc"
# Act
output = DdsImagePlugin._validate(prefix)
output = DdsImagePlugin._accept(prefix)
# Assert
assert output
def test__validate_false():
def test__accept_false():
"""Check invalid prefix"""
# Arrange
prefix = b"something invalid"
# Act
output = DdsImagePlugin._validate(prefix)
output = DdsImagePlugin._accept(prefix)
# Assert
assert not output

8
src/PIL/BlpImagePlugin.py
View File

@ -417,9 +417,11 @@ class BLP2Decoder(_BLPBaseDecoder):
self.set_as_raw(bytes(data))
Image.register_open(
BlpImageFile.format, BlpImageFile, lambda p: p[:4] in (b"BLP1", b"BLP2")
)
def _accept(prefix):
return prefix[:4] in (b"BLP1", b"BLP2")
Image.register_open(BlpImageFile.format, BlpImageFile, _accept)
Image.register_extension(BlpImageFile.format, ".blp")
Image.register_decoder("BLP1", BLP1Decoder)

4
src/PIL/DdsImagePlugin.py
View File

@ -182,9 +182,9 @@ class DdsImageFile(ImageFile.ImageFile):
pass
def _validate(prefix):
def _accept(prefix):
return prefix[:4] == b"DDS "
Image.register_open(DdsImageFile.format, DdsImageFile, _validate)
Image.register_open(DdsImageFile.format, DdsImageFile, _accept)
Image.register_extension(DdsImageFile.format, ".dds")

4
src/PIL/FtexImagePlugin.py
View File

@ -98,9 +98,9 @@ class FtexImageFile(ImageFile.ImageFile):
pass
def _validate(prefix):
def _accept(prefix):
return prefix[:4] == MAGIC
Image.register_open(FtexImageFile.format, FtexImageFile, _validate)
Image.register_open(FtexImageFile.format, FtexImageFile, _accept)
Image.register_extensions(FtexImageFile.format, [".ftc", ".ftu"])

6
src/PIL/IcnsImagePlugin.py
View File

@ -359,7 +359,11 @@ def _save(im, fp, filename):
fp.write(f.read())
Image.register_open(IcnsImageFile.format, IcnsImageFile, lambda x: x[:4] == b"icns")
def _accept(prefix):
return prefix[:4] == b"icns"
Image.register_open(IcnsImageFile.format, IcnsImageFile, _accept)
Image.register_extension(IcnsImageFile.format, ".icns")
if sys.platform == "darwin":