mirror of
https://chromium.googlesource.com/chromium/tools/depot_tools.git
synced 2026-01-11 18:51:29 +00:00
switch to 4 space indent
Leave the recipes/ code at 2 space to match the rest of the recipes
project in other repos.
Reformatted using:
files=( $(
git ls-tree -r --name-only HEAD | \
grep -Ev -e '^(third_party|recipes)/' | \
grep '\.py$';
git grep -l '#!/usr/bin/env.*python' | grep -v '\.py$'
) )
parallel ./yapf -i -- "${files[@]}"
~/chromiumos/chromite/contrib/reflow_overlong_comments "${files[@]}"
The files that still had strings that were too long were manually
reformatted because they were easy and only a few issues.
autoninja.py
clang_format.py
download_from_google_storage.py
fix_encoding.py
gclient_utils.py
git_cache.py
git_common.py
git_map_branches.py
git_reparent_branch.py
gn.py
my_activity.py
owners_finder.py
presubmit_canned_checks.py
reclient_helper.py
reclientreport.py
roll_dep.py
rustfmt.py
siso.py
split_cl.py
subcommand.py
subprocess2.py
swift_format.py
upload_to_google_storage.py
These files still had lines (strings) that were too long, so the pylint
warnings were suppressed with a TODO.
auth.py
gclient.py
gclient_eval.py
gclient_paths.py
gclient_scm.py
gerrit_util.py
git_cl.py
presubmit_canned_checks.py
presubmit_support.py
scm.py
Change-Id: Ia6535c4f2c48d46b589ec1e791dde6c6b2ea858f
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/4836379
Commit-Queue: Josip Sokcevic <sokcevic@chromium.org>
Auto-Submit: Mike Frysinger <vapier@chromium.org>
Reviewed-by: Josip Sokcevic <sokcevic@chromium.org>
This commit is contained in:
251
owners_client.py
251
owners_client.py
@@ -10,7 +10,7 @@ import git_common
|
||||
|
||||
|
||||
class OwnersClient(object):
|
||||
"""Interact with OWNERS files in a repository.
|
||||
"""Interact with OWNERS files in a repository.
|
||||
|
||||
This class allows you to interact with OWNERS files in a repository both the
|
||||
Gerrit Code-Owners plugin REST API, and the owners database implemented by
|
||||
@@ -23,164 +23,167 @@ class OwnersClient(object):
|
||||
All code should use this class to interact with OWNERS files instead of the
|
||||
owners database in owners.py
|
||||
"""
|
||||
# '*' means that everyone can approve.
|
||||
EVERYONE = '*'
|
||||
# '*' means that everyone can approve.
|
||||
EVERYONE = '*'
|
||||
|
||||
# Possible status of a file.
|
||||
# - INSUFFICIENT_REVIEWERS: The path needs owners approval, but none of its
|
||||
# owners is currently a reviewer of the change.
|
||||
# - PENDING: An owner of this path has been added as reviewer, but approval
|
||||
# has not been given yet.
|
||||
# - APPROVED: The path has been approved by an owner.
|
||||
APPROVED = 'APPROVED'
|
||||
PENDING = 'PENDING'
|
||||
INSUFFICIENT_REVIEWERS = 'INSUFFICIENT_REVIEWERS'
|
||||
# Possible status of a file.
|
||||
# - INSUFFICIENT_REVIEWERS: The path needs owners approval, but none of its
|
||||
# owners is currently a reviewer of the change.
|
||||
# - PENDING: An owner of this path has been added as reviewer, but approval
|
||||
# has not been given yet.
|
||||
# - APPROVED: The path has been approved by an owner.
|
||||
APPROVED = 'APPROVED'
|
||||
PENDING = 'PENDING'
|
||||
INSUFFICIENT_REVIEWERS = 'INSUFFICIENT_REVIEWERS'
|
||||
|
||||
def ListOwners(self, path):
|
||||
"""List all owners for a file.
|
||||
def ListOwners(self, path):
|
||||
"""List all owners for a file.
|
||||
|
||||
The returned list is sorted so that better owners appear first.
|
||||
"""
|
||||
raise Exception('Not implemented')
|
||||
raise Exception('Not implemented')
|
||||
|
||||
def BatchListOwners(self, paths):
|
||||
"""List all owners for a group of files.
|
||||
def BatchListOwners(self, paths):
|
||||
"""List all owners for a group of files.
|
||||
|
||||
Returns a dictionary {path: [owners]}.
|
||||
"""
|
||||
with git_common.ScopedPool(kind='threads') as pool:
|
||||
return dict(pool.imap_unordered(
|
||||
lambda p: (p, self.ListOwners(p)), paths))
|
||||
with git_common.ScopedPool(kind='threads') as pool:
|
||||
return dict(
|
||||
pool.imap_unordered(lambda p: (p, self.ListOwners(p)), paths))
|
||||
|
||||
def GetFilesApprovalStatus(self, paths, approvers, reviewers):
|
||||
"""Check the approval status for the given paths.
|
||||
def GetFilesApprovalStatus(self, paths, approvers, reviewers):
|
||||
"""Check the approval status for the given paths.
|
||||
|
||||
Utility method to check for approval status when a change has not yet been
|
||||
created, given reviewers and approvers.
|
||||
|
||||
See GetChangeApprovalStatus for description of the returned value.
|
||||
"""
|
||||
approvers = set(approvers)
|
||||
if approvers:
|
||||
approvers.add(self.EVERYONE)
|
||||
reviewers = set(reviewers)
|
||||
if reviewers:
|
||||
reviewers.add(self.EVERYONE)
|
||||
status = {}
|
||||
owners_by_path = self.BatchListOwners(paths)
|
||||
for path, owners in owners_by_path.items():
|
||||
owners = set(owners)
|
||||
if owners.intersection(approvers):
|
||||
status[path] = self.APPROVED
|
||||
elif owners.intersection(reviewers):
|
||||
status[path] = self.PENDING
|
||||
else:
|
||||
status[path] = self.INSUFFICIENT_REVIEWERS
|
||||
return status
|
||||
approvers = set(approvers)
|
||||
if approvers:
|
||||
approvers.add(self.EVERYONE)
|
||||
reviewers = set(reviewers)
|
||||
if reviewers:
|
||||
reviewers.add(self.EVERYONE)
|
||||
status = {}
|
||||
owners_by_path = self.BatchListOwners(paths)
|
||||
for path, owners in owners_by_path.items():
|
||||
owners = set(owners)
|
||||
if owners.intersection(approvers):
|
||||
status[path] = self.APPROVED
|
||||
elif owners.intersection(reviewers):
|
||||
status[path] = self.PENDING
|
||||
else:
|
||||
status[path] = self.INSUFFICIENT_REVIEWERS
|
||||
return status
|
||||
|
||||
def ScoreOwners(self, paths, exclude=None):
|
||||
"""Get sorted list of owners for the given paths."""
|
||||
if not paths:
|
||||
return []
|
||||
exclude = exclude or []
|
||||
owners = []
|
||||
queues = self.BatchListOwners(paths).values()
|
||||
for i in range(max(len(q) for q in queues)):
|
||||
for q in queues:
|
||||
if i < len(q) and q[i] not in owners and q[i] not in exclude:
|
||||
owners.append(q[i])
|
||||
return owners
|
||||
def ScoreOwners(self, paths, exclude=None):
|
||||
"""Get sorted list of owners for the given paths."""
|
||||
if not paths:
|
||||
return []
|
||||
exclude = exclude or []
|
||||
owners = []
|
||||
queues = self.BatchListOwners(paths).values()
|
||||
for i in range(max(len(q) for q in queues)):
|
||||
for q in queues:
|
||||
if i < len(q) and q[i] not in owners and q[i] not in exclude:
|
||||
owners.append(q[i])
|
||||
return owners
|
||||
|
||||
def SuggestOwners(self, paths, exclude=None):
|
||||
"""Suggest a set of owners for the given paths."""
|
||||
exclude = exclude or []
|
||||
def SuggestOwners(self, paths, exclude=None):
|
||||
"""Suggest a set of owners for the given paths."""
|
||||
exclude = exclude or []
|
||||
|
||||
paths_by_owner = {}
|
||||
owners_by_path = self.BatchListOwners(paths)
|
||||
for path, owners in owners_by_path.items():
|
||||
for owner in owners:
|
||||
paths_by_owner.setdefault(owner, set()).add(path)
|
||||
paths_by_owner = {}
|
||||
owners_by_path = self.BatchListOwners(paths)
|
||||
for path, owners in owners_by_path.items():
|
||||
for owner in owners:
|
||||
paths_by_owner.setdefault(owner, set()).add(path)
|
||||
|
||||
selected = []
|
||||
missing = set(paths)
|
||||
for owner in self.ScoreOwners(paths, exclude=exclude):
|
||||
missing_len = len(missing)
|
||||
missing.difference_update(paths_by_owner[owner])
|
||||
if missing_len > len(missing):
|
||||
selected.append(owner)
|
||||
if not missing:
|
||||
break
|
||||
selected = []
|
||||
missing = set(paths)
|
||||
for owner in self.ScoreOwners(paths, exclude=exclude):
|
||||
missing_len = len(missing)
|
||||
missing.difference_update(paths_by_owner[owner])
|
||||
if missing_len > len(missing):
|
||||
selected.append(owner)
|
||||
if not missing:
|
||||
break
|
||||
|
||||
return selected
|
||||
|
||||
return selected
|
||||
|
||||
class GerritClient(OwnersClient):
|
||||
"""Implement OwnersClient using OWNERS REST API."""
|
||||
def __init__(self, host, project, branch):
|
||||
super(GerritClient, self).__init__()
|
||||
"""Implement OwnersClient using OWNERS REST API."""
|
||||
def __init__(self, host, project, branch):
|
||||
super(GerritClient, self).__init__()
|
||||
|
||||
self._host = host
|
||||
self._project = project
|
||||
self._branch = branch
|
||||
self._owners_cache = {}
|
||||
self._best_owners_cache = {}
|
||||
self._host = host
|
||||
self._project = project
|
||||
self._branch = branch
|
||||
self._owners_cache = {}
|
||||
self._best_owners_cache = {}
|
||||
|
||||
# Seed used by Gerrit to shuffle code owners that have the same score. Can
|
||||
# be used to make the sort order stable across several requests, e.g. to get
|
||||
# the same set of random code owners for different file paths that have the
|
||||
# same code owners.
|
||||
self._seed = random.getrandbits(30)
|
||||
# Seed used by Gerrit to shuffle code owners that have the same score.
|
||||
# Can be used to make the sort order stable across several requests,
|
||||
# e.g. to get the same set of random code owners for different file
|
||||
# paths that have the same code owners.
|
||||
self._seed = random.getrandbits(30)
|
||||
|
||||
def _FetchOwners(self, path, cache, highest_score_only=False):
|
||||
# Always use slashes as separators.
|
||||
path = path.replace(os.sep, '/')
|
||||
if path not in cache:
|
||||
# GetOwnersForFile returns a list of account details sorted by order of
|
||||
# best reviewer for path. If owners have the same score, the order is
|
||||
# random, seeded by `self._seed`.
|
||||
data = gerrit_util.GetOwnersForFile(self._host,
|
||||
self._project,
|
||||
self._branch,
|
||||
path,
|
||||
resolve_all_users=False,
|
||||
highest_score_only=highest_score_only,
|
||||
seed=self._seed)
|
||||
cache[path] = [
|
||||
d['account']['email'] for d in data['code_owners']
|
||||
if 'account' in d and 'email' in d['account']
|
||||
]
|
||||
# If owned_by_all_users is true, add everyone as an owner at the end of
|
||||
# the owners list.
|
||||
if data.get('owned_by_all_users', False):
|
||||
cache[path].append(self.EVERYONE)
|
||||
return cache[path]
|
||||
def _FetchOwners(self, path, cache, highest_score_only=False):
|
||||
# Always use slashes as separators.
|
||||
path = path.replace(os.sep, '/')
|
||||
if path not in cache:
|
||||
# GetOwnersForFile returns a list of account details sorted by order
|
||||
# of best reviewer for path. If owners have the same score, the
|
||||
# order is random, seeded by `self._seed`.
|
||||
data = gerrit_util.GetOwnersForFile(
|
||||
self._host,
|
||||
self._project,
|
||||
self._branch,
|
||||
path,
|
||||
resolve_all_users=False,
|
||||
highest_score_only=highest_score_only,
|
||||
seed=self._seed)
|
||||
cache[path] = [
|
||||
d['account']['email'] for d in data['code_owners']
|
||||
if 'account' in d and 'email' in d['account']
|
||||
]
|
||||
# If owned_by_all_users is true, add everyone as an owner at the end
|
||||
# of the owners list.
|
||||
if data.get('owned_by_all_users', False):
|
||||
cache[path].append(self.EVERYONE)
|
||||
return cache[path]
|
||||
|
||||
def ListOwners(self, path):
|
||||
return self._FetchOwners(path, self._owners_cache)
|
||||
def ListOwners(self, path):
|
||||
return self._FetchOwners(path, self._owners_cache)
|
||||
|
||||
def ListBestOwners(self, path):
|
||||
return self._FetchOwners(path,
|
||||
self._best_owners_cache,
|
||||
highest_score_only=True)
|
||||
def ListBestOwners(self, path):
|
||||
return self._FetchOwners(path,
|
||||
self._best_owners_cache,
|
||||
highest_score_only=True)
|
||||
|
||||
def BatchListBestOwners(self, paths):
|
||||
"""List only the higest-scoring owners for a group of files.
|
||||
def BatchListBestOwners(self, paths):
|
||||
"""List only the higest-scoring owners for a group of files.
|
||||
|
||||
Returns a dictionary {path: [owners]}.
|
||||
"""
|
||||
with git_common.ScopedPool(kind='threads') as pool:
|
||||
return dict(
|
||||
pool.imap_unordered(lambda p: (p, self.ListBestOwners(p)), paths))
|
||||
with git_common.ScopedPool(kind='threads') as pool:
|
||||
return dict(
|
||||
pool.imap_unordered(lambda p: (p, self.ListBestOwners(p)),
|
||||
paths))
|
||||
|
||||
|
||||
def GetCodeOwnersClient(host, project, branch):
|
||||
"""Get a new OwnersClient.
|
||||
"""Get a new OwnersClient.
|
||||
|
||||
Uses GerritClient and raises an exception if code-owners plugin is not
|
||||
available."""
|
||||
if gerrit_util.IsCodeOwnersEnabledOnHost(host):
|
||||
return GerritClient(host, project, branch)
|
||||
raise Exception(
|
||||
'code-owners plugin is not enabled. Ask your host admin to enable it '
|
||||
'on %s. Read more about code-owners at '
|
||||
'https://chromium-review.googlesource.com/'
|
||||
'plugins/code-owners/Documentation/index.html.' % host)
|
||||
if gerrit_util.IsCodeOwnersEnabledOnHost(host):
|
||||
return GerritClient(host, project, branch)
|
||||
raise Exception(
|
||||
'code-owners plugin is not enabled. Ask your host admin to enable it '
|
||||
'on %s. Read more about code-owners at '
|
||||
'https://chromium-review.googlesource.com/'
|
||||
'plugins/code-owners/Documentation/index.html.' % host)
|
||||
|
||||
Reference in New Issue
Block a user