From 297560195686f822d3091ca795c0385a09f195ea Mon Sep 17 00:00:00 2001 From: Bob Mottram Date: Sat, 14 Sep 2019 18:16:03 +0100 Subject: [PATCH] Avatar cache filenames --- webinterface.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/webinterface.py b/webinterface.py index 85f3ae453..7c67cee05 100644 --- a/webinterface.py +++ b/webinterface.py @@ -47,16 +47,18 @@ def updateAvatarImageCache(session,baseDir: str,httpPrefix: str,actor: str,avata """ if not avatarUrl: return None - avatarImageFilename=baseDir+'/cache/avatars/'+avatarUrl.replace('/','#') + if avatarUrl.endswith('.png'): + sessionHeaders = {'Accept': 'image/png'} + avatarImageFilename=baseDir+'/cache/avatars/'+actor.replace('/','#')+'.png' + elif avatarUrl.endswith('.jpg') or avatarUrl.endswith('.jpeg'): + sessionHeaders = {'Accept': 'image/jpeg'} + avatarImageFilename=baseDir+'/cache/avatars/'+actor.replace('/','#')+'.jpg' + elif avatarUrl.endswith('.gif'): + sessionHeaders = {'Accept': 'image/gif'} + avatarImageFilename=baseDir+'/cache/avatars/'+actor.replace('/','#')+'.gif' + else: + return None if not os.path.isfile(avatarImageFilename) or force: - if avatarUrl.endswith('.png'): - sessionHeaders = {'Accept': 'image/png'} - elif avatarUrl.endswith('.jpg') or avatarUrl.endswith('.jpeg'): - sessionHeaders = {'Accept': 'image/jpeg'} - elif avatarUrl.endswith('.gif'): - sessionHeaders = {'Accept': 'image/gif'} - else: - return None try: result=session.get(avatarUrl, headers=sessionHeaders, params=None) with open(avatarImageFilename, 'wb') as f: