mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-04-20 16:00:29 +00:00
Code Issues Projects Releases Wiki Activity

Support path-like objects in Attachment.save.

Browse Source 
Fix #1958
This commit is contained in:
Rapptz 2019-03-13 21:58:56 -04:00
parent 8f7ce7c64e
commit 6ffd079cb9
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
discord/message.py
View File

@ -27,6 +27,7 @@ DEALINGS IN THE SOFTWARE.
import asyncio
import datetime
import re
import io
from . import utils
from .reaction import Reaction
@ -84,7 +85,7 @@ class Attachment:
Parameters
-----------
fp: Union[BinaryIO, str]
fp: Union[BinaryIO, :class:`os.PathLike`]
The file-like object to save this attachment to or the filename
to use. If a filename is passed then a file is created with that
filename and used instead.
@ -112,14 +113,14 @@ class Attachment:
"""
url = self.proxy_url if use_cached else self.url
data = await self._http.get_attachment(url)
if isinstance(fp, str):
with open(fp, 'wb') as f:
return f.write(data)
else:
if isinstance(fp, io.IOBase) and fp.writable():
written = fp.write(data)
if seek_begin:
fp.seek(0)
return written
else:
with open(fp, 'wb') as f:
return f.write(data)
class Message:
r"""Represents a message from Discord.