velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

images: add dummy IconManager gif handling

This commit is contained in:
Luna Mendes 2018-11-21 18:29:19 -03:00
parent 7485e95a9e
commit 2140f572e8
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
litecord/images.py
View File

@ -228,6 +228,11 @@ class IconManager:
return await self.generic_get( return await self.generic_get(
'guild', guild_id, icon_hash, **kwargs) 'guild', guild_id, icon_hash, **kwargs)
async def _put_gif(self, scope: str, key,
raw_data: bytes, **kwargs) -> Icon:
"""Insert a gif icon"""
raise NotImplementedError
async def put(self, scope: str, key: str, async def put(self, scope: str, key: str,
b64_data: str, **kwargs) -> Icon: b64_data: str, **kwargs) -> Icon:
"""Insert an icon.""" """Insert an icon."""
@ -235,6 +240,8 @@ class IconManager:
return _invalid(kwargs) return _invalid(kwargs)
mime, raw_data = parse_data_uri(b64_data) mime, raw_data = parse_data_uri(b64_data)
# TODO: filter mimes
data_fd = BytesIO(raw_data) data_fd = BytesIO(raw_data)
# get an extension for the given data uri # get an extension for the given data uri
@ -243,6 +250,11 @@ class IconManager:
if 'bsize' in kwargs and len(raw_data) > kwargs['bsize']: if 'bsize' in kwargs and len(raw_data) > kwargs['bsize']:
return _invalid(kwargs) return _invalid(kwargs)
# size management is different for gif files
# as they're composed of multiple frames.
if mime == 'image/gif':
return await self._put_gif(scope, key, raw_data, **kwargs)
if 'size' in kwargs: if 'size' in kwargs:
image = Image.open(data_fd) image = Image.open(data_fd)