source: TI12-security/trunk/python/ndg_security_server/ndg/security/server/initCredReposDb.py @ 6069

#!/usr/bin/env python
"""NDG Security - initialisation script for MySQL Credential Repository
database

Use with care!  It initialise all the tables in the database.

NERC Data Grid Project
"""
__author__ = "P J Kershaw"
__date__ = "25/04/06"
__copyright__ = "(C) 2009 Science and Technology Facilities Council"
__license__ = "BSD - see LICENSE file in top-level directory"
__contact__ = "Philip.Kershaw@stfc.ac.uk"
__author__ = "P J Kershaw"
__revision__ = "$Id: $"

# Command line processing
import sys
import os
import optparse
import getpass

from ndg.security.server.Session import *


def main():
    parser = optparse.OptionParser()
    parser.add_option("-u",
                      "--username",
                      dest="username",
                      help="Database username")

    parser.add_option("-n",
                      "--hostname",
                      dest="hostname",
                      help="database hostname - default is \"localhost\"")

    parser.add_option("-D",
                      "--database-name",
                      dest="dbName",
                      help="database name - default is \"ndgCredentialRepository\"")

    (opt, args) = parser.parse_args()

    if not opt.username:
        sys.stderr.write("Error, No username set.\n\n")
        parser.print_help()
        sys.exit(1)

    if not opt.hostname:
        opt.hostname = "localhost"

    if not opt.dbName:
        opt.dbName = "ndgCredentialRepository"


    # Obtain from prompt
    try:
        password = getpass.getpass(prompt="Database password: ")
    except KeyboardInterrupt:
        sys.exit(1)


    try:
        dbURI = "mysql://%s:%s@%s/%s" % \
                            (opt.username, password, opt.hostname, opt.dbName)
        credentialRepository = SessionManagerCredentialRepository(dbURI=dbURI)
    except Exception, e:
        sys.stderr.write("%s\n" % str(e))
        sys.exit(1)

    # This method prompts the user for confirmation of table initialisation
    try:
        credentialRepository._initTables()
    except Exception, e:
        sys.stderr.write("Error creating tables: %s\n" % str(e))

if __name__ == "__main__":
    main()