R3127/scripts/saveBDStudy.py
2141fd5fce36features/zeo

saveBDStudy.py
View Options

			#!/usr/bin/env python3

			import BlackDynamite as BD
			import os
			import 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"] is 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 isinstance(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()