Index: git_cl.py
|
diff --git a/git_cl.py b/git_cl.py
|
index bb31fa0ad9705de247d21e310e4e7f4fe16d7e27..d8fd642246aaff5e6f64f02b96b497517d4f74aa 100755
|
--- a/git_cl.py
|
+++ b/git_cl.py
|
@@ -17,9 +17,11 @@
|
import logging
|
import optparse
|
import os
|
+import Queue
|
import re
|
import stat
|
import sys
|
+import tempfile
|
import textwrap
|
import time
|
import traceback
|
@@ -2897,8 +2899,20 @@
|
'error': Fore.WHITE,
|
}.get(status, Fore.WHITE)
|
|
+def fetch_cl_status(branch, auth_config=None):
|
+ """Fetches information for an issue and returns (branch, issue, status)."""
|
+ cl = Changelist(branchref=branch, auth_config=auth_config)
|
+ url = cl.GetIssueURL()
|
+ status = cl.GetStatus()
|
+
|
+ if url and (not status or status == 'error'):
|
+ # The issue probably doesn't exist anymore.
|
+ url += ' (broken)'
|
+
|
+ return (branch, url, status)
|
+
|
def get_cl_statuses(
|
- changes, fine_grained, max_processes=None):
|
+ branches, fine_grained, max_processes=None, auth_config=None):
|
"""Returns a blocking iterable of (branch, issue, color) for given branches.
|
|
If fine_grained is true, this will fetch CL statuses from the server.
|
@@ -2914,21 +2928,23 @@
|
if fine_grained:
|
# Process one branch synchronously to work through authentication, then
|
# spawn processes to process all the other branches in parallel.
|
- if changes:
|
- fetch = lambda cl: (cl, cl.GetStatus())
|
- yield fetch(changes[0])
|
-
|
- changes_to_fetch = changes[1:]
|
+ if branches:
|
+ fetch = lambda branch: fetch_cl_status(branch, auth_config=auth_config)
|
+ yield fetch(branches[0])
|
+
|
+ branches_to_fetch = branches[1:]
|
pool = ThreadPool(
|
- min(max_processes, len(changes_to_fetch))
|
+ min(max_processes, len(branches_to_fetch))
|
if max_processes is not None
|
- else len(changes_to_fetch))
|
- for x in pool.imap_unordered(fetch, changes_to_fetch):
|
+ else len(branches_to_fetch))
|
+ for x in pool.imap_unordered(fetch, branches_to_fetch):
|
yield x
|
else:
|
# Do not use GetApprovingReviewers(), since it requires an HTTP request.
|
- for cl in changes:
|
- yield (cl, 'waiting' if cl.GetIssueURL() else 'error')
|
+ for b in branches:
|
+ cl = Changelist(branchref=b, auth_config=auth_config)
|
+ url = cl.GetIssueURL()
|
+ yield (b, url, 'waiting' if url else 'error')
|
|
|
def upload_branch_deps(cl, args):
|
@@ -3081,27 +3097,25 @@
|
print('No local branch found.')
|
return 0
|
|
- changes = [
|
+ changes = (
|
Changelist(branchref=b, auth_config=auth_config)
|
- for b in branches.splitlines()]
|
+ for b in branches.splitlines())
|
+ # TODO(tandrii): refactor to use CLs list instead of branches list.
|
+ branches = [c.GetBranch() for c in changes]
|
+ alignment = max(5, max(len(b) for b in branches))
|
print 'Branches associated with reviews:'
|
- output = get_cl_statuses(changes,
|
+ output = get_cl_statuses(branches,
|
fine_grained=not options.fast,
|
- max_processes=options.maxjobs)
|
+ max_processes=options.maxjobs,
|
+ auth_config=auth_config)
|
|
branch_statuses = {}
|
- alignment = max(5, max(len(ShortBranchName(c.GetBranch())) for c in changes))
|
- for cl in sorted(changes, key=lambda c: c.GetBranch()):
|
- branch = cl.GetBranch()
|
+ alignment = max(5, max(len(ShortBranchName(b)) for b in branches))
|
+ for branch in sorted(branches):
|
while branch not in branch_statuses:
|
- c, status = output.next()
|
- branch_statuses[c.GetBranch()] = status
|
- status = branch_statuses.pop(branch)
|
- url = cl.GetIssueURL()
|
- if url and (not status or status == 'error'):
|
- # The issue probably doesn't exist anymore.
|
- url += ' (broken)'
|
-
|
+ b, i, status = output.next()
|
+ branch_statuses[b] = (i, status)
|
+ issue_url, status = branch_statuses.pop(branch)
|
color = color_for_status(status)
|
reset = Fore.RESET
|
if not setup_color.IS_TTY:
|
@@ -3109,8 +3123,8 @@
|
reset = ''
|
status_str = '(%s)' % status if status else ''
|
print ' %*s : %s%s %s%s' % (
|
- alignment, ShortBranchName(branch), color, url,
|
- status_str, reset)
|
+ alignment, ShortBranchName(branch), color, issue_url, status_str,
|
+ reset)
|
|
cl = Changelist(auth_config=auth_config)
|
print
|
|