1
1
import sys
2
2
from typing import Any
3
+ from typing import Callable
3
4
from typing import Iterable
4
5
from typing import Iterator
5
6
from typing import List
@@ -19,6 +20,8 @@ _TChunkedIteratorResult = TypeVar(
19
20
"_TChunkedIteratorResult" , bound = ChunkedIteratorResult
20
21
)
21
22
23
+ def tuplegetter (* indexes : Any ) -> Callable [[Any ], Tuple [Any , ...]]: ...
24
+
22
25
class ResultMetaData :
23
26
@property
24
27
def keys (self ) -> RMKeyView : ...
@@ -50,7 +53,7 @@ class _WithKeys:
50
53
def keys (self ) -> RMKeyView : ...
51
54
52
55
class Result (_WithKeys , ResultInternal ):
53
- def __init__ (self , cursor_metadata : Any ) -> None : ...
56
+ def __init__ (self , cursor_metadata : ResultMetaData ) -> None : ...
54
57
def yield_per (self : _TResult , num : Any ) -> _TResult : ...
55
58
def unique (
56
59
self : _TResult , strategy : Optional [object ] = ...
@@ -134,7 +137,10 @@ class IteratorResult(Result):
134
137
iterator : Any = ...
135
138
raw : Any = ...
136
139
def __init__ (
137
- self , cursor_metadata : Any , iterator : Any , raw : Optional [Any ] = ...
140
+ self ,
141
+ cursor_metadata : ResultMetaData ,
142
+ iterator : Any ,
143
+ raw : Optional [Any ] = ...,
138
144
) -> None : ...
139
145
140
146
def null_result () -> IteratorResult : ...
@@ -146,7 +152,7 @@ class ChunkedIteratorResult(IteratorResult):
146
152
dynamic_yield_per : Any = ...
147
153
def __init__ (
148
154
self ,
149
- cursor_metadata : Any ,
155
+ cursor_metadata : ResultMetaData ,
150
156
chunks : Any ,
151
157
source_supports_scalars : bool = ...,
152
158
raw : Optional [Any ] = ...,
@@ -158,5 +164,7 @@ class ChunkedIteratorResult(IteratorResult):
158
164
159
165
class MergedResult (IteratorResult ):
160
166
closed : bool = ...
161
- def __init__ (self , cursor_metadata : Any , results : Result ) -> None : ...
167
+ def __init__ (
168
+ self , cursor_metadata : ResultMetaData , results : Result
169
+ ) -> None : ...
162
170
def close (self ) -> None : ...
0 commit comments