Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix 588: CLI command add schedule for-storage not working without --as-job #589

Merged
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
5 changes: 2 additions & 3 deletions flexmeasures/cli/data_add.py
Expand Up @@ -1000,7 +1000,6 @@ def add_schedule_for_storage(
roundtrip_efficiency = roundtrip_efficiency.magnitude / 100.0

scheduling_kwargs = dict(
sensor=power_sensor,
start=start,
end=end,
belief_time=server_now(),
Expand All @@ -1019,11 +1018,11 @@ def add_schedule_for_storage(
},
)
if as_job:
job = create_scheduling_job(**scheduling_kwargs)
job = create_scheduling_job(sensor=power_sensor, **scheduling_kwargs)
if job:
print(f"New scheduling job {job.id} has been added to the queue.")
else:
success = make_schedule(**scheduling_kwargs)
success = make_schedule(sensor_id=power_sensor.id, **scheduling_kwargs)
if success:
print("New schedule is stored.")

Expand Down
11 changes: 6 additions & 5 deletions flexmeasures/data/scripts/data_gen.py
Expand Up @@ -303,9 +303,9 @@ def depopulate_prognoses(
% db.engine
)

# Clear all jobs
num_forecasting_jobs_deleted = app.queues["forecasting"].empty()
num_scheduling_jobs_deleted = app.queues["scheduling"].empty()
if not sensor_id:
nhoening marked this conversation as resolved.
Show resolved Hide resolved
num_forecasting_jobs_deleted = app.queues["forecasting"].empty()
num_scheduling_jobs_deleted = app.queues["scheduling"].empty()

# Clear all forecasts (data with positive horizon)
query = db.session.query(TimedBelief).filter(
Expand All @@ -315,8 +315,9 @@ def depopulate_prognoses(
query = query.filter(TimedBelief.sensor_id == sensor_id)
num_forecasts_deleted = query.delete()

click.echo("Deleted %d Forecast Jobs" % num_forecasting_jobs_deleted)
click.echo("Deleted %d Schedule Jobs" % num_scheduling_jobs_deleted)
if not sensor_id:
click.echo("Deleted %d Forecast Jobs" % num_forecasting_jobs_deleted)
click.echo("Deleted %d Schedule Jobs" % num_scheduling_jobs_deleted)
click.echo("Deleted %d forecasts (ex-ante beliefs)" % num_forecasts_deleted)


Expand Down