4 import multiprocessing as mp
5 import powder.experiment as pexp
15 format="[%(asctime)s] %(name)s:%(levelname)s: %(message)s"
20 """Instantiates a Powder experiment based on the provided Powder profile
21 by default the project is 'osc' and the profile is 'ubuntu-20'.
22 It returns the IPv4 address if a profile starts successfully on Powder
25 # default Powder experiment credentials
27 PROFILE_NAME = 'ubuntu-20'
28 EXPERIMENT_NAME_PREFIX = 'osctest-'
30 SUCCEEDED = 0 # all steps succeeded
31 FAILED = 1 # on of the steps failed
33 def __init__(self, experiment_name=None):
34 if experiment_name is not None:
35 self.experiment_name = experiment_name
37 self.experiment_name = self.EXPERIMENT_NAME_PREFIX + self._random_string()
40 self.project_name = os.environ['PROJECT']
42 self.project_name = self.PROJECT_NAME
45 self.profile_name = os.environ['PROFILE']
47 self.profile_name = self.PROFILE_NAME
50 if not self._start_powder_experiment():
51 return self._finish(self.FAILED)
53 return self._finish(self.SUCCEEDED)
55 def _random_string(self, strlen=7):
56 characters = string.ascii_lowercase + string.digits
57 return ''.join(random.choice(characters) for i in range(strlen))
59 def _start_powder_experiment(self):
60 logging.info('Instantiating Powder experiment...')
61 self.exp = pexp.PowderExperiment(experiment_name=self.experiment_name,
62 project_name=self.project_name,
63 profile_name=self.profile_name)
65 exp_status = self.exp.start_and_wait()
66 if exp_status != self.exp.EXPERIMENT_READY:
67 logging.error('Failed to start experiment.')
72 def _finish(self, test_status):
73 if test_status == self.FAILED:
74 logging.info('The experiment could not be started... maybe the resources were unavailable.')
75 return test_status, None
76 elif test_status == self.SUCCEEDED:
77 logging.info('The experiment successfully started.')
78 return test_status, self.exp.ipv4