sleep.py 776 B

12345678910111213141516171819202122232425
  1. import time
  2. from core.NeuronModule import NeuronModule, MissingParameterException
  3. class Sleep(NeuronModule):
  4. def __init__(self, **kwargs):
  5. super(Sleep, self).__init__(**kwargs)
  6. self.seconds = kwargs.get('seconds', None)
  7. # check parameters
  8. if self._is_parameters_ok():
  9. time.sleep(self.seconds)
  10. def _is_parameters_ok(self):
  11. """
  12. Check if received parameters are ok to perform operations in the neuron
  13. :return: true if parameters are ok, raise an exception otherwise
  14. .. raises:: MissingParameterException
  15. """
  16. if self.seconds is None:
  17. raise MissingParameterException("You must set a number of seconds as parameter")