From 75dc0b8a53a4105caf2d362bac75a7a23c856e5c Mon Sep 17 00:00:00 2001
From: SebastiaanZ <33516116+SebastiaanZ@users.noreply.github.com>
Date: Tue, 23 Apr 2019 16:59:34 +0200
Subject: Changing the send_attachments utility to add WebHook support

---
 bot/utils/messages.py | 24 +++++++++++++++++++-----
 1 file changed, 19 insertions(+), 5 deletions(-)

diff --git a/bot/utils/messages.py b/bot/utils/messages.py
index fc38b0127..b285444c2 100644
--- a/bot/utils/messages.py
+++ b/bot/utils/messages.py
@@ -1,9 +1,9 @@
 import asyncio
 import contextlib
 from io import BytesIO
-from typing import Sequence
+from typing import Sequence, Union
 
-from discord import Embed, File, Message, TextChannel
+from discord import Embed, File, Message, TextChannel, Webhook
 from discord.abc import Snowflake
 from discord.errors import HTTPException
 
@@ -78,7 +78,7 @@ async def wait_for_deletion(
         await message.delete()
 
 
-async def send_attachments(message: Message, destination: TextChannel):
+async def send_attachments(message: Message, destination: Union[TextChannel, Webhook]):
     """
     Re-uploads each attachment in a message to the given channel.
 
@@ -97,7 +97,14 @@ async def send_attachments(message: Message, destination: TextChannel):
             if attachment.size <= MAX_SIZE - 512:
                 with BytesIO() as file:
                     await attachment.save(file)
-                    await destination.send(file=File(file, filename=attachment.filename))
+                    if isinstance(destination, TextChannel):
+                        await destination.send(file=File(file, filename=attachment.filename))
+                    else:
+                        await destination.send(
+                            file=File(file, filename=attachment.filename),
+                            username=message.author.display_name,
+                            avatar_url=message.author.avatar_url
+                        )
             else:
                 large.append(attachment)
         except HTTPException as e:
@@ -109,4 +116,11 @@ async def send_attachments(message: Message, destination: TextChannel):
     if large:
         embed = Embed(description=f"\n".join(f"[{attachment.filename}]({attachment.url})" for attachment in large))
         embed.set_footer(text="Attachments exceed upload size limit.")
-        await destination.send(embed=embed)
+        if isinstance(destination, TextChannel):
+            await destination.send(embed=embed)
+        else:
+            await destination.send(
+                embed=embed,
+                username=message.author.display_name,
+                avatar_url=message.author.avatar_url
+            )
-- 
cgit v1.2.3