~wgrant/launchpad/bootstrap-db-from-scratch

#!/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).

"""Bootstrap a Launchpad database from scratch.

Usage: bootstrap-lp-db <admin username> <admin email address> <openid suffix>

This script will populate an empty Launchpad database with just the

celebrities and other magic required to not have everything crash.

It also creates a user with the given username, email address and OpenID

identifier suffix as the root of everything. The OpenID identifier is arbitrary

if using testopenid.dev, but should match your identity provider otherwise.

"""

import _pythonpath

import os

import sys

import transaction

from zope.component import getUtility

from lp.services.scripts import execute_zcml_for_scripts

from lp.registry.interfaces.distribution import IDistributionSet

from lp.registry.interfaces.person import (

    IPersonSet,

    PersonCreationRationale,

    TeamMembershipPolicy,

    )

from lp.registry.interfaces.product import IProductSet

from lp.registry.model.karma import KarmaAction, KarmaCategory

from lp.services.database.interfaces import IStore

from lp.services.identity.interfaces.account import AccountStatus

from lp.services.identity.interfaces.emailaddress import EmailAddressStatus

from lp.services.identity.interfaces.emailaddress import EmailAddressStatus

from lp.services.openid.model.openididentifier import OpenIdIdentifier

from lp.services.statistics.interfaces.statistic import (

    ILaunchpadStatisticSet)

from lp.services.worlddata.interfaces.language import ILanguageSet

from lp.testing.dbuser import switch_dbuser

# Shut up, pyflakes.

_pythonpath = _pythonpath

