Skip to content

Commit

Permalink
Raise a TypeError when Image() is called
Browse files Browse the repository at this point in the history
  • Loading branch information
radarhere committed Oct 13, 2023
1 parent 4ecf1df commit c4cdab4
Show file tree
Hide file tree
Showing 3 changed files with 36 additions and 21 deletions.
4 changes: 4 additions & 0 deletions Tests/test_image.py
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,10 @@ def test_sanity(self):
# with pytest.raises(MemoryError):
# Image.new("L", (1000000, 1000000))

def test_direct(self):
with pytest.raises(TypeError):
Image.Image()

def test_repr_pretty(self):
class Pretty:
def text(self, text):
Expand Down
51 changes: 31 additions & 20 deletions src/PIL/Image.py
Original file line number Diff line number Diff line change
Expand Up @@ -481,16 +481,11 @@ class Image:
_close_exclusive_fp_after_loading = True

def __init__(self):
# FIXME: take "new" parameters / other image?
# FIXME: turn mode and size into delegating properties?
self.im = None
self._mode = ""
self._size = (0, 0)
self.palette = None
self.info = {}
self.readonly = 0
self.pyaccess = None
self._exif = None
msg = (
"Images should not be instantiated directly."
"Use the module new() function instead."
)
raise TypeError(msg)

@property
def width(self):
Expand All @@ -508,8 +503,24 @@ def size(self):
def mode(self):
return self._mode

def _prepare(self):
self.im = None
self._mode = ""
self._size = (0, 0)
self.palette = None
self.info = {}
self.readonly = 0
self.pyaccess = None
self._exif = None

@classmethod
def _init(cls):
self = cls.__new__(cls)
self._prepare()
return self

def _new(self, im):
new = Image()
new = Image._init()
new.im = im
new._mode = im.mode
new._size = im.size
Expand Down Expand Up @@ -695,7 +706,7 @@ def __getstate__(self):
return [self.info, self.mode, self.size, self.getpalette(), im_data]

def __setstate__(self, state):
Image.__init__(self)
self._prepare()
info, mode, size, palette, data = state
self.info = info
self._mode = mode
Expand Down Expand Up @@ -980,7 +991,7 @@ def convert_transparency(m, v):
else:
# get the new transparency color.
# use existing conversions
trns_im = Image()._new(core.new(self.mode, (1, 1)))
trns_im = Image._init()._new(core.new(self.mode, (1, 1)))
if self.mode == "P":
trns_im.putpalette(self.palette)
if isinstance(t, tuple):
Expand Down Expand Up @@ -2874,7 +2885,7 @@ class ImageTransformHandler:
def _wedge():
"""Create greyscale wedge (for debugging only)"""

return Image()._new(core.wedge("L"))
return Image._init()._new(core.wedge("L"))


def _check_size(size):
Expand Down Expand Up @@ -2918,7 +2929,7 @@ def new(mode, size, color=0):

if color is None:
# don't initialize
return Image()._new(core.new(mode, size))
return Image._init()._new(core.new(mode, size))

if isinstance(color, str):
# css3-style specifier
Expand All @@ -2927,7 +2938,7 @@ def new(mode, size, color=0):

color = ImageColor.getcolor(color, mode)

im = Image()
im = Image._init()
if mode == "P" and isinstance(color, (list, tuple)) and len(color) in [3, 4]:
# RGB or RGBA value for a P image
from . import ImagePalette
Expand Down Expand Up @@ -3547,7 +3558,7 @@ def effect_mandelbrot(size, extent, quality):
(x0, y0, x1, y1).
:param quality: Quality.
"""
return Image()._new(core.effect_mandelbrot(size, extent, quality))
return Image._init()._new(core.effect_mandelbrot(size, extent, quality))


def effect_noise(size, sigma):
Expand All @@ -3558,7 +3569,7 @@ def effect_noise(size, sigma):
(width, height).
:param sigma: Standard deviation of noise.
"""
return Image()._new(core.effect_noise(size, sigma))
return Image._init()._new(core.effect_noise(size, sigma))


def linear_gradient(mode):
Expand All @@ -3567,7 +3578,7 @@ def linear_gradient(mode):
:param mode: Input mode.
"""
return Image()._new(core.linear_gradient(mode))
return Image._init()._new(core.linear_gradient(mode))


def radial_gradient(mode):
Expand All @@ -3576,7 +3587,7 @@ def radial_gradient(mode):
:param mode: Input mode.
"""
return Image()._new(core.radial_gradient(mode))
return Image._init()._new(core.radial_gradient(mode))


# --------------------------------------------------------------------
Expand Down
2 changes: 1 addition & 1 deletion src/PIL/ImageFile.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ class ImageFile(Image.Image):
"""Base class for image file format handlers."""

def __init__(self, fp=None, filename=None):
super().__init__()
super()._prepare()

self._min_frame = 0

Expand Down

0 comments on commit c4cdab4

Please sign in to comment.