Skip to content

Commit 109c6b1

Browse files
committed
not_landed: remove duplicate request to get user nicks
1 parent 14fb24a commit 109c6b1

1 file changed

Lines changed: 8 additions & 24 deletions

File tree

bugbot/rules/not_landed.py

Lines changed: 8 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -263,37 +263,18 @@ def get_bz_userid(self, phids):
263263
return {}
264264

265265
def handler(user, data):
266-
data[str(user["id"])] = user["name"]
266+
data[str(user["id"])] = (user["name"], user["nick"])
267267

268268
data = {}
269269
BugzillaUser(
270270
user_names=list(users.values()),
271-
include_fields=["id", "name"],
271+
include_fields=["id", "name", "nick"],
272272
user_handler=handler,
273273
user_data=data,
274274
).wait()
275275

276276
return {phid: data[id] for phid, id in users.items()}
277277

278-
def get_nicks(self, nicknames):
279-
def handler(user, data):
280-
data[user["name"]] = user["nick"]
281-
282-
users = set(nicknames.values())
283-
data = {}
284-
if users:
285-
BugzillaUser(
286-
user_names=list(users),
287-
include_fields=["name", "nick"],
288-
user_handler=handler,
289-
user_data=data,
290-
).wait()
291-
292-
for bugid, name in nicknames.items():
293-
nicknames[bugid] = (name, data[name])
294-
295-
return nicknames
296-
297278
def get_bz_params(self, date):
298279
self.date = lmdutils.get_date_ymd(date)
299280
fields = ["flags", "depends_on"]
@@ -340,7 +321,6 @@ def get_bugs(self, date="today", bug_ids=[]):
340321

341322
bz_reviewers = self.get_bz_userid(reviewers_phid)
342323
all_reviewers = set(bz_reviewers.keys())
343-
nicknames = self.get_nicks(nicknames)
344324

345325
for bugid, data in bugs_patch.items():
346326
res[bugid] = d = bugs[bugid]
@@ -349,7 +329,7 @@ def get_bugs(self, date="today", bug_ids=[]):
349329
nickname = d["nickname"]
350330

351331
if not assignee:
352-
assignee, nickname = nicknames[bugid]
332+
assignee, nickname = bz_reviewers[nicknames[bugid]]
353333

354334
if not assignee:
355335
continue
@@ -360,7 +340,11 @@ def get_bugs(self, date="today", bug_ids=[]):
360340
if common:
361341
reviewer = random.choice(list(common))
362342
self.add_auto_ni(
363-
bugid, {"mail": bz_reviewers[reviewer], "nickname": None}
343+
bugid,
344+
{
345+
"mail": bz_reviewers[reviewer][0],
346+
"nickname": bz_reviewers[reviewer][1],
347+
},
364348
)
365349

366350
return res

0 commit comments

Comments
 (0)