aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/Jinja2/py3/jinja2/async_utils.py
diff options
context:
space:
mode:
authorshadchin <shadchin@yandex-team.ru>2022-02-10 16:44:39 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:44:39 +0300
commite9656aae26e0358d5378e5b63dcac5c8dbe0e4d0 (patch)
tree64175d5cadab313b3e7039ebaa06c5bc3295e274 /contrib/python/Jinja2/py3/jinja2/async_utils.py
parent2598ef1d0aee359b4b6d5fdd1758916d5907d04f (diff)
downloadydb-e9656aae26e0358d5378e5b63dcac5c8dbe0e4d0.tar.gz
Restoring authorship annotation for <shadchin@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'contrib/python/Jinja2/py3/jinja2/async_utils.py')
-rw-r--r--contrib/python/Jinja2/py3/jinja2/async_utils.py136
1 files changed, 68 insertions, 68 deletions
diff --git a/contrib/python/Jinja2/py3/jinja2/async_utils.py b/contrib/python/Jinja2/py3/jinja2/async_utils.py
index 2d826cfb23b..35e6cb10902 100644
--- a/contrib/python/Jinja2/py3/jinja2/async_utils.py
+++ b/contrib/python/Jinja2/py3/jinja2/async_utils.py
@@ -1,75 +1,75 @@
-import inspect
-import typing as t
-from functools import wraps
-
-from .utils import _PassArg
-from .utils import pass_eval_context
-
-V = t.TypeVar("V")
-
-
-def async_variant(normal_func): # type: ignore
- def decorator(async_func): # type: ignore
- pass_arg = _PassArg.from_obj(normal_func)
- need_eval_context = pass_arg is None
-
- if pass_arg is _PassArg.environment:
-
- def is_async(args: t.Any) -> bool:
- return t.cast(bool, args[0].is_async)
-
- else:
-
- def is_async(args: t.Any) -> bool:
- return t.cast(bool, args[0].environment.is_async)
-
- @wraps(normal_func)
- def wrapper(*args, **kwargs): # type: ignore
- b = is_async(args)
-
- if need_eval_context:
- args = args[1:]
-
- if b:
- return async_func(*args, **kwargs)
-
- return normal_func(*args, **kwargs)
-
- if need_eval_context:
- wrapper = pass_eval_context(wrapper)
-
- wrapper.jinja_async_variant = True
- return wrapper
-
- return decorator
-
-
+import inspect
+import typing as t
+from functools import wraps
+
+from .utils import _PassArg
+from .utils import pass_eval_context
+
+V = t.TypeVar("V")
+
+
+def async_variant(normal_func): # type: ignore
+ def decorator(async_func): # type: ignore
+ pass_arg = _PassArg.from_obj(normal_func)
+ need_eval_context = pass_arg is None
+
+ if pass_arg is _PassArg.environment:
+
+ def is_async(args: t.Any) -> bool:
+ return t.cast(bool, args[0].is_async)
+
+ else:
+
+ def is_async(args: t.Any) -> bool:
+ return t.cast(bool, args[0].environment.is_async)
+
+ @wraps(normal_func)
+ def wrapper(*args, **kwargs): # type: ignore
+ b = is_async(args)
+
+ if need_eval_context:
+ args = args[1:]
+
+ if b:
+ return async_func(*args, **kwargs)
+
+ return normal_func(*args, **kwargs)
+
+ if need_eval_context:
+ wrapper = pass_eval_context(wrapper)
+
+ wrapper.jinja_async_variant = True
+ return wrapper
+
+ return decorator
+
+
_common_primitives = {int, float, bool, str, list, dict, tuple, type(None)}
-async def auto_await(value: t.Union[t.Awaitable["V"], "V"]) -> "V":
+async def auto_await(value: t.Union[t.Awaitable["V"], "V"]) -> "V":
# Avoid a costly call to isawaitable
if type(value) in _common_primitives:
return t.cast("V", value)
- if inspect.isawaitable(value):
- return await t.cast("t.Awaitable[V]", value)
-
- return t.cast("V", value)
-
-
-async def auto_aiter(
- iterable: "t.Union[t.AsyncIterable[V], t.Iterable[V]]",
-) -> "t.AsyncIterator[V]":
- if hasattr(iterable, "__aiter__"):
- async for item in t.cast("t.AsyncIterable[V]", iterable):
- yield item
- else:
- for item in t.cast("t.Iterable[V]", iterable):
- yield item
-
-
-async def auto_to_list(
- value: "t.Union[t.AsyncIterable[V], t.Iterable[V]]",
-) -> t.List["V"]:
- return [x async for x in auto_aiter(value)]
+ if inspect.isawaitable(value):
+ return await t.cast("t.Awaitable[V]", value)
+
+ return t.cast("V", value)
+
+
+async def auto_aiter(
+ iterable: "t.Union[t.AsyncIterable[V], t.Iterable[V]]",
+) -> "t.AsyncIterator[V]":
+ if hasattr(iterable, "__aiter__"):
+ async for item in t.cast("t.AsyncIterable[V]", iterable):
+ yield item
+ else:
+ for item in t.cast("t.Iterable[V]", iterable):
+ yield item
+
+
+async def auto_to_list(
+ value: "t.Union[t.AsyncIterable[V], t.Iterable[V]]",
+) -> t.List["V"]:
+ return [x async for x in auto_aiter(value)]