engine-db-tool.py

#!/usr/bin/python
#
# Copyright (C) 2011-2012
#
# Douglas Schilling Landgraf <dougsland@redhat.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 2 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

import subprocess
import sys
import getopt
import datetime
import os

class Operations:

	def __init__(self):
		self.path = None
		self.selectedOption = None
		self.VERSION = "1.0.0"

	def checkUser(self):
		if os.geteuid() != 0:
			print "You must be root to run this script."
			sys.exit(2)

	def backup(self, path):
		currTime = datetime.datetime.now()

		bkpPath = path +  "/dump_ENGINEDB_BACKUP_" + currTime.strftime("%Y-%m-%d-%H:%M") + ".sql"

		print "Backuping database: %s" % (bkpPath)
		cmd = "pg_dump -C -E UTF8 --column-inserts" \
			" --disable-dollar-quoting --disable-triggers -U postgres --format=p -f " + "\"" + bkpPath + "\"" + " engine"
		ret = self.runCmd(cmd)
		if ret[0] != 0:
			print "cannot backup engine database, please verify!"
			print "Possible options:"
			print "\t - pg_dump command not installed?"
			print "\t - there is no engine database"
			print "\t - are you running postgresql service?"
			sys.exit(-1)

	def restore(self, sqlFile):
		cmd = "/usr/bin/dropdb -U postgres engine"
		self.runCmd(cmd)

		cmd = "/usr/bin/createdb -U postgres engine"
		self.runCmd(cmd)

		print "Restoring database: %s" % (sqlFile)
		cmd = "/usr/bin/psql -U postgres -d engine -w < " + "\"" + sqlFile + "\""
		ret = self.runCmd(cmd)
		if ret[0] != 0:
			print "cannot restore engine database, please verify!"
			print "Possible options:"
			print "\t - have you provided a sql file ?"
			print "\t - psql command not installed?"
			print "\t - there is no engine database"
			print "\t - are you running postgresql service?"
			sys.exit(-1)

	def runCmd(self, cmd):
		process = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
		(output, error) = process.communicate()

		# Let's return the exit value
		return process.poll(), output, error

	def stopJboss(self):
		print "Stopping jboss-as service..."
		self.runCmd("/bin/systemctl stop jboss-as.service")
		ret = self.runCmd("/bin/systemctl status jboss-as.service")

                if not "Active: inactive" in ret[1]:
			print "jboss-as still running, please stop it before continue!"
			sys.exit(-1)

		return 0

	def startJboss(self):
		print "Starting jboss-as service..."
		self.runCmd("/bin/systemctl start jboss-as.service")
		ret = self.runCmd("/bin/systemctl status jboss-as.service")

                if not "Active: active" in ret[1]:
			print "cannot start jboss-as service, please verify!"
			sys.exit(-1)

		return 0

	def usage(self):
		print "Usage: " +  sys.argv[0] + " [OPTIONS]"
		print "\t--backup	 \tBackup engine* databases" 
		print "\t--restore	 \tRestore database"
		print "\t--path		 \tPath to backup or SQL to restore"
		print "\t--version       \tList version release" 
		print "\t--help          \tThis help menu\n"

		print "Example:"
		print "\t" + sys.argv[0] + " --backup --path=/tmp"
		print "\t" + sys.argv[0] + " --restore --path=/tmp/dump_ENGINEDB_BACKUP_2011-11-13-05:29.sql"
		sys.exit(0)

if __name__ == "__main__":

	run = Operations()

	try:
		opts, args = getopt.getopt(sys.argv[1:], "rbp:hv", ["restore", "backup", "path=", "help", "version"])
	except getopt.GetoptError, err:
		# print help info and exit:
		print(err) 
		run.usage()

	for o, a in opts:
		if o in ("-r", "--restore"):
			run.selectedOption = "restore"
		elif o in ("-h", "--help"):
			run.usage()
		elif o in ("-b", "--backup"):
			run.selectedOption = "backup"
		elif o in ("-p", "--path"):
			run.path = a
		elif o in ("-V", "--version"):
			print run.VERSION
			sys.exit(0)
		else:
			assert False, "unhandled option"

	argc = len(sys.argv)

	run.checkUser()

	if argc != 3:
		run.usage()

	if run.selectedOption == None or run.path == None:
		run.usage()

	if os.path.exists(run.path) == False:
		print "path %s is doesn't exist" % (run.path)

	if run.selectedOption == "restore" and os.path.isdir(run.path) == True:
		print "for restore mode, you must provide a sql file"
		sys.exit(-1)
		

	# Stop Jbossas service
	run.stopJboss()

	if run.selectedOption == "backup":
		run.backup(run.path)
	elif run.selectedOption == "restore":
		run.restore(run.path)

	# Start Jbossas service
	run.startJboss()

	print "Done"