1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
|
import os
from typing import (
Any,
IO,
NamedTuple,
Optional,
Type,
TypeVar,
)
class FileStream(NamedTuple):
name: str
fd: IO
def close(self) -> None:
self.fd.close()
def attach_file(path: str, name: Optional[str] = None) -> FileStream:
"""
Attach a path to a request payload object.
:type path: str
:param path: Path to file to attach to payload.
:type name: str
:param name: Name to give file, if different than actual pathname.
:rtype: object
:return: Returns an object compatible with requests post operation and
capable of being closed with a ``close()`` method.
"""
if name is None:
name = os.path.basename(path)
return FileStream(name, open(path, "rb"))
T = TypeVar("T")
def abstractclass(decorated_cls: Type[T]) -> Type[T]:
"""
Decorator that marks a class as abstract even without any abstract method
Adapted from https://stackoverflow.com/a/49013561/4503125
"""
def clsnew(cls: Type[T], *args: Any, **kwargs: Any) -> T:
# assert issubclass(cls, decorated_cls)
if cls is decorated_cls:
cls_name = getattr(decorated_cls, "__name__", str(decorated_cls))
raise TypeError(f"Can't instantiate abstract class {cls_name}")
return super(decorated_cls, cls).__new__(cls) # type: ignore[misc]
decorated_cls.__new__ = clsnew # type: ignore[assignment]
return decorated_cls
__all__ = (
"abstractclass",
"attach_file",
)
|