PapaTutuWawa/xmpp-discord-bridge
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

avatar: Guard against a non-specified type

Browse Source
This commit is contained in:
PapaTutuWawa 2021-11-05 22:24:11 +01:00
parent 3d3686c301
commit 40fcd404a4
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
xmpp_discord_bridge/avatar.py
View File

@ -20,8 +20,13 @@ class AvatarManager:
self._logger = logging.getLogger("xmpp.avatar")
def _save_avatar(self, jid, data, type_):
self._logger.debug("Got type_ of '%s'", type_)
if not type_:
# TODO: Maybe guess?
self._logger.info("Got VCARD with empty photo type. Assuming image/png")
type_ = "image/png"
if len(type_.split("/")) < 2:
self._logger.debug("Got type_ of '%s'", type_)
self._logger.debug("COMPAT! Just assuming image/png")
type_ = "image/png"
filename = hashlib.sha1(data).hexdigest() + "." + type_.split("/")[1]