showard | d1ee1dd | 2009-01-07 21:33:08 +0000 | [diff] [blame] | 1 | import common |
| 2 | from autotest_lib.client.common_lib import global_config |
| 3 | |
| 4 | CONFIG_SECTION = 'SCHEDULER' |
| 5 | |
| 6 | class SchedulerConfig(object): |
| 7 | """ |
| 8 | Contains configuration that can be changed during scheduler execution. |
| 9 | """ |
showard | 8dbd05a | 2010-01-12 18:54:59 +0000 | [diff] [blame] | 10 | FIELDS = {'max_processes_per_drone': 'max_processes_per_drone', |
| 11 | 'max_processes_started_per_cycle': 'max_jobs_started_per_cycle', |
| 12 | 'clean_interval': 'clean_interval_minutes', |
| 13 | 'max_parse_processes': 'max_parse_processes', |
| 14 | 'tick_pause_sec': 'tick_pause_sec', |
| 15 | 'max_transfer_processes': 'max_transfer_processes', |
| 16 | 'secs_to_wait_for_atomic_group_hosts': |
showard | 7718256 | 2009-06-10 00:16:05 +0000 | [diff] [blame] | 17 | 'secs_to_wait_for_atomic_group_hosts', |
showard | 8dbd05a | 2010-01-12 18:54:59 +0000 | [diff] [blame] | 18 | 'reverify_period_minutes': 'reverify_period_minutes', |
showard | d1ee1dd | 2009-01-07 21:33:08 +0000 | [diff] [blame] | 19 | } |
| 20 | |
showard | 7718256 | 2009-06-10 00:16:05 +0000 | [diff] [blame] | 21 | |
showard | d1ee1dd | 2009-01-07 21:33:08 +0000 | [diff] [blame] | 22 | def __init__(self): |
| 23 | self.read_config() |
| 24 | |
| 25 | |
| 26 | def read_config(self): |
| 27 | config = global_config.global_config |
| 28 | config.parse_config_file() |
| 29 | for field, config_option in self.FIELDS.iteritems(): |
| 30 | setattr(self, field, config.get_config_value(CONFIG_SECTION, |
| 31 | config_option, |
| 32 | type=int)) |
| 33 | |
showard | 7718256 | 2009-06-10 00:16:05 +0000 | [diff] [blame] | 34 | |
showard | d1ee1dd | 2009-01-07 21:33:08 +0000 | [diff] [blame] | 35 | config = SchedulerConfig() |