@quantod/qq 0.3.7 → 1.0.0

package/src/sdk-templates/python.ts

@@ -0,0 +1,27 @@
+export function renderPythonSdk(host: string, port: number, token: string): string {
+  return `import urllib.request, json as _json, urllib.parse
+class BridgeError(Exception): pass
+_BRIDGE_URL = 'http://${host}:${port}'
+_BRIDGE_HEADERS = {'Authorization': 'Bearer ${token}', 'Content-Type': 'application/json'}
+def _bridge_req(method, path, body=None):
+    data = _json.dumps(body).encode() if body is not None else None
+    req = urllib.request.Request(_BRIDGE_URL + path, data=data, headers=_BRIDGE_HEADERS, method=method)
+    try:
+        with urllib.request.urlopen(req) as r:
+            return _json.loads(r.read()) if r.status != 204 else None
+    except urllib.error.HTTPError as e:
+        return _json.loads(e.read())
+    except Exception as e:
+        raise BridgeError('Connection failed — call get_sdk again to refresh credentials') from e
+def qq_push(pipeline, stage, id=None, payload=None, **opts):
+    return _bridge_req('POST', f'/pipelines/{pipeline}/push', {'stage': stage, 'id': id, 'payload': payload, **opts})
+def qq_claim(pipeline, stage, **opts):
+    return _bridge_req('POST', f'/pipelines/{pipeline}/claim', {'stage': stage, **opts})
+def qq_release(pipeline, seq, **opts):
+    return _bridge_req('POST', f'/pipelines/{pipeline}/release/{seq}', opts)
+def qq_batch_read(pipeline, stage, **opts):
+    qs = urllib.parse.urlencode({'stage': stage, **opts})
+    return _bridge_req('GET', f'/pipelines/{pipeline}/items?{qs}')
+def qq_status(pipeline):
+    return _bridge_req('GET', f'/pipelines/{pipeline}/status')`;
+}