juni/aikaterna-cogs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update picwelcome to use other image types

Browse Source 
Picwelcome can now use GIF images.
This commit is contained in:
aikaterna
2018-04-01 11:49:24 -07:00
committed by GitHub
parent eaee1acf21
commit 7dd97fb2c0
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
picwelcome/picwelcome.py
View File

@@ -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 = {}