matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Add spoiler attribute to File

Browse Source
This commit is contained in:
Josh 2021-01-17 15:18:10 +10:00 committed by GitHub
parent b7c7200f4d
commit fce67c32c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
discord/file.py
View File

@ -58,7 +58,7 @@ class File:
Whether the attachment is a spoiler. Whether the attachment is a spoiler.
""" """
__slots__ = ('fp', 'filename', '_original_pos', '_owner', '_closer') __slots__ = ('fp', 'filename', 'spoiler', '_original_pos', '_owner', '_closer')
def __init__(self, fp, filename=None, *, spoiler=False): def __init__(self, fp, filename=None, *, spoiler=False):
self.fp = fp self.fp = fp
@ -92,6 +92,8 @@ class File:
if spoiler and self.filename is not None and not self.filename.startswith('SPOILER_'): if spoiler and self.filename is not None and not self.filename.startswith('SPOILER_'):
self.filename = 'SPOILER_' + self.filename self.filename = 'SPOILER_' + self.filename
self.spoiler = spoiler or (self.filename is not None and self.filename.startswith('SPOILER_'))
def reset(self, *, seek=True): def reset(self, *, seek=True):
# The `seek` parameter is needed because # The `seek` parameter is needed because
# the retry-loop is iterated over multiple times # the retry-loop is iterated over multiple times