saveBDStudy.py
2461e10f6e78master

saveBDStudy.py
View Options

			#!/usr/bin/env python

			import BlackDynamite as BD
			import os, sys
			import subprocess

			def saveSchema(params):
			if ("out_file" not in params):
			out_file = os.path.join("./", params["study"] + ".db")
			else:
			out_file = params["out_file"]

			r, ext = os.path.splitext(out_file)
			if (not ext == ".gz"):
			out_file += ".gz"

			print "Saving study " + params["study"] + " to file " + out_file

			command = "pg_dump --host " + params["host"] + " --schema=" + params["study"] + " -C -f " + out_file + " --compress=9"

			if (params["verbose"] == True):
			command += " --verbose"
			command = command.strip().split(" ")
			ret = subprocess.call(command)
			if not ret == 0:
			sys.exit("pg_dump error")


			def main(argv = None):
			if (type(argv) == str):
			argv = argv.split()

			parser = BD.BDParser()
			parser.register_params(
			group = "saveBDStudy.py",
			params = {"out_file": str, "verbose": bool, "study":str},
			help = {"out_file":"Specify the filename where to save the study",
			"verbose": "Activate the verbose mode of pg_dump",
			"study": "specify the study to backup. \
			If none provided all studies are backed up"})



			params = parser.parseBDParameters(argv = argv)
			params["should_not_check_study"] = True
			mybase = BD.Base(**params)

			if ("study" in params):
			saveSchema(params)
			else:
			if "out_file" in params:
			del params["out_file"]
			sch_list = mybase.getSchemaList()
			for s in sch_list:
			params["study"] = s
			saveSchema(params)


			if __name__ == '__main__':
			main()