From b0f1035ae8cef117aa95b45b3fb61e609dfed249 Mon Sep 17 00:00:00 2001 From: Roger Philibert Date: Thu, 29 Mar 2012 23:07:28 +0200 Subject: [PATCH] [boobmsg] Use url suffix to save photos with their extension --- weboob/applications/boobmsg/boobmsg.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/weboob/applications/boobmsg/boobmsg.py b/weboob/applications/boobmsg/boobmsg.py index f142da21..26b8c263 100644 --- a/weboob/applications/boobmsg/boobmsg.py +++ b/weboob/applications/boobmsg/boobmsg.py @@ -455,7 +455,11 @@ class Boobmsg(ReplApplication): # Write photo to temporary files tmp_files = [] for photo in contact.photos.values(): - f = NamedTemporaryFile(suffix='.jpg') + suffix = '.jpg' + if '.' in photo.url.split('/')[-1]: + suffix = '.%s' % photo.url.split('/')[-1].split('.')[-1] + f = NamedTemporaryFile(suffix=suffix) + photo = backend.fillobj(photo, 'data') f.write(photo.data) tmp_files.append(f)