From 6b3e20277723cf5335e0c1981a5fa198964435bb Mon Sep 17 00:00:00 2001
From: Lurenaud <lucien.renaud@pal-robotics.com>
Date: Mon, 6 Feb 2023 19:18:30 +0100
Subject: [PATCH] fix printing

---
 scripts/run.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/scripts/run.py b/scripts/run.py
index 4e5a59a..87f0a3b 100644
--- a/scripts/run.py
+++ b/scripts/run.py
@@ -5,6 +5,7 @@ import sys
 import pathlib
 import subprocess
 from rich.progress import track
+from rich import print
 
 # folder input
 input = next(os.walk(sys.argv[1]))
@@ -20,8 +21,8 @@ currentPath = str(pathlib.Path().resolve())
 partFiles = [s for s in files if ".SLDPRT" in s]    # get only parts files
 drwgFiles = [s for s in files if ".SLDDRW" in s]    # get only drawings files
 
-# print("Parts: ", partFiles)
-# print("Drawings: ", drwgFiles)
+print("\nParts: \n", partFiles)
+print("\nDrawings: \n", drwgFiles, "\n")
 
 partWithoutExention = [os.path.splitext(x)[0] for x in partFiles]   # remove extension
 drwgWithoutExtension =  [os.path.splitext(x)[0] for x in drwgFiles]   # remove extension
@@ -40,17 +41,17 @@ outputFolder = str(path) + "\\export\\"
 # create export directory
 if not os.path.exists(outputFolder):
     os.makedirs(outputFolder)
-    print("Export folder created")
+    print("\nExport folder created\n")
 
 # part files
-for i in track(partWithoutExention, description="Exporting step files..."):
+for i in track(partWithoutExention, description="Exporting step files...", total=len(partWithoutExention)):
     input = str(i)
     commandstep = "PowerShell -NoProfile -ExecutionPolicy Bypass -File " + currentPath + '\scripts\export-step.ps1 "' + path + "\\" + input + '" "' + outputFolder + input + '"'
     # print(commandstep)
     subprocess.run(commandstep, capture_output=True)
 
 # drawings files
-for i in track(drwgWithoutExtension, description="Exporting PDF and DXF..."):
+for i in track(drwgWithoutExtension, description="Exporting PDF and DXF...", total = len(drwgWithoutExtension)):
     input = str(i)
     commandPdfDxf = "PowerShell -NoProfile -ExecutionPolicy Bypass -File " + currentPath + '\scripts\export-pdf-dxf.ps1 "' + path + "\\" + input + '" "' + outputFolder + input + '"'
     # print(commandPdfDxf)