CVEDB: Improve indexing and add foreign key (#902)

Author: Niraj-Kamdar

cve_bin_tool/cvedb.py

@@ -301,13 +301,12 @@ def init_database(self):
             versionStartIncluding TEXT,
             versionStartExcluding TEXT,
             versionEndIncluding TEXT,
-            versionEndExcluding TEXT
+            versionEndExcluding TEXT,
+            FOREIGN KEY(cve_number) REFERENCES cve_severity(cve_number)
         )
         """
-        index_range = "CREATE INDEX IF NOT EXISTS product_index ON cve_range (cve_number, vendor, product)"
         cursor.execute(cve_data_create)
         cursor.execute(version_range_create)
-        cursor.execute(index_range)
 
         # Check that latest schema is being used
         if not self.latest_schema(cursor):
@@ -444,6 +443,11 @@ def populate_db(self):
 
         # supplemental data gets added here
         self.supplement_curl()
+
+        # create index
+        index_range = "CREATE INDEX IF NOT EXISTS product_index ON cve_range (vendor, product, version)"
+        cursor.execute(index_range)
+
         self.db_close()
 
     def parse_node(self, node):