bec/bec_atlas
0
0
Fork 0
mirror of https://github.com/bec-project/bec_atlas.git synced 2025-07-14 07:01:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: do not raise if deployment does not exist in db

Browse Source
This commit is contained in:
wakonig_k
2025-01-13 17:41:14 +01:00
parent ef637cc3a8
commit d1c42aa044
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/bec_atlas/ingestor/data_ingestor.py
View File

@ -194,6 +194,8 @@ class DataIngestor:
out = self.datasource.db["sessions"].find_one( out = self.datasource.db["sessions"].find_one(
{"name": "_default_", "deployment_id": ObjectId(deployment_id)} {"name": "_default_", "deployment_id": ObjectId(deployment_id)}
) )
if out is None:
return None
return out["_id"] return out["_id"]
def update_scan_status(self, msg: messages.ScanStatusMessage, deployment_id: str): def update_scan_status(self, msg: messages.ScanStatusMessage, deployment_id: str):
@ -215,6 +217,9 @@ class DataIngestor:
if session_id == "_default_": if session_id == "_default_":
session_id = self.get_default_session_id(deployment_id) session_id = self.get_default_session_id(deployment_id)
if session_id is None:
logger.error("Default session not found.")
return
# scans are indexed by the scan_id, hence we can use find_one and search by the ObjectId # scans are indexed by the scan_id, hence we can use find_one and search by the ObjectId
data = self.datasource.db["scans"].find_one({"_id": msg.scan_id}) data = self.datasource.db["scans"].find_one({"_id": msg.scan_id})