fix: Add warning for slow schema updates, bump version to 3.2 (#2447)

* fix: Add warning for slow schema updates, bump version to 3.2

* fix: make latest_schema work if no arguments specified

* fix: improve messages per Anthony's comments

Author: terriko

cve_bin_tool/cvedb.py

@@ -157,9 +157,17 @@ def get_cvelist_if_stale(self) -> None:
                 self.time_of_last_update = datetime.datetime.today()
 
     def latest_schema(
-        self, table_name: str, table_schema: str, cursor: sqlite3.Cursor | None = None
+        self,
+        table_name: str = "",
+        table_schema: str = "",
+        cursor: sqlite3.Cursor | None = None,
     ) -> bool:
         """Check database is using latest schema"""
+        if table_name == "":
+            # If no table specified, check cve_range (the last one changed)
+            _, range_schema = self.table_schemas()
+            return self.latest_schema("cve_range", range_schema)
+
         self.LOGGER.debug("Check database is using latest schema")
         cursor = self.db_open_and_get_cursor()
         schema_check = f"SELECT * FROM {table_name} WHERE 1=0"
@@ -265,13 +273,19 @@ def init_database(self) -> None:
         # Check schema on cve_severity
         if not self.latest_schema("cve_severity", severity_schema, cursor):
             # Recreate table using latest schema
-            self.LOGGER.info("Upgrading database cve_severity to latest schema")
+            self.LOGGER.info("Upgrading cve_severity data. This may take some time.")
+            self.LOGGER.info(
+                "If this step hangs, try using `-u now` to get a fresh db."
+            )
             cursor.execute("DROP TABLE cve_severity")
             cursor.execute(cve_data_create)
 
         # Check schema on cve_range
         if not self.latest_schema("cve_range", range_schema, cursor):
-            self.LOGGER.info("Upgrading database cve_range to latest schema")
+            self.LOGGER.info("Upgrading cve_range data. This may take some time.")
+            self.LOGGER.info(
+                "If this step hangs, try using `-u now` to get a fresh db."
+            )
             cursor.execute("DROP TABLE cve_range")
             cursor.execute(version_range_create)
 

cve_bin_tool/version.py

@@ -8,7 +8,7 @@
 
 from cve_bin_tool.log import LOGGER
 
-VERSION: str = "3.2rc0"
+VERSION: str = "3.2"
 
 
 def check_latest_version():