def createRetryRun(user_id, reduction_run, overwrite=None, script=None,
variables=None, delay=0, description=''):
"""
Create a run ready for re-running based on the run provided.
If variables (RunVariable) are provided, copy them and associate
them with the new one, otherwise use the previous run's.
If a script (as a string) is supplied then use it, otherwise use the previous run's.
"""
from reduction_variables.utils import InstrumentVariablesUtils
run_last_updated = reduction_run.last_updated
# find the previous run version, so we don't create a duplicate
last_version = -1
# pylint:disable=no-member
previous_run = ReductionRun.objects.filter(experiment=reduction_run.experiment,
run_number=reduction_run.run_number) \
.order_by("-run_version").first()
last_version = previous_run.run_version
# get the script to use:
script_text = script if script is not None else reduction_run.script
# create the run object and save it
new_job = ReductionRun(instrument=reduction_run.instrument,
run_number=reduction_run.run_number,
run_name=description,
run_version=last_version + 1,
experiment=reduction_run.experiment,
started_by=user_id,
status=StatusUtils().get_queued(),
script=script_text,
overwrite=overwrite)
# Check record is safe to save
try:
new_job.full_clean()
# pylint:disable=catching-non-exception
except django.core.exceptions as exception:
LOGGER.error(traceback.format_exc())
LOGGER.error(exception)
raise
# Attempt to save
try:
new_job.save()
except ValueError as exception:
# This usually indicates a F.K. constraint mismatch. Maybe we didn't get a record in?
LOGGER.error(traceback.format_exc())
LOGGER.error(exception)
raise
reduction_run.retry_run = new_job
reduction_run.retry_when = timezone.now().replace(microsecond=0) + datetime.timedelta(
seconds=delay if delay else 0)
reduction_run.save()
# pylint:disable=no-member
ReductionRun.objects.filter(id=reduction_run.id).update(last_updated=run_last_updated)
# copy the previous data locations
# pylint:disable=no-member
for data_location in reduction_run.data_location.all():
new_data_location = DataLocation(file_path=data_location.file_path,
reduction_run=new_job)
new_data_location.save()
new_job.data_location.add(new_data_location)
if variables is not None:
# associate the variables with the new run
for var in variables:
var.reduction_run = new_job
var.save()
else:
# provide variables if they aren't already
InstrumentVariablesUtils().create_variables_for_run(new_job)
return new_job
