about summary refs log tree commit diff
path: root/src/ci/scripts/free_disk_space_windows_util.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/ci/scripts/free_disk_space_windows_util.py')
-rw-r--r--src/ci/scripts/free_disk_space_windows_util.py29
1 files changed, 29 insertions, 0 deletions
diff --git a/src/ci/scripts/free_disk_space_windows_util.py b/src/ci/scripts/free_disk_space_windows_util.py
new file mode 100644
index 00000000000..488187864c2
--- /dev/null
+++ b/src/ci/scripts/free_disk_space_windows_util.py
@@ -0,0 +1,29 @@
+"""
+Utilities for Windows disk space cleanup scripts.
+"""
+
+import os
+from pathlib import Path
+import sys
+
+
+def get_temp_dir() -> Path:
+    """Get the temporary directory set by GitHub Actions."""
+    return Path(os.environ.get("RUNNER_TEMP"))
+
+
+def get_pid_file() -> Path:
+    return get_temp_dir() / "free-disk-space.pid"
+
+
+def get_log_file() -> Path:
+    return get_temp_dir() / "free-disk-space.log"
+
+
+def run_main(main_fn):
+    exit_code = 1
+    try:
+        exit_code = main_fn()
+    except Exception as e:
+        print(f"::error::{e}")
+    sys.exit(exit_code)