From ce684d22a93b56f87e04b94db97aa581e90260e9 Mon Sep 17 00:00:00 2001
From: LawyZheng <lawyzheng1106@gmail.com>
Date: Tue, 26 Nov 2024 17:04:18 +0800
Subject: [PATCH] add remove dir function (#1263)

---
 skyvern/forge/sdk/api/files.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/skyvern/forge/sdk/api/files.py b/skyvern/forge/sdk/api/files.py
index 8fb378bb..98f47f87 100644
--- a/skyvern/forge/sdk/api/files.py
+++ b/skyvern/forge/sdk/api/files.py
@@ -2,6 +2,7 @@ import hashlib
 import mimetypes
 import os
 import re
+import shutil
 import tempfile
 import zipfile
 from pathlib import Path
@@ -180,3 +181,25 @@ def create_named_temporary_file(delete: bool = True) -> tempfile._TemporaryFileW
     temp_dir = SettingsManager.get_settings().TEMP_PATH
     create_folder_if_not_exist(temp_dir)
     return tempfile.NamedTemporaryFile(dir=temp_dir, delete=delete)
+
+
+def clean_up_dir(dir: str) -> None:
+    if not os.path.exists(dir):
+        return
+
+    if os.path.isfile(dir):
+        os.unlink(dir)
+        return
+
+    for item in os.listdir(dir):
+        item_path = os.path.join(dir, item)
+        if os.path.isfile(item_path) or os.path.islink(item_path):
+            os.unlink(item_path)
+        elif os.path.isdir(item_path):
+            shutil.rmtree(item_path)
+
+    return
+
+
+def clean_up_skyvern_temp_dir() -> None:
+    return clean_up_dir(get_skyvern_temp_dir())