From 7dd97fb2c052bb28ae919d33f3bd6ad3596288f2 Mon Sep 17 00:00:00 2001
From: aikaterna <20862007+aikaterna@users.noreply.github.com>
Date: Sun, 1 Apr 2018 11:49:24 -0700
Subject: [PATCH] Update picwelcome to use other image types

Picwelcome can now use GIF images.
---
 picwelcome/picwelcome.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/picwelcome/picwelcome.py b/picwelcome/picwelcome.py
index c2de3d3..f377216 100644
--- a/picwelcome/picwelcome.py
+++ b/picwelcome/picwelcome.py
@@ -109,24 +109,24 @@ class PicWelcome:
             print(e)
 
         serverimage = Image
-
         if success:
             try:
                 async with aiohttp.get(bg_url) as r:
                     image = await r.content.read()
                     if not os.path.exists('data/picwelcome/{}'.format(server.id)):
                         os.makedirs('data/picwelcome/{}'.format(server.id))
-                serverbg = 'data/picwelcome/{}/serverpic.png'.format(server.id)
+                file_suffix = bg_url.rsplit('.', 1)[1]
+                serverbg = 'data/picwelcome/{}/serverpic.{}'.format(server.id, file_suffix)
                 with open(serverbg, 'wb') as f:
                     f.write(image)
                     serverimage = Image.open(serverbg).convert('RGBA')
                     success = True
-
             except Exception as e:
                 success = False
                 print(e)
+
             if success:
-                self.settings[server.id]['PICTURE'] = "data/picwelcome/" + ctx.message.server.id + "/serverpic.png"
+                self.settings[server.id]['PICTURE'] = "data/picwelcome/{}/serverpic.{}".format(ctx.message.server.id, file_suffix)
                 await self.save_settings()
                 await self.bot.say('Welcome image for this server set to uploaded file.')
             else:
@@ -152,12 +152,10 @@ class PicWelcome:
         serverpicture = self.settings[server.id]["PICTURE"]
         await self.bot.send_file(channel_object, serverpicture)
 
-
 def check_folders():
     if not os.path.exists('data/picwelcome/'):
         os.mkdir('data/picwelcome/')
 
-
 def check_files():
     if not dataIO.is_valid_json('data/picwelcome/settings.json'):
         defaults = {}