test

2023-02-25 12:52:07 -07:00
parent e81f9d92e3
commit 48af1e8eb8
1 changed files with 64 additions and 65 deletions
@@ -1,13 +1,12 @@
 import os
-
-# import sys
-# import grpc
+import sys
+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/")
-# 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 release_runner_pb2, release_runner_pb2_grpc
+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 run_command_pb2, run_command_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
-# channel = grpc.insecure_channel(server_url)
-#
-## GetRunner stub
-# get_runner_stub = get_runner_pb2_grpc.GetRunnerServiceStub(channel)
-#
-## GetRunner Request
-# request = get_runner_pb2.GetRunnerRequest()
-#
-## Call GetRunner
-# get_runner_resp = None
-# try:
-#    get_runner_resp = get_runner_stub.GetRunner(request)
-# except grpc._channel._InactiveRpcError as e:
-#    print(f"Error: {e.details()}")
-#    sys.exit(-1)
-#
-## Print ID of allocated runner
-# print(get_runner_resp.id)
-#
-## RunCommand stub
-# run_command_stub = run_command_pb2_grpc.RunCommandServiceStub(channel)
-#
-## RunCommand request
-# request = run_command_pb2.RunCommandRequest(
-#    id=get_runner_resp.id,
-#    command="ls",
-#    args=["-lah"],
-# )
-#
-## Call RunCommand
-# run_command_resp = None
-# try:
-#    run_command_resp = run_command_stub.RunCommand(request)
-# except grpc._channel._InactiveRpcError as e:
-#    print(f"Error: {e.details()}")
-#    sys.exit(-1)
-#
-# print(f"return code: {run_command_resp.return_code}")
-# print(f"stdout: {run_command_resp.stdout}")
-# print(f"stderr: {run_command_resp.stderr}")
-#
-#
-## ReleaseRunner stub
-# release_runner_stub = release_runner_pb2_grpc.ReleaseRunnerServiceStub(channel)
-#
-## ReleaseRunner request
-# request = release_runner_pb2.ReleaseRunnerRequest(
-#    id=get_runner_resp.id,
-# )
-#
-## Call ReleaseRunner
-# release_runner_resp = None
-# try:
-#    release_runner_resp = release_runner_stub.ReleaseRunner(request)
-# except grpc._channel._InactiveRpcError as e:
-#    print(f"Error: {e.details()}")
-#    sys.exit(-1)
+# create GRPc connection
+channel = grpc.insecure_channel(server_url)
+
+# GetRunner stub
+get_runner_stub = get_runner_pb2_grpc.GetRunnerServiceStub(channel)
+
+# GetRunner Request
+request = get_runner_pb2.GetRunnerRequest()
+
+# Call GetRunner
+get_runner_resp = None
+try:
+    get_runner_resp = get_runner_stub.GetRunner(request)
+except grpc._channel._InactiveRpcError as e:
+    print(f"Error: {e.details()}")
+    sys.exit(-1)
+
+# Print ID of allocated runner
+print(get_runner_resp.id)
+
+# RunCommand stub
+run_command_stub = run_command_pb2_grpc.RunCommandServiceStub(channel)
+
+# RunCommand request
+request = run_command_pb2.RunCommandRequest(
+    id=get_runner_resp.id,
+    command="ls",
+    args=["-lah"],
+)
+
+# Call RunCommand
+run_command_resp = None
+try:
+    run_command_resp = run_command_stub.RunCommand(request)
+except grpc._channel._InactiveRpcError as e:
+    print(f"Error: {e.details()}")
+    sys.exit(-1)
+
+print(f"return code: {run_command_resp.return_code}")
+print(f"stdout: {run_command_resp.stdout}")
+print(f"stderr: {run_command_resp.stderr}")
+
+
+# ReleaseRunner stub
+release_runner_stub = release_runner_pb2_grpc.ReleaseRunnerServiceStub(channel)
+
+# ReleaseRunner request
+request = release_runner_pb2.ReleaseRunnerRequest(
+    id=get_runner_resp.id,
+)
+
+# Call ReleaseRunner
+release_runner_resp = None
+try:
+    release_runner_resp = release_runner_stub.ReleaseRunner(request)
+except grpc._channel._InactiveRpcError as e:
+    print(f"Error: {e.details()}")
+    sys.exit(-1)