epss fix (#292)

* fix for cve-search/cve-search#1103 on re-import * fix for cve-search/cve-search#1099
cve-search · Jun 28, 2024 · 06c4047 · 06c4047
1 parent fec39f7
commit 06c4047
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 6 deletions.
diff --git a/CveXplore/.schema_version b/CveXplore/.schema_version
@@ -1,4 +1,4 @@
 {
- "version": "1.8",
+ "version": "1.9",
  "rebuild_needed": true
 }
diff --git a/CveXplore/core/database_actions/db_action.py b/CveXplore/core/database_actions/db_action.py
@@ -13,7 +13,7 @@ class DatabaseAction(object):
 
  actions = collections.namedtuple("Actions", "InsertOne UpdateOne")(0, 1)
 
- def __init__(self, action: actions, doc: dict):
+ def __init__(self, action: actions, doc: dict, upsert: bool = True):
  """
  Create a DatabaseAction object.
 
@@ -23,6 +23,7 @@ def __init__(self, action: actions, doc: dict):
  """
  self.action = action
  self.doc = doc
+ self.upsert = upsert
 
  @property
  def entry(self):
@@ -37,4 +38,6 @@ def entry(self):
  if self.action == self.actions.InsertOne:
  return InsertOne(self.doc)
  elif self.action == self.actions.UpdateOne:
- return UpdateOne({"id": self.doc["id"]}, {"$set": self.doc}, upsert=True)
+ return UpdateOne(
+ {"id": self.doc["id"]}, {"$set": self.doc}, upsert=self.upsert
+ )
diff --git a/CveXplore/core/database_maintenance/sources_process.py b/CveXplore/core/database_maintenance/sources_process.py
@@ -1048,8 +1048,7 @@ def process_item(self, item):
  if epss is not None:
  self.queue.put(
  DatabaseAction(
- action=DatabaseAction.actions.UpdateOne,
- doc=epss,
+ action=DatabaseAction.actions.UpdateOne, doc=epss, upsert=False
  )
  )
 

diff --git a/requirements/modules/mongodb.txt b/requirements/modules/mongodb.txt
@@ -1 +1 @@
-pymongo>=4.5.0
+pymongo>=4.8.0