~launchpad-pqm/launchpad/devel

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#!/usr/bin/env python
# Copyright 2005 Canonical Ltd.  All rights reserved.


import sys
import logging
import optparse
import MySQLdb

import _pythonpath

from zope.component import getUtility
from canonical.lp import initZopeless
from canonical.launchpad.scripts import (
    execute_zcml_for_scripts, logger_options, logger)

from canonical.launchpad.scripts import bugzilla


def make_connection(options):
    kws = {}
    if options.db_name is not None:
        kws['db'] = options.db_name
    if options.db_user is not None:
        kws['user'] = options.db_user
    if options.db_password is not None:
        kws['passwd'] = options.db_passwd
    if options.db_host is not None:
        kws['host'] = options.db_host

    return MySQLdb.connect(**kws)
        
def main(argv):
    parser = optparse.OptionParser(
        description="This script imports bugs from a Bugzilla into Launchpad.")

    parser.add_option('--component', metavar='COMPONENT', action='append',
                      help='Limit to this bugzilla component',
                      type='string', dest='component', default=[])
    parser.add_option('--status', metavar='STATUS,...', action='store',
                      help='Only import bugs with the given status',
                      type='string', dest='status',
                      default=None)

    # MySQL connection details
    parser.add_option('-d', '--dbname', metavar='DB', action='store',
                      help='The MySQL database name',
                      type='string', dest='db_name', default='bugs_warty')
    parser.add_option('-U', '--username', metavar='USER', action='store',
                      help='The MySQL user name',
                      type='string', dest='db_user', default=None)
    parser.add_option('-p', '--password', metavar='PASSWORD', action='store',
                      help='The MySQL password',
                      type='string', dest='db_password', default=None)
    parser.add_option('-H', '--host', metavar='HOST', action='store',
                      help='The MySQL database host',
                      type='string', dest='db_host', default=None)

    # logging options
    logger_options(parser, logging.INFO)
    
    options, args = parser.parse_args(argv[1:])
    if options.status is not None:
        options.status = options.status.split(',')
    else:
        options.status = []

    logger(options, 'canonical.launchpad.scripts.bugzilla')

    ztm = initZopeless()
    execute_zcml_for_scripts()

    db = make_connection(options)
    bz = bugzilla.Bugzilla(db)

    bz.importBugs(ztm,
                  product=['Ubuntu'],
                  component=options.component,
                  status=options.status)

    bz.processDuplicates(ztm)

if __name__ == '__main__':
    sys.exit(main(sys.argv))