possible fix for cve-search/cve-search#1103

cve-search · Jun 26, 2024 · 24f11ca · 24f11ca
1 parent fec39f7
commit 24f11ca
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
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
  )
  )