mirror of
https://github.com/allaunthefox/Research-Stack.git
synced 2026-07-31 03:05:21 +00:00
fd863af6fd
- .devcontainer/Dockerfile: add PostgreSQL client libs, OpenSSL/libffi headers, gfortran/BLAS for scipy, rclone; install full Python dependency set (boto3, psycopg2-binary, fastapi, uvicorn, notion-client, httpx, pytest, numpy, scipy, etc.) in uv-managed venv; add rclone S3 gateway init script as ENTRYPOINT - .devcontainer/devcontainer.json: switch from build to pre-built image (localhost/research
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Bridge wrapper — called by the Rust sync daemon to invoke Python infra surfaces.
|
|
|
|
Usage: python3 bridge_wrapper.py <infra_dir>
|
|
Reads JSON request from stdin, writes JSON response to stdout.
|
|
|
|
Request format:
|
|
{"module": "ene_rds_wiki_layer", "payload": {"op": "recent", "limit": 5}}
|
|
|
|
Response format:
|
|
{"ok": true, "data": {...}} or {"ok": false, "error": "..."}
|
|
"""
|
|
|
|
import inspect
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
|
|
def find_target_class(mod):
|
|
"""Find the first class in a module that has a handle_request method."""
|
|
for _, obj in inspect.getmembers(mod, inspect.isclass):
|
|
if hasattr(obj, "handle_request") and callable(getattr(obj, "handle_request", None)):
|
|
return obj
|
|
return None
|
|
|
|
|
|
def main():
|
|
infra_dir = sys.argv[1] if len(sys.argv) > 1 else "."
|
|
sys.path.insert(0, infra_dir)
|
|
os.chdir(infra_dir)
|
|
|
|
req = json.load(sys.stdin)
|
|
module_name = req.get("module", "")
|
|
payload = req.get("payload", {})
|
|
|
|
if not module_name:
|
|
json.dump({"ok": False, "error": "missing 'module' field"}, sys.stdout)
|
|
return
|
|
|
|
try:
|
|
mod = __import__(module_name, fromlist=[""])
|
|
except Exception as e:
|
|
json.dump({"ok": False, "error": f"import failed: {e}"}, sys.stdout)
|
|
return
|
|
|
|
target_class = find_target_class(mod)
|
|
if target_class is None:
|
|
# Fallback: try calling a module-level handle_request
|
|
if hasattr(mod, "handle_request") and callable(mod.handle_request):
|
|
try:
|
|
result = mod.handle_request(payload)
|
|
json.dump({"ok": True, "data": result}, sys.stdout)
|
|
except Exception as e:
|
|
json.dump({"ok": False, "error": f"handle_request error: {e}"}, sys.stdout)
|
|
return
|
|
json.dump(
|
|
{"ok": False, "error": f"No class with handle_request in {module_name}"},
|
|
sys.stdout,
|
|
)
|
|
return
|
|
|
|
try:
|
|
instance = target_class()
|
|
result = instance.handle_request(payload)
|
|
json.dump({"ok": True, "data": result}, sys.stdout)
|
|
except Exception as e:
|
|
json.dump({"ok": False, "error": f"handle_request error: {e}"}, sys.stdout)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|