Allow removing an embed in Message.edit

1 files changed, 20 insertions, 5 deletions

diff --git a/discord/message.py b/discord/message.py
index acf4da2c..d4367f23 100644
--- a/discord/message.py
+++ b/discord/message.py
@@ -416,7 +416,7 @@ class Message:
         yield from self._state.http.delete_message(self.channel.id, self.id)
 
     @asyncio.coroutine
-    def edit(self, *, content: str = None, embed: Embed = None):
+    def edit(self, **fields):
         """|coro|
 
         Edits the message.
@@ -427,8 +427,9 @@ class Message:
         -----------
         content: str
             The new content to replace the message with.
-        embed: :class:`Embed`
+        embed: Optional[:class:`Embed`]
             The new embed to replace the original with.
+            Could be ``None`` to remove the embed.
 
         Raises
         -------
@@ -436,9 +437,23 @@ class Message:
             Editing the message failed.
         """
 
-        content = str(content) if content else None
-        embed = embed.to_dict() if embed else None
-        data = yield from self._state.http.edit_message(self.id, self.channel.id, content, embed=embed)
+        try:
+            content = fields['content']
+        except KeyError:
+            pass
+        else:
+            if content is not None:
+                fields['content'] = str(content)
+
+        try:
+            embed = fields['embed']
+        except KeyError:
+            pass
+        else:
+            if embed is not None:
+                fields['embed'] = embed.to_dict()
+
+        data = yield from self._state.http.edit_message(self.id, self.channel.id, **fields)
         self._update(channel=self.channel, data=data)
 
     @asyncio.coroutine