def main(arguments):

    """Run the script."""

    if len(arguments) != 3:

        print __doc__

        return 2

    execute_zcml_for_scripts()

    username, email_address, openid_suffix = [

        arg.decode('utf-8') for arg in arguments]

    transaction.begin()

    # Create the person named on the commandline first. It'll own the

    # admin team.

    person, email = getUtility(IPersonSet).createPersonAndEmail(

        email_address, PersonCreationRationale.OWNER_CREATED_LAUNCHPAD,

        name=username, displayname=username)

    person.account.reactivate(u"when bootstrapping the database.")

    person.setPreferredEmail(email)

    identifier = OpenIdIdentifier()

    identifier.account = person.account

    identifier.identifier = openid_suffix

    IStore(OpenIdIdentifier).add(identifier)

    # Create celebrities. Lots of them.

    # There are a few that we don't create:

    # debbugs, debian, gnome_bugzilla, obsolete_junk, savannah_tracker,

    # sourceforge_tracker, ubuntu_bugzilla, ubuntu_archive_mirror,

    # ubuntu_cdimage_mirror.

    # Create the admin team, the owner of most of the other celebrities.

    admins = getUtility(IPersonSet).newTeam(

        person, 'admins', 'Launchpad Administrators',

        membership_policy=TeamMembershipPolicy.RESTRICTED)

    teams = {

        'launchpad-buildd-admins': 'Launchpad Builder Administrators',

        'commercial-admins': 'Commercial Admins',

        'hwdb-team': 'Hardware Database Team',

        'launchpad': 'Launchpad Developers',

        'launchpad-beta-testers': 'Launchpad Beta Testers',

        'launchpad-ppa-self-admins': 'Launchpad PPA Self Admins',

        'registry': 'Registry Admins',

        'rosetta-admins': 'Rosetta Experts',

        'ubuntu-security': 'Ubuntu Security Team',

        'techboard': 'Ubuntu Technical Board',

        'vcs-imports': 'VCS Imports Team',

        }

    for name, displayname in teams.iteritems():

        getUtility(IPersonSet).newTeam(

            admins, name, displayname,

            membership_policy=TeamMembershipPolicy.RESTRICTED)

    people = {

        'bug-importer': 'Bug Importer',

        'bug-watch-updater': 'Bug Watch Updater',

        'janitor': 'Launchpad Janitor',

        'katie': 'Launchpad Autosync',

        'ppa-key-guard': 'PPA Key Guard',

        'software-center-agent': 'Software Center Agent',

        }

    for name, displayname in people.iteritems():

        getUtility(IPersonSet).createPersonAndEmail(

            '%s@example.com' % name, PersonCreationRationale.UNKNOWN,

            name=name, displayname=displayname, hide_email_addresses=True)

    english = getUtility(ILanguageSet).createLanguage(

        'en', 'English', 'English', 2, 'n != 1')

    launchpad_developers = getUtility(IPersonSet).getByName('launchpad')

    launchpad = getUtility(IProductSet).createProduct(

        launchpad_developers, 'launchpad', 'Launchpad', 'Launchpad itself',

        'Launchpad is Launchpad.', 'Launchpad.',

        registrant=launchpad_developers)

    ubuntu = getUtility(IDistributionSet).new(

        'ubuntu', 'Ubuntu', 'Ubuntu', 'Ubuntu', 'Ubuntu', 'ubuntu.com',

        admins, admins, admins)

    # BugTask:+editstatus crashes if Ubuntu isn't set to use Bugs

    # officially, as it's hardcoded as a default for a widget that only

    # allows official_malone pillars.

    ubuntu.official_malone = True

    # Product:+index and various other places assume that

    # ubuntu.currentseries exists. So give it a series.

    # populate-ubuntu-from-scratch.py can create a more complete set

    # later if desired.

    ubuntu.newSeries(

        name='warty', title='Warty Warthog', displayname='Warty',

        summary='Ubuntu 4.10 is good.', description='4.10 is awesome.',

        version='4.10', previous_series=None, registrant=admins)

    # Create some important karmaactions.

    karmacategories = {

        "code": ("Bazaar Branches", "This category for all karma associated "

                 "with branches and the links between branches and other "

                 "Launchpad entities.", {

            "branchmergerejected": (

                15, "Branch merge rejected", "User rejected a proposed "

                "branch merge."),

            "bugbranchcreated": (

                5, "Branch linked to a bug", "User linked a branch to a bug."),

            "specbranchcreated": (

                5, "Branch linked to a blueprint", "User linked a branch to "

                "a blueprint."),

            "branchmergeproposed": (

                1, "Branch merge proposed", "User proposed a branch for "

                "merging."),

            "codereviewcomment": (

                0, "Commented on a review", "User commented on a code "

                "review."),

            "branchmergerejectedown": (

                0, "Branch merge rejected", "User rejected their own proposed "

                "branch merge."),

            "revisionadded": (

                1, "Revision added", "A new revision by the user is available "

                "through Launchpad."),

            "codereviewreviewercomment": (

                10, "Reviewer commented on a review", "Reviewer commented on "

                "a code review."),

            "branchmergeapprovedown": (

                0, "Branch merge approved", "User approved their own branch "

                "for merging."),

            "branchmergeapproved": (

                15, "Branch merge approved", "User approved a branch for "

                "merging."),

            "branchcreated": (

                1, "New branch registered", "User registered a new branch.")

            }),

        "translations": ("Translations in Rosetta", "This categor covers all "

                         "actions related to translation using the Rosetta "

                         "web translation portal. Creating new translation "

                         "projects, submitting new translations and editing "

                         "existing translations will all earn karma.", {

            "translationtemplateimport": (

                1, "Import of Translation Template", "The user updated a "

                "translation template, providing a newer version to be "

                "imported in Rosetta."),

            "translationimportupstream": (

                0, "Upstream Translation Imported", "The user imported a set "

                "of upstream translations into Rosetta"),

            "translationtemplatedescriptionchanged": (

                2, "Edited Translation Template Description", "The user "

                "updated the description of a specific translation template."),

            "translationreview": (

                1, "Translation Review", "The user has completed a review of "

                "suggested translations."),

            "translationsuggestionapproved": (

                1, "Translation Suggestion Approved", "The user approved a "

                "translation suggestion that was previously contributed by "

                "someone else."),

            "translationsuggestionadded": (

                1, "Translation Suggestion", "The user contributed a new "

                "suggested translation. That may not yet have been accepted, "

                "but is valued nonetheless.")

            }),

        "bugs": ("Bug Management", "This karma category covers work in the "

                 "Malone bug tracking system, such as filing, closing and "

                 "otherwise working with bugs.", {

            "bugcreated": (

                10, "New Bug Filed", "The user filed a new bug report. This "

                "is distinct from creating a new \"task\" for an existing "

                "bug, on, say, an upstream product."),

            "bugmarkedasduplicate": (

                5, "Bug Marked as Duplicate", "The user has marked a bug as "

                "a duplicate of another bug. This greatly reduces the amount "

                "of time developers need to spend reviewing existing bug "

                "lists."),

            "bugaccepted": (

                5, "Bug Accepted", "The user has marked a bug as accepted."),

            "bugrejected": (

                3, "Bug Rejected", "The user has rejected a bug."),

            "bugtaskimportancechanged": (

                1, "Bug importance changed", "Updating the importance of a "

                "bug to a specific context. (The importance of a bug can vary "

                "depending on where the buggy code is being used.)"),

            "bugtaskprioritychanged": (

                1, "Bug Priority Changed", "The user has updated the priority "

                "of a particular bug task. Note that bug task has a distinct "

                "priority, because each of them will likely have a different "

                "developer responsible for them."),

            "bugextrefadded": (

                10, "Bug External Reference Added", "The user provided a URL "

                "to information which is relevant to this bug, for example, "

                "to a mailing list archive where it is discussed, or to a "

                "detailed problem report."),

            "bugtitlechanged": (

                1, "Edited Bug Title", "The user edited the title of the bug "

                "to provide a clearer idea of the core issue."),

            "bugdescriptionchanged": (

                3, "Edited Bug Description", "The user edited the bug "

                "description to describe more clearly the specific symptoms "

                "and expected outcomes for the bug. This will also improve "

                "the ability of other users to find this bug report and "

                "avoid creating duplicates."),

            "bugwatchadded": (

                10, "Bug Watch Added", "The user has linked an existing bug "

                "in Launchpad to an external bug tracker, to indicate that "

                "the bug is being tracked in that bug tracker too."),

            "bugfixed": (

                10, "Bug Marked as Fixed", "The user marked a bug as fixed."),

            "bugtaskcreated": (

                10, "Bug Task Created", "The user has created a new task on a "

                "bug. This means that they have indicated that the same bug "

                "exists in another place (for example, upstream) and have "

                "reported that in Malone."),

            "bugsummarychanged": (

                2, "Edited Bug Summary", "The user edited the bug summary. "

                "This will specifically help users searching for existing "

                "bugs in Malone."),

            "bugcommentadded": (

                0, "Bug Comment Added", "The user commented on an existing "

                "bug in Malone."),

            "bugcverefadded": (

                15, "Bug CVE Link Added", "The user has linked a bug report "

                "to a specific entry in the CVE database.")

            }),

        "answers": ("Answer Tracker", "This is the category for all karma "

                    "associated with helping with users questions in the "

                    "Launchpad Answer Tracker. Help solve users problems to "

                    "earn this karma.", {

            "questionownersolved": (

                3, "Solved own question", "User post a message explaining how "

                "he solved his own problem."),

            "questionrejected": (

                0, "Rejected question", "User rejected a question in the "

                "Answer Tracker."),

            "questionanswered": (

                15, "Answered question", "User posed a message that was "

                "accepted by the question owner as answering the question."),

            "questiongaveanswer": (

                0, "Gave answer on a question", "User post a message "

                "containing an answer to a question in the Answer Tracker. "

                "This is distinct from having that message confirmed as "

                "solving the problem."),

            "questionasked": (

                1, "Asked question", "User asked a question in the Answer "

                "Tracker."),

            "questiongaveinfo": (

                0, "Gave more information on a question", "User replied to a "

                "message asking for more information on a question in the "

                "Answer Tracker."),

            "questiontitlechanged": (

                1, "Question title changed", "User changed the title of a "

                "question in the Answer Tracker"),

            "faqcreated": (

                5, "FAQ created", "User create a new FAQ in Launchpad."),

            "questionansweraccepted": (

                5, "Question owner accepted answer", "User accepted one of "

                "the message as the actual answer to his question."),

            "faqedited": (

                1, "FAQ edited", "User updated the details of a FAQ in "

                "Launchpad."),

            "questionlinkedtobug": (

                5, "Question linked to a bug", "User linked a question in the "

                "Answer Tracker to a bug."),

            "questionreopened": (

                0, "Reopened question", "User posed a message to reopen his "

                "question in the Answer Tracker."),

            "questiondescriptionchanged": (

                3, "Question description changed", "User changed the "

                "description of a question in the Answer Tracker"),

            "questionrequestedinfo": (

                0, "Requested for information on a question", "User post a "

                "message requesting for more information from a question "

                "owner in the Answer Tracker."),

            "questioncommentadded": (

                0, "Comment made on a question", "User made a comment on a "

                "ticket in Launchpad.")

            }),

        "soyuz": ("Soyuz", "This category is for all karma associated with "

                  "Soyuz actions.", {

            "ppauploadaccepted": (

                1, "PPA package upload accepted", "A user uploaded a package "

                "to a PPA."),

            "distributionuploadaccepted": (

                10, "Distribution package upload accepted", "A user uploaded "

                "a package to a distribution."),

            "sponsoruploadaccepted": (

                5, "Sponsored package upload accepted", "A package sponsor "

                "uploaded a package to a distribution.")

            }),

        "specs": ("Specification Tracking", "This category includes all karma "

                  "associated with the Launchpad specification tracking "

                  "system.", {

            "specpendingapproval": (

                15, "Specification is Pending Approval", "The user has set "

                "the status of the spec to PendingApproval"),

            "specsummarychanged": (

                2, "Edited Specification Summary", "The user edited the "

                "summary of a specification."),

            "spectitlechanged": (

                2, "Edited Specification Title", "The user edited the title "

                "of a specification."),

            "specdraft": (

                3, "Specification Drafting", "The user has changed the status "

                "of the specification to indicate that drafting has begun."),

            "specseries": (

                5, "Targeted Specification to Series", "The user has "

                "targetted a specification to a particular series."),

            "specbugadded": (

                10, "Linked Bug to Specification", "The user has indicated "

                "that a particular bug is related to a specification."),

            "specpriority": (

                5, "Updated Specification Priority", "The user has changed "

                "the priority of a specification to match the requirements "

                "of the project."),

            "specbugremoved": (

                10, "Removed Bug from Specification", "The user has indicated "

                "that a particular bug is not related to a specification."),

            "specrelease": (

                5, "Targeted Specification to Release", "The user has "

                "targetted a specification to a particular distribution "

                "release."),

            "specmilestone": (

                5, "Targeted Specification to Milestone", "The user has "

                "targetted a specification to a particular milestone."),

            "specurlchanged": (

                2, "Specification URL Updated", "The user edited the URL of a "

                "specification."),

            "specreviewed": (

                10, "Specification Review", "The user has completed a review "

                "of a specification."),

            "addspec": (

                30, "Registered Specification", "The user has registered a "

                "new specification in the Launchpad spec tracker.")

            })

    }

    store = IStore(KarmaAction)

    for cname in karmacategories:

        ctitle, csummary, actions = karmacategories[cname]

        c = KarmaCategory(name=cname, title=ctitle, summary=csummary)

        store.add(c)

        for aname in actions:

            apoints, atitle, asummary = actions[aname]

            store.add(KarmaAction(

                name=aname, title=atitle, summary=asummary, points=apoints,

                category=c))

    switch_dbuser('statistician')

    launchpad_stats = getUtility(ILaunchpadStatisticSet)

    launchpad_stats.updateStatistics(transaction)

    getUtility(IPersonSet).updateStatistics()

    transaction.commit()

    return 0

if __name__ == '__main__':

    sys.exit(main(sys.argv[1:]))