diff options
Diffstat (limited to 'git_command.py')
| -rw-r--r-- | git_command.py | 63 |
1 files changed, 28 insertions, 35 deletions
diff --git a/git_command.py b/git_command.py index 513b9ebf..5988cc28 100644 --- a/git_command.py +++ b/git_command.py | |||
| @@ -72,6 +72,8 @@ def terminate_ssh_clients(): | |||
| 72 | pass | 72 | pass |
| 73 | _ssh_clients = [] | 73 | _ssh_clients = [] |
| 74 | 74 | ||
| 75 | _git_version = None | ||
| 76 | |||
| 75 | class _GitCall(object): | 77 | class _GitCall(object): |
| 76 | def version(self): | 78 | def version(self): |
| 77 | p = GitCommand(None, ['--version'], capture_stdout=True) | 79 | p = GitCommand(None, ['--version'], capture_stdout=True) |
| @@ -79,6 +81,21 @@ class _GitCall(object): | |||
| 79 | return p.stdout | 81 | return p.stdout |
| 80 | return None | 82 | return None |
| 81 | 83 | ||
| 84 | def version_tuple(self): | ||
| 85 | global _git_version | ||
| 86 | |||
| 87 | if _git_version is None: | ||
| 88 | ver_str = git.version() | ||
| 89 | if ver_str.startswith('git version '): | ||
| 90 | _git_version = tuple( | ||
| 91 | map(lambda x: int(x), | ||
| 92 | ver_str[len('git version '):].strip().split('-')[0].split('.')[0:3] | ||
| 93 | )) | ||
| 94 | else: | ||
| 95 | print >>sys.stderr, 'fatal: "%s" unsupported' % ver_str | ||
| 96 | sys.exit(1) | ||
| 97 | return _git_version | ||
| 98 | |||
| 82 | def __getattr__(self, name): | 99 | def __getattr__(self, name): |
| 83 | name = name.replace('_','-') | 100 | name = name.replace('_','-') |
| 84 | def fun(*cmdv): | 101 | def fun(*cmdv): |
| @@ -88,23 +105,9 @@ class _GitCall(object): | |||
| 88 | return fun | 105 | return fun |
| 89 | git = _GitCall() | 106 | git = _GitCall() |
| 90 | 107 | ||
| 91 | _git_version = None | ||
| 92 | |||
| 93 | def git_require(min_version, fail=False): | 108 | def git_require(min_version, fail=False): |
| 94 | global _git_version | 109 | git_version = git.version_tuple() |
| 95 | 110 | if min_version <= git_version: | |
| 96 | if _git_version is None: | ||
| 97 | ver_str = git.version() | ||
| 98 | if ver_str.startswith('git version '): | ||
| 99 | _git_version = tuple( | ||
| 100 | map(lambda x: int(x), | ||
| 101 | ver_str[len('git version '):].strip().split('.')[0:3] | ||
| 102 | )) | ||
| 103 | else: | ||
| 104 | print >>sys.stderr, 'fatal: "%s" unsupported' % ver_str | ||
| 105 | sys.exit(1) | ||
| 106 | |||
| 107 | if min_version <= _git_version: | ||
| 108 | return True | 111 | return True |
| 109 | if fail: | 112 | if fail: |
| 110 | need = '.'.join(map(lambda x: str(x), min_version)) | 113 | need = '.'.join(map(lambda x: str(x), min_version)) |
| @@ -144,6 +147,12 @@ class GitCommand(object): | |||
| 144 | if ssh_proxy: | 147 | if ssh_proxy: |
| 145 | _setenv(env, 'REPO_SSH_SOCK', ssh_sock()) | 148 | _setenv(env, 'REPO_SSH_SOCK', ssh_sock()) |
| 146 | _setenv(env, 'GIT_SSH', _ssh_proxy()) | 149 | _setenv(env, 'GIT_SSH', _ssh_proxy()) |
| 150 | if 'http_proxy' in env and 'darwin' == sys.platform: | ||
| 151 | s = "'http.proxy=%s'" % (env['http_proxy'],) | ||
| 152 | p = env.get('GIT_CONFIG_PARAMETERS') | ||
| 153 | if p is not None: | ||
| 154 | s = p + ' ' + s | ||
| 155 | _setenv(env, 'GIT_CONFIG_PARAMETERS', s) | ||
| 147 | 156 | ||
| 148 | if project: | 157 | if project: |
| 149 | if not cwd: | 158 | if not cwd: |
| @@ -218,26 +227,10 @@ class GitCommand(object): | |||
| 218 | self.stdin = p.stdin | 227 | self.stdin = p.stdin |
| 219 | 228 | ||
| 220 | def Wait(self): | 229 | def Wait(self): |
| 221 | p = self.process | ||
| 222 | |||
| 223 | if p.stdin: | ||
| 224 | p.stdin.close() | ||
| 225 | self.stdin = None | ||
| 226 | |||
| 227 | if p.stdout: | ||
| 228 | self.stdout = p.stdout.read() | ||
| 229 | p.stdout.close() | ||
| 230 | else: | ||
| 231 | p.stdout = None | ||
| 232 | |||
| 233 | if p.stderr: | ||
| 234 | self.stderr = p.stderr.read() | ||
| 235 | p.stderr.close() | ||
| 236 | else: | ||
| 237 | p.stderr = None | ||
| 238 | |||
| 239 | try: | 230 | try: |
| 240 | rc = p.wait() | 231 | p = self.process |
| 232 | (self.stdout, self.stderr) = p.communicate() | ||
| 233 | rc = p.returncode | ||
| 241 | finally: | 234 | finally: |
| 242 | _remove_ssh_client(p) | 235 | _remove_ssh_client(p) |
| 243 | return rc | 236 | return rc |
