fix os error

Lab-Notebooks · Oct 1, 2023 · d69cf14 · d69cf14
1 parent f32211e
commit d69cf14
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/jobrunner/api/_commands.py b/jobrunner/api/_commands.py
@@ -97,7 +97,11 @@ def clean(workdir_list):
     """
     # Get base directory
     basedir = os.getcwd()
-    print(os.get_terminal_size().columns * "—")
+
+    try:
+        print(os.get_terminal_size().columns * "—")
+    except:
+        print(100 * "—")
 
     # run cleanup
     for workdir in workdir_list:
@@ -121,7 +125,11 @@ def archive(tag, workdir_list):
     """
     # Get base directory
     basedir = os.getcwd()
-    print(os.get_terminal_size().columns * "—")
+
+    try:
+        print(os.get_terminal_size().columns * "—")
+    except:
+        print(100 * "—")
 
     # loop over workdir_list
     for workdir in workdir_list:

diff --git a/jobrunner/lib/_console.py b/jobrunner/lib/_console.py
@@ -13,7 +13,11 @@ def DisplayTree(basedir, workdir):
     """
     Display tree information on console
     """
-    print(os.get_terminal_size().columns * "—")
+    try:
+        print(os.get_terminal_size().columns * "—")
+    except:
+        print(100 * "—")
+
     print(f"{lib.Color.purple}ROOT:{lib.Color.end} {basedir}")
     print(f'{lib.Color.purple}NODE:{lib.Color.end} {workdir.replace(basedir,"<ROOT>")}')