mirror of https://gitlab.com/litecord/litecord.git
images: add always_icon kwarg to put()
- guilds: use always_icon
This commit is contained in:
parent
1e368f8cc9
commit
630db1e1e9
|
|
@ -96,7 +96,7 @@ async def put_guild_icon(guild_id: int, icon: str):
|
||||||
else None)
|
else None)
|
||||||
|
|
||||||
return await app.icons.put(
|
return await app.icons.put(
|
||||||
'guild', guild_id, encoded, size=(128, 128))
|
'guild', guild_id, encoded, size=(128, 128), always_icon=True)
|
||||||
|
|
||||||
|
|
||||||
@bp.route('', methods=['POST'])
|
@bp.route('', methods=['POST'])
|
||||||
|
|
|
||||||
|
|
@ -138,6 +138,14 @@ def _gen_update_sql(scope: str) -> str:
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
def _invalid(kwargs: dict):
|
||||||
|
"""Send an invalid value."""
|
||||||
|
if kwargs.get('always_icon', False):
|
||||||
|
return None
|
||||||
|
|
||||||
|
return Icon(None, None, '')
|
||||||
|
|
||||||
|
|
||||||
class IconManager:
|
class IconManager:
|
||||||
"""Main icon manager."""
|
"""Main icon manager."""
|
||||||
def __init__(self, app):
|
def __init__(self, app):
|
||||||
|
|
@ -205,7 +213,7 @@ class IconManager:
|
||||||
b64_data: str, **kwargs) -> Icon:
|
b64_data: str, **kwargs) -> Icon:
|
||||||
"""Insert an icon."""
|
"""Insert an icon."""
|
||||||
if b64_data is None:
|
if b64_data is None:
|
||||||
return None
|
return _invalid(kwargs)
|
||||||
|
|
||||||
mime, raw_data = parse_data_uri(b64_data)
|
mime, raw_data = parse_data_uri(b64_data)
|
||||||
data_fd = BytesIO(raw_data)
|
data_fd = BytesIO(raw_data)
|
||||||
|
|
@ -214,7 +222,7 @@ class IconManager:
|
||||||
extension = _get_ext(mime)
|
extension = _get_ext(mime)
|
||||||
|
|
||||||
if 'bsize' in kwargs and len(raw_data) > kwargs['bsize']:
|
if 'bsize' in kwargs and len(raw_data) > kwargs['bsize']:
|
||||||
return None
|
return _invalid(kwargs)
|
||||||
|
|
||||||
if 'size' in kwargs:
|
if 'size' in kwargs:
|
||||||
image = Image.open(data_fd)
|
image = Image.open(data_fd)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue