修复任务执行前命令错误提示

whyour · Sep 8, 2024 · 69fcf45 · 69fcf45
1 parent c370450
commit 69fcf45
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 6 deletions.
diff --git a/shell/preload/sitecustomize.js b/shell/preload/sitecustomize.js
@@ -1,5 +1,6 @@
 const { execSync } = require('child_process');
 require(`./env.js`);
+
 function expandRange(rangeStr, max) {
   const tempRangeStr = rangeStr
     .trim()
@@ -30,8 +31,6 @@ function run() {
     task_before,
   } = process.env;
 
-  require(file_task_before_js);
-
   try {
     const splitStr = '__sitecustomize__';
     const fileName = process.argv[1].replace(`${dir_scripts}/`, '');
@@ -60,10 +59,19 @@ function run() {
     }
   } catch (error) {
     if (!error.message.includes('spawnSync /bin/sh E2BIG')) {
-      console.log(`run task before error: `, error);
+      console.log(`❌ run task before error: `, error);
+    } else {
+      console.log(
+        `❌ The environment variable is too large. It is recommended to use task_before.js instead of task_before.sh\n`,
+      );
+    }
+    if (task_before) {
+      console.log('执行前置命令结束\n');
     }
   }
 
+  require(file_task_before_js);
+
   if (envParam && numParam) {
     const array = (process.env[envParam] || '').split('&');
     const runArr = expandRange(numParam, array.length);

diff --git a/shell/preload/sitecustomize.py b/shell/preload/sitecustomize.py
@@ -35,8 +35,6 @@ def expand_range(range_str, max_value):
 
 
 def run():
-    import task_before
-
     try:
         split_str = "__sitecustomize__"
         file_name = sys.argv[0].replace(f"{os.getenv('dir_scripts')}/", "")
@@ -68,10 +66,18 @@ def run():
     except OSError as error:
         error_message = str(error)
         if "Argument list too long" not in error_message:
-            print(f"run task before error: {error}")
+            print(f"❌ run task before error: {error}")
+        else:
+            print(
+                "❌ The environment variable is too large. It is recommended to use task_before.py instead of task_before.sh\n"
+            )
+        if task_before:
+            print("执行前置命令结束")
     except Exception as error:
         print(f"run task before error: {error}")
 
+    import task_before
+
     env_param = os.getenv("envParam")
     num_param = os.getenv("numParam")