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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
|
#!/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).
# This script runs through the set of Debbugs watches, and tries to
# syncronise each of those to the malone bug which is watching it.
import _pythonpath
import os
import sys
import email
import logging
# zope bits
from zope.component import getUtility
from canonical.database.constants import UTC_NOW
from lp.app.interfaces.launchpad import ILaunchpadCelebrities
from lp.bugs.interfaces.bug import IBugSet
from lp.bugs.interfaces.bugtask import (
BugTaskSearchParams,
IBugTaskSet,
)
from lp.bugs.interfaces.bugwatch import IBugWatchSet
from lp.bugs.interfaces.cve import ICveSet
from lp.bugs.scripts import debbugs
from lp.services.scripts.base import (LaunchpadCronScript,
LaunchpadScriptFailure)
from lp.services.messages.interfaces.message import (
InvalidEmailMessage,
IMessageSet,
)
# setup core values and defaults
debbugs_location_default = '/srv/bugs-mirror.debian.org/'
class DebWatchUpdater(LaunchpadCronScript):
loglevel = logging.WARNING
def add_my_options(self):
self.parser.add_option('--max', action='store', type='int', dest='max',
default=None, help="The maximum number of bugs to synchronise.")
self.parser.add_option('--debbugs', action='store', type='string',
dest='debbugs',
default=debbugs_location_default,
help="The location of your debbugs database.")
def main(self):
if not os.path.exists(os.path.join(self.options.debbugs, 'index/index.db')):
raise LaunchpadScriptFailure('%s is not a debbugs db.'
% self.options.debbugs)
self.txn.begin()
self.debbugs_db = debbugs.Database(self.options.debbugs)
self.sync()
self.txn.commit()
self.logger.info('Done!')
def sync(self):
changedcounter = 0
self.logger.info('Finding existing debbugs watches...')
debbugs_tracker = getUtility(ILaunchpadCelebrities).debbugs
debwatches = debbugs_tracker.watches
previousimportset = set([b.remotebug for b in debwatches])
self.logger.info('%d debbugs previously imported.' % len(previousimportset))
target_watches = [watch for watch in debwatches if watch.needscheck]
self.logger.info('%d debbugs watches to syncronise.' % len(target_watches))
self.logger.info('Sorting bug watches...')
target_watches.sort(key=lambda a: a.remotebug)
self.logger.info('Syncing bug watches...')
for watch in target_watches:
if self.sync_watch(watch):
changedcounter += 1
self.txn.commit()
if self.options.max:
if changedcounter >= self.options.max:
self.logger.info('Synchronised %d bugs!' % changedcounter)
return
def sync_watch(self, watch):
# keep track of whether or not something changed
waschanged = False
# find the bug in malone
malone_bug = watch.bug
# find the bug in debbugs
debian_bug = self.debbugs_db[int(watch.remotebug)]
bugset = getUtility(IBugSet)
bugtaskset = getUtility(IBugTaskSet)
bugwatchset = getUtility(IBugWatchSet)
messageset = getUtility(IMessageSet)
debian = getUtility(ILaunchpadCelebrities).debian
debbugs_tracker = getUtility(ILaunchpadCelebrities).debbugs
# make sure we have tasks for all the debian package linkages, and also
# make sure we have updated their status and severity appropriately
for packagename in debian_bug.packagelist():
try:
srcpkgname, binpkgname = debian.guessPackageNames(packagename)
except ValueError:
self.logger.error(sys.exc_value)
continue
search_params = BugTaskSearchParams(user=None, bug=malone_bug,
sourcepackagename=srcpkgname)
search_params.setDistribution(debian)
bugtasks = bugtaskset.search(search_params)
if len(bugtasks) == 0:
# we need a new task to link the bug to the debian package
self.logger.info('Linking %d and debian %s/%s' % (
malone_bug.id, srcpkgname.name, binpkgname.name))
# XXX: kiko 2007-02-03:
# This code is completely untested and broken.
bugtask = malone_bug.addTask(
owner=malone_bug.owner, distribution=debian,
sourcepackagename=srcpkgname)
bugtask.bugwatch = watch
waschanged = True
else:
assert len(bugtasks) == 1, 'Should only find a single task'
bugtask = bugtasks[0]
status = bugtask.status
if status <> bugtask.setStatusFromDebbugs(debian_bug.status):
waschanged = True
severity = bugtask.severity
if severity <> bugtask.setSeverityFromDebbugs(debian_bug.severity):
waschanged = True
known_msg_ids = set([msg.rfc822msgid for msg in malone_bug.messages])
for raw_msg in debian_bug.comments:
# parse it so we can extract the message id easily
message = email.message_from_string(raw_msg)
# see if we already have imported a message with this id for this
# bug
message_id = message['message-id']
if message_id in known_msg_ids:
# Skipping msg that is already imported
continue
# make sure this message is in the db
msg = None
try:
msg = messageset.fromEmail(raw_msg, parsed_message=message,
create_missing_persons=True)
except InvalidEmailMessage:
self.logger.error('Invalid email: %s' % sys.exc_value)
if msg is None:
continue
# create the link between the bug and this message
bugmsg = malone_bug.linkMessage(msg)
# ok, this is a new message for this bug, so in effect something has
# changed
waschanged = True
# now we need to analyse the message for useful data
watches = bugwatchset.fromMessage(msg, malone_bug)
for watch in watches:
self.logger.info('New watch for #%s on %s' % (watch.bug.id, watch.url))
waschanged = True
# and also for CVE ref clues
prior_cves = set(malone_bug.cves)
cveset = getUtility(ICveSet)
cves = cveset.inMessage(msg)
for cve in cves:
malone_bug.linkCVE(cve)
if cve not in prior_cves:
self.logger.info('CVE-%s (%s) found for Malone #%s' % (
cve.sequence, cve.status.name, malone_bug.id))
# now we know about this message for this bug
known_msg_ids.add(message_id)
# and best we commit, so that we can see the email that the
# librarian has created in the db
self.txn.commit()
# Mark all merged bugs as duplicates of the lowest-numbered bug
if (len(debian_bug.mergedwith) > 0 and
min(debian_bug.mergedwith) > debian_bug.id):
for merged_id in debian_bug.mergedwith:
merged_bug = bugset.queryByRemoteBug(debbugs_tracker, merged_id)
if merged_bug is not None:
# Bug has been imported already
if merged_bug.duplicateof == malone_bug:
# we already know about this
continue
elif merged_bug.duplicateof is not None:
# Interesting, we think it's a dup of something else
self.logger.warning('Debbugs thinks #%d is a dup of #%d' % (
merged_bug.id, merged_bug.duplicateof))
continue
# Go ahead and merge it
self.logger.info("Malone #%d is a duplicate of Malone #%d" % (
merged_bug.id, malone_bug.id))
merged_bug.duplicateof = malone_bug.id
# the dup status has changed
waschanged = True
# make a note of the remote watch status, if it has changed
if watch.remotestatus <> debian_bug.status:
watch.remotestatus = debian_bug.status
waschanged = True
# update the watch date details
watch.lastchecked = UTC_NOW
if waschanged:
watch.lastchanged = UTC_NOW
self.logger.info('Watch on Malone #%d changed.' % watch.bug.id)
return waschanged
if __name__ == '__main__':
script = DebWatchUpdater('launchpad-debbugs-sync')
script.lock_and_run()
|