def wait()

in pygenie/jobs/running.py [0:0]

• 21 lines of code
• 11 McCabe index (conditional complexity)

    def wait(self, sleep_seconds=10, suppress_stream=False, until_running=False,
             job_timeout=None, kill_after_job_timeout=False):
        """
        Blocking call that will wait for the job to complete.

        Example:
            >>> running_job.wait()

        Args:
            sleep_seconds (int, optional): The number of seconds to sleep while
                polling to get job status (default: 10).
            suppress_stream (bool, optional): If True, do not write anything to
                the sys stream (stderr/stdout) while waiting for the job to
                complete (default: False).
            until_running (bool, optional): If True, only block until the job
                status becomes 'RUNNING' (default: False).
            job_timeout (int, optional): The number of seconds to wait for the
                job to finish (default: None).
            kill_after_job_timeout (bool, optional): Whether to kill the job or
                not after the job_timeout (default: False).

        Returns:
            :py:class:`RunningJob`: self
        """

        i = 0

        statuses = {s for s in RUNNING_STATUSES \
            if not until_running or s.upper() != 'RUNNING'}

        start_time = time.time()

        while self._adapter.get_status(self._job_id).upper() in statuses:
            if i % 3 == 0 and not suppress_stream:
                self._write_to_stream('.')
            time.sleep(sleep_seconds)

            # handle client-side job timeout
            if (job_timeout is not None) and (time.time() - start_time > job_timeout):
                if kill_after_job_timeout:
                    self.kill()
                    return self
                else:
                    raise JobTimeoutError("Timed out while waiting for job {} to finish" \
                        .format(self.job_id))

            i += 1

        if not suppress_stream:
            self._write_to_stream('\n')

        return self