saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-16 09:40:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Actually download the stream

Browse source
This commit is contained in:
Tyler Levy Conde 2024-09-16 11:42:51 -06:00 • committed by Daniel Wozniak
parent 4bb6e93cab
commit 7df3bbfe66
1 changed files with 25 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
salt/client/ssh/__init__.py
View file

@ -1265,16 +1265,35 @@ class Single:
# Download the relenv tarball if not already cached
if not os.path.exists(tarball_path):
log.info(f"Downloading relenv tarball from {relenv_url} to {tarball_path}")
result = salt.utils.http.query(relenv_url, stream=True, local_file=tarball_path)
if result.get("status") != 200:
log.error(f"Failed to download relenv tarball from {relenv_url}")
try:
with salt.utils.files.fopen(tarball_path, 'wb') as tarball_file:
# Define streaming callback to write data chunks to the file
def stream_callback(chunk):
tarball_file.write(chunk)
# Download the file using salt.utils.http.query
result = salt.utils.http.query(
url=relenv_url,
method='GET',
stream=True,
streaming_callback=stream_callback,
raise_error=True
)
# Check if the download was successful
if result.get("status") != 200:
log.error(f"Failed to download relenv tarball from {relenv_url}")
return False
except Exception as e:
log.error(f"Error during relenv tarball download: {e}")
return False
# Send the tarball to the target machine
self.shell.send(tarball_path, os.path.join(self.thin_dir, "salt-relenv.tar.xz"))
return True
stdout, stderr, retcode = self.shell.send(tarball_path, os.path.join(self.thin_dir, "salt-relenv.tar.xz"))
log.info(stdout.strip())
return retcode == 0
def deploy(self):
"""