diff options
| author | Marco Capitani <m.capitani@nextworks.it> | 2017-11-15 15:50:55 +0100 |
|---|---|---|
| committer | Marco Capitani <m.capitani@nextworks.it> | 2017-11-15 15:50:55 +0100 |
| commit | 0801eba70daacddd5692543dea013e85a82ef76b (patch) | |
| tree | 3ee971d14f79e37520052d5ae517bcb5e0628b93 | |
| parent | 28a89b20caea996429515aba69d9332d690e8033 (diff) | |
| download | rumba-0801eba70daacddd5692543dea013e85a82ef76b.tar.gz rumba-0801eba70daacddd5692543dea013e85a82ef76b.zip | |
utils: add swap_out after prompt in ExperimentManager
| -rw-r--r-- | rumba/utils.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/rumba/utils.py b/rumba/utils.py index 5348b0a..2fc623c 100644 --- a/rumba/utils.py +++ b/rumba/utils.py @@ -8,7 +8,11 @@ logger = log.get_logger(__name__) class ExperimentManager(object): - def __init__(self, experiment, do_swap_out=True): + PROMPT = 1 + AUTO = 2 + NO = 3 + + def __init__(self, experiment, do_swap_out=AUTO): assert isinstance(experiment, model.Experiment), \ 'An experiment instance is required.' self.experiment = experiment @@ -18,7 +22,11 @@ class ExperimentManager(object): pass def __exit__(self, exc_type, exc_val, exc_tb): - if self.do_swap_out: + if self.do_swap_out == self.PROMPT: + logger.info('Press ENTER to start swap out.') + input('') + if self.do_swap_out == self.PROMPT \ + or self.do_swap_out == self.AUTO: self.experiment.swap_out() if exc_val is not None: logger.error('Something went wrong. Got %s: %s', |