mirror of
https://github.com/stoatchat/python-client-sdk.git
synced 2026-07-01 20:44:04 -04:00
Zomatree
39 lines
1.3 KiB
Python
Executable File
39 lines
1.3 KiB
Python
Executable File
from __future__ import annotations
|
|
|
|
import io
|
|
from typing import Optional, Union, cast
|
|
|
|
__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):
|
|
self.f: io.BufferedIOBase
|
|
|
|
if isinstance(file, str):
|
|
self.f = open(file, "rb")
|
|
else:
|
|
self.f = io.BytesIO(file)
|
|
|
|
if filename is None and isinstance(file, str):
|
|
filename = cast(Optional[str], self.f.name)
|
|
|
|
self.spoiler: bool = spoiler or (bool(filename) and filename.startswith("SPOILER_"))
|
|
|
|
if self.spoiler and (filename and not filename.startswith("SPOILER_")):
|
|
filename = f"SPOILER_{filename}"
|
|
|
|
self.filename: str | None = filename
|