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