| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
| 5 | 5 |
| 6 # Copyright (C) 2008 Evan Martin <martine@danga.com> | 6 # Copyright (C) 2008 Evan Martin <martine@danga.com> |
| 7 | 7 |
| 8 """A git-command for integrating reviews on Rietveld and Gerrit.""" | 8 """A git-command for integrating reviews on Rietveld.""" |
| 9 | 9 |
| 10 from distutils.version import LooseVersion | 10 from distutils.version import LooseVersion |
| 11 from multiprocessing.pool import ThreadPool | 11 from multiprocessing.pool import ThreadPool |
| 12 import base64 | 12 import base64 |
| 13 import collections | 13 import collections |
| 14 import glob | 14 import glob |
| 15 import httplib | 15 import httplib |
| 16 import json | 16 import json |
| 17 import logging | 17 import logging |
| 18 import optparse | 18 import optparse |
| (...skipping 21 matching lines...) Expand all Loading... |
| 40 from third_party import colorama | 40 from third_party import colorama |
| 41 from third_party import httplib2 | 41 from third_party import httplib2 |
| 42 from third_party import upload | 42 from third_party import upload |
| 43 import auth | 43 import auth |
| 44 from luci_hacks import trigger_luci_job as luci_trigger | 44 from luci_hacks import trigger_luci_job as luci_trigger |
| 45 import clang_format | 45 import clang_format |
| 46 import commit_queue | 46 import commit_queue |
| 47 import dart_format | 47 import dart_format |
| 48 import fix_encoding | 48 import fix_encoding |
| 49 import gclient_utils | 49 import gclient_utils |
| 50 import gerrit_util | |
| 51 import git_cache | 50 import git_cache |
| 52 import git_common | 51 import git_common |
| 53 import git_footers | 52 import git_footers |
| 54 import owners | 53 import owners |
| 55 import owners_finder | 54 import owners_finder |
| 56 import presubmit_support | 55 import presubmit_support |
| 57 import rietveld | 56 import rietveld |
| 58 import scm | 57 import scm |
| 59 import subcommand | 58 import subcommand |
| 60 import subprocess2 | 59 import subprocess2 |
| (...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 150 | 149 |
| 151 def ask_for_data(prompt): | 150 def ask_for_data(prompt): |
| 152 try: | 151 try: |
| 153 return raw_input(prompt) | 152 return raw_input(prompt) |
| 154 except KeyboardInterrupt: | 153 except KeyboardInterrupt: |
| 155 # Hide the exception. | 154 # Hide the exception. |
| 156 sys.exit(1) | 155 sys.exit(1) |
| 157 | 156 |
| 158 | 157 |
| 159 def git_set_branch_value(key, value): | 158 def git_set_branch_value(key, value): |
| 160 branch = GetCurrentBranch() | 159 branch = Changelist().GetBranch() |
| 161 if not branch: | 160 if not branch: |
| 162 return | 161 return |
| 163 | 162 |
| 164 cmd = ['config'] | 163 cmd = ['config'] |
| 165 if isinstance(value, int): | 164 if isinstance(value, int): |
| 166 cmd.append('--int') | 165 cmd.append('--int') |
| 167 git_key = 'branch.%s.%s' % (branch, key) | 166 git_key = 'branch.%s.%s' % (branch, key) |
| 168 RunGit(cmd + [git_key, str(value)]) | 167 RunGit(cmd + [git_key, str(value)]) |
| 169 | 168 |
| 170 | 169 |
| 171 def git_get_branch_default(key, default): | 170 def git_get_branch_default(key, default): |
| 172 branch = GetCurrentBranch() | 171 branch = Changelist().GetBranch() |
| 173 if branch: | 172 if branch: |
| 174 git_key = 'branch.%s.%s' % (branch, key) | 173 git_key = 'branch.%s.%s' % (branch, key) |
| 175 (_, stdout) = RunGitWithCode(['config', '--int', '--get', git_key]) | 174 (_, stdout) = RunGitWithCode(['config', '--int', '--get', git_key]) |
| 176 try: | 175 try: |
| 177 return int(stdout.strip()) | 176 return int(stdout.strip()) |
| 178 except ValueError: | 177 except ValueError: |
| 179 pass | 178 pass |
| 180 return default | 179 return default |
| 181 | 180 |
| 182 | 181 |
| (...skipping 624 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 807 def _GetBranchConfig(self, branch_name, param, **kwargs): | 806 def _GetBranchConfig(self, branch_name, param, **kwargs): |
| 808 return self._GetConfig('branch.' + branch_name + '.' + param, **kwargs) | 807 return self._GetConfig('branch.' + branch_name + '.' + param, **kwargs) |
| 809 | 808 |
| 810 def _GetConfig(self, param, **kwargs): | 809 def _GetConfig(self, param, **kwargs): |
| 811 self.LazyUpdateIfNeeded() | 810 self.LazyUpdateIfNeeded() |
| 812 return RunGit(['config', param], **kwargs).strip() | 811 return RunGit(['config', param], **kwargs).strip() |
| 813 | 812 |
| 814 | 813 |
| 815 def ShortBranchName(branch): | 814 def ShortBranchName(branch): |
| 816 """Convert a name like 'refs/heads/foo' to just 'foo'.""" | 815 """Convert a name like 'refs/heads/foo' to just 'foo'.""" |
| 817 return branch.replace('refs/heads/', '', 1) | 816 return branch.replace('refs/heads/', '') |
| 818 | |
| 819 | |
| 820 def GetCurrentBranchRef(): | |
| 821 """Returns branch ref (e.g., refs/heads/master) or None.""" | |
| 822 return RunGit(['symbolic-ref', 'HEAD'], | |
| 823 stderr=subprocess2.VOID, error_ok=True).strip() or None | |
| 824 | |
| 825 | |
| 826 def GetCurrentBranch(): | |
| 827 """Returns current branch or None. | |
| 828 | |
| 829 For refs/heads/* branches, returns just last part. For others, full ref. | |
| 830 """ | |
| 831 branchref = GetCurrentBranchRef() | |
| 832 if branchref: | |
| 833 return ShortBranchName(branchref) | |
| 834 return None | |
| 835 | 817 |
| 836 | 818 |
| 837 class Changelist(object): | 819 class Changelist(object): |
| 838 """Changelist works with one changelist in local branch. | 820 def __init__(self, branchref=None, issue=None, auth_config=None): |
| 839 | |
| 840 Supports two codereview backends: Rietveld or Gerrit, selected at object | |
| 841 creation. | |
| 842 | |
| 843 Not safe for concurrent multi-{thread,process} use. | |
| 844 """ | |
| 845 | |
| 846 def __init__(self, branchref=None, issue=None, codereview=None, **kwargs): | |
| 847 """Create a new ChangeList instance. | |
| 848 | |
| 849 If issue is given, the codereview must be given too. | |
| 850 | |
| 851 If `codereview` is given, it must be 'rietveld' or 'gerrit'. | |
| 852 Otherwise, it's decided based on current configuration of the local branch, | |
| 853 with default being 'rietveld' for backwards compatibility. | |
| 854 See _load_codereview_impl for more details. | |
| 855 | |
| 856 **kwargs will be passed directly to codereview implementation. | |
| 857 """ | |
| 858 # Poke settings so we get the "configure your server" message if necessary. | 821 # Poke settings so we get the "configure your server" message if necessary. |
| 859 global settings | 822 global settings |
| 860 if not settings: | 823 if not settings: |
| 861 # Happens when git_cl.py is used as a utility library. | 824 # Happens when git_cl.py is used as a utility library. |
| 862 settings = Settings() | 825 settings = Settings() |
| 863 | 826 settings.GetDefaultServerUrl() |
| 864 if issue: | |
| 865 assert codereview, 'codereview must be known, if issue is known' | |
| 866 | |
| 867 self.branchref = branchref | 827 self.branchref = branchref |
| 868 if self.branchref: | 828 if self.branchref: |
| 869 self.branch = ShortBranchName(self.branchref) | 829 self.branch = ShortBranchName(self.branchref) |
| 870 else: | 830 else: |
| 871 self.branch = None | 831 self.branch = None |
| 832 self.rietveld_server = None |
| 872 self.upstream_branch = None | 833 self.upstream_branch = None |
| 873 self.lookedup_issue = False | 834 self.lookedup_issue = False |
| 874 self.issue = issue or None | 835 self.issue = issue or None |
| 875 self.has_description = False | 836 self.has_description = False |
| 876 self.description = None | 837 self.description = None |
| 877 self.lookedup_patchset = False | 838 self.lookedup_patchset = False |
| 878 self.patchset = None | 839 self.patchset = None |
| 879 self.cc = None | 840 self.cc = None |
| 880 self.watchers = () | 841 self.watchers = () |
| 842 self._auth_config = auth_config |
| 843 self._props = None |
| 881 self._remote = None | 844 self._remote = None |
| 845 self._rpc_server = None |
| 882 | 846 |
| 883 self._codereview_impl = None | 847 @property |
| 884 self._load_codereview_impl(codereview, **kwargs) | 848 def auth_config(self): |
| 885 | 849 return self._auth_config |
| 886 def _load_codereview_impl(self, codereview=None, **kwargs): | |
| 887 if codereview: | |
| 888 codereview = codereview.lower() | |
| 889 if codereview == 'gerrit': | |
| 890 self._codereview_impl = _GerritChangelistImpl(self, **kwargs) | |
| 891 elif codereview == 'rietveld': | |
| 892 self._codereview_impl = _RietveldChangelistImpl(self, **kwargs) | |
| 893 else: | |
| 894 assert codereview in ('rietveld', 'gerrit') | |
| 895 return | |
| 896 | |
| 897 # Automatic selection based on issue number set for a current branch. | |
| 898 # Rietveld takes precedence over Gerrit. | |
| 899 assert not self.issue | |
| 900 # Whether we find issue or not, we are doing the lookup. | |
| 901 self.lookedup_issue = True | |
| 902 for cls in [_RietveldChangelistImpl, _GerritChangelistImpl]: | |
| 903 setting = cls.IssueSetting(self.GetBranch()) | |
| 904 issue = RunGit(['config', setting], error_ok=True).strip() | |
| 905 if issue: | |
| 906 self._codereview_impl = cls(self, **kwargs) | |
| 907 self.issue = int(issue) | |
| 908 return | |
| 909 | |
| 910 # No issue is set for this branch, so decide based on repo-wide settings. | |
| 911 return self._load_codereview_impl( | |
| 912 codereview='gerrit' if settings.GetIsGerrit() else 'rietveld') | |
| 913 | |
| 914 | 850 |
| 915 def GetCCList(self): | 851 def GetCCList(self): |
| 916 """Return the users cc'd on this CL. | 852 """Return the users cc'd on this CL. |
| 917 | 853 |
| 918 Return is a string suitable for passing to gcl with the --cc flag. | 854 Return is a string suitable for passing to gcl with the --cc flag. |
| 919 """ | 855 """ |
| 920 if self.cc is None: | 856 if self.cc is None: |
| 921 base_cc = settings.GetDefaultCCList() | 857 base_cc = settings.GetDefaultCCList() |
| 922 more_cc = ','.join(self.watchers) | 858 more_cc = ','.join(self.watchers) |
| 923 self.cc = ','.join(filter(None, (base_cc, more_cc))) or '' | 859 self.cc = ','.join(filter(None, (base_cc, more_cc))) or '' |
| 924 return self.cc | 860 return self.cc |
| 925 | 861 |
| 926 def GetCCListWithoutDefault(self): | 862 def GetCCListWithoutDefault(self): |
| 927 """Return the users cc'd on this CL excluding default ones.""" | 863 """Return the users cc'd on this CL excluding default ones.""" |
| 928 if self.cc is None: | 864 if self.cc is None: |
| 929 self.cc = ','.join(self.watchers) | 865 self.cc = ','.join(self.watchers) |
| 930 return self.cc | 866 return self.cc |
| 931 | 867 |
| 932 def SetWatchers(self, watchers): | 868 def SetWatchers(self, watchers): |
| 933 """Set the list of email addresses that should be cc'd based on the changed | 869 """Set the list of email addresses that should be cc'd based on the changed |
| 934 files in this CL. | 870 files in this CL. |
| 935 """ | 871 """ |
| 936 self.watchers = watchers | 872 self.watchers = watchers |
| 937 | 873 |
| 938 def GetBranch(self): | 874 def GetBranch(self): |
| 939 """Returns the short branch name, e.g. 'master'.""" | 875 """Returns the short branch name, e.g. 'master'.""" |
| 940 if not self.branch: | 876 if not self.branch: |
| 941 branchref = GetCurrentBranchRef() | 877 branchref = RunGit(['symbolic-ref', 'HEAD'], |
| 878 stderr=subprocess2.VOID, error_ok=True).strip() |
| 942 if not branchref: | 879 if not branchref: |
| 943 return None | 880 return None |
| 944 self.branchref = branchref | 881 self.branchref = branchref |
| 945 self.branch = ShortBranchName(self.branchref) | 882 self.branch = ShortBranchName(self.branchref) |
| 946 return self.branch | 883 return self.branch |
| 947 | 884 |
| 948 def GetBranchRef(self): | 885 def GetBranchRef(self): |
| 949 """Returns the full branch name, e.g. 'refs/heads/master'.""" | 886 """Returns the full branch name, e.g. 'refs/heads/master'.""" |
| 950 self.GetBranch() # Poke the lazy loader. | 887 self.GetBranch() # Poke the lazy loader. |
| 951 return self.branchref | 888 return self.branchref |
| (...skipping 23 matching lines...) Expand all Loading... |
| 975 if 'origin/master' in remote_branches: | 912 if 'origin/master' in remote_branches: |
| 976 # Fall back on origin/master if it exits. | 913 # Fall back on origin/master if it exits. |
| 977 remote = 'origin' | 914 remote = 'origin' |
| 978 upstream_branch = 'refs/heads/master' | 915 upstream_branch = 'refs/heads/master' |
| 979 elif 'origin/trunk' in remote_branches: | 916 elif 'origin/trunk' in remote_branches: |
| 980 # Fall back on origin/trunk if it exists. Generally a shared | 917 # Fall back on origin/trunk if it exists. Generally a shared |
| 981 # git-svn clone | 918 # git-svn clone |
| 982 remote = 'origin' | 919 remote = 'origin' |
| 983 upstream_branch = 'refs/heads/trunk' | 920 upstream_branch = 'refs/heads/trunk' |
| 984 else: | 921 else: |
| 985 DieWithError( | 922 DieWithError("""Unable to determine default branch to diff against. |
| 986 'Unable to determine default branch to diff against.\n' | 923 Either pass complete "git diff"-style arguments, like |
| 987 'Either pass complete "git diff"-style arguments, like\n' | 924 git cl upload origin/master |
| 988 ' git cl upload origin/master\n' | 925 or verify this branch is set up to track another (via the --track argument to |
| 989 'or verify this branch is set up to track another \n' | 926 "git checkout -b ...").""") |
| 990 '(via the --track argument to "git checkout -b ...").') | |
| 991 | 927 |
| 992 return remote, upstream_branch | 928 return remote, upstream_branch |
| 993 | 929 |
| 994 def GetCommonAncestorWithUpstream(self): | 930 def GetCommonAncestorWithUpstream(self): |
| 995 upstream_branch = self.GetUpstreamBranch() | 931 upstream_branch = self.GetUpstreamBranch() |
| 996 if not BranchExists(upstream_branch): | 932 if not BranchExists(upstream_branch): |
| 997 DieWithError('The upstream for the current branch (%s) does not exist ' | 933 DieWithError('The upstream for the current branch (%s) does not exist ' |
| 998 'anymore.\nPlease fix it and try again.' % self.GetBranch()) | 934 'anymore.\nPlease fix it and try again.' % self.GetBranch()) |
| 999 return git_common.get_or_create_merge_base(self.GetBranch(), | 935 return git_common.get_or_create_merge_base(self.GetBranch(), |
| 1000 upstream_branch) | 936 upstream_branch) |
| (...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1122 # If URL is pointing to a local directory, it is probably a git cache. | 1058 # If URL is pointing to a local directory, it is probably a git cache. |
| 1123 if os.path.isdir(url): | 1059 if os.path.isdir(url): |
| 1124 url = RunGit(['config', 'remote.%s.url' % remote], | 1060 url = RunGit(['config', 'remote.%s.url' % remote], |
| 1125 error_ok=True, | 1061 error_ok=True, |
| 1126 cwd=url).strip() | 1062 cwd=url).strip() |
| 1127 return url | 1063 return url |
| 1128 | 1064 |
| 1129 def GetIssue(self): | 1065 def GetIssue(self): |
| 1130 """Returns the issue number as a int or None if not set.""" | 1066 """Returns the issue number as a int or None if not set.""" |
| 1131 if self.issue is None and not self.lookedup_issue: | 1067 if self.issue is None and not self.lookedup_issue: |
| 1132 issue = RunGit(['config', | 1068 issue = RunGit(['config', self._IssueSetting()], error_ok=True).strip() |
| 1133 self._codereview_impl.IssueSetting(self.GetBranch())], | |
| 1134 error_ok=True).strip() | |
| 1135 self.issue = int(issue) or None if issue else None | 1069 self.issue = int(issue) or None if issue else None |
| 1136 self.lookedup_issue = True | 1070 self.lookedup_issue = True |
| 1137 return self.issue | 1071 return self.issue |
| 1138 | 1072 |
| 1073 def GetRietveldServer(self): |
| 1074 if not self.rietveld_server: |
| 1075 # If we're on a branch then get the server potentially associated |
| 1076 # with that branch. |
| 1077 if self.GetIssue(): |
| 1078 rietveld_server_config = self._RietveldServer() |
| 1079 if rietveld_server_config: |
| 1080 self.rietveld_server = gclient_utils.UpgradeToHttps(RunGit( |
| 1081 ['config', rietveld_server_config], error_ok=True).strip()) |
| 1082 if not self.rietveld_server: |
| 1083 self.rietveld_server = settings.GetDefaultServerUrl() |
| 1084 return self.rietveld_server |
| 1085 |
| 1086 def GetGerritServer(self): |
| 1087 # We don't support multiple Gerrit servers, and assume it to be same as |
| 1088 # origin, except with a '-review' suffix for first subdomain. |
| 1089 parts = urlparse.urlparse(self.GetRemoteUrl()).netloc.split('.') |
| 1090 parts[0] = parts[0] + '-review' |
| 1091 return 'https://%s' % '.'.join(parts) |
| 1092 |
| 1139 def GetIssueURL(self): | 1093 def GetIssueURL(self): |
| 1140 """Get the URL for a particular issue.""" | 1094 """Get the URL for a particular issue.""" |
| 1141 issue = self.GetIssue() | 1095 if not self.GetIssue(): |
| 1142 if not issue: | |
| 1143 return None | 1096 return None |
| 1144 return '%s/%s' % (self._codereview_impl.GetCodereviewServer(), issue) | 1097 if settings.GetIsGerrit(): |
| 1098 return '%s/%s' % (self.GetGerritServer(), self.GetIssue()) |
| 1099 return '%s/%s' % (self.GetRietveldServer(), self.GetIssue()) |
| 1145 | 1100 |
| 1146 def GetDescription(self, pretty=False): | 1101 def GetDescription(self, pretty=False): |
| 1147 if not self.has_description: | 1102 if not self.has_description: |
| 1148 if self.GetIssue(): | 1103 if self.GetIssue(): |
| 1149 self.description = self._codereview_impl.FetchDescription() | 1104 issue = self.GetIssue() |
| 1105 try: |
| 1106 self.description = self.RpcServer().get_description(issue).strip() |
| 1107 except urllib2.HTTPError as e: |
| 1108 if e.code == 404: |
| 1109 DieWithError( |
| 1110 ('\nWhile fetching the description for issue %d, received a ' |
| 1111 '404 (not found)\n' |
| 1112 'error. It is likely that you deleted this ' |
| 1113 'issue on the server. If this is the\n' |
| 1114 'case, please run\n\n' |
| 1115 ' git cl issue 0\n\n' |
| 1116 'to clear the association with the deleted issue. Then run ' |
| 1117 'this command again.') % issue) |
| 1118 else: |
| 1119 DieWithError( |
| 1120 '\nFailed to fetch issue description. HTTP error %d' % e.code) |
| 1121 except urllib2.URLError as e: |
| 1122 print >> sys.stderr, ( |
| 1123 'Warning: Failed to retrieve CL description due to network ' |
| 1124 'failure.') |
| 1125 self.description = '' |
| 1126 |
| 1150 self.has_description = True | 1127 self.has_description = True |
| 1151 if pretty: | 1128 if pretty: |
| 1152 wrapper = textwrap.TextWrapper() | 1129 wrapper = textwrap.TextWrapper() |
| 1153 wrapper.initial_indent = wrapper.subsequent_indent = ' ' | 1130 wrapper.initial_indent = wrapper.subsequent_indent = ' ' |
| 1154 return wrapper.fill(self.description) | 1131 return wrapper.fill(self.description) |
| 1155 return self.description | 1132 return self.description |
| 1156 | 1133 |
| 1157 def GetPatchset(self): | 1134 def GetPatchset(self): |
| 1158 """Returns the patchset number as a int or None if not set.""" | 1135 """Returns the patchset number as a int or None if not set.""" |
| 1159 if self.patchset is None and not self.lookedup_patchset: | 1136 if self.patchset is None and not self.lookedup_patchset: |
| 1160 patchset = RunGit(['config', self._codereview_impl.PatchsetSetting()], | 1137 patchset = RunGit(['config', self._PatchsetSetting()], |
| 1161 error_ok=True).strip() | 1138 error_ok=True).strip() |
| 1162 self.patchset = int(patchset) or None if patchset else None | 1139 self.patchset = int(patchset) or None if patchset else None |
| 1163 self.lookedup_patchset = True | 1140 self.lookedup_patchset = True |
| 1164 return self.patchset | 1141 return self.patchset |
| 1165 | 1142 |
| 1166 def SetPatchset(self, patchset): | 1143 def SetPatchset(self, patchset): |
| 1167 """Set this branch's patchset. If patchset=0, clears the patchset.""" | 1144 """Set this branch's patchset. If patchset=0, clears the patchset.""" |
| 1168 patchset_setting = self._codereview_impl.PatchsetSetting() | |
| 1169 if patchset: | 1145 if patchset: |
| 1170 RunGit(['config', patchset_setting, str(patchset)]) | 1146 RunGit(['config', self._PatchsetSetting(), str(patchset)]) |
| 1171 self.patchset = patchset | 1147 self.patchset = patchset |
| 1172 else: | 1148 else: |
| 1173 RunGit(['config', '--unset', patchset_setting], | 1149 RunGit(['config', '--unset', self._PatchsetSetting()], |
| 1174 stderr=subprocess2.PIPE, error_ok=True) | 1150 stderr=subprocess2.PIPE, error_ok=True) |
| 1175 self.patchset = None | 1151 self.patchset = None |
| 1176 | 1152 |
| 1153 def GetMostRecentPatchset(self): |
| 1154 return self.GetIssueProperties()['patchsets'][-1] |
| 1155 |
| 1156 def GetPatchSetDiff(self, issue, patchset): |
| 1157 return self.RpcServer().get( |
| 1158 '/download/issue%s_%s.diff' % (issue, patchset)) |
| 1159 |
| 1160 def GetIssueProperties(self): |
| 1161 if self._props is None: |
| 1162 issue = self.GetIssue() |
| 1163 if not issue: |
| 1164 self._props = {} |
| 1165 else: |
| 1166 self._props = self.RpcServer().get_issue_properties(issue, True) |
| 1167 return self._props |
| 1168 |
| 1169 def GetApprovingReviewers(self): |
| 1170 return get_approving_reviewers(self.GetIssueProperties()) |
| 1171 |
| 1172 def AddComment(self, message): |
| 1173 return self.RpcServer().add_comment(self.GetIssue(), message) |
| 1174 |
| 1177 def SetIssue(self, issue=None): | 1175 def SetIssue(self, issue=None): |
| 1178 """Set this branch's issue. If issue isn't given, clears the issue.""" | 1176 """Set this branch's issue. If issue isn't given, clears the issue.""" |
| 1179 issue_setting = self._codereview_impl.IssueSetting(self.GetBranch()) | |
| 1180 codereview_setting = self._codereview_impl.GetCodereviewServerSetting() | |
| 1181 if issue: | 1177 if issue: |
| 1182 self.issue = issue | 1178 self.issue = issue |
| 1183 RunGit(['config', issue_setting, str(issue)]) | 1179 RunGit(['config', self._IssueSetting(), str(issue)]) |
| 1184 codereview_server = self._codereview_impl.GetCodereviewServer() | 1180 if not settings.GetIsGerrit() and self.rietveld_server: |
| 1185 if codereview_server: | 1181 RunGit(['config', self._RietveldServer(), self.rietveld_server]) |
| 1186 RunGit(['config', codereview_setting, codereview_server]) | |
| 1187 else: | 1182 else: |
| 1188 current_issue = self.GetIssue() | 1183 current_issue = self.GetIssue() |
| 1189 if current_issue: | 1184 if current_issue: |
| 1190 RunGit(['config', '--unset', issue_setting]) | 1185 RunGit(['config', '--unset', self._IssueSetting()]) |
| 1191 self.issue = None | 1186 self.issue = None |
| 1192 self.SetPatchset(None) | 1187 self.SetPatchset(None) |
| 1193 | 1188 |
| 1194 def GetChange(self, upstream_branch, author): | 1189 def GetChange(self, upstream_branch, author): |
| 1195 if not self.GitSanityChecks(upstream_branch): | 1190 if not self.GitSanityChecks(upstream_branch): |
| 1196 DieWithError('\nGit sanity check failure') | 1191 DieWithError('\nGit sanity check failure') |
| 1197 | 1192 |
| 1198 root = settings.GetRelativeRoot() | 1193 root = settings.GetRelativeRoot() |
| 1199 if not root: | 1194 if not root: |
| 1200 root = '.' | 1195 root = '.' |
| (...skipping 29 matching lines...) Expand all Loading... |
| 1230 return presubmit_support.GitChange( | 1225 return presubmit_support.GitChange( |
| 1231 name, | 1226 name, |
| 1232 description, | 1227 description, |
| 1233 absroot, | 1228 absroot, |
| 1234 files, | 1229 files, |
| 1235 issue, | 1230 issue, |
| 1236 patchset, | 1231 patchset, |
| 1237 author, | 1232 author, |
| 1238 upstream=upstream_branch) | 1233 upstream=upstream_branch) |
| 1239 | 1234 |
| 1240 def UpdateDescription(self, description): | |
| 1241 self.description = description | |
| 1242 return self._codereview_impl.UpdateDescriptionRemote(description) | |
| 1243 | |
| 1244 def RunHook(self, committing, may_prompt, verbose, change): | |
| 1245 """Calls sys.exit() if the hook fails; returns a HookResults otherwise.""" | |
| 1246 try: | |
| 1247 return presubmit_support.DoPresubmitChecks(change, committing, | |
| 1248 verbose=verbose, output_stream=sys.stdout, input_stream=sys.stdin, | |
| 1249 default_presubmit=None, may_prompt=may_prompt, | |
| 1250 rietveld_obj=self._codereview_impl.GetRieveldObjForPresubmit()) | |
| 1251 except presubmit_support.PresubmitFailure, e: | |
| 1252 DieWithError( | |
| 1253 ('%s\nMaybe your depot_tools is out of date?\n' | |
| 1254 'If all fails, contact maruel@') % e) | |
| 1255 | |
| 1256 # Forward methods to codereview specific implementation. | |
| 1257 | |
| 1258 def CloseIssue(self): | |
| 1259 return self._codereview_impl.CloseIssue() | |
| 1260 | |
| 1261 def GetStatus(self): | |
| 1262 return self._codereview_impl.GetStatus() | |
| 1263 | |
| 1264 def GetCodereviewServer(self): | |
| 1265 return self._codereview_impl.GetCodereviewServer() | |
| 1266 | |
| 1267 def GetApprovingReviewers(self): | |
| 1268 return self._codereview_impl.GetApprovingReviewers() | |
| 1269 | |
| 1270 def GetMostRecentPatchset(self): | |
| 1271 return self._codereview_impl.GetMostRecentPatchset() | |
| 1272 | |
| 1273 def __getattr__(self, attr): | |
| 1274 # This is because lots of untested code accesses Rietveld-specific stuff | |
| 1275 # directly, and it's hard to fix for sure. So, just let it work, and fix | |
| 1276 # on a cases by case basis. | |
| 1277 return getattr(self._codereview_impl, attr) | |
| 1278 | |
| 1279 | |
| 1280 class _ChangelistCodereviewBase(object): | |
| 1281 """Abstract base class encapsulating codereview specifics of a changelist.""" | |
| 1282 def __init__(self, changelist): | |
| 1283 self._changelist = changelist # instance of Changelist | |
| 1284 | |
| 1285 def __getattr__(self, attr): | |
| 1286 # Forward methods to changelist. | |
| 1287 # TODO(tandrii): maybe clean up _GerritChangelistImpl and | |
| 1288 # _RietveldChangelistImpl to avoid this hack? | |
| 1289 return getattr(self._changelist, attr) | |
| 1290 | |
| 1291 def GetStatus(self): | 1235 def GetStatus(self): |
| 1292 """Apply a rough heuristic to give a simple summary of an issue's review | 1236 """Apply a rough heuristic to give a simple summary of an issue's review |
| 1293 or CQ status, assuming adherence to a common workflow. | 1237 or CQ status, assuming adherence to a common workflow. |
| 1294 | |
| 1295 Returns None if no issue for this branch, or specific string keywords. | |
| 1296 """ | |
| 1297 raise NotImplementedError() | |
| 1298 | |
| 1299 def GetCodereviewServer(self): | |
| 1300 """Returns server URL without end slash, like "https://codereview.com".""" | |
| 1301 raise NotImplementedError() | |
| 1302 | |
| 1303 def FetchDescription(self): | |
| 1304 """Fetches and returns description from the codereview server.""" | |
| 1305 raise NotImplementedError() | |
| 1306 | |
| 1307 def GetCodereviewServerSetting(self): | |
| 1308 """Returns git config setting for the codereview server.""" | |
| 1309 raise NotImplementedError() | |
| 1310 | |
| 1311 @staticmethod | |
| 1312 def IssueSetting(branch): | |
| 1313 """Returns name of git config setting which stores issue number for a given | |
| 1314 branch.""" | |
| 1315 raise NotImplementedError() | |
| 1316 | |
| 1317 def PatchsetSetting(self): | |
| 1318 """Returns name of git config setting which stores issue number.""" | |
| 1319 raise NotImplementedError() | |
| 1320 | |
| 1321 def GetRieveldObjForPresubmit(self): | |
| 1322 # This is an unfortunate Rietveld-embeddedness in presubmit. | |
| 1323 # For non-Rietveld codereviews, this probably should return a dummy object. | |
| 1324 raise NotImplementedError() | |
| 1325 | |
| 1326 def UpdateDescriptionRemote(self, description): | |
| 1327 """Update the description on codereview site.""" | |
| 1328 raise NotImplementedError() | |
| 1329 | |
| 1330 def CloseIssue(self): | |
| 1331 """Closes the issue.""" | |
| 1332 raise NotImplementedError() | |
| 1333 | |
| 1334 def GetApprovingReviewers(self): | |
| 1335 """Returns a list of reviewers approving the change. | |
| 1336 | |
| 1337 Note: not necessarily committers. | |
| 1338 """ | |
| 1339 raise NotImplementedError() | |
| 1340 | |
| 1341 def GetMostRecentPatchset(self): | |
| 1342 """Returns the most recent patchset number from the codereview site.""" | |
| 1343 raise NotImplementedError() | |
| 1344 | |
| 1345 | |
| 1346 class _RietveldChangelistImpl(_ChangelistCodereviewBase): | |
| 1347 def __init__(self, changelist, auth_config=None, rietveld_server=None): | |
| 1348 super(_RietveldChangelistImpl, self).__init__(changelist) | |
| 1349 assert settings, 'must be initialized in _ChangelistCodereviewBase' | |
| 1350 settings.GetDefaultServerUrl() | |
| 1351 | |
| 1352 self._rietveld_server = rietveld_server | |
| 1353 self._auth_config = auth_config | |
| 1354 self._props = None | |
| 1355 self._rpc_server = None | |
| 1356 | |
| 1357 def GetAuthConfig(self): | |
| 1358 return self._auth_config | |
| 1359 | |
| 1360 def GetCodereviewServer(self): | |
| 1361 if not self._rietveld_server: | |
| 1362 # If we're on a branch then get the server potentially associated | |
| 1363 # with that branch. | |
| 1364 if self.GetIssue(): | |
| 1365 rietveld_server_setting = self.GetCodereviewServerSetting() | |
| 1366 if rietveld_server_setting: | |
| 1367 self._rietveld_server = gclient_utils.UpgradeToHttps(RunGit( | |
| 1368 ['config', rietveld_server_setting], error_ok=True).strip()) | |
| 1369 if not self._rietveld_server: | |
| 1370 self._rietveld_server = settings.GetDefaultServerUrl() | |
| 1371 return self._rietveld_server | |
| 1372 | |
| 1373 def FetchDescription(self): | |
| 1374 issue = self.GetIssue() | |
| 1375 assert issue | |
| 1376 try: | |
| 1377 return self.RpcServer().get_description(issue).strip() | |
| 1378 except urllib2.HTTPError as e: | |
| 1379 if e.code == 404: | |
| 1380 DieWithError( | |
| 1381 ('\nWhile fetching the description for issue %d, received a ' | |
| 1382 '404 (not found)\n' | |
| 1383 'error. It is likely that you deleted this ' | |
| 1384 'issue on the server. If this is the\n' | |
| 1385 'case, please run\n\n' | |
| 1386 ' git cl issue 0\n\n' | |
| 1387 'to clear the association with the deleted issue. Then run ' | |
| 1388 'this command again.') % issue) | |
| 1389 else: | |
| 1390 DieWithError( | |
| 1391 '\nFailed to fetch issue description. HTTP error %d' % e.code) | |
| 1392 except urllib2.URLError as e: | |
| 1393 print >> sys.stderr, ( | |
| 1394 'Warning: Failed to retrieve CL description due to network ' | |
| 1395 'failure.') | |
| 1396 return '' | |
| 1397 | |
| 1398 def GetMostRecentPatchset(self): | |
| 1399 return self.GetIssueProperties()['patchsets'][-1] | |
| 1400 | |
| 1401 def GetPatchSetDiff(self, issue, patchset): | |
| 1402 return self.RpcServer().get( | |
| 1403 '/download/issue%s_%s.diff' % (issue, patchset)) | |
| 1404 | |
| 1405 def GetIssueProperties(self): | |
| 1406 if self._props is None: | |
| 1407 issue = self.GetIssue() | |
| 1408 if not issue: | |
| 1409 self._props = {} | |
| 1410 else: | |
| 1411 self._props = self.RpcServer().get_issue_properties(issue, True) | |
| 1412 return self._props | |
| 1413 | |
| 1414 def GetApprovingReviewers(self): | |
| 1415 return get_approving_reviewers(self.GetIssueProperties()) | |
| 1416 | |
| 1417 def AddComment(self, message): | |
| 1418 return self.RpcServer().add_comment(self.GetIssue(), message) | |
| 1419 | |
| 1420 def GetStatus(self): | |
| 1421 """Apply a rough heuristic to give a simple summary of an issue's review | |
| 1422 or CQ status, assuming adherence to a common workflow. | |
| 1423 | 1238 |
| 1424 Returns None if no issue for this branch, or one of the following keywords: | 1239 Returns None if no issue for this branch, or one of the following keywords: |
| 1425 * 'error' - error from review tool (including deleted issues) | 1240 * 'error' - error from review tool (including deleted issues) |
| 1426 * 'unsent' - not sent for review | 1241 * 'unsent' - not sent for review |
| 1427 * 'waiting' - waiting for review | 1242 * 'waiting' - waiting for review |
| 1428 * 'reply' - waiting for owner to reply to review | 1243 * 'reply' - waiting for owner to reply to review |
| 1429 * 'lgtm' - LGTM from at least one approved reviewer | 1244 * 'lgtm' - LGTM from at least one approved reviewer |
| 1430 * 'commit' - in the commit queue | 1245 * 'commit' - in the commit queue |
| 1431 * 'closed' - closed | 1246 * 'closed' - closed |
| 1432 """ | 1247 """ |
| (...skipping 24 matching lines...) Expand all Loading... |
| 1457 messages = props.get('messages') or [] | 1272 messages = props.get('messages') or [] |
| 1458 | 1273 |
| 1459 if not messages: | 1274 if not messages: |
| 1460 # No message was sent. | 1275 # No message was sent. |
| 1461 return 'unsent' | 1276 return 'unsent' |
| 1462 if messages[-1]['sender'] != props.get('owner_email'): | 1277 if messages[-1]['sender'] != props.get('owner_email'): |
| 1463 # Non-LGTM reply from non-owner | 1278 # Non-LGTM reply from non-owner |
| 1464 return 'reply' | 1279 return 'reply' |
| 1465 return 'waiting' | 1280 return 'waiting' |
| 1466 | 1281 |
| 1467 def UpdateDescriptionRemote(self, description): | 1282 def RunHook(self, committing, may_prompt, verbose, change): |
| 1283 """Calls sys.exit() if the hook fails; returns a HookResults otherwise.""" |
| 1284 |
| 1285 try: |
| 1286 return presubmit_support.DoPresubmitChecks(change, committing, |
| 1287 verbose=verbose, output_stream=sys.stdout, input_stream=sys.stdin, |
| 1288 default_presubmit=None, may_prompt=may_prompt, |
| 1289 rietveld_obj=self.RpcServer()) |
| 1290 except presubmit_support.PresubmitFailure, e: |
| 1291 DieWithError( |
| 1292 ('%s\nMaybe your depot_tools is out of date?\n' |
| 1293 'If all fails, contact maruel@') % e) |
| 1294 |
| 1295 def UpdateDescription(self, description): |
| 1296 self.description = description |
| 1468 return self.RpcServer().update_description( | 1297 return self.RpcServer().update_description( |
| 1469 self.GetIssue(), self.description) | 1298 self.GetIssue(), self.description) |
| 1470 | 1299 |
| 1471 def CloseIssue(self): | 1300 def CloseIssue(self): |
| 1301 """Updates the description and closes the issue.""" |
| 1472 return self.RpcServer().close_issue(self.GetIssue()) | 1302 return self.RpcServer().close_issue(self.GetIssue()) |
| 1473 | 1303 |
| 1474 def SetFlag(self, flag, value): | 1304 def SetFlag(self, flag, value): |
| 1475 """Patchset must match.""" | 1305 """Patchset must match.""" |
| 1476 if not self.GetPatchset(): | 1306 if not self.GetPatchset(): |
| 1477 DieWithError('The patchset needs to match. Send another patchset.') | 1307 DieWithError('The patchset needs to match. Send another patchset.') |
| 1478 try: | 1308 try: |
| 1479 return self.RpcServer().set_flag( | 1309 return self.RpcServer().set_flag( |
| 1480 self.GetIssue(), self.GetPatchset(), flag, value) | 1310 self.GetIssue(), self.GetPatchset(), flag, value) |
| 1481 except urllib2.HTTPError, e: | 1311 except urllib2.HTTPError, e: |
| 1482 if e.code == 404: | 1312 if e.code == 404: |
| 1483 DieWithError('The issue %s doesn\'t exist.' % self.GetIssue()) | 1313 DieWithError('The issue %s doesn\'t exist.' % self.GetIssue()) |
| 1484 if e.code == 403: | 1314 if e.code == 403: |
| 1485 DieWithError( | 1315 DieWithError( |
| 1486 ('Access denied to issue %s. Maybe the patchset %s doesn\'t ' | 1316 ('Access denied to issue %s. Maybe the patchset %s doesn\'t ' |
| 1487 'match?') % (self.GetIssue(), self.GetPatchset())) | 1317 'match?') % (self.GetIssue(), self.GetPatchset())) |
| 1488 raise | 1318 raise |
| 1489 | 1319 |
| 1490 def RpcServer(self): | 1320 def RpcServer(self): |
| 1491 """Returns an upload.RpcServer() to access this review's rietveld instance. | 1321 """Returns an upload.RpcServer() to access this review's rietveld instance. |
| 1492 """ | 1322 """ |
| 1493 if not self._rpc_server: | 1323 if not self._rpc_server: |
| 1494 self._rpc_server = rietveld.CachingRietveld( | 1324 self._rpc_server = rietveld.CachingRietveld( |
| 1495 self.GetCodereviewServer(), | 1325 self.GetRietveldServer(), |
| 1496 self._auth_config or auth.make_auth_config()) | 1326 self._auth_config or auth.make_auth_config()) |
| 1497 return self._rpc_server | 1327 return self._rpc_server |
| 1498 | 1328 |
| 1499 @staticmethod | 1329 def _IssueSetting(self): |
| 1500 def IssueSetting(branch): | 1330 """Return the git setting that stores this change's issue.""" |
| 1501 return 'branch.%s.rietveldissue' % branch | 1331 return 'branch.%s.rietveldissue' % self.GetBranch() |
| 1502 | 1332 |
| 1503 def PatchsetSetting(self): | 1333 def _PatchsetSetting(self): |
| 1504 """Return the git setting that stores this change's most recent patchset.""" | 1334 """Return the git setting that stores this change's most recent patchset.""" |
| 1505 return 'branch.%s.rietveldpatchset' % self.GetBranch() | 1335 return 'branch.%s.rietveldpatchset' % self.GetBranch() |
| 1506 | 1336 |
| 1507 def GetCodereviewServerSetting(self): | 1337 def _RietveldServer(self): |
| 1508 """Returns the git setting that stores this change's rietveld server.""" | 1338 """Returns the git setting that stores this change's rietveld server.""" |
| 1509 branch = self.GetBranch() | 1339 branch = self.GetBranch() |
| 1510 if branch: | 1340 if branch: |
| 1511 return 'branch.%s.rietveldserver' % branch | 1341 return 'branch.%s.rietveldserver' % branch |
| 1512 return None | 1342 return None |
| 1513 | 1343 |
| 1514 def GetRieveldObjForPresubmit(self): | |
| 1515 return self.RpcServer() | |
| 1516 | |
| 1517 | |
| 1518 class _GerritChangelistImpl(_ChangelistCodereviewBase): | |
| 1519 def __init__(self, changelist, auth_config=None): | |
| 1520 # auth_config is Rietveld thing, kept here to preserve interface only. | |
| 1521 super(_GerritChangelistImpl, self).__init__(changelist) | |
| 1522 self._change_id = None | |
| 1523 self._gerrit_server = None # e.g. https://chromium-review.googlesource.com | |
| 1524 self._gerrit_host = None # e.g. chromium-review.googlesource.com | |
| 1525 | |
| 1526 def _GetGerritHost(self): | |
| 1527 # Lazy load of configs. | |
| 1528 self.GetCodereviewServer() | |
| 1529 return self._gerrit_host | |
| 1530 | |
| 1531 def GetCodereviewServer(self): | |
| 1532 if not self._gerrit_server: | |
| 1533 # If we're on a branch then get the server potentially associated | |
| 1534 # with that branch. | |
| 1535 if self.GetIssue(): | |
| 1536 gerrit_server_setting = self.GetCodereviewServerSetting() | |
| 1537 if gerrit_server_setting: | |
| 1538 self._gerrit_server = RunGit(['config', gerrit_server_setting], | |
| 1539 error_ok=True).strip() | |
| 1540 if self._gerrit_server: | |
| 1541 self._gerrit_host = urlparse.urlparse(self._gerrit_server).netloc | |
| 1542 if not self._gerrit_server: | |
| 1543 # We assume repo to be hosted on Gerrit, and hence Gerrit server | |
| 1544 # has "-review" suffix for lowest level subdomain. | |
| 1545 parts = urlparse.urlparse(self.GetRemoteUrl()).netloc.split('.') | |
| 1546 parts[0] = parts[0] + '-review' | |
| 1547 self._gerrit_host = '.'.join(parts) | |
| 1548 self._gerrit_server = 'https://%s' % self._gerrit_host | |
| 1549 return self._gerrit_server | |
| 1550 | |
| 1551 @staticmethod | |
| 1552 def IssueSetting(branch): | |
| 1553 return 'branch.%s.gerritissue' % branch | |
| 1554 | |
| 1555 def PatchsetSetting(self): | |
| 1556 """Return the git setting that stores this change's most recent patchset.""" | |
| 1557 return 'branch.%s.gerritpatchset' % self.GetBranch() | |
| 1558 | |
| 1559 def GetCodereviewServerSetting(self): | |
| 1560 """Returns the git setting that stores this change's Gerrit server.""" | |
| 1561 branch = self.GetBranch() | |
| 1562 if branch: | |
| 1563 return 'branch.%s.gerritserver' % branch | |
| 1564 return None | |
| 1565 | |
| 1566 def GetRieveldObjForPresubmit(self): | |
| 1567 class ThisIsNotRietveldIssue(object): | |
| 1568 def __nonzero__(self): | |
| 1569 # This is a hack to make presubmit_support think that rietveld is not | |
| 1570 # defined, yet still ensure that calls directly result in a decent | |
| 1571 # exception message below. | |
| 1572 return False | |
| 1573 | |
| 1574 def __getattr__(self, attr): | |
| 1575 print( | |
| 1576 'You aren\'t using Rietveld at the moment, but Gerrit.\n' | |
| 1577 'Using Rietveld in your PRESUBMIT scripts won\'t work.\n' | |
| 1578 'Please, either change your PRESUBIT to not use rietveld_obj.%s,\n' | |
| 1579 'or use Rietveld for codereview.\n' | |
| 1580 'See also http://crbug.com/579160.' % attr) | |
| 1581 raise NotImplementedError() | |
| 1582 return ThisIsNotRietveldIssue() | |
| 1583 | |
| 1584 def GetStatus(self): | |
| 1585 # TODO(tandrii) | |
| 1586 raise NotImplementedError() | |
| 1587 | |
| 1588 def GetMostRecentPatchset(self): | |
| 1589 data = gerrit_util.GetChangeDetail(self._GetGerritHost(), self.GetIssue(), | |
| 1590 ['CURRENT_REVISION']) | |
| 1591 return data['revisions'][data['current_revision']]['_number'] | |
| 1592 | |
| 1593 def FetchDescription(self): | |
| 1594 data = gerrit_util.GetChangeDetail(self._GetGerritHost(), self.GetIssue(), | |
| 1595 ['COMMIT_FOOTERS', 'CURRENT_REVISION']) | |
| 1596 return data['revisions'][data['current_revision']]['commit_with_footers'] | |
| 1597 | |
| 1598 def UpdateDescriptionRemote(self, description): | |
| 1599 # TODO(tandrii) | |
| 1600 raise NotImplementedError() | |
| 1601 | |
| 1602 def CloseIssue(self): | |
| 1603 gerrit_util.AbandonChange(self._GetGerritHost(), self.GetIssue(), msg='') | |
| 1604 | |
| 1605 | 1344 |
| 1606 class ChangeDescription(object): | 1345 class ChangeDescription(object): |
| 1607 """Contains a parsed form of the change description.""" | 1346 """Contains a parsed form of the change description.""" |
| 1608 R_LINE = r'^[ \t]*(TBR|R)[ \t]*=[ \t]*(.*?)[ \t]*$' | 1347 R_LINE = r'^[ \t]*(TBR|R)[ \t]*=[ \t]*(.*?)[ \t]*$' |
| 1609 BUG_LINE = r'^[ \t]*(BUG)[ \t]*=[ \t]*(.*?)[ \t]*$' | 1348 BUG_LINE = r'^[ \t]*(BUG)[ \t]*=[ \t]*(.*?)[ \t]*$' |
| 1610 | 1349 |
| 1611 def __init__(self, description): | 1350 def __init__(self, description): |
| 1612 self._description_lines = (description or '').strip().splitlines() | 1351 self._description_lines = (description or '').strip().splitlines() |
| 1613 | 1352 |
| 1614 @property # www.logilab.org/ticket/89786 | 1353 @property # www.logilab.org/ticket/89786 |
| (...skipping 525 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2140 return 0 | 1879 return 0 |
| 2141 | 1880 |
| 2142 branches = RunGit(['for-each-ref', '--format=%(refname)', 'refs/heads']) | 1881 branches = RunGit(['for-each-ref', '--format=%(refname)', 'refs/heads']) |
| 2143 if not branches: | 1882 if not branches: |
| 2144 print('No local branch found.') | 1883 print('No local branch found.') |
| 2145 return 0 | 1884 return 0 |
| 2146 | 1885 |
| 2147 changes = ( | 1886 changes = ( |
| 2148 Changelist(branchref=b, auth_config=auth_config) | 1887 Changelist(branchref=b, auth_config=auth_config) |
| 2149 for b in branches.splitlines()) | 1888 for b in branches.splitlines()) |
| 2150 # TODO(tandrii): refactor to use CLs list instead of branches list. | |
| 2151 branches = [c.GetBranch() for c in changes] | 1889 branches = [c.GetBranch() for c in changes] |
| 2152 alignment = max(5, max(len(b) for b in branches)) | 1890 alignment = max(5, max(len(b) for b in branches)) |
| 2153 print 'Branches associated with reviews:' | 1891 print 'Branches associated with reviews:' |
| 2154 output = get_cl_statuses(branches, | 1892 output = get_cl_statuses(branches, |
| 2155 fine_grained=not options.fast, | 1893 fine_grained=not options.fast, |
| 2156 max_processes=options.maxjobs, | 1894 max_processes=options.maxjobs, |
| 2157 auth_config=auth_config) | 1895 auth_config=auth_config) |
| 2158 | 1896 |
| 2159 branch_statuses = {} | 1897 branch_statuses = {} |
| 2160 alignment = max(5, max(len(ShortBranchName(b)) for b in branches)) | 1898 alignment = max(5, max(len(ShortBranchName(b)) for b in branches)) |
| (...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2256 options, args = parser.parse_args(args) | 1994 options, args = parser.parse_args(args) |
| 2257 auth_config = auth.extract_auth_config_from_options(options) | 1995 auth_config = auth.extract_auth_config_from_options(options) |
| 2258 | 1996 |
| 2259 issue = None | 1997 issue = None |
| 2260 if options.issue: | 1998 if options.issue: |
| 2261 try: | 1999 try: |
| 2262 issue = int(options.issue) | 2000 issue = int(options.issue) |
| 2263 except ValueError: | 2001 except ValueError: |
| 2264 DieWithError('A review issue id is expected to be a number') | 2002 DieWithError('A review issue id is expected to be a number') |
| 2265 | 2003 |
| 2266 cl = Changelist(issue=issue, codereview='rietveld', auth_config=auth_config) | 2004 cl = Changelist(issue=issue, auth_config=auth_config) |
| 2267 | 2005 |
| 2268 if options.comment: | 2006 if options.comment: |
| 2269 cl.AddComment(options.comment) | 2007 cl.AddComment(options.comment) |
| 2270 return 0 | 2008 return 0 |
| 2271 | 2009 |
| 2272 data = cl.GetIssueProperties() | 2010 data = cl.GetIssueProperties() |
| 2273 summary = [] | 2011 summary = [] |
| 2274 for message in sorted(data.get('messages', []), key=lambda x: x['date']): | 2012 for message in sorted(data.get('messages', []), key=lambda x: x['date']): |
| 2275 summary.append({ | 2013 summary.append({ |
| 2276 'date': message['date'], | 2014 'date': message['date'], |
| (...skipping 358 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2635 remote_branch = remote_branch.replace('refs/remotes/', 'refs/') | 2373 remote_branch = remote_branch.replace('refs/remotes/', 'refs/') |
| 2636 # If a pending prefix exists then replace refs/ with it. | 2374 # If a pending prefix exists then replace refs/ with it. |
| 2637 if pending_prefix: | 2375 if pending_prefix: |
| 2638 remote_branch = remote_branch.replace('refs/', pending_prefix) | 2376 remote_branch = remote_branch.replace('refs/', pending_prefix) |
| 2639 return remote_branch | 2377 return remote_branch |
| 2640 | 2378 |
| 2641 | 2379 |
| 2642 def RietveldUpload(options, args, cl, change): | 2380 def RietveldUpload(options, args, cl, change): |
| 2643 """upload the patch to rietveld.""" | 2381 """upload the patch to rietveld.""" |
| 2644 upload_args = ['--assume_yes'] # Don't ask about untracked files. | 2382 upload_args = ['--assume_yes'] # Don't ask about untracked files. |
| 2645 upload_args.extend(['--server', cl.GetCodereviewServer()]) | 2383 upload_args.extend(['--server', cl.GetRietveldServer()]) |
| 2646 # TODO(tandrii): refactor this ugliness into _RietveldChangelistImpl. | 2384 upload_args.extend(auth.auth_config_to_command_options(cl.auth_config)) |
| 2647 upload_args.extend(auth.auth_config_to_command_options( | |
| 2648 cl._codereview_impl.GetAuthConfig())) | |
| 2649 if options.emulate_svn_auto_props: | 2385 if options.emulate_svn_auto_props: |
| 2650 upload_args.append('--emulate_svn_auto_props') | 2386 upload_args.append('--emulate_svn_auto_props') |
| 2651 | 2387 |
| 2652 change_desc = None | 2388 change_desc = None |
| 2653 | 2389 |
| 2654 if options.email is not None: | 2390 if options.email is not None: |
| 2655 upload_args.extend(['--email', options.email]) | 2391 upload_args.extend(['--email', options.email]) |
| 2656 | 2392 |
| 2657 if cl.GetIssue(): | 2393 if cl.GetIssue(): |
| 2658 if options.title: | 2394 if options.title: |
| (...skipping 222 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2881 | 2617 |
| 2882 # Default to diffing against common ancestor of upstream branch | 2618 # Default to diffing against common ancestor of upstream branch |
| 2883 base_branch = cl.GetCommonAncestorWithUpstream() | 2619 base_branch = cl.GetCommonAncestorWithUpstream() |
| 2884 args = [base_branch, 'HEAD'] | 2620 args = [base_branch, 'HEAD'] |
| 2885 | 2621 |
| 2886 # Make sure authenticated to Rietveld before running expensive hooks. It is | 2622 # Make sure authenticated to Rietveld before running expensive hooks. It is |
| 2887 # a fast, best efforts check. Rietveld still can reject the authentication | 2623 # a fast, best efforts check. Rietveld still can reject the authentication |
| 2888 # during the actual upload. | 2624 # during the actual upload. |
| 2889 if not settings.GetIsGerrit() and auth_config.use_oauth2: | 2625 if not settings.GetIsGerrit() and auth_config.use_oauth2: |
| 2890 authenticator = auth.get_authenticator_for_host( | 2626 authenticator = auth.get_authenticator_for_host( |
| 2891 cl.GetCodereviewServer(), auth_config) | 2627 cl.GetRietveldServer(), auth_config) |
| 2892 if not authenticator.has_cached_credentials(): | 2628 if not authenticator.has_cached_credentials(): |
| 2893 raise auth.LoginRequiredError(cl.GetCodereviewServer()) | 2629 raise auth.LoginRequiredError(cl.GetRietveldServer()) |
| 2894 | 2630 |
| 2895 # Apply watchlists on upload. | 2631 # Apply watchlists on upload. |
| 2896 change = cl.GetChange(base_branch, None) | 2632 change = cl.GetChange(base_branch, None) |
| 2897 watchlist = watchlists.Watchlists(change.RepositoryRoot()) | 2633 watchlist = watchlists.Watchlists(change.RepositoryRoot()) |
| 2898 files = [f.LocalPath() for f in change.AffectedFiles()] | 2634 files = [f.LocalPath() for f in change.AffectedFiles()] |
| 2899 if not options.bypass_watchlists: | 2635 if not options.bypass_watchlists: |
| 2900 cl.SetWatchers(watchlist.GetWatchersForPaths(files)) | 2636 cl.SetWatchers(watchlist.GetWatchersForPaths(files)) |
| 2901 | 2637 |
| 2902 if not options.bypass_hooks: | 2638 if not options.bypass_hooks: |
| 2903 if options.reviewers or options.tbr_owners: | 2639 if options.reviewers or options.tbr_owners: |
| (...skipping 573 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 3477 return PatchIssue(issue_arg, options.reject, options.nocommit, | 3213 return PatchIssue(issue_arg, options.reject, options.nocommit, |
| 3478 options.directory, auth_config) | 3214 options.directory, auth_config) |
| 3479 | 3215 |
| 3480 | 3216 |
| 3481 def PatchIssue(issue_arg, reject, nocommit, directory, auth_config): | 3217 def PatchIssue(issue_arg, reject, nocommit, directory, auth_config): |
| 3482 # PatchIssue should never be called with a dirty tree. It is up to the | 3218 # PatchIssue should never be called with a dirty tree. It is up to the |
| 3483 # caller to check this, but just in case we assert here since the | 3219 # caller to check this, but just in case we assert here since the |
| 3484 # consequences of the caller not checking this could be dire. | 3220 # consequences of the caller not checking this could be dire. |
| 3485 assert(not git_common.is_dirty_git_tree('apply')) | 3221 assert(not git_common.is_dirty_git_tree('apply')) |
| 3486 | 3222 |
| 3487 # TODO(tandrii): implement for Gerrit. | |
| 3488 if type(issue_arg) is int or issue_arg.isdigit(): | 3223 if type(issue_arg) is int or issue_arg.isdigit(): |
| 3489 # Input is an issue id. Figure out the URL. | 3224 # Input is an issue id. Figure out the URL. |
| 3490 issue = int(issue_arg) | 3225 issue = int(issue_arg) |
| 3491 cl = Changelist(issue=issue, codereview='rietveld', auth_config=auth_config) | 3226 cl = Changelist(issue=issue, auth_config=auth_config) |
| 3492 patchset = cl.GetMostRecentPatchset() | 3227 patchset = cl.GetMostRecentPatchset() |
| 3493 patch_data = cl._codereview_impl.GetPatchSetDiff(issue, patchset) | 3228 patch_data = cl.GetPatchSetDiff(issue, patchset) |
| 3494 else: | 3229 else: |
| 3495 # Assume it's a URL to the patch. Default to https. | 3230 # Assume it's a URL to the patch. Default to https. |
| 3496 issue_url = gclient_utils.UpgradeToHttps(issue_arg) | 3231 issue_url = gclient_utils.UpgradeToHttps(issue_arg) |
| 3497 match = re.match(r'(.*?)/download/issue(\d+)_(\d+).diff', issue_url) | 3232 match = re.match(r'(.*?)/download/issue(\d+)_(\d+).diff', issue_url) |
| 3498 if not match: | 3233 if not match: |
| 3499 DieWithError('Must pass an issue ID or full URL for ' | 3234 DieWithError('Must pass an issue ID or full URL for ' |
| 3500 '\'Download raw patch set\'') | 3235 '\'Download raw patch set\'') |
| 3501 issue = int(match.group(2)) | 3236 issue = int(match.group(2)) |
| 3502 cl = Changelist(issue=issue, codereview='rietveld', | 3237 cl = Changelist(issue=issue, auth_config=auth_config) |
| 3503 rietvled_server=match.group(1), auth_config=auth_config) | 3238 cl.rietveld_server = match.group(1) |
| 3504 patchset = int(match.group(3)) | 3239 patchset = int(match.group(3)) |
| 3505 patch_data = urllib2.urlopen(issue_arg).read() | 3240 patch_data = urllib2.urlopen(issue_arg).read() |
| 3506 | 3241 |
| 3507 # Switch up to the top-level directory, if necessary, in preparation for | 3242 # Switch up to the top-level directory, if necessary, in preparation for |
| 3508 # applying the patch. | 3243 # applying the patch. |
| 3509 top = settings.GetRelativeRoot() | 3244 top = settings.GetRelativeRoot() |
| 3510 if top: | 3245 if top: |
| 3511 os.chdir(top) | 3246 os.chdir(top) |
| 3512 | 3247 |
| 3513 # Git patches have a/ at the beginning of source paths. We strip that out | 3248 # Git patches have a/ at the beginning of source paths. We strip that out |
| (...skipping 23 matching lines...) Expand all Loading... |
| 3537 except subprocess2.CalledProcessError: | 3272 except subprocess2.CalledProcessError: |
| 3538 print 'Failed to apply the patch' | 3273 print 'Failed to apply the patch' |
| 3539 return 1 | 3274 return 1 |
| 3540 | 3275 |
| 3541 # If we had an issue, commit the current state and register the issue. | 3276 # If we had an issue, commit the current state and register the issue. |
| 3542 if not nocommit: | 3277 if not nocommit: |
| 3543 RunGit(['commit', '-m', (cl.GetDescription() + '\n\n' + | 3278 RunGit(['commit', '-m', (cl.GetDescription() + '\n\n' + |
| 3544 'patch from issue %(i)s at patchset ' | 3279 'patch from issue %(i)s at patchset ' |
| 3545 '%(p)s (http://crrev.com/%(i)s#ps%(p)s)' | 3280 '%(p)s (http://crrev.com/%(i)s#ps%(p)s)' |
| 3546 % {'i': issue, 'p': patchset})]) | 3281 % {'i': issue, 'p': patchset})]) |
| 3547 cl = Changelist(codereview='rietveld', auth_config=auth_config, | 3282 cl = Changelist(auth_config=auth_config) |
| 3548 rietveld_server=cl.GetCodereviewServer()) | |
| 3549 cl.SetIssue(issue) | 3283 cl.SetIssue(issue) |
| 3550 cl.SetPatchset(patchset) | 3284 cl.SetPatchset(patchset) |
| 3551 print "Committed patch locally." | 3285 print "Committed patch locally." |
| 3552 else: | 3286 else: |
| 3553 print "Patch applied to index." | 3287 print "Patch applied to index." |
| 3554 return 0 | 3288 return 0 |
| 3555 | 3289 |
| 3556 | 3290 |
| 3557 def CMDrebase(parser, args): | 3291 def CMDrebase(parser, args): |
| 3558 """Rebases current branch on top of svn repo.""" | 3292 """Rebases current branch on top of svn repo.""" |
| (...skipping 714 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 4273 if __name__ == '__main__': | 4007 if __name__ == '__main__': |
| 4274 # These affect sys.stdout so do it outside of main() to simplify mocks in | 4008 # These affect sys.stdout so do it outside of main() to simplify mocks in |
| 4275 # unit testing. | 4009 # unit testing. |
| 4276 fix_encoding.fix_encoding() | 4010 fix_encoding.fix_encoding() |
| 4277 colorama.init() | 4011 colorama.init() |
| 4278 try: | 4012 try: |
| 4279 sys.exit(main(sys.argv[1:])) | 4013 sys.exit(main(sys.argv[1:])) |
| 4280 except KeyboardInterrupt: | 4014 except KeyboardInterrupt: |
| 4281 sys.stderr.write('interrupted\n') | 4015 sys.stderr.write('interrupted\n') |
| 4282 sys.exit(1) | 4016 sys.exit(1) |
| OLD | NEW |