relikd/icnsutil
1
0
Fork 0
Code Issues Pull Requests Releases Activity

feat: allow ArgbImage to load from Pillow image

Browse Source
This commit is contained in:
relikd
2023-03-12 17:33:32 +01:00
parent 176b675316
commit bf4efb42d8
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
icnsutil/ArgbImage.py
View File

@@ -38,6 +38,7 @@ class ArgbImage:
*, *,
data: Optional[bytes] = None, data: Optional[bytes] = None,
file: Optional[str] = None, file: Optional[str] = None,
image: Optional['Image.Image'] = None,
mask: Union[bytes, str, None] = None, mask: Union[bytes, str, None] = None,
) -> None: ) -> None:
''' '''
@@ -50,6 +51,8 @@ class ArgbImage:
self.load_file(file) self.load_file(file)
elif data: elif data:
self.load_data(data) self.load_data(data)
elif image:
self._load_pillow_image(image)
else: else:
raise AttributeError('Neither data nor file provided.') raise AttributeError('Neither data nor file provided.')
if mask: if mask:
@@ -128,7 +131,10 @@ class ArgbImage:
def _load_png(self, fname: str) -> None: def _load_png(self, fname: str) -> None:
if not PIL_ENABLED: if not PIL_ENABLED:
raise ImportError('Install Pillow to support PNG conversion.') raise ImportError('Install Pillow to support PNG conversion.')
img = Image.open(fname, mode='r').convert('RGBA') self._load_pillow_image(Image.open(fname, mode='r'))
def _load_pillow_image(self, image: 'Image.Image') -> None:
img = image.convert('RGBA')
self.size = img.size self.size = img.size
self.channels = 4 self.channels = 4
self.a = [] self.a = []