~launchpad-pqm/launchpad/devel

#!/usr/bin/python -S

#

# Copyright 2009 Canonical Ltd.  This software is licensed under the

# GNU Affero General Public License version 3 (see the file LICENSE).

"""Create dummy Bazaar branches for all HOSTED branches in the database.

The Launchpad sample data includes a number of HOSTED branches that users can

branch from, push to and view on the website. However, some of these things

will break if we are missing the actual Bazaar branches, so we have a script

to create them.

NOTE: This script will delete any existing sample data branches, so that the

sample data on the filesystem is consistent with the sample data in the

database.

"""

import _pythonpath

import os

import shutil

import sys

import tempfile

import transaction

from zope.component import getUtility

# Until we completely kill canonical.launchpad.database.__init__.py and remove

# all of the import *'s we need to import it first here.

import canonical.launchpad.database

from lp.code.enums import BranchType

from lp.code.model.branch import Branch

from lp.codehosting.tests.helpers import make_bazaar_branch_and_tree

from canonical.config import config

from canonical.database.sqlbase import sqlvalues

from canonical.launchpad.scripts import execute_zcml_for_scripts

def main(argv):

    os.environ['BZR_HOME'] = tempfile.mkdtemp()

    if os.path.exists(config.codehosting.hosted_branches_root):

        shutil.rmtree(config.codehosting.hosted_branches_root)

    execute_zcml_for_scripts()

    try:

        branches = Branch.select(

            "Branch.branch_type = %s" % sqlvalues(BranchType.HOSTED))

        for branch in branches:

            make_bazaar_branch_and_tree(branch)

    finally:

        transaction.abort()

    print "Created %d branches based on sample data." % len(list(branches))

if __name__ == '__main__':

    main(sys.argv)