python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-04-07 10:54:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Include filename in state

Browse Source
This commit is contained in:
Andrew Murray 2025-04-02 11:14:58 +11:00
parent 7c56b383ff
commit 1103e82d17
3 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Tests/test_pickle.py
View File

@ -81,6 +81,7 @@ def test_pickle_jpeg() -> None:
unpickled_image = pickle.loads(pickle.dumps(image))
# Assert
assert unpickled_image.filename == "Tests/images/hopper.jpg"
assert len(unpickled_image.layer) == 3
assert unpickled_image.layers == 3

4
src/PIL/ImageFile.py
View File

@ -252,8 +252,12 @@ class ImageFile(Image.Image):
return Image.MIME.get(self.format.upper())
return None
def __getstate__(self) -> list[Any]:
return super().__getstate__() + [self.filename]
def __setstate__(self, state: list[Any]) -> None:
self.tile = []
self.filename = state[5]
super().__setstate__(state)
def verify(self) -> None:

2
src/PIL/JpegImagePlugin.py
View File

@ -403,8 +403,8 @@ class JpegImageFile(ImageFile.ImageFile):
return super().__getstate__() + [self.layers, self.layer]
def __setstate__(self, state: list[Any]) -> None:
self.layers, self.layer = state[6:]
super().__setstate__(state)
self.layers, self.layer = state[5:]
def load_read(self, read_bytes: int) -> bytes:
"""