Skip to content

Commit

Permalink
individual messages for advance_run_number and get_run_number
Browse files Browse the repository at this point in the history
  • Loading branch information
NichtJens committed Apr 2, 2024
1 parent 3759d6f commit f171c40
Showing 1 changed file with 11 additions and 6 deletions.
17 changes: 11 additions & 6 deletions sf_daq_broker/broker_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,11 +100,15 @@ def set_pvlist(self, request, remote_ip):
return res


def get_current_run_number(self, request, remote_ip, increment_run_number=False):
return self.advance_run_number(request=request, remote_ip=remote_ip, increment_run_number=increment_run_number)
def get_current_run_number(self, request, remote_ip):
return self._get_run_number(request, remote_ip, False)


def advance_run_number(self, request, remote_ip, increment_run_number=True):
def advance_run_number(self, request, remote_ip):
return self._get_run_number(request, remote_ip, True)


def _get_run_number(self, request, remote_ip, increment_run_number):
validate.request_has(request, "pgroup")

beamline = get_beamline(remote_ip)
Expand All @@ -119,12 +123,13 @@ def advance_run_number(self, request, remote_ip, increment_run_number=True):

validate.pgroup_is_not_closed(daq_directory, path_to_pgroup)

next_run = get_run_number(daq_directory, increment_run_number=increment_run_number)
run_number = get_run_number(daq_directory, increment_run_number=increment_run_number)
action = "advanced" if increment_run_number else "retrieved"

res = {
"status": "ok",
"message": f"successfully retrieved run number for {beamline} {pgroup}",
"run_number": next_run
"message": f"successfully {action} run number for {beamline} {pgroup}",
"run_number": run_number
}
return res

Expand Down

0 comments on commit f171c40

Please sign in to comment.