Support reading directly from a config

This commit is contained in:
Erik Johnston 2015-05-08 13:45:58 +01:00
parent 326121aec4
commit 4fa0f53521

View file

@ -723,6 +723,9 @@ if __name__ == "__main__":
postgres_config = yaml.safe_load(args.postgres_config) postgres_config = yaml.safe_load(args.postgres_config)
if "database" in postgres_config:
postgres_config = postgres_config["database"]
if "name" not in postgres_config: if "name" not in postgres_config:
sys.stderr.write("Malformed database config: no 'name'") sys.stderr.write("Malformed database config: no 'name'")
sys.exit(2) sys.exit(2)