Refactor config file reading.

This commit is contained in:
Tommi Virtanen 2011-07-26 13:16:34 -07:00
parent 7a98a496e4
commit 367fd3981a

View file

@ -84,6 +84,13 @@ def nuke_prefixed_buckets():
print 'Done with cleanup of test buckets.' print 'Done with cleanup of test buckets.'
def read_config(fp):
config = bunch.Bunch()
g = yaml.safe_load_all(fp)
for new in g:
config.update(bunch.bunchify(new))
return config
def setup(): def setup():
global s3, config, prefix global s3, config, prefix
s3.clear() s3.clear()
@ -97,9 +104,7 @@ def setup():
+ 'variable S3TEST_CONF to a config file.', + 'variable S3TEST_CONF to a config file.',
) )
with file(path) as f: with file(path) as f:
g = yaml.safe_load_all(f) config.update(read_config(f))
for new in g:
config.update(bunch.bunchify(new))
# These 3 should always be present. # These 3 should always be present.
if 's3' not in config: if 's3' not in config: