1import logging 2from autotest_lib.client.common_lib import enum, global_config 3 4# Changing this file has consequences that need to be understood. 5# Adding a protection level to the enum requires you to append your change to 6# the end of the enum or a database migration needs to be added to migrate 7# older protections to match the layout of the new enum. 8# Removing a protection level from the enum requires a database migration to 9# update the integer values in the DB and migrate hosts that use the removed 10# protection to a default protection level. 11# IF THIS IS NOT DONE HOSTS' PROTECTION LEVELS WILL BE CHANGED RANDOMLY. 12 13Protection = enum.Enum('No protection', # Repair can do anything to 14 # this host. 15 'Repair software only', # repair should try to fix any 16 # software problem 17 'Repair filesystem only', # Repair should only try to 18 # recover the file system. 19 'Do not repair', # Repair should not touch this 20 # host. 21 'Do not verify', # Don't even try to verify 22 # this host 23 ) 24 25running_client = global_config.global_config.check_stand_alone_client_run() 26 27try: 28 _bad_value = object() 29 default_protection = global_config.global_config.get_config_value( 30 'HOSTS', 'default_protection', default=_bad_value) 31 if default_protection == _bad_value: 32 if not running_client: 33 raise global_config.ConfigError( 34 'No HOSTS.default_protection defined in global_config.ini') 35 else: 36 default = Protection.get_value(default_protection) 37 38# It is OK to have an empty global configuration object (stand alone client) 39# so we trap this exception. 40except global_config.ConfigError: 41 pass 42 43choices = Protection.choices() 44