Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 7 additions & 5 deletions allure-python-commons/src/_allure.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from functools import wraps
from typing import Any, Callable, TypeVar, Union, overload
from typing import Any, Callable, TypeVar, overload

from allure_commons._core import plugin_manager
from allure_commons.types import LabelType, LinkType, ParameterMode
Expand Down Expand Up @@ -162,14 +162,16 @@ def manual():


@overload
def step(title: str) -> "StepContext": ...
def step(title: str) -> "StepContext":
...


@overload
def step(title: _TFunc) -> _TFunc: ...
def step(title: _TFunc) -> _TFunc:
...


def step(title: Union[str, _TFunc]) -> Union["StepContext", _TFunc]:
def step(title):
if callable(title):
return StepContext(title.__name__, {})(title)
else:
Expand Down Expand Up @@ -199,7 +201,7 @@ def impl(*a, **kw):
with StepContext(self.title.format(*args, **params), params):
return func(*a, **kw)

return impl
return impl # type: ignore


class Attach:
Expand Down