python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-05-29 02:03:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

Allow writing IFDRational to BYTE tag

Browse Source
This commit is contained in:
Andrew Murray 2023-01-13 21:02:42 +11:00
parent 145b80be56
commit e48aead015
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Tests/test_file_tiff_metadata.py
View File

@ -202,14 +202,15 @@ def test_writing_other_types_to_ascii(value, expected, tmp_path):
assert reloaded.tag_v2[271] == expected assert reloaded.tag_v2[271] == expected
def test_writing_int_to_bytes(tmp_path): @pytest.mark.parametrize("value", (1, IFDRational(1)))
def test_writing_other_types_to_bytes(value, tmp_path):
im = hopper() im = hopper()
info = TiffImagePlugin.ImageFileDirectory_v2() info = TiffImagePlugin.ImageFileDirectory_v2()
tag = TiffTags.TAGS_V2[700] tag = TiffTags.TAGS_V2[700]
assert tag.type == TiffTags.BYTE assert tag.type == TiffTags.BYTE
info[700] = 1 info[700] = value
out = str(tmp_path / "temp.tiff") out = str(tmp_path / "temp.tiff")
im.save(out, tiffinfo=info) im.save(out, tiffinfo=info)

2
src/PIL/TiffImagePlugin.py
View File

@ -722,6 +722,8 @@ class ImageFileDirectory_v2(MutableMapping):
@_register_writer(1) # Basic type, except for the legacy API. @_register_writer(1) # Basic type, except for the legacy API.
def write_byte(self, data): def write_byte(self, data):
if isinstance(data, IFDRational):
data = int(data)
if isinstance(data, int): if isinstance(data, int):
data = bytes((data,)) data = bytes((data,))
return data return data