Skip to content

Commit 64be60e

Browse files
committed
chore: Texture2D - add assertions for typecheck
1 parent d289b72 commit 64be60e

File tree

3 files changed

+5
-0
lines changed

3 files changed

+5
-0
lines changed

UnityPy/classes/legacy_patch/Texture2D.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ def _Texture2d_set_image(
2424

2525
if not isinstance(img, Image.Image):
2626
img = Image.open(img)
27+
assert isinstance(img, Image.Image)
2728

2829
platform = self.object_reader.platform if self.object_reader is not None else 0
2930
img_data, tex_format = Texture2DConverter.image_to_texture2d(
@@ -70,6 +71,7 @@ def _Texture2D_get_image_data(self: Texture2D):
7071
if self.m_StreamData:
7172
from ...helpers.ResourceReader import get_resource_data
7273

74+
assert self.object_reader is not None
7375
return get_resource_data(
7476
self.m_StreamData.path,
7577
self.object_reader.assets_file,

UnityPy/classes/legacy_patch/Texture2DArray.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,11 @@ def _Texture2DArray_get_images(self: Texture2DArray) -> List[Image.Image]:
1515
texture_format = GRAPHICS_TO_TEXTURE_MAP.get(GraphicsFormat(self.m_Format))
1616
if not texture_format:
1717
raise NotImplementedError(f"GraphicsFormat {self.m_Format} not supported yet")
18+
assert self.object_reader is not None
1819

1920
image_data = self.image_data
2021
if image_data is None:
22+
assert self.m_StreamData is not None
2123
image_data = get_resource_data(
2224
self.m_StreamData.path,
2325
self.object_reader.assets_file,

UnityPy/export/Texture2DConverter.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -280,6 +280,7 @@ def get_image_from_texture2d(
280280
:return: PIL.Image object
281281
:rtype: Image
282282
"""
283+
assert texture_2d.object_reader is not None
283284
return parse_image_data(
284285
texture_2d.get_image_data(),
285286
texture_2d.m_Width,

0 commit comments

Comments
 (0)