nodepyx 1.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (184) hide show
  1. package/LICENSE +22 -0
  2. package/README.md +399 -0
  3. package/binding.gyp +73 -0
  4. package/dist/core/PyCallable.d.ts +65 -0
  5. package/dist/core/PyCallable.d.ts.map +1 -0
  6. package/dist/core/PyCallable.js +109 -0
  7. package/dist/core/PyCallable.js.map +1 -0
  8. package/dist/core/PyContext.d.ts +76 -0
  9. package/dist/core/PyContext.d.ts.map +1 -0
  10. package/dist/core/PyContext.js +228 -0
  11. package/dist/core/PyContext.js.map +1 -0
  12. package/dist/core/PyIterator.d.ts +84 -0
  13. package/dist/core/PyIterator.d.ts.map +1 -0
  14. package/dist/core/PyIterator.js +243 -0
  15. package/dist/core/PyIterator.js.map +1 -0
  16. package/dist/core/PyModule.d.ts +55 -0
  17. package/dist/core/PyModule.d.ts.map +1 -0
  18. package/dist/core/PyModule.js +172 -0
  19. package/dist/core/PyModule.js.map +1 -0
  20. package/dist/core/PyProxy.d.ts +65 -0
  21. package/dist/core/PyProxy.d.ts.map +1 -0
  22. package/dist/core/PyProxy.js +483 -0
  23. package/dist/core/PyProxy.js.map +1 -0
  24. package/dist/core/PyRuntime.d.ts +105 -0
  25. package/dist/core/PyRuntime.d.ts.map +1 -0
  26. package/dist/core/PyRuntime.js +438 -0
  27. package/dist/core/PyRuntime.js.map +1 -0
  28. package/dist/env/CondaManager.d.ts +118 -0
  29. package/dist/env/CondaManager.d.ts.map +1 -0
  30. package/dist/env/CondaManager.js +401 -0
  31. package/dist/env/CondaManager.js.map +1 -0
  32. package/dist/env/PackageInstaller.d.ts +233 -0
  33. package/dist/env/PackageInstaller.d.ts.map +1 -0
  34. package/dist/env/PackageInstaller.js +609 -0
  35. package/dist/env/PackageInstaller.js.map +1 -0
  36. package/dist/env/PythonDetector.d.ts +103 -0
  37. package/dist/env/PythonDetector.d.ts.map +1 -0
  38. package/dist/env/PythonDetector.js +381 -0
  39. package/dist/env/PythonDetector.js.map +1 -0
  40. package/dist/env/VenvManager.d.ts +117 -0
  41. package/dist/env/VenvManager.d.ts.map +1 -0
  42. package/dist/env/VenvManager.js +331 -0
  43. package/dist/env/VenvManager.js.map +1 -0
  44. package/dist/index.d.ts +169 -0
  45. package/dist/index.d.ts.map +1 -0
  46. package/dist/index.js +393 -0
  47. package/dist/index.js.map +1 -0
  48. package/dist/plugins/Plugin.interface.d.ts +41 -0
  49. package/dist/plugins/Plugin.interface.d.ts.map +1 -0
  50. package/dist/plugins/Plugin.interface.js +12 -0
  51. package/dist/plugins/Plugin.interface.js.map +1 -0
  52. package/dist/plugins/PluginManager.d.ts +26 -0
  53. package/dist/plugins/PluginManager.d.ts.map +1 -0
  54. package/dist/plugins/PluginManager.js +174 -0
  55. package/dist/plugins/PluginManager.js.map +1 -0
  56. package/dist/plugins/builtin/NumpyPlugin.d.ts +17 -0
  57. package/dist/plugins/builtin/NumpyPlugin.d.ts.map +1 -0
  58. package/dist/plugins/builtin/NumpyPlugin.js +41 -0
  59. package/dist/plugins/builtin/NumpyPlugin.js.map +1 -0
  60. package/dist/plugins/builtin/PandasPlugin.d.ts +19 -0
  61. package/dist/plugins/builtin/PandasPlugin.d.ts.map +1 -0
  62. package/dist/plugins/builtin/PandasPlugin.js +57 -0
  63. package/dist/plugins/builtin/PandasPlugin.js.map +1 -0
  64. package/dist/plugins/builtin/TorchPlugin.d.ts +23 -0
  65. package/dist/plugins/builtin/TorchPlugin.d.ts.map +1 -0
  66. package/dist/plugins/builtin/TorchPlugin.js +50 -0
  67. package/dist/plugins/builtin/TorchPlugin.js.map +1 -0
  68. package/dist/plugins/index.d.ts +7 -0
  69. package/dist/plugins/index.d.ts.map +1 -0
  70. package/dist/plugins/index.js +12 -0
  71. package/dist/plugins/index.js.map +1 -0
  72. package/dist/serialization/DataFrameBridge.d.ts +141 -0
  73. package/dist/serialization/DataFrameBridge.d.ts.map +1 -0
  74. package/dist/serialization/DataFrameBridge.js +355 -0
  75. package/dist/serialization/DataFrameBridge.js.map +1 -0
  76. package/dist/serialization/MsgPackSerializer.d.ts +45 -0
  77. package/dist/serialization/MsgPackSerializer.d.ts.map +1 -0
  78. package/dist/serialization/MsgPackSerializer.js +242 -0
  79. package/dist/serialization/MsgPackSerializer.js.map +1 -0
  80. package/dist/serialization/NumpyBridge.d.ts +96 -0
  81. package/dist/serialization/NumpyBridge.d.ts.map +1 -0
  82. package/dist/serialization/NumpyBridge.js +323 -0
  83. package/dist/serialization/NumpyBridge.js.map +1 -0
  84. package/dist/serialization/Serializer.d.ts +78 -0
  85. package/dist/serialization/Serializer.d.ts.map +1 -0
  86. package/dist/serialization/Serializer.js +281 -0
  87. package/dist/serialization/Serializer.js.map +1 -0
  88. package/dist/types/PythonTypeMapper.d.ts +87 -0
  89. package/dist/types/PythonTypeMapper.d.ts.map +1 -0
  90. package/dist/types/PythonTypeMapper.js +449 -0
  91. package/dist/types/PythonTypeMapper.js.map +1 -0
  92. package/dist/types/StubCache.d.ts +109 -0
  93. package/dist/types/StubCache.d.ts.map +1 -0
  94. package/dist/types/StubCache.js +333 -0
  95. package/dist/types/StubCache.js.map +1 -0
  96. package/dist/types/TypeGenerator.d.ts +139 -0
  97. package/dist/types/TypeGenerator.d.ts.map +1 -0
  98. package/dist/types/TypeGenerator.js +372 -0
  99. package/dist/types/TypeGenerator.js.map +1 -0
  100. package/dist/types/addon.d.ts +114 -0
  101. package/dist/types/addon.d.ts.map +1 -0
  102. package/dist/types/addon.js +32 -0
  103. package/dist/types/addon.js.map +1 -0
  104. package/dist/types/config.d.ts +175 -0
  105. package/dist/types/config.d.ts.map +1 -0
  106. package/dist/types/config.js +35 -0
  107. package/dist/types/config.js.map +1 -0
  108. package/dist/types/index.d.ts +10 -0
  109. package/dist/types/index.d.ts.map +1 -0
  110. package/dist/types/index.js +12 -0
  111. package/dist/types/index.js.map +1 -0
  112. package/dist/types/python.d.ts +235 -0
  113. package/dist/types/python.d.ts.map +1 -0
  114. package/dist/types/python.js +7 -0
  115. package/dist/types/python.js.map +1 -0
  116. package/dist/utils/ErrorTranslator.d.ts +83 -0
  117. package/dist/utils/ErrorTranslator.d.ts.map +1 -0
  118. package/dist/utils/ErrorTranslator.js +210 -0
  119. package/dist/utils/ErrorTranslator.js.map +1 -0
  120. package/dist/utils/Logger.d.ts +27 -0
  121. package/dist/utils/Logger.d.ts.map +1 -0
  122. package/dist/utils/Logger.js +115 -0
  123. package/dist/utils/Logger.js.map +1 -0
  124. package/dist/utils/MemoryMonitor.d.ts +44 -0
  125. package/dist/utils/MemoryMonitor.d.ts.map +1 -0
  126. package/dist/utils/MemoryMonitor.js +143 -0
  127. package/dist/utils/MemoryMonitor.js.map +1 -0
  128. package/package.json +177 -0
  129. package/python/error_handler.py +433 -0
  130. package/python/nodepyx_runtime.py +575 -0
  131. package/python/serializer.py +379 -0
  132. package/python/type_inspector.py +288 -0
  133. package/scripts/build-native.js +68 -0
  134. package/scripts/download-prebuilds.js +99 -0
  135. package/scripts/generate-stubs.js +405 -0
  136. package/scripts/install.js +260 -0
  137. package/src/core/PyCallable.ts +137 -0
  138. package/src/core/PyContext.ts +296 -0
  139. package/src/core/PyIterator.ts +294 -0
  140. package/src/core/PyModule.ts +194 -0
  141. package/src/core/PyProxy.ts +605 -0
  142. package/src/core/PyRuntime.ts +504 -0
  143. package/src/env/CondaManager.ts +451 -0
  144. package/src/env/PackageInstaller.ts +738 -0
  145. package/src/env/PythonDetector.ts +414 -0
  146. package/src/env/VenvManager.ts +396 -0
  147. package/src/index.ts +425 -0
  148. package/src/native/gil_guard.cpp +26 -0
  149. package/src/native/gil_guard.h +175 -0
  150. package/src/native/nodepyx_addon.cpp +886 -0
  151. package/src/native/python_bridge.cpp +790 -0
  152. package/src/native/python_bridge.h +257 -0
  153. package/src/native/thread_pool.cpp +336 -0
  154. package/src/native/thread_pool.h +175 -0
  155. package/src/native/type_converter.cpp +901 -0
  156. package/src/native/type_converter.h +272 -0
  157. package/src/nextjs/PyProvider.tsx +123 -0
  158. package/src/nextjs/index.ts +21 -0
  159. package/src/nextjs/usePython.ts +106 -0
  160. package/src/nextjs/withnodepyx.ts +88 -0
  161. package/src/plugins/Plugin.interface.ts +51 -0
  162. package/src/plugins/PluginManager.ts +155 -0
  163. package/src/plugins/builtin/NumpyPlugin.ts +36 -0
  164. package/src/plugins/builtin/PandasPlugin.ts +49 -0
  165. package/src/plugins/builtin/TorchPlugin.ts +56 -0
  166. package/src/plugins/index.ts +7 -0
  167. package/src/serialization/DataFrameBridge.ts +398 -0
  168. package/src/serialization/MsgPackSerializer.ts +220 -0
  169. package/src/serialization/NumpyBridge.ts +332 -0
  170. package/src/serialization/Serializer.ts +320 -0
  171. package/src/types/PythonTypeMapper.ts +495 -0
  172. package/src/types/StubCache.ts +340 -0
  173. package/src/types/TypeGenerator.ts +491 -0
  174. package/src/types/addon.ts +170 -0
  175. package/src/types/config.ts +226 -0
  176. package/src/types/index.ts +55 -0
  177. package/src/types/python.ts +309 -0
  178. package/src/types/stubs/numpy.d.ts +441 -0
  179. package/src/types/stubs/pandas.d.ts +575 -0
  180. package/src/types/stubs/sklearn.d.ts +728 -0
  181. package/src/types/stubs/torch.d.ts +694 -0
  182. package/src/utils/ErrorTranslator.ts +220 -0
  183. package/src/utils/Logger.ts +119 -0
  184. package/src/utils/MemoryMonitor.ts +175 -0
@@ -0,0 +1,76 @@
1
+ /**
2
+ * nodepyx — PyContext
3
+ * Manages an isolated Python execution context (namespace).
4
+ * Useful for running code in isolation without polluting the global namespace.
5
+ */
6
+ import type { NativeAddon } from '../types/addon';
7
+ import { PyProxy } from './PyProxy';
8
+ /**
9
+ * PyContext — an isolated Python execution environment.
10
+ *
11
+ * @example
12
+ * ```typescript
13
+ * const ctx = await PyContext.create(addon);
14
+ *
15
+ * // Run code in isolation
16
+ * await ctx.exec('x = 42');
17
+ * await ctx.exec('y = x * 2');
18
+ * const result = await ctx.eval('x + y');
19
+ * console.log(result); // 126
20
+ *
21
+ * // Import into context
22
+ * const pd = await ctx.import('pandas');
23
+ * await ctx.exec("df = pd.DataFrame({'a': [1,2,3]})");
24
+ * const df = await ctx.get('df');
25
+ * ```
26
+ */
27
+ export declare class PyContext {
28
+ private readonly _addon;
29
+ private readonly _namespacePyId;
30
+ private _variables;
31
+ private _destroyed;
32
+ private constructor();
33
+ /**
34
+ * Create a new isolated Python context.
35
+ */
36
+ static create(addon: NativeAddon): Promise<PyContext>;
37
+ /**
38
+ * Execute Python code in this context.
39
+ */
40
+ exec(code: string): Promise<void>;
41
+ /**
42
+ * Evaluate a Python expression in this context and return the result.
43
+ */
44
+ eval(expression: string): Promise<unknown>;
45
+ /**
46
+ * Import a module into this context.
47
+ */
48
+ import(moduleName: string): Promise<PyProxy>;
49
+ /**
50
+ * Get a variable from the context namespace.
51
+ */
52
+ get(name: string): Promise<unknown>;
53
+ /**
54
+ * Set a variable in the context namespace.
55
+ */
56
+ set(name: string, value: unknown): Promise<void>;
57
+ /**
58
+ * Check if a name is defined in the context.
59
+ */
60
+ has(name: string): Promise<boolean>;
61
+ /**
62
+ * Delete a variable from the context.
63
+ */
64
+ delete(name: string): Promise<void>;
65
+ /**
66
+ * Clear all variables in the context.
67
+ */
68
+ clear(): Promise<void>;
69
+ /**
70
+ * Destroy the context and free resources.
71
+ */
72
+ destroy(): Promise<void>;
73
+ private _ensureNotDestroyed;
74
+ [Symbol.asyncDispose](): Promise<void>;
75
+ }
76
+ //# sourceMappingURL=PyContext.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"PyContext.d.ts","sourceRoot":"","sources":["../../src/core/PyContext.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAElD,OAAO,EAAE,OAAO,EAAoB,MAAM,WAAW,CAAC;AAMtD;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,SAAS;IACpB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAc;IACrC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAa;IAC5C,OAAO,CAAC,UAAU,CAA0B;IAC5C,OAAO,CAAC,UAAU,CAAS;IAE3B,OAAO;IAMP;;OAEG;WACU,MAAM,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC;IA2B3D;;OAEG;IACG,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBvC;;OAEG;IACG,IAAI,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAyChD;;OAEG;IACG,MAAM,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IA+BlD;;OAEG;IACG,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAmDzC;;OAEG;IACG,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IActD;;OAEG;IACG,GAAG,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAUzC;;OAEG;IACG,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAMzC;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAK5B;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ9B,OAAO,CAAC,mBAAmB;IAM3B,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC;CAGvC"}
@@ -0,0 +1,228 @@
1
+ "use strict";
2
+ /**
3
+ * nodepyx — PyContext
4
+ * Manages an isolated Python execution context (namespace).
5
+ * Useful for running code in isolation without polluting the global namespace.
6
+ */
7
+ Object.defineProperty(exports, "__esModule", { value: true });
8
+ exports.PyContext = void 0;
9
+ const PyProxy_1 = require("./PyProxy");
10
+ const ErrorTranslator_1 = require("../utils/ErrorTranslator");
11
+ const Logger_1 = require("../utils/Logger");
12
+ const logger = new Logger_1.Logger('PyContext');
13
+ /**
14
+ * PyContext — an isolated Python execution environment.
15
+ *
16
+ * @example
17
+ * ```typescript
18
+ * const ctx = await PyContext.create(addon);
19
+ *
20
+ * // Run code in isolation
21
+ * await ctx.exec('x = 42');
22
+ * await ctx.exec('y = x * 2');
23
+ * const result = await ctx.eval('x + y');
24
+ * console.log(result); // 126
25
+ *
26
+ * // Import into context
27
+ * const pd = await ctx.import('pandas');
28
+ * await ctx.exec("df = pd.DataFrame({'a': [1,2,3]})");
29
+ * const df = await ctx.get('df');
30
+ * ```
31
+ */
32
+ class PyContext {
33
+ _addon;
34
+ _namespacePyId;
35
+ _variables;
36
+ _destroyed = false;
37
+ constructor(addon, namespacePyId) {
38
+ this._addon = addon;
39
+ this._namespacePyId = namespacePyId;
40
+ this._variables = new Map();
41
+ }
42
+ /**
43
+ * Create a new isolated Python context.
44
+ */
45
+ static async create(addon) {
46
+ // Create a fresh namespace dict in Python
47
+ const result = await addon.evalPython('{}');
48
+ if (!result.success) {
49
+ if (result.error) {
50
+ throw (0, ErrorTranslator_1.translatePythonError)({
51
+ type: result.error.type,
52
+ message: result.error.message,
53
+ traceback: result.error.traceback || '',
54
+ });
55
+ }
56
+ throw new Error('Failed to create Python context');
57
+ }
58
+ if (!result.objectId) {
59
+ throw new Error('Context namespace object ID not returned');
60
+ }
61
+ return new PyContext(addon, result.objectId);
62
+ }
63
+ /**
64
+ * Execute Python code in this context.
65
+ */
66
+ async exec(code) {
67
+ this._ensureNotDestroyed();
68
+ const result = await this._addon.runPythonCode(code);
69
+ if (!result.success && result.error) {
70
+ throw (0, ErrorTranslator_1.translatePythonError)({
71
+ type: result.error.type,
72
+ message: result.error.message,
73
+ traceback: result.error.traceback || '',
74
+ });
75
+ }
76
+ }
77
+ /**
78
+ * Evaluate a Python expression in this context and return the result.
79
+ */
80
+ async eval(expression) {
81
+ this._ensureNotDestroyed();
82
+ const result = await this._addon.evalPython(expression);
83
+ if (!result.success) {
84
+ if (result.error) {
85
+ throw (0, ErrorTranslator_1.translatePythonError)({
86
+ type: result.error.type,
87
+ message: result.error.message,
88
+ traceback: result.error.traceback || '',
89
+ });
90
+ }
91
+ throw new Error('Evaluation failed');
92
+ }
93
+ if (result.isObject && result.objectId) {
94
+ return PyProxy_1.PyProxy._createFromObjectId(result.objectId, [], this._addon);
95
+ }
96
+ if (result.resultJson !== undefined) {
97
+ try {
98
+ return JSON.parse(result.resultJson);
99
+ }
100
+ catch {
101
+ return result.resultJson;
102
+ }
103
+ }
104
+ return null;
105
+ }
106
+ /**
107
+ * Import a module into this context.
108
+ */
109
+ async import(moduleName) {
110
+ this._ensureNotDestroyed();
111
+ const result = await this._addon.importModule(moduleName);
112
+ if (!result.success) {
113
+ if (result.error) {
114
+ throw (0, ErrorTranslator_1.translatePythonError)({
115
+ type: result.error.type,
116
+ message: result.error.message,
117
+ traceback: result.error.traceback || '',
118
+ });
119
+ }
120
+ throw new Error(`Failed to import module '${moduleName}'`);
121
+ }
122
+ if (!result.objectId) {
123
+ throw new Error('Module object ID not returned');
124
+ }
125
+ return PyProxy_1.PyProxy._createFromObjectId(result.objectId, [], this._addon);
126
+ }
127
+ /**
128
+ * Get a variable from the context namespace.
129
+ */
130
+ async get(name) {
131
+ this._ensureNotDestroyed();
132
+ const pyId = this._variables.get(name);
133
+ if (pyId) {
134
+ return PyProxy_1.PyProxy._createFromObjectId(pyId, [], this._addon);
135
+ }
136
+ // Try to get from Python global namespace
137
+ const result = await this._addon.evalPython(name);
138
+ if (!result.success) {
139
+ if (result.error?.type === 'NameError') {
140
+ return undefined;
141
+ }
142
+ if (result.error) {
143
+ throw (0, ErrorTranslator_1.translatePythonError)({
144
+ type: result.error.type,
145
+ message: result.error.message,
146
+ traceback: result.error.traceback || '',
147
+ });
148
+ }
149
+ return undefined;
150
+ }
151
+ if (result.isObject && result.objectId) {
152
+ return PyProxy_1.PyProxy._createFromObjectId(result.objectId, [], this._addon);
153
+ }
154
+ if (result.resultJson !== undefined) {
155
+ try {
156
+ return JSON.parse(result.resultJson);
157
+ }
158
+ catch {
159
+ return result.resultJson;
160
+ }
161
+ }
162
+ return undefined;
163
+ }
164
+ /**
165
+ * Set a variable in the context namespace.
166
+ */
167
+ async set(name, value) {
168
+ this._ensureNotDestroyed();
169
+ if (value instanceof PyProxy_1.PyProxy) {
170
+ const internal = value[PyProxy_1.PYPROXY_INTERNAL];
171
+ this._variables.set(name, internal.objectId);
172
+ return;
173
+ }
174
+ // For primitives and serializable values: set via exec
175
+ const serialized = JSON.stringify(value);
176
+ await this.exec(`${name} = __import__('json').loads('${serialized.replace(/'/g, "\\'")}')`);
177
+ }
178
+ /**
179
+ * Check if a name is defined in the context.
180
+ */
181
+ async has(name) {
182
+ this._ensureNotDestroyed();
183
+ try {
184
+ const val = await this.get(name);
185
+ return val !== undefined;
186
+ }
187
+ catch {
188
+ return false;
189
+ }
190
+ }
191
+ /**
192
+ * Delete a variable from the context.
193
+ */
194
+ async delete(name) {
195
+ this._ensureNotDestroyed();
196
+ this._variables.delete(name);
197
+ await this.exec(`del ${name}`).catch(() => { }); // ignore if not defined
198
+ }
199
+ /**
200
+ * Clear all variables in the context.
201
+ */
202
+ async clear() {
203
+ this._ensureNotDestroyed();
204
+ this._variables.clear();
205
+ }
206
+ /**
207
+ * Destroy the context and free resources.
208
+ */
209
+ async destroy() {
210
+ if (this._destroyed) {
211
+ return;
212
+ }
213
+ this._destroyed = true;
214
+ this._variables.clear();
215
+ this._addon.decRef(this._namespacePyId);
216
+ logger.debug('PyContext destroyed');
217
+ }
218
+ _ensureNotDestroyed() {
219
+ if (this._destroyed) {
220
+ throw new Error('PyContext has been destroyed');
221
+ }
222
+ }
223
+ [Symbol.asyncDispose]() {
224
+ return this.destroy();
225
+ }
226
+ }
227
+ exports.PyContext = PyContext;
228
+ //# sourceMappingURL=PyContext.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"PyContext.js","sourceRoot":"","sources":["../../src/core/PyContext.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAIH,uCAAsD;AACtD,8DAAgE;AAChE,4CAAyC;AAEzC,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC,WAAW,CAAC,CAAC;AAEvC;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAa,SAAS;IACH,MAAM,CAAc;IACpB,cAAc,CAAa;IACpC,UAAU,CAA0B;IACpC,UAAU,GAAG,KAAK,CAAC;IAE3B,YAAoB,KAAkB,EAAE,aAAyB;QAC/D,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,UAAU,GAAG,IAAI,GAAG,EAAE,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAkB;QACpC,0CAA0C;QAC1C,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,IAAI,CAKzC,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAA,sCAAoB,EAAC;oBACzB,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;oBACvB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;oBAC7B,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE;iBACxC,CAAC,CAAC;YACL,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,IAAI,SAAS,CAAC,KAAK,EAAE,MAAM,CAAC,QAAsB,CAAC,CAAC;IAC7D,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,IAAY;QACrB,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAGlD,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YACpC,MAAM,IAAA,sCAAoB,EAAC;gBACzB,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;gBACvB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;gBAC7B,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE;aACxC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,UAAkB;QAC3B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAMrD,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAA,sCAAoB,EAAC;oBACzB,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;oBACvB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;oBAC7B,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE;iBACxC,CAAC,CAAC;YACL,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACvC,OAAO,iBAAO,CAAC,mBAAmB,CAChC,MAAM,CAAC,QAAsB,EAC7B,EAAE,EACF,IAAI,CAAC,MAAM,CACZ,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACpC,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACvC,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,MAAM,CAAC,UAAU,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,UAAkB;QAC7B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,UAAU,CAIvD,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAA,sCAAoB,EAAC;oBACzB,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;oBACvB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;oBAC7B,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE;iBACxC,CAAC,CAAC;YACL,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,4BAA4B,UAAU,GAAG,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QAED,OAAO,iBAAO,CAAC,mBAAmB,CAChC,MAAM,CAAC,QAAsB,EAC7B,EAAE,EACF,IAAI,CAAC,MAAM,CACZ,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,GAAG,CAAC,IAAY;QACpB,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3B,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEvC,IAAI,IAAI,EAAE,CAAC;YACT,OAAO,iBAAO,CAAC,mBAAmB,CAAC,IAAI,EAAE,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5D,CAAC;QAED,0CAA0C;QAC1C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAM/C,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,MAAM,CAAC,KAAK,EAAE,IAAI,KAAK,WAAW,EAAE,CAAC;gBACvC,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAA,sCAAoB,EAAC;oBACzB,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;oBACvB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;oBAC7B,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE;iBACxC,CAAC,CAAC;YACL,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACvC,OAAO,iBAAO,CAAC,mBAAmB,CAChC,MAAM,CAAC,QAAsB,EAC7B,EAAE,EACF,IAAI,CAAC,MAAM,CACZ,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YACpC,IAAI,CAAC;gBACH,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACvC,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,MAAM,CAAC,UAAU,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,GAAG,CAAC,IAAY,EAAE,KAAc;QACpC,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3B,IAAI,KAAK,YAAY,iBAAO,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,KAAK,CAAC,0BAAgB,CAAC,CAAC;YACzC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAC7C,OAAO;QACT,CAAC;QAED,uDAAuD;QACvD,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,gCAAgC,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IAC9F,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,GAAG,CAAC,IAAY;QACpB,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACjC,OAAO,GAAG,KAAK,SAAS,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,IAAY;QACvB,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC3B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC7B,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;IAC1E,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC3B,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAAA,OAAO;QAAA,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,cAAwB,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACtC,CAAC;IAEO,mBAAmB;QACzB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,CAAC,MAAM,CAAC,YAAY,CAAC;QACnB,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;CACF;AArQD,8BAqQC"}
@@ -0,0 +1,84 @@
1
+ /**
2
+ * nodepyx — PyIterator
3
+ * Wraps a Python iterator for use with JavaScript's for await...of.
4
+ * Provides both async iteration and manual .next() calls.
5
+ */
6
+ import type { NativeAddon } from '../types/addon';
7
+ import type { PyObjectId } from '../types/python';
8
+ import { PyProxy } from './PyProxy';
9
+ export interface PyIteratorOptions {
10
+ /** Maximum number of items to yield (0 = no limit) */
11
+ limit?: number;
12
+ /** Whether to wrap Python objects in PyProxy */
13
+ wrapObjects?: boolean;
14
+ /** Timeout per .next() call (ms) */
15
+ timeout?: number;
16
+ }
17
+ /**
18
+ * PyIterator — manages a Python iterator lifecycle.
19
+ *
20
+ * @example
21
+ * ```typescript
22
+ * const iterator = await PyIterator.create(pyList, addon);
23
+ *
24
+ * for await (const item of iterator) {
25
+ * console.log(item);
26
+ * }
27
+ *
28
+ * // Or manually:
29
+ * const { value, done } = await iterator.next();
30
+ * ```
31
+ */
32
+ export declare class PyIterator<T = unknown> implements AsyncIterable<T>, AsyncIterator<T> {
33
+ private readonly _iteratorId;
34
+ private readonly _addon;
35
+ private readonly _objectId;
36
+ private readonly _options;
37
+ private _done;
38
+ private _position;
39
+ private _destroyed;
40
+ private constructor();
41
+ /**
42
+ * Create a PyIterator from a PyProxy.
43
+ */
44
+ static create(source: PyProxy, options?: PyIteratorOptions): Promise<PyIterator>;
45
+ /**
46
+ * Create a PyIterator from an object ID and iterator ID directly.
47
+ */
48
+ static fromIds(iteratorId: number, objectId: PyObjectId, addon: NativeAddon, options?: PyIteratorOptions): PyIterator;
49
+ /** Current position in iteration */
50
+ get position(): number;
51
+ /** Whether iteration is complete */
52
+ get isDone(): boolean;
53
+ next(): Promise<IteratorResult<T>>;
54
+ return(value?: T): Promise<IteratorResult<T>>;
55
+ throw(error?: unknown): Promise<IteratorResult<T>>;
56
+ [Symbol.asyncIterator](): AsyncIterator<T>;
57
+ /**
58
+ * Collect all items into an array.
59
+ * Use with care for large iterators!
60
+ */
61
+ toArray(maxItems?: number): Promise<T[]>;
62
+ /**
63
+ * Process items in batches.
64
+ */
65
+ batch(batchSize: number): AsyncGenerator<T[]>;
66
+ /**
67
+ * Map items through a transform function.
68
+ */
69
+ map<U>(fn: (item: T, index: number) => U | Promise<U>): AsyncGenerator<U>;
70
+ /**
71
+ * Filter items.
72
+ */
73
+ filter(fn: (item: T, index: number) => boolean | Promise<boolean>): AsyncGenerator<T>;
74
+ /**
75
+ * Take first N items.
76
+ */
77
+ take(n: number): Promise<T[]>;
78
+ /**
79
+ * Destroy the underlying Python iterator (frees resources).
80
+ * Called automatically when iteration completes.
81
+ */
82
+ destroy(): Promise<void>;
83
+ }
84
+ //# sourceMappingURL=PyIterator.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"PyIterator.d.ts","sourceRoot":"","sources":["../../src/core/PyIterator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAElD,OAAO,EAAE,OAAO,EAAoB,MAAM,WAAW,CAAC;AAKtD,MAAM,WAAW,iBAAiB;IAChC,sDAAsD;IACtD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,gDAAgD;IAChD,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,oCAAoC;IACpC,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;;;;;;;;;;;;;GAcG;AACH,qBAAa,UAAU,CAAC,CAAC,GAAG,OAAO,CAAE,YAAW,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,CAAC;IAChF,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAc;IACrC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAa;IACvC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAoB;IAC7C,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,UAAU,CAAS;IAE3B,OAAO;IAYP;;OAEG;WACU,MAAM,CACjB,MAAM,EAAE,OAAO,EACf,OAAO,CAAC,EAAE,iBAAiB,GAC1B,OAAO,CAAC,UAAU,CAAC;IAgCtB;;OAEG;IACH,MAAM,CAAC,OAAO,CACZ,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,UAAU,EACpB,KAAK,EAAE,WAAW,EAClB,OAAO,CAAC,EAAE,iBAAiB,GAC1B,UAAU;IAIb,oCAAoC;IACpC,IAAI,QAAQ,IAAI,MAAM,CAErB;IAED,oCAAoC;IACpC,IAAI,MAAM,IAAI,OAAO,CAEpB;IAIK,IAAI,IAAI,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IA4ElC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAK7C,KAAK,CAAC,KAAK,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;IAOxD,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC;IAM1C;;;OAGG;IACG,OAAO,CAAC,QAAQ,SAAS,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAS9C;;OAEG;IACI,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,cAAc,CAAC,CAAC,EAAE,CAAC;IAcpD;;OAEG;IACI,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC;IAOhF;;OAEG;IACI,MAAM,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC;IAO5F;;OAEG;IACG,IAAI,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAUnC;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAU/B"}
@@ -0,0 +1,243 @@
1
+ "use strict";
2
+ /**
3
+ * nodepyx — PyIterator
4
+ * Wraps a Python iterator for use with JavaScript's for await...of.
5
+ * Provides both async iteration and manual .next() calls.
6
+ */
7
+ Object.defineProperty(exports, "__esModule", { value: true });
8
+ exports.PyIterator = void 0;
9
+ const ErrorTranslator_1 = require("../utils/ErrorTranslator");
10
+ const PyProxy_1 = require("./PyProxy");
11
+ const Logger_1 = require("../utils/Logger");
12
+ const logger = new Logger_1.Logger('PyIterator');
13
+ /**
14
+ * PyIterator — manages a Python iterator lifecycle.
15
+ *
16
+ * @example
17
+ * ```typescript
18
+ * const iterator = await PyIterator.create(pyList, addon);
19
+ *
20
+ * for await (const item of iterator) {
21
+ * console.log(item);
22
+ * }
23
+ *
24
+ * // Or manually:
25
+ * const { value, done } = await iterator.next();
26
+ * ```
27
+ */
28
+ class PyIterator {
29
+ _iteratorId;
30
+ _addon;
31
+ _objectId;
32
+ _options;
33
+ _done = false;
34
+ _position = 0;
35
+ _destroyed = false;
36
+ constructor(iteratorId, objectId, addon, options = {}) {
37
+ this._iteratorId = iteratorId;
38
+ this._objectId = objectId;
39
+ this._addon = addon;
40
+ this._options = options;
41
+ }
42
+ /**
43
+ * Create a PyIterator from a PyProxy.
44
+ */
45
+ static async create(source, options) {
46
+ const internal = source[PyProxy_1.PYPROXY_INTERNAL];
47
+ const addon = internal.addon;
48
+ const result = await addon.createIterator(internal.objectId, internal.path);
49
+ if (!result.success) {
50
+ if (result.error) {
51
+ throw (0, ErrorTranslator_1.translatePythonError)({
52
+ type: result.error.type,
53
+ message: result.error.message,
54
+ traceback: result.error.traceback || '',
55
+ });
56
+ }
57
+ throw new Error('Failed to create Python iterator');
58
+ }
59
+ let iteratorId;
60
+ try {
61
+ const parsed = JSON.parse(result.resultJson || '{}');
62
+ if (!parsed.iteratorId) {
63
+ throw new Error('no iteratorId');
64
+ }
65
+ iteratorId = parsed.iteratorId;
66
+ }
67
+ catch {
68
+ throw new Error('Failed to parse iterator ID from: ' + result.resultJson);
69
+ }
70
+ return new PyIterator(iteratorId, internal.objectId, addon, options);
71
+ }
72
+ /**
73
+ * Create a PyIterator from an object ID and iterator ID directly.
74
+ */
75
+ static fromIds(iteratorId, objectId, addon, options) {
76
+ return new PyIterator(iteratorId, objectId, addon, options);
77
+ }
78
+ /** Current position in iteration */
79
+ get position() {
80
+ return this._position;
81
+ }
82
+ /** Whether iteration is complete */
83
+ get isDone() {
84
+ return this._done;
85
+ }
86
+ // ─── AsyncIterator Protocol ────────────────────────────────────────────────
87
+ async next() {
88
+ if (this._done || this._destroyed) {
89
+ return { value: undefined, done: true };
90
+ }
91
+ // Check limit
92
+ if (this._options.limit && this._position >= this._options.limit) {
93
+ await this.destroy();
94
+ return { value: undefined, done: true };
95
+ }
96
+ try {
97
+ const result = await this._addon.iteratorNext(this._iteratorId);
98
+ if (!result.success) {
99
+ if (result.error) {
100
+ throw (0, ErrorTranslator_1.translatePythonError)({
101
+ type: result.error.type,
102
+ message: result.error.message,
103
+ traceback: result.error.traceback || '',
104
+ });
105
+ }
106
+ this._done = true;
107
+ return { value: undefined, done: true };
108
+ }
109
+ let parsed;
110
+ try {
111
+ parsed = JSON.parse(result.resultJson || '{"done":true}');
112
+ }
113
+ catch {
114
+ this._done = true;
115
+ return { value: undefined, done: true };
116
+ }
117
+ if (parsed.done) {
118
+ this._done = true;
119
+ return { value: undefined, done: true };
120
+ }
121
+ this._position++;
122
+ // Deserialize the value
123
+ const rawValue = parsed.value;
124
+ let value;
125
+ if (rawValue !== null &&
126
+ typeof rawValue === 'object' &&
127
+ '__pyref__' in rawValue) {
128
+ // It's a Python object reference
129
+ if (this._options.wrapObjects !== false) {
130
+ const refId = rawValue.__pyref__;
131
+ value = PyProxy_1.PyProxy._createFromObjectId(refId, [], this._addon);
132
+ }
133
+ else {
134
+ value = rawValue;
135
+ }
136
+ }
137
+ else {
138
+ value = rawValue;
139
+ }
140
+ return { value, done: false };
141
+ }
142
+ catch (err) {
143
+ this._done = true;
144
+ throw err;
145
+ }
146
+ }
147
+ async return(value) {
148
+ await this.destroy();
149
+ return { value: value, done: true };
150
+ }
151
+ async throw(error) {
152
+ await this.destroy();
153
+ throw error;
154
+ }
155
+ // ─── AsyncIterable Protocol ────────────────────────────────────────────────
156
+ [Symbol.asyncIterator]() {
157
+ return this;
158
+ }
159
+ // ─── Utilities ─────────────────────────────────────────────────────────────
160
+ /**
161
+ * Collect all items into an array.
162
+ * Use with care for large iterators!
163
+ */
164
+ async toArray(maxItems = 100000) {
165
+ const items = [];
166
+ for await (const item of this) {
167
+ items.push(item);
168
+ if (items.length >= maxItems) {
169
+ break;
170
+ }
171
+ }
172
+ return items;
173
+ }
174
+ /**
175
+ * Process items in batches.
176
+ */
177
+ async *batch(batchSize) {
178
+ let current = [];
179
+ for await (const item of this) {
180
+ current.push(item);
181
+ if (current.length >= batchSize) {
182
+ yield current;
183
+ current = [];
184
+ }
185
+ }
186
+ if (current.length > 0) {
187
+ yield current;
188
+ }
189
+ }
190
+ /**
191
+ * Map items through a transform function.
192
+ */
193
+ async *map(fn) {
194
+ let i = 0;
195
+ for await (const item of this) {
196
+ yield await fn(item, i++);
197
+ }
198
+ }
199
+ /**
200
+ * Filter items.
201
+ */
202
+ async *filter(fn) {
203
+ let i = 0;
204
+ for await (const item of this) {
205
+ if (await fn(item, i++)) {
206
+ yield item;
207
+ }
208
+ }
209
+ }
210
+ /**
211
+ * Take first N items.
212
+ */
213
+ async take(n) {
214
+ const items = [];
215
+ for await (const item of this) {
216
+ items.push(item);
217
+ if (items.length >= n) {
218
+ break;
219
+ }
220
+ }
221
+ await this.destroy();
222
+ return items;
223
+ }
224
+ /**
225
+ * Destroy the underlying Python iterator (frees resources).
226
+ * Called automatically when iteration completes.
227
+ */
228
+ async destroy() {
229
+ if (this._destroyed) {
230
+ return;
231
+ }
232
+ this._destroyed = true;
233
+ this._done = true;
234
+ try {
235
+ await this._addon.destroyIterator(this._iteratorId);
236
+ }
237
+ catch (err) {
238
+ logger.debug('Error destroying iterator', err);
239
+ }
240
+ }
241
+ }
242
+ exports.PyIterator = PyIterator;
243
+ //# sourceMappingURL=PyIterator.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"PyIterator.js","sourceRoot":"","sources":["../../src/core/PyIterator.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAIH,8DAAgE;AAChE,uCAAsD;AACtD,4CAAyC;AAEzC,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC,YAAY,CAAC,CAAC;AAWxC;;;;;;;;;;;;;;GAcG;AACH,MAAa,UAAU;IACJ,WAAW,CAAS;IACpB,MAAM,CAAc;IACpB,SAAS,CAAa;IACtB,QAAQ,CAAoB;IACrC,KAAK,GAAG,KAAK,CAAC;IACd,SAAS,GAAG,CAAC,CAAC;IACd,UAAU,GAAG,KAAK,CAAC;IAE3B,YACE,UAAkB,EAClB,QAAoB,EACpB,KAAkB,EAClB,UAA6B,EAAE;QAE/B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CACjB,MAAe,EACf,OAA2B;QAE3B,MAAM,QAAQ,GAAG,MAAM,CAAC,0BAAgB,CAAC,CAAC;QAC1C,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QAE7B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,cAAc,CACvC,QAAQ,CAAC,QAAkB,EAC3B,QAAQ,CAAC,IAAI,CAC8F,CAAC;QAE9G,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,MAAM,IAAA,sCAAoB,EAAC;oBACzB,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;oBACvB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;oBAC7B,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE;iBACxC,CAAC,CAAC;YACL,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,UAAkB,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,IAAI,IAAI,CAA4B,CAAC;YAChF,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;gBAAA,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YAAA,CAAC;YAC3D,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QACjC,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,KAAK,CAAC,oCAAoC,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;QAC5E,CAAC;QAED,OAAO,IAAI,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;IACvE,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,OAAO,CACZ,UAAkB,EAClB,QAAoB,EACpB,KAAkB,EAClB,OAA2B;QAE3B,OAAO,IAAI,UAAU,CAAC,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAED,oCAAoC;IACpC,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,oCAAoC;IACpC,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,oLAAoL;IAEpL,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAClC,OAAO,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC1D,CAAC;QAED,cAAc;QACd,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;YACjE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;YACrB,OAAO,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC1D,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,WAAW,CAI7D,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBACjB,MAAM,IAAA,sCAAoB,EAAC;wBACzB,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI;wBACvB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;wBAC7B,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE;qBACxC,CAAC,CAAC;gBACL,CAAC;gBACD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;gBAClB,OAAO,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YAC1D,CAAC;YAED,IAAI,MAA2C,CAAC;YAChD,IAAI,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,IAAI,eAAe,CAAwC,CAAC;YACnG,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;gBAClB,OAAO,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YAC1D,CAAC;YAED,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAChB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;gBAClB,OAAO,EAAE,KAAK,EAAE,SAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YAC1D,CAAC;YAED,IAAI,CAAC,SAAS,EAAE,CAAC;YAEjB,wBAAwB;YACxB,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC;YAC9B,IAAI,KAAQ,CAAC;YAEb,IACE,QAAQ,KAAK,IAAI;gBACjB,OAAO,QAAQ,KAAK,QAAQ;gBAC5B,WAAW,IAAK,QAAoC,EACpD,CAAC;gBACD,iCAAiC;gBACjC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;oBACxC,MAAM,KAAK,GAAI,QAAkC,CAAC,SAAS,CAAC;oBAC5D,KAAK,GAAG,iBAAO,CAAC,mBAAmB,CACjC,KAAmB,EACnB,EAAE,EACF,IAAI,CAAC,MAAM,CACI,CAAC;gBACpB,CAAC;qBAAM,CAAC;oBACN,KAAK,GAAG,QAAa,CAAC;gBACxB,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,KAAK,GAAG,QAAa,CAAC;YACxB,CAAC;YAED,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QAChC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;YAClB,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAS;QACpB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,EAAE,KAAK,EAAE,KAAU,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAe;QACzB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,MAAM,KAAK,CAAC;IACd,CAAC;IAED,oLAAoL;IAEpL,CAAC,MAAM,CAAC,aAAa,CAAC;QACpB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,8MAA8M;IAE9M;;;OAGG;IACH,KAAK,CAAC,OAAO,CAAC,QAAQ,GAAG,MAAM;QAC7B,MAAM,KAAK,GAAQ,EAAE,CAAC;QACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YAC9B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,IAAI,KAAK,CAAC,MAAM,IAAI,QAAQ,EAAE,CAAC;gBAAA,MAAM;YAAA,CAAC;QACxC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,CAAC,KAAK,CAAC,SAAiB;QAC5B,IAAI,OAAO,GAAQ,EAAE,CAAC;QACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YAC9B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,IAAI,OAAO,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;gBAChC,MAAM,OAAO,CAAC;gBACd,OAAO,GAAG,EAAE,CAAC;YACf,CAAC;QACH,CAAC;QACD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,OAAO,CAAC;QAChB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,CAAC,GAAG,CAAI,EAA8C;QAC1D,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YAC9B,MAAM,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,CAAC,MAAM,CAAC,EAA0D;QACtE,IAAI,CAAC,GAAG,CAAC,CAAC;QACV,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YAC9B,IAAI,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC;gBAAA,MAAM,IAAI,CAAC;YAAA,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,CAAS;QAClB,MAAM,KAAK,GAAQ,EAAE,CAAC;QACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YAC9B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;gBAAA,MAAM;YAAA,CAAC;QACjC,CAAC;QACD,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAAA,OAAO;QAAA,CAAC;QAC9B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACvB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,GAAG,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;CACF;AA9PD,gCA8PC"}