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):
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