Merge pull request #8859 from radarhere/filename

Include filename in ImageFile state

Author: hugovk

Tests/test_pickle.py

@@ -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
 

src/PIL/Image.py

@@ -2543,7 +2543,9 @@ def save(
         from . import ImageFile
 
         # may mutate self!
-        if isinstance(self, ImageFile.ImageFile) and filename == self.filename:
+        if isinstance(self, ImageFile.ImageFile) and os.path.abspath(
+            filename
+        ) == os.path.abspath(self.filename):
             self._ensure_mutable()
         else:
             self.load()

src/PIL/ImageFile.py

@@ -252,8 +252,12 @@ def get_format_mimetype(self) -> str | None:
             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:

src/PIL/JpegImagePlugin.py

@@ -403,8 +403,8 @@ def __getstate__(self) -> list[Any]:
         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:
         """