diff options
author | Marco Capitani <m.capitani@nextworks.it> | 2017-11-13 12:18:55 +0100 |
---|---|---|
committer | Sander Vrijders <sander.vrijders@ugent.be> | 2017-11-13 15:48:27 +0000 |
commit | 510bb2116ea3e6c0d2559fe1b0c465f010d74bfb (patch) | |
tree | 4f1d56841da5e01398557b33e209b58317ab5227 /examples/mouse.py | |
parent | 9d15d1117c80ee9313d0dc391488efb9bb060a4a (diff) | |
download | rumba-510bb2116ea3e6c0d2559fe1b0c465f010d74bfb.tar.gz rumba-510bb2116ea3e6c0d2559fe1b0c465f010d74bfb.zip |
utils, examples: Add ExperimentManager CM
Adds an ExperimentManager context manager for scripts:
- No more ugly stack traces after execution if something goes wrong
- No more ugly and easy to forget try-finally blocks
Diffstat (limited to 'examples/mouse.py')
-rwxr-xr-x | examples/mouse.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/examples/mouse.py b/examples/mouse.py index 5ff5c6a..026b3bc 100755 --- a/examples/mouse.py +++ b/examples/mouse.py @@ -3,6 +3,7 @@ # An example script using the rumba package from rumba.model import * +from rumba.utils import ExperimentManager # import testbed plugins import rumba.testbeds.emulab as emulab @@ -15,7 +16,6 @@ import rumba.prototypes.ouroboros as our import rumba.prototypes.rlite as rl import rumba.prototypes.irati as irati - log.set_logging_level('DEBUG') @@ -102,8 +102,6 @@ exp = rl.Experiment(tb, nodes = [a, b, c, d, e, f, g, h, i, j, k, l, m, n]) print(exp) -try: +with ExperimentManager(exp): exp.swap_in() exp.bootstrap_prototype() -finally: - exp.swap_out() |