python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2024-12-25 01:16:16 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added support for pathlib Path objects to open and save

Browse Source
This commit is contained in:
Andrew Murray 2015-08-05 20:54:33 +10:00
parent 6ffa876d33
commit 457d39832d
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
PIL/Image.py
View File

@ -1653,13 +1653,15 @@ class Image(object):
may have been created, and may contain partial data. may have been created, and may contain partial data.
""" """
filename = ""
if isPath(fp): if isPath(fp):
filename = fp filename = fp
else: elif sys.version_info >= (3, 4):
if hasattr(fp, "name") and isPath(fp.name): from pathlib import Path
if isinstance(fp, Path):
filename = str(fp.resolve())
elif hasattr(fp, "name") and isPath(fp.name):
filename = fp.name filename = fp.name
else:
filename = ""
# may mutate self! # may mutate self!
self.load() self.load()
@ -1687,8 +1689,8 @@ class Image(object):
else: else:
save_handler = SAVE[format.upper()] save_handler = SAVE[format.upper()]
if isPath(fp): if filename:
fp = builtins.open(fp, "wb") fp = builtins.open(filename, "wb")
close = 1 close = 1
else: else:
close = 0 close = 0
@ -2277,11 +2279,15 @@ def open(fp, mode="r"):
if mode != "r": if mode != "r":
raise ValueError("bad mode %r" % mode) raise ValueError("bad mode %r" % mode)
filename = ""
if isPath(fp): if isPath(fp):
filename = fp filename = fp
fp = builtins.open(fp, "rb") elif sys.version_info >= (3, 4):
else: from pathlib import Path
filename = "" if isinstance(fp, Path):
filename = str(fp.resolve())
if filename:
fp = builtins.open(filename, "rb")
try: try:
fp.seek(0) fp.seek(0)