sdk: more cluster fixes for python

Author: koush

sdk/package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@scrypted/sdk",
-  "version": "0.3.78",
+  "version": "0.3.84",
   "description": "",
   "main": "dist/src/index.js",
   "exports": {

sdk/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@scrypted/types",
-  "version": "0.3.72",
+  "version": "0.3.77",
   "description": "",
   "main": "dist/index.js",
   "author": "",

sdk/types/package-lock.json

@@ -15,6 +15,7 @@ def terminate(self):
 deviceManager: DeviceManager = None
 systemManager: SystemManager = None
 mediaManager: MediaManager = None
+clusterManager: ClusterManager = None
 zip: ZipFile = None
 remote: Any = None
 api: Any
@@ -28,6 +29,7 @@ def __init__(self) -> None:
         self.systemManager: SystemManager = None
         self.deviceManager: DeviceManager = None
         self.mediaManager: MediaManager = None
+        self.clusterManager: ClusterManager = None
         self.zip: ZipFile = None
         self.remote: Any = None
         self.api: Any = None
@@ -52,13 +54,16 @@ def sdk_init2(scryptedStatic: ScryptedStatic):
     global systemManager
     global deviceManager
     global mediaManager
+    global clusterManager
     global sdk
     global api
     global fork
     sdk  = scryptedStatic
     systemManager = sdk.systemManager
     deviceManager = sdk.deviceManager
     mediaManager = sdk.mediaManager
+    if hasattr(sdk, 'clusterManager'):
+        clusterManager = sdk.clusterManager
     zip = sdk.zip
     remote = sdk.remote
     api = sdk.api

sdk/types/package.json

@@ -1,9 +1,9 @@
 from __future__ import annotations
 from enum import Enum
 try:
-    from typing import TypedDict
+    from typing import TypedDict, Mapping
 except:
-    from typing_extensions import TypedDict
+    from typing_extensions import TypedDict, Mapping
 from typing import Union, Any, AsyncGenerator
 
 from .other import *
@@ -299,6 +299,16 @@ class AudioStreamOptions(TypedDict):
     profile: str
     sampleRate: float
 
+class ClusterFork(TypedDict):
+
+    clusterWorkerId: str  # The id of the cluster worker id that will execute this fork.
+    filename: str  # The filename to execute in the fork. Not supported in all runtimes.
+    id: str  # The id of the device that is associated with this fork.
+    labels: Any  # The labels used to select the cluster worker that will execute this fork.
+    name: str  # The name of this fork. This will be used to set the thread name
+    nativeId: str  # The native id of the mixin that is associated with this fork.
+    runtime: str  # The runtime to use for this fork. If not specified, the current runtime will be used.
+
 class HttpResponseOptions(TypedDict):
 
     code: float
@@ -459,6 +469,13 @@ class ClusterForkInterfaceOptions(TypedDict):
     id: str  # The id of the device that is associated with this fork.
     nativeId: str  # The native id of the mixin that is associated with this fork.
 
+class ClusterWorker(TypedDict):
+
+    forks: list[ClusterFork]
+    id: str
+    labels: list[str]
+    name: str
+
 class ColorHsv(TypedDict):
     """Represents an HSV color value component."""
 
@@ -1821,7 +1838,7 @@ def getClusterMode(self) -> Any | Any:
     def getClusterWorkerId(self) -> str:
         pass
 
-    async def getClusterWorkers(self) -> str:
+    async def getClusterWorkers(self) -> Mapping[str, ClusterWorker]:
         pass
 
 

sdk/types/scrypted_python/scrypted_sdk/__init__.py

@@ -95,6 +95,8 @@ function toPythonType(type: any): string {
         return type.types.map((type: any) => toPythonType(type)).join(' | ')
     if (type.name === 'AsyncGenerator')
         return `AsyncGenerator[${toPythonType(type.typeArguments[0])}, None]`;
+    if (type.name === 'Record')
+        return `Mapping[${toPythonType(type.typeArguments[0])}, ${toPythonType(type.typeArguments[1])}]`;
     type = type.typeArguments?.[0]?.name || type.name || type;
     if (parameterTypes.has(type))
         return 'Any';
@@ -344,9 +346,9 @@ ${toDocstring(td)}
 const pythonTypes = `from __future__ import annotations
 from enum import Enum
 try:
-    from typing import TypedDict
+    from typing import TypedDict, Mapping
 except:
-    from typing_extensions import TypedDict
+    from typing_extensions import TypedDict, Mapping
 from typing import Union, Any, AsyncGenerator
 
 from .other import *