Quiet Mode should not generate any output

cve_bin_tool/cli.py

@@ -260,9 +260,9 @@ def main(argv=None):
             )
         )
 
-    with CVEScanner(score=score) as cve_scanner:
-        total_files: int = 0
+    with CVEScanner(score=score, error_mode=error_mode) as cve_scanner:
         triage_data: TriageData
+        total_files: int = 0
         parsed_data: Dict[tuple, TriageData] = {}
 
         if args.input_file:

cve_bin_tool/cve_scanner.py

@@ -11,6 +11,7 @@
 from rich.panel import Panel
 
 from .cvedb import DISK_LOCATION_DEFAULT, DBNAME
+from .error_handler import ErrorMode
 from .input_engine import Remarks, TriageData
 from .log import LOGGER
 
@@ -53,8 +54,14 @@ class CVEScanner:
     CONSOLE: Console = Console(file=sys.stderr)
     ALPHA_TO_NUM: Dict[str, int] = dict(zip(ascii_lowercase, range(26)))
 
-    def __init__(self, score: int = 0, logger: Logger = None):
+    def __init__(
+        self,
+        score: int = 0,
+        logger: Logger = None,
+        error_mode: ErrorMode = ErrorMode.TruncTrace,
+    ):
         self.logger = logger or LOGGER.getChild(self.__class__.__name__)
+        self.error_mode = error_mode
         self.score = score
         self.products_with_cve = 0
         self.products_without_cve = 0
@@ -185,11 +192,16 @@ def get_cves(
             if cves:
                 self.products_with_cve += 1
                 self.logger.info(f"Known CVEs in {product}-{version}")
-                self.CONSOLE.print(
-                    Columns(
-                        [Panel(f"[yellow]{cve.cve_number}[/yellow]") for cve in cves]
-                    ),
-                )
+                # error_mode.value will only be greater than 1 if quiet mode.
+                if self.error_mode.value > 1:
+                    self.CONSOLE.print(
+                        Columns(
+                            [
+                                Panel(f"[yellow]{cve.cve_number}[/yellow]")
+                                for cve in cves
+                            ]
+                        ),
+                    )
             else:
                 # No cves found for (product, vendor, version) tuple in the NVD database.
                 self.products_without_cve += 1

cve_bin_tool/cvedb.py

@@ -210,12 +210,19 @@ async def refresh(self):
                 ],
             )
         )
-        total_tasks = len(nvd_metadata) + 1
-        for task in track(
-            asyncio.as_completed(tasks),
-            description="Downloading CVEs...",
-            total=total_tasks,
-        ):
+        total_tasks = len(tasks)
+
+        # error_mode.value will only be greater than 1 if quiet mode.
+        if self.error_mode.value > 1:
+            iter_tasks = track(
+                asyncio.as_completed(tasks),
+                description="Downloading CVEs...",
+                total=total_tasks,
+            )
+        else:
+            iter_tasks = asyncio.as_completed(tasks)
+
+        for task in iter_tasks:
             await task
         self.was_updated = True
         await self.session.close()
@@ -310,7 +317,13 @@ def populate_db(self):
         """
         del_cve_range = "DELETE from cve_range where CVE_number=?"
 
-        for year in track(self.nvd_years(), description="Updating CVEs from NVD..."):
+        # error_mode.value will only be greater than 1 if quiet mode.
+        if self.error_mode.value > 1:
+            years = track(self.nvd_years(), description="Updating CVEs from NVD...")
+        else:
+            years = self.nvd_years()
+
+        for year in years:
             cve_data = self.load_nvd_year(year)
             self.LOGGER.debug(
                 f'Time = {datetime.datetime.today().strftime("%H:%M:%S")}'

cve_bin_tool/egg_updater.py

@@ -26,6 +26,7 @@ def update_egg():
         cwd = os.getcwd()
         os.chdir(os.path.join(os.path.dirname(__file__), ".."))
         sys.stdout = f
+        sys.stderr = f
         dist = Distribution(
             dict(
                 script_name="setup.py",
@@ -56,6 +57,7 @@ def update_egg():
         dist.parse_command_line()
         dist.run_commands()
         sys.stdout = sys.__stdout__
+        sys.stderr = sys.__stderr__
         os.chdir(cwd)
 
 

requirements.txt

@@ -1,5 +1,5 @@
 jsonschema
-rich
+rich==4.0.0
 plotly
 jinja2
 beautifulsoup4