Chromium Code Reviews| Index: recipe_engine/fetch.py |
| diff --git a/recipe_engine/fetch.py b/recipe_engine/fetch.py |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..ad535c859b1f21fb72effc4a8a0747d61477fa9a |
| --- /dev/null |
| +++ b/recipe_engine/fetch.py |
| @@ -0,0 +1,74 @@ |
| +# Copyright 2016 The Chromium Authors. All rights reserved. |
|
nodir
2016/05/20 23:03:49
is this file about git? it seems so. Maybe rename
Paweł Hajdan Jr.
2016/05/24 16:58:06
This won't be necessarily exclusive to git. For ex
nodir
2016/05/24 17:46:31
Acknowledged.
|
| +# Use of this source code is governed by a BSD-style license that can be |
| +# found in the LICENSE file. |
| + |
| +import logging |
| +import os |
| +import sys |
| + |
| +from .third_party import subprocess42 |
| + |
| + |
| +class FetchError(Exception): |
| + pass |
| + |
| + |
| +class UncleanFilesystemError(FetchError): |
| + pass |
| + |
| + |
| +class FetchNotAllowedError(FetchError): |
| + pass |
| + |
| + |
| +def _run_git(checkout_dir, *args): |
| + if sys.platform.startswith(('win', 'cygwin')): |
| + cmd = ['git.bat'] |
| + else: |
| + cmd = ['git'] |
| + |
| + if checkout_dir is not None: |
| + cmd += ['-C', checkout_dir] |
| + cmd += list(args) |
| + |
| + logging.info('Running: %s', cmd) |
| + return subprocess42.check_output(cmd) |
| + |
| + |
| +def _cleanup_pyc(path): |
|
nodir
2016/05/20 23:03:49
why cleanup_pyc is in a file related to fetching?
Paweł Hajdan Jr.
2016/05/24 16:58:06
Good point. Moved back.
|
| + """Removes any .pyc files from |path|'s directory tree. |
| + This ensures we always use the fresh code. |
|
nodir
2016/05/20 23:03:49
there should be a blank line between first line of
Paweł Hajdan Jr.
2016/05/24 16:58:06
Done.
|
| + """ |
| + for root, dirs, files in os.walk(path): |
| + for f in files: |
| + if f.endswith('.pyc'): |
| + os.unlink(os.path.join(root, f)) |
| + |
| + |
| +def fetch_from_git(repo, revision, checkout_dir, allow_fetch): |
|
nodir
2016/05/20 23:03:49
it is strange to see "allow_fetch" parameter in a
Paweł Hajdan Jr.
2016/05/24 16:58:06
Yeah, I was also thinking it's somewhat ugly but m
|
| + """Fetches given |repo| at |revision| to |checkout_dir| using git. |
| + Network operations are performed only if |allow_fetch| is True. |
|
nodir
2016/05/20 23:03:49
here too
Paweł Hajdan Jr.
2016/05/24 16:58:06
Done.
|
| + """ |
| + logging.info('Freshening repository %s in %s', repo, checkout_dir) |
| + |
| + if not os.path.isdir(checkout_dir): |
| + if allow_fetch: |
|
nodir
2016/05/20 23:03:49
inverse condition and unindent the _run_git call
Paweł Hajdan Jr.
2016/05/24 16:58:06
Done.
|
| + _run_git(None, 'clone', '-q', repo, checkout_dir) |
| + else: |
| + raise FetchNotAllowedError( |
| + 'need to clone %s but fetch not allowed' % repo) |
| + elif not os.path.isdir(os.path.join(checkout_dir, '.git')): |
| + raise UncleanFilesystemError( |
| + '%s exists but is not a git repo' % checkout_dir) |
| + |
|
nodir
2016/05/20 23:03:49
what if checkout dir exists but points to a differ
Paweł Hajdan Jr.
2016/05/24 16:58:06
Good point. Out of scope for this CL (this is just
nodir
2016/05/24 17:46:31
It is likely to succeed because default value of r
Paweł Hajdan Jr.
2016/05/25 00:15:41
Yeah, fair enough. I added code to verify it in th
|
| + try: |
| + _run_git(checkout_dir, 'rev-parse', '-q', '--verify', |
| + '%s^{commit}' % revision) |
| + except subprocess42.CalledProcessError: |
| + if allow_fetch: |
|
nodir
2016/05/20 23:03:49
same here
Paweł Hajdan Jr.
2016/05/24 16:58:06
Done.
|
| + _run_git(checkout_dir, 'fetch') |
| + else: |
| + raise FetchNotAllowedError( |
| + 'need to fetch %s but fetch not allowed' % repo) |
| + _run_git(checkout_dir, 'reset', '-q', '--hard', revision) |
| + _cleanup_pyc(checkout_dir) |
|
nodir
2016/05/20 23:03:49
pyc files have nothing to do with git. This functi
Paweł Hajdan Jr.
2016/05/24 16:58:06
Yup, done.
|