mirror of
https://github.com/stoatchat/python-client-sdk.git
synced 2026-07-22 10:35:25 -04:00
Zomatree
79 lines
2.1 KiB
Python
Executable File
79 lines
2.1 KiB
Python
Executable File
import inspect
|
|
from operator import attrgetter
|
|
from typing import Any, Callable, Coroutine, Iterable, TypeVar, Union
|
|
|
|
from typing_extensions import ParamSpec
|
|
|
|
__all__ = ("Missing", "copy_doc", "maybe_coroutine", "get")
|
|
|
|
class _Missing:
|
|
def __repr__(self):
|
|
return "<Missing>"
|
|
|
|
Missing = _Missing()
|
|
|
|
T = TypeVar("T")
|
|
|
|
def copy_doc(from_t: T) -> Callable[[T], T]:
|
|
def inner(to_t: T) -> T:
|
|
to_t.__doc__ = from_t.__doc__
|
|
return to_t
|
|
|
|
return inner
|
|
|
|
R_T = TypeVar("R_T")
|
|
P = ParamSpec("P")
|
|
|
|
# its impossible to type this function correctly for a couple reasons:
|
|
# 1. isawaitable doesnt narrow while keeping typevars - there is an open pr for this (typeshed#5658) but it cant be merged because mypy doesnt support the feature fully
|
|
# 2. typeguard doesnt narrow for the negative case which is dumb imo, so `value` would stay being a union even after the if statement (PEP 647 - "The type is not narrowed in the negative case")
|
|
|
|
async def maybe_coroutine(func: Callable[P, Union[R_T, Coroutine[Any, Any, R_T]]], *args: P.args, **kwargs: P.kwargs) -> R_T:
|
|
value = func(*args, **kwargs)
|
|
|
|
if inspect.isawaitable(value):
|
|
value = await value
|
|
|
|
return value # type: ignore
|
|
|
|
|
|
def get(iterable: Iterable[T], **attrs: Any) -> T:
|
|
"""A convenience function to help get a value from an iterable with a specific attribute
|
|
|
|
Examples
|
|
---------
|
|
|
|
.. code-block:: python
|
|
:emphasize-lines: 3
|
|
|
|
from revolt import utils
|
|
|
|
channel = utils.get(server.channels, name="General")
|
|
await channel.send("Hello general chat.")
|
|
|
|
Parameters
|
|
-----------
|
|
iterable: Iterable
|
|
The values to search though
|
|
**attrs: Any
|
|
The attributes to check
|
|
|
|
Returns
|
|
--------
|
|
Any
|
|
The value from the iterable with the met attributes
|
|
|
|
Raises
|
|
-------
|
|
LookupError
|
|
Raises when none of the values in the iterable matches the attributes
|
|
|
|
"""
|
|
converted = [(attrgetter(attr.replace('__', '.')), value) for attr, value in attrs.items()]
|
|
|
|
for elem in iterable:
|
|
if all(pred(elem) == value for pred, value in converted):
|
|
return elem
|
|
|
|
raise LookupError
|