Skip to content

Commit acfb144

Browse files
Copilotjoshuaunity
andauthored
refactor: eliminate duplicated save/enqueue logic in save_and_enqueue fallback
Agent-Logs-Url: https://github.com/FlexMeasures/flexmeasures/sessions/a4ce9ca2-10ca-4304-8df3-9cb99864f075 Co-authored-by: joshuaunity <[email protected]>
1 parent cb133ee commit acfb144

1 file changed

Lines changed: 5 additions & 7 deletions

File tree

flexmeasures/api/common/utils/api_utils.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
)
2626
from flexmeasures.data.models.generic_assets import GenericAsset
2727
from flexmeasures.data.models.time_series import Sensor
28-
from flexmeasures.data.utils import save_to_db
2928
from flexmeasures.auth.policy import check_access
3029
from flexmeasures.api.common.responses import (
3130
invalid_replacement,
@@ -171,12 +170,11 @@ def save_and_enqueue(
171170
)
172171

173172
# Attempt to save directly (fallback when no ingestion queue or workers are available)
174-
status = save_to_db(data, save_changed_beliefs_only=save_changed_beliefs_only)
175-
db.session.commit()
176-
177-
# Only enqueue forecasting jobs upon successfully saving new data
178-
if status[:7] == "success" and status != "success_but_nothing_new":
179-
enqueue_forecasting_jobs(forecasting_jobs)
173+
status = add_beliefs_to_db_and_enqueue_forecasting_jobs(
174+
data,
175+
forecasting_jobs=forecasting_jobs,
176+
save_changed_beliefs_only=save_changed_beliefs_only,
177+
)
180178

181179
# Pick a response
182180
if status == "success":

0 commit comments

Comments
 (0)