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
This commit is contained in:
Brian S. Stephan 2016-06-27 10:24:40 -05:00
parent e2ed53d031
commit 5fc2c0c7f6
1 changed files with 9 additions and 2 deletions

View File

@ -57,12 +57,19 @@ class GitlabBot(object):
def random_reviews_done_message(self): def random_reviews_done_message(self):
return "{0:s} {1:s}".format(self.NEW_ACCEPTER_FORMAT, random.choice(self.NOTE_COMMENT)) 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) project_obj = self.client.projects.get(project.project_id)
if not project_obj: if not project_obj:
return 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) log.debug("scanning merge request '%s'", merge_request.title)
# moved this method into here # moved this method into here