diff options
Diffstat (limited to 'client/bin/test_config.py')
| -rw-r--r-- | client/bin/test_config.py | 132 |
1 files changed, 66 insertions, 66 deletions
diff --git a/client/bin/test_config.py b/client/bin/test_config.py index e9268138..d8a53370 100644 --- a/client/bin/test_config.py +++ b/client/bin/test_config.py @@ -13,79 +13,79 @@ from autotest_lib.client.common_lib import utils __all__ = ['config_loader'] class config_loader: - """Base class of the configuration parser""" - def __init__(self, cfg, tmpdir = '/tmp'): - """\ - Instantiate ConfigParser and provide the file like object that we'll - use to read configuration data from. - Args: - * cfg: Where we'll get configuration data. It can be either: - * A URL containing the file - * A valid file path inside the filesystem - * A string containing configuration data - * tmpdir: Where we'll dump the temporary conf files. The default - is the /tmp directory. - """ - # Base Parser - self.parser = ConfigParser() - # File is already a file like object - if hasattr(cfg, 'read'): - self.cfg = cfg - self.parser.readfp(self.cfg) - elif isinstance(cfg, types.StringTypes): - # Config file is a URL. Download it to a temp dir - if cfg.startswith('http') or cfg.startswith('ftp'): - self.cfg = path.join(tmpdir, path.basename(cfg)) - utils.urlretrieve(cfg, self.cfg) - self.parser.read(self.cfg) - # Config is a valid filesystem path to a file. - elif path.exists(path.abspath(cfg)): - if path.isfile(cfg): - self.cfg = path.abspath(cfg) - self.parser.read(self.cfg) - else: - e_msg = 'Invalid config file path: %s' % cfg - raise IOError(e_msg) - # Config file is just a string, convert it to a python file like - # object using StringIO - else: - self.cfg = StringIO(cfg) - self.parser.readfp(self.cfg) + """Base class of the configuration parser""" + def __init__(self, cfg, tmpdir = '/tmp'): + """\ + Instantiate ConfigParser and provide the file like object that we'll + use to read configuration data from. + Args: + * cfg: Where we'll get configuration data. It can be either: + * A URL containing the file + * A valid file path inside the filesystem + * A string containing configuration data + * tmpdir: Where we'll dump the temporary conf files. The default + is the /tmp directory. + """ + # Base Parser + self.parser = ConfigParser() + # File is already a file like object + if hasattr(cfg, 'read'): + self.cfg = cfg + self.parser.readfp(self.cfg) + elif isinstance(cfg, types.StringTypes): + # Config file is a URL. Download it to a temp dir + if cfg.startswith('http') or cfg.startswith('ftp'): + self.cfg = path.join(tmpdir, path.basename(cfg)) + utils.urlretrieve(cfg, self.cfg) + self.parser.read(self.cfg) + # Config is a valid filesystem path to a file. + elif path.exists(path.abspath(cfg)): + if path.isfile(cfg): + self.cfg = path.abspath(cfg) + self.parser.read(self.cfg) + else: + e_msg = 'Invalid config file path: %s' % cfg + raise IOError(e_msg) + # Config file is just a string, convert it to a python file like + # object using StringIO + else: + self.cfg = StringIO(cfg) + self.parser.readfp(self.cfg) - def get(self, section, name, default=None): - """Get the value of a option. + def get(self, section, name, default=None): + """Get the value of a option. - Section of the config file and the option name. - You can pass a default value if the option doesn't exist. - """ - if not self.parser.has_option(section, name): - return default - return self.parser.get(section, name) + Section of the config file and the option name. + You can pass a default value if the option doesn't exist. + """ + if not self.parser.has_option(section, name): + return default + return self.parser.get(section, name) - def set(self, section, option, value): - """Set an option. + def set(self, section, option, value): + """Set an option. - This change is not persistent unless saved with 'save()'. - """ - if not self.parser.has_section(section): - self.parser.add_section(section) - return self.parser.set(section, name, value) + This change is not persistent unless saved with 'save()'. + """ + if not self.parser.has_section(section): + self.parser.add_section(section) + return self.parser.set(section, name, value) - def remove(self, section, name): - """Remove an option.""" - if self.parser.has_section(section): - self.parser.remove_option(section, name) + def remove(self, section, name): + """Remove an option.""" + if self.parser.has_section(section): + self.parser.remove_option(section, name) - def save(self): - """Save the configuration file with all modifications""" - if not self.filename: - return - fileobj = file(self.filename, 'w') - try: - self.parser.write(fileobj) - finally: - fileobj.close() + def save(self): + """Save the configuration file with all modifications""" + if not self.filename: + return + fileobj = file(self.filename, 'w') + try: + self.parser.write(fileobj) + finally: + fileobj.close() |