github/prometheus-pve-sd
0
0
Fork 0
mirror of https://github.com/thegeeklab/prometheus-pve-sd.git synced 2024-11-21 08:50:40 +00:00
Code Issues Projects Releases Wiki Activity

use context manager for tempfile

Browse Source
This commit is contained in:
Robert Kaussow 2024-10-21 20:25:28 +02:00 committed by renovate[bot]
parent 607f201888
commit 468d452996
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
prometheuspvesd/cli.py
View File

@ -174,11 +174,10 @@ class PrometheusSD:
output.append(host.to_sd_json())
# Write to tmp file and move after write
temp_file = tempfile.NamedTemporaryFile(mode="w", prefix="prometheus-pve-sd", delete=False)
with temp_file as tf:
with tempfile.NamedTemporaryFile(mode="w", prefix="prometheus-pve-sd", delete=False) as tf:
json.dump(output, tf, indent=4)
shutil.move(temp_file.name, self.config.config["output_file"])
shutil.move(tf.name, self.config.config["output_file"])
chmod(self.config.config["output_file"], int(self.config.config["output_file_mode"], 8))
def _terminate(self, signal, frame): # noqa