From 5fc2c0c7f6124770d19356dbac44fe7b92e340e3 Mon Sep 17 00:00:00 2001 From: "Brian S. Stephan" Date: Mon, 27 Jun 2016 10:24:40 -0500 Subject: [PATCH] gitlab bot: merge request IDs arg to scanner for use cases when we know the merge request(s) to scan, allow passing them as an optional argument --- dr_botzo/gitlab_bot/lib.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/dr_botzo/gitlab_bot/lib.py b/dr_botzo/gitlab_bot/lib.py index 9d07152..603b06d 100644 --- a/dr_botzo/gitlab_bot/lib.py +++ b/dr_botzo/gitlab_bot/lib.py @@ -57,12 +57,19 @@ class GitlabBot(object): def random_reviews_done_message(self): return "{0:s} {1:s}".format(self.NEW_ACCEPTER_FORMAT, random.choice(self.NOTE_COMMENT)) - def scan_project_for_reviews(self, project): + def scan_project_for_reviews(self, project, merge_request_ids=None): project_obj = self.client.projects.get(project.project_id) if not project_obj: return - for merge_request in project_obj.mergerequests.list(state='opened'): + if merge_request_ids: + merge_requests = [] + for merge_request_id in merge_request_ids: + merge_requests.append(project_obj.mergerequests.get(id=merge_request_id)) + else: + merge_requests = project_obj.mergerequests.list(state='opened') + + for merge_request in merge_requests: log.debug("scanning merge request '%s'", merge_request.title) # moved this method into here