Open Source Projects Europe

import os

import sys

import shutil

import logging

from pylons import c

from ming.orm import session

from allura import model as M

log = logging.getLogger(__name__)

MONGO_HOME=os.environ.get('MONGO_HOME', '/usr')

MONGO_DUMP=os.path.join(MONGO_HOME, 'bin/mongodump')

MONGO_RESTORE=os.path.join(MONGO_HOME, 'bin/mongorestore')

def main():

    if len(sys.argv) > 1:

        shortnames = sys.argv[1:]

    else:

        shortnames = [ p.shortname for p in M.Project.query.find(dict(is_root=True)) ]

    for pname in shortnames:

        migrate_project_database(pname)

def migrate_project_database(pname):

    c.project = project = M.Project.query.get(shortname=pname)

    if project is None:

        log.fatal('Project %s not found', pname)

        print 'Project %s not found' % pname

        return 2

    if project.database_uri is not None:

        log.fatal('Project %s is already migrated to %s', pname, project.database_uri)

        print 'Project %s already migrated to %s' % (pname, project.database_uri)

        return 2

    M.ProjectRole.query.update(dict(project_id=None), dict(project_id=c.project._id))

    conn = M.session.main_doc_session.db.connection

    host = '%s:%s' % (conn.host, conn.port)

    dirname = os.tempnam()

    try:

        log.info('Backing up %s to %s', pname, dirname)

        assert 0 == os.system('%s --host %s --db %s -o %s' % (

                MONGO_DUMP, host, project.database, dirname))

        assert 0 == os.system('%s --host %s --db project-data %s/%s ' % (

                MONGO_RESTORE, host, dirname, project.database))

        database = project.database

        for p in M.Project.query.find(dict(database=database)):

            p.database = ''

            p.database_uri = 'ming://%s/project-data' % host

        project.ensure_project_indexes()

        session(project).flush()

        conn.drop_database(database)

    finally:

        if os.path.exists(dirname):

            shutil.rmtree(dirname)

    return 0

def pm(etype, value, tb): # pragma no cover

    import pdb, traceback

    try:

        from IPython.ipapi import make_session; make_session()

        from IPython.Debugger import Pdb

        sys.stderr.write('Entering post-mortem IPDB shell\n')

        p = Pdb(color_scheme='Linux')

        p.reset()

        p.setup(None, tb)

        p.print_stack_trace()

        sys.stderr.write('%s: %s\n' % ( etype, value))

        p.cmdloop()

        p.forget()

        # p.interaction(None, tb)

    except ImportError:

        sys.stderr.write('Entering post-mortem PDB shell\n')

        traceback.print_exception(etype, value, tb)

        pdb.post_mortem(tb)

sys.excepthook = pm

if __name__ == '__main__':

    sys.exit(main())