|
1 | | -from types import CoroutineType |
2 | 1 | from typing import ( |
3 | 2 | Any, |
4 | | - Coroutine, |
5 | 3 | Dict, |
6 | 4 | Generic, |
7 | 5 | List, |
8 | 6 | Literal, |
9 | 7 | Optional, |
10 | 8 | TypeVar, |
11 | 9 | Union, |
12 | | - overload, |
13 | 10 | ) |
14 | 11 |
|
15 | 12 | import pydantic |
|
23 | 20 |
|
24 | 21 | _ReturnType = TypeVar("_ReturnType") |
25 | 22 | _FuncParams = ParamSpec("_FuncParams") |
26 | | -_T2 = TypeVar("_T2") |
27 | 23 |
|
28 | 24 |
|
29 | 25 | class DumpedStep(pydantic.BaseModel): |
@@ -65,30 +61,6 @@ def __init__( |
65 | 61 | if task: |
66 | 62 | self.call_next(task) |
67 | 63 |
|
68 | | - @overload |
69 | | - def call_next( |
70 | | - self: "Pipeline[_FuncParams, _ReturnType]", |
71 | | - task: Union[ |
72 | | - AsyncKicker[Any, Coroutine[Any, Any, _T2]], |
73 | | - AsyncKicker[Any, "CoroutineType[Any, Any, _T2]"], |
74 | | - AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], |
75 | | - AsyncTaskiqDecoratedTask[Any, "CoroutineType[Any, Any, _T2]"], |
76 | | - ], |
77 | | - param_name: Union[Optional[str], Literal[-1]] = None, |
78 | | - **additional_kwargs: Any, |
79 | | - ) -> "Pipeline[_FuncParams, _T2]": ... |
80 | | - |
81 | | - @overload |
82 | | - def call_next( |
83 | | - self: "Pipeline[_FuncParams, _ReturnType]", |
84 | | - task: Union[ |
85 | | - AsyncKicker[Any, _T2], |
86 | | - AsyncTaskiqDecoratedTask[Any, _T2], |
87 | | - ], |
88 | | - param_name: Union[Optional[str], Literal[-1]] = None, |
89 | | - **additional_kwargs: Any, |
90 | | - ) -> "Pipeline[_FuncParams, _T2]": ... |
91 | | - |
92 | 64 | def call_next( |
93 | 65 | self, |
94 | 66 | task: Union[ |
@@ -126,28 +98,6 @@ def call_next( |
126 | 98 | ) |
127 | 99 | return self |
128 | 100 |
|
129 | | - @overload |
130 | | - def call_after( |
131 | | - self: "Pipeline[_FuncParams, _ReturnType]", |
132 | | - task: Union[ |
133 | | - AsyncKicker[Any, Coroutine[Any, Any, _T2]], |
134 | | - AsyncKicker[Any, "CoroutineType[Any, Any, _T2]"], |
135 | | - AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], |
136 | | - AsyncTaskiqDecoratedTask[Any, "CoroutineType[Any, Any, _T2]"], |
137 | | - ], |
138 | | - **additional_kwargs: Any, |
139 | | - ) -> "Pipeline[_FuncParams, _T2]": ... |
140 | | - |
141 | | - @overload |
142 | | - def call_after( |
143 | | - self: "Pipeline[_FuncParams, _ReturnType]", |
144 | | - task: Union[ |
145 | | - AsyncKicker[Any, _T2], |
146 | | - AsyncTaskiqDecoratedTask[Any, _T2], |
147 | | - ], |
148 | | - **additional_kwargs: Any, |
149 | | - ) -> "Pipeline[_FuncParams, _T2]": ... |
150 | | - |
151 | 101 | def call_after( |
152 | 102 | self, |
153 | 103 | task: Union[ |
@@ -182,34 +132,6 @@ def call_after( |
182 | 132 | ) |
183 | 133 | return self |
184 | 134 |
|
185 | | - @overload |
186 | | - def map( |
187 | | - self: "Pipeline[_FuncParams, _ReturnType]", |
188 | | - task: Union[ |
189 | | - AsyncKicker[Any, Coroutine[Any, Any, _T2]], |
190 | | - AsyncKicker[Any, "CoroutineType[Any, Any, _T2]"], |
191 | | - AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, _T2]], |
192 | | - AsyncTaskiqDecoratedTask[Any, "CoroutineType[Any, Any, _T2]"], |
193 | | - ], |
194 | | - param_name: Optional[str] = None, |
195 | | - skip_errors: bool = False, |
196 | | - check_interval: float = 0.5, |
197 | | - **additional_kwargs: Any, |
198 | | - ) -> "Pipeline[_FuncParams, List[_T2]]": ... |
199 | | - |
200 | | - @overload |
201 | | - def map( |
202 | | - self: "Pipeline[_FuncParams, _ReturnType]", |
203 | | - task: Union[ |
204 | | - AsyncKicker[Any, _T2], |
205 | | - AsyncTaskiqDecoratedTask[Any, _T2], |
206 | | - ], |
207 | | - param_name: Optional[str] = None, |
208 | | - skip_errors: bool = False, |
209 | | - check_interval: float = 0.5, |
210 | | - **additional_kwargs: Any, |
211 | | - ) -> "Pipeline[_FuncParams, List[_T2]]": ... |
212 | | - |
213 | 135 | def map( |
214 | 136 | self, |
215 | 137 | task: Union[ |
@@ -253,34 +175,6 @@ def map( |
253 | 175 | ) |
254 | 176 | return self |
255 | 177 |
|
256 | | - @overload |
257 | | - def filter( |
258 | | - self: "Pipeline[_FuncParams, _ReturnType]", |
259 | | - task: Union[ |
260 | | - AsyncKicker[Any, Coroutine[Any, Any, bool]], |
261 | | - AsyncKicker[Any, "CoroutineType[Any, Any, bool]"], |
262 | | - AsyncTaskiqDecoratedTask[Any, Coroutine[Any, Any, bool]], |
263 | | - AsyncTaskiqDecoratedTask[Any, "CoroutineType[Any, Any, bool]"], |
264 | | - ], |
265 | | - param_name: Optional[str] = None, |
266 | | - skip_errors: bool = False, |
267 | | - check_interval: float = 0.5, |
268 | | - **additional_kwargs: Any, |
269 | | - ) -> "Pipeline[_FuncParams, _ReturnType]": ... |
270 | | - |
271 | | - @overload |
272 | | - def filter( |
273 | | - self: "Pipeline[_FuncParams, _ReturnType]", |
274 | | - task: Union[ |
275 | | - AsyncKicker[Any, bool], |
276 | | - AsyncTaskiqDecoratedTask[Any, bool], |
277 | | - ], |
278 | | - param_name: Optional[str] = None, |
279 | | - skip_errors: bool = False, |
280 | | - check_interval: float = 0.5, |
281 | | - **additional_kwargs: Any, |
282 | | - ) -> "Pipeline[_FuncParams, _ReturnType]": ... |
283 | | - |
284 | 178 | def filter( |
285 | 179 | self, |
286 | 180 | task: Union[ |
|
0 commit comments