46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""
|
|
Drop a database.
|
|
"""
|
|
|
|
import common
|
|
|
|
# TODO turn template1 in a parameter
|
|
# This should be exposed from openerp (currently in
|
|
# openerp/service/web_services.py).
|
|
def drop_database(database_name):
|
|
import openerp
|
|
db = openerp.sql_db.db_connect('template1')
|
|
cr = db.cursor()
|
|
cr.autocommit(True) # avoid transaction block
|
|
try:
|
|
# TODO option for doing this.
|
|
# Try to terminate all other connections that might prevent
|
|
# dropping the database
|
|
try:
|
|
cr.execute("""SELECT pg_terminate_backend(procpid)
|
|
FROM pg_stat_activity
|
|
WHERE datname = %s AND
|
|
procpid != pg_backend_pid()""",
|
|
(database_name,))
|
|
except Exception:
|
|
pass
|
|
|
|
try:
|
|
cr.execute('DROP DATABASE "%s"' % database_name)
|
|
except Exception, e:
|
|
print "Can't drop %s" % (database_name,)
|
|
finally:
|
|
cr.close()
|
|
|
|
def run(args):
|
|
assert args.database
|
|
drop_database(args.database)
|
|
|
|
def add_parser(subparsers):
|
|
parser = subparsers.add_parser('drop',
|
|
description='Drop a database.')
|
|
parser.add_argument('-d', '--database', metavar='DATABASE',
|
|
**common.required_or_default('DATABASE', 'the database to create'))
|
|
|
|
parser.set_defaults(run=run)
|