Synnefo » snf-ganeti

import time

  def _InterruptibleDelay(self):

    """Delays but provides the mechanisms necessary to interrupt the delay as

    needed.

    """

    socket_wrapper = TestSocketWrapper()

    sock, path = socket_wrapper.Create()

    self.Log(constants.ELOG_DELAY_TEST, (path,))

    try:

      sock.settimeout(self.op.duration)

      start = time.time()

      (conn, _) = sock.accept()

    except socket.timeout, _:

      # If we timed out, all is well

      return False

    finally:

      # Destroys the original socket, but the new connection is still usable

      socket_wrapper.Destroy()

    try:

      # Change to remaining time

      time_to_go = self.op.duration - (time.time() - start)

      self.Log(constants.ELOG_MESSAGE,

               "Received connection, time to go is %d" % time_to_go)

      if time_to_go < 0:

        time_to_go = 0

      # pylint: disable=E1101

      # Instance of '_socketobject' has no ... member

      conn.settimeout(time_to_go)

      conn.recv(1)

      # pylint: enable=E1101

    except socket.timeout, _:

      # A second timeout can occur if no data is sent

      return False

    finally:

      conn.close()

    self.Log(constants.ELOG_MESSAGE,

             "Interrupted, time spent waiting: %d" % (time.time() - start))

    # Reaching this point means we were interrupted

    return True

  def _UninterruptibleDelay(self):

    """Delays without allowing interruptions.

  def _TestDelay(self):

    """Do the actual sleep.

    @rtype: bool

    @return: Whether the delay was interrupted

    """

    if self.op.interruptible:

      return self._InterruptibleDelay()

    else:

      self._UninterruptibleDelay()

      return False

      i = self._TestDelay()

        # Break in case of interruption

        if self._TestDelay():

          break