Defer db load (#79)

Signed-off-by: Prabhu Subramanian <[email protected]>

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "appthreat-vulnerability-db"
-version = "5.5.5"
+version = "5.5.6"
 description = "AppThreat's vulnerability database and package search library with a built-in file based storage. OSV, CVE, GitHub, npm are the primary sources of vulnerabilities."
 authors = [
     {name = "Team AppThreat", email = "[email protected]"},

vdb/lib/nvd.py

@@ -34,8 +34,6 @@
 now = datetime.datetime.now()
 # Size of the stream to read and write to the file
 download_chunk_size = 128
-# Create database
-db = dbLib.get()
 
 json_lib = orjson if ORJSON_AVAILABLE else json
 
@@ -57,6 +55,8 @@ class NvdSource(VulnerabilitySource):
 
     def download_all(self, local_store=True):
         """Download all historic cve data"""
+        # Create database
+        self.db = dbLib.get()
         data_list = []
         for y in range(now.year, int(start_year) - 1, -1):
             data = self.fetch(y)
@@ -68,6 +68,8 @@ def download_all(self, local_store=True):
 
     def download_recent(self, local_store=True):
         """Method which downloads the recent CVE gzip from NVD"""
+        # Create database
+        self.db = dbLib.get()
         data = self.fetch("recent")
         if local_store:
             self.store(data)
@@ -114,10 +116,10 @@ def refresh(self):
 
     def store(self, data, reindex=True):
         """Store data in the database"""
-        docs = dbLib.store(db, data, reindex)
+        docs = dbLib.store(self.db, data, reindex)
         return docs
 
-    def bulk_search():
+    def bulk_search(self):
         """
         Bulk search the resource instead of downloading the information
         :return: Vulnerability result