import os # 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 print("Hello world!") print("Hello world!") run_id = os.environ.get("CURSORIUS_RUN_ID") src_dir = os.environ.get("CURSORIUS_SRC_DIR") server_url = os.environ.get("CURSORIUS_SERVER_URL") 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)