detection/__main__.py

@@ -9,6 +9,7 @@ from .pyc_detection import disassemble_pyc
 from .utils import *
 import sys
 from colorama import init, Fore, Style
+from tqdm import tqdm
 
 PYCDC_FLAG = True
 PYCDC_ADDR_FLAG = True
@@ -361,11 +362,17 @@ def process_path(
 ):
     results = {"high": [], "medium": [], "low": [], "none": []}
     if os.path.isdir(path):
+        # 获取所有文件
+        all_files = []
         for root, dirs, files in os.walk(path):
             for file in files:
                 file_extension = os.path.splitext(file)[1]
                 if file_extension in SUPPORTED_EXTENSIONS:
                     file_path = os.path.join(root, file)
+                    all_files.append(file_path)
+        # 扫描动画
+        for file_path in tqdm(all_files, desc="Scanning files", unit="file"):
+            file_extension = os.path.splitext(file_path)[1]
             file_results = checkModeAndDetect(
                 mode, file_path, file_extension, pycdc_addr
             )
@@ -379,9 +386,16 @@ def process_path(
                             ]
                         )
     elif os.path.isfile(path):
+        file_results = ""
         file_extension = os.path.splitext(path)[1]
         if file_extension in SUPPORTED_EXTENSIONS:
-            file_results = checkModeAndDetect(mode, path, file_extension, pycdc_addr)
+            # 扫描动画
+            with tqdm(total=100, desc="Scanning file", unit="%", ncols=100) as pbar:
+                for i in range(10):
+                    file_results = checkModeAndDetect(
+                        mode, path, file_extension, pycdc_addr
+                    )
+                    pbar.update(10)  # Update the progress bar by 10%
             if file_results is not None:
                 for key in file_results:
                     if key != "none":  # Exclude 'none' risk level