diff options
| -rw-r--r-- | discord/message.py | 39 |
1 files changed, 37 insertions, 2 deletions
diff --git a/discord/message.py b/discord/message.py index c82c5d18..980a38a2 100644 --- a/discord/message.py +++ b/discord/message.py @@ -112,8 +112,7 @@ class Attachment: :class:`int` The number of bytes written. """ - url = self.proxy_url if use_cached else self.url - data = await self._http.get_from_cdn(url) + data = await self.read(use_cached=use_cached) if isinstance(fp, io.IOBase) and fp.writable(): written = fp.write(data) if seek_begin: @@ -123,6 +122,42 @@ class Attachment: with open(fp, 'wb') as f: return f.write(data) + async def read(self, *, use_cached=False): + """|coro| + + Retrieves the content of this attachment as a :class:`bytes` object. + + Parameters + ----------- + use_cached: :class:`bool` + Whether to use :attr:`proxy_url` rather than :attr:`url` when downloading + the attachment. This will allow attachments to be saved after deletion + more often, compared to the regular URL which is generally deleted right + after the message is deleted. Note that this can still fail to download + deleted attachments if too much time has passed and it does not work + on some type of attachments. + + .. versionadded:: 1.1.0 + + Raises + ------ + HTTPException + Downloading the attachment failed. + Forbidden + You do not have permissions to access this attachment + NotFound + The attachment was deleted. + + Returns + ------- + :class:`bytes` + The contents of the attachment. + """ + url = self.proxy_url if use_cached else self.url + data = await self._http.get_from_cdn(url) + return data + + class Message: r"""Represents a message from Discord. |