test

.cursorius/build.py

@@ -1,13 +1,12 @@
 import os
-
+import sys
-# import sys
+import grpc
-# import grpc
 
 # HACK: need to fix how protobuf is generated to get imports working
-# sys.path.append("/usr/local/lib/python3.11/site-packages/cursorius_pipeline_api_v2/")
+sys.path.append("/usr/local/lib/python3.11/site-packages/cursorius_pipeline_api_v2/")
-# from cursorius_pipeline_api_v2 import get_runner_pb2, get_runner_pb2_grpc
+from cursorius_pipeline_api_v2 import get_runner_pb2, get_runner_pb2_grpc
-# from cursorius_pipeline_api_v2 import run_command_pb2, run_command_pb2_grpc
+from cursorius_pipeline_api_v2 import run_command_pb2, run_command_pb2_grpc
-# from cursorius_pipeline_api_v2 import release_runner_pb2, release_runner_pb2_grpc
+from cursorius_pipeline_api_v2 import release_runner_pb2, release_runner_pb2_grpc
 
 
 print("Hello world!")
@@ -21,61 +20,61 @@ print(f"{run_id=}")
 print(f"{src_dir=}")
 print(f"{server_url=}")
 
-## create GRPc connection
+# create GRPc connection
-# channel = grpc.insecure_channel(server_url)
+channel = grpc.insecure_channel(server_url)
-#
+
-## GetRunner stub
+# GetRunner stub
-# get_runner_stub = get_runner_pb2_grpc.GetRunnerServiceStub(channel)
+get_runner_stub = get_runner_pb2_grpc.GetRunnerServiceStub(channel)
-#
+
-## GetRunner Request
+# GetRunner Request
-# request = get_runner_pb2.GetRunnerRequest()
+request = get_runner_pb2.GetRunnerRequest()
-#
+
-## Call GetRunner
+# Call GetRunner
-# get_runner_resp = None
+get_runner_resp = None
-# try:
+try:
-#    get_runner_resp = get_runner_stub.GetRunner(request)
+    get_runner_resp = get_runner_stub.GetRunner(request)
-# except grpc._channel._InactiveRpcError as e:
+except grpc._channel._InactiveRpcError as e:
-#    print(f"Error: {e.details()}")
+    print(f"Error: {e.details()}")
-#    sys.exit(-1)
+    sys.exit(-1)
-#
+
-## Print ID of allocated runner
+# Print ID of allocated runner
-# print(get_runner_resp.id)
+print(get_runner_resp.id)
-#
+
-## RunCommand stub
+# RunCommand stub
-# run_command_stub = run_command_pb2_grpc.RunCommandServiceStub(channel)
+run_command_stub = run_command_pb2_grpc.RunCommandServiceStub(channel)
-#
+
-## RunCommand request
+# RunCommand request
-# request = run_command_pb2.RunCommandRequest(
+request = run_command_pb2.RunCommandRequest(
-#    id=get_runner_resp.id,
+    id=get_runner_resp.id,
-#    command="ls",
+    command="ls",
-#    args=["-lah"],
+    args=["-lah"],
-# )
+)
-#
+
-## Call RunCommand
+# Call RunCommand
-# run_command_resp = None
+run_command_resp = None
-# try:
+try:
-#    run_command_resp = run_command_stub.RunCommand(request)
+    run_command_resp = run_command_stub.RunCommand(request)
-# except grpc._channel._InactiveRpcError as e:
+except grpc._channel._InactiveRpcError as e:
-#    print(f"Error: {e.details()}")
+    print(f"Error: {e.details()}")
-#    sys.exit(-1)
+    sys.exit(-1)
-#
+
-# print(f"return code: {run_command_resp.return_code}")
+print(f"return code: {run_command_resp.return_code}")
-# print(f"stdout: {run_command_resp.stdout}")
+print(f"stdout: {run_command_resp.stdout}")
-# print(f"stderr: {run_command_resp.stderr}")
+print(f"stderr: {run_command_resp.stderr}")
-#
+
-#
+
-## ReleaseRunner stub
+# ReleaseRunner stub
-# release_runner_stub = release_runner_pb2_grpc.ReleaseRunnerServiceStub(channel)
+release_runner_stub = release_runner_pb2_grpc.ReleaseRunnerServiceStub(channel)
-#
+
-## ReleaseRunner request
+# ReleaseRunner request
-# request = release_runner_pb2.ReleaseRunnerRequest(
+request = release_runner_pb2.ReleaseRunnerRequest(
-#    id=get_runner_resp.id,
+    id=get_runner_resp.id,
-# )
+)
-#
+
-## Call ReleaseRunner
+# Call ReleaseRunner
-# release_runner_resp = None
+release_runner_resp = None
-# try:
+try:
-#    release_runner_resp = release_runner_stub.ReleaseRunner(request)
+    release_runner_resp = release_runner_stub.ReleaseRunner(request)
-# except grpc._channel._InactiveRpcError as e:
+except grpc._channel._InactiveRpcError as e:
-#    print(f"Error: {e.details()}")
+    print(f"Error: {e.details()}")
-#    sys.exit(-1)
+    sys.exit(-1)