From cb10388c07db937416c974a44cbd9a93d91b580b Mon Sep 17 00:00:00 2001 From: Prabhu Subramanian Date: Sat, 16 Mar 2024 17:35:03 +0000 Subject: [PATCH] Store to sqlite Signed-off-by: Prabhu Subramanian --- vdb/lib/cve.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/vdb/lib/cve.py b/vdb/lib/cve.py index 3486e5d..34bef1c 100644 --- a/vdb/lib/cve.py +++ b/vdb/lib/cve.py @@ -353,6 +353,7 @@ def store5(self, data: list[CVE]): cve_id = d.cveMetadata.cveId if d.containers.cna and d.containers.cna.affected: for affected in d.containers.cna.affected.root: + cve_id = cve_id.model_dump(mode="python") versions: list[Versions] = affected.versions source_data = d.model_dump(mode="json", exclude_defaults=True, @@ -360,7 +361,8 @@ def store5(self, data: list[CVE]): exclude_none=True) self.db_conn.execute( "INSERT INTO cve_data values(?, ?, ?, ?, ?, ?);", ( - cve_id.model_dump(mode="python"), affected.vendor, + cve_id, + affected.vendor, affected.product, affected.packageName, pickle.dumps(source_data), @@ -368,7 +370,8 @@ def store5(self, data: list[CVE]): cleaned_versions = [v.model_dump(mode="json", exclude_none=True) for v in versions] self.index_conn.execute( "INSERT INTO cve_index values(?, ?, ?, ?, json(?));", ( - cve_id.model_dump(mode="python"), affected.vendor, + cve_id, + affected.vendor, affected.product, affected.packageName, orjson.dumps(cleaned_versions).decode("utf-8", "ignore")