mirror of
https://github.com/stoatchat/python-client-sdk.git
synced 2026-07-01 20:44:04 -04:00
Zomatree
35 lines
1.2 KiB
Python
Executable File
35 lines
1.2 KiB
Python
Executable File
import io
|
|
import os
|
|
from typing import Optional, Union
|
|
|
|
__all__ = ("File",)
|
|
|
|
class File:
|
|
"""Respresents a file about to be uploaded to revolt
|
|
|
|
Parameters
|
|
-----------
|
|
file: Union[str, bytes]
|
|
The name of the file or the content of the file in bytes, text files will be need to be encoded
|
|
filename: Optional[str]
|
|
The filename of the file when being uploaded, this will default to the name of the file if one exists
|
|
spoiler: bool
|
|
Determines if the file will be a spoiler, this prefexes the filename with `SPOILER_`
|
|
"""
|
|
__slots__ = ("f", "spoiler", "filename")
|
|
|
|
def __init__(self, file: Union[str, bytes], *, filename: Optional[str] = None, spoiler: bool = False):
|
|
if isinstance(file, str):
|
|
self.f = open(file, "rb")
|
|
elif isinstance(file, bytes):
|
|
self.f = io.BytesIO(file)
|
|
|
|
if filename is None and isinstance(file, str):
|
|
filename = self.f.name
|
|
|
|
self.spoiler = spoiler or (filename and filename.startswith("SPOILER_"))
|
|
|
|
if self.spoiler and (filename and not filename.startswith("SPOILER_")):
|
|
filename = f"SPOILER_{filename}"
|
|
|
|
self.filename = filename |