async def pull(self, bundle_id: str, src_path: str, dest_path: str) -> None:
async with self.stub.pull.open() as stream:
request = request = PullRequest(
bundle_id=bundle_id,
src_path=src_path,
# not sending the destination to remote companion
# so it streams the file back
dst_path=dest_path if self.is_local else None,
)
await stream.send_message(request)
await stream.end()
if self.is_local:
await stream.recv_message()
else:
await drain_untar(generate_bytes(stream), output_path=dest_path)
self.logger.info(f"pulled file to {dest_path}")
async def daemon(client: CompanionClient,
request: PullRequest) -> PullResponse:
destination = request.dst_path
async with client.stub.pull.open() as stream:
if not client.is_local:
# not sending the destination to remote companion
# so it streams the file back
request = PullRequest(bundle_id=request.bundle_id,
src_path=request.src_path,
dst_path=None)
await stream.send_message(request, end=True)
if client.is_local:
await stream.recv_message()
else:
await drain_untar(generate_bytes(stream), output_path=destination)
client.logger.info(f"pulled file to {destination}")
return PullResponse(payload=Payload(file_path=destination))
async def client(client: CompanionClient, bundle_id: str, src_path: str,
dest_path: str) -> None:
await client.stub.pull(
PullRequest(bundle_id=bundle_id, src_path=src_path,
dst_path=dest_path))