Added type hints to GdImageFile

This commit is contained in:
Andrew Murray 2024-01-17 23:10:37 +11:00
parent 20830c541e
commit d8c7af0157

View File

@ -27,6 +27,8 @@
""" """
from __future__ import annotations from __future__ import annotations
from io import BytesIO
from . import ImageFile, ImagePalette, UnidentifiedImageError from . import ImageFile, ImagePalette, UnidentifiedImageError
from ._binary import i16be as i16 from ._binary import i16be as i16
from ._binary import i32be as i32 from ._binary import i32be as i32
@ -43,8 +45,10 @@ class GdImageFile(ImageFile.ImageFile):
format = "GD" format = "GD"
format_description = "GD uncompressed images" format_description = "GD uncompressed images"
def _open(self): def _open(self) -> None:
# Header # Header
assert self.fp is not None
s = self.fp.read(1037) s = self.fp.read(1037)
if i16(s) not in [65534, 65535]: if i16(s) not in [65534, 65535]:
@ -76,7 +80,7 @@ class GdImageFile(ImageFile.ImageFile):
] ]
def open(fp, mode="r"): def open(fp: BytesIO, mode: str = "r") -> GdImageFile:
""" """
Load texture from a GD image file. Load texture from a GD image file.