Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(642)

Unified Diff: recipe_engine/fetch.py

Issue 1997023002: recipe engine: add remote_run command (Closed) Base URL: https://github.com/luci/recipes-py.git@master
Patch Set: 80cols Created 4 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « no previous file | recipe_engine/package.py » ('j') | recipe_engine/remote_run.py » ('J')
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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.
« no previous file with comments | « no previous file | recipe_engine/package.py » ('j') | recipe_engine/remote_run.py » ('J')

Powered by Google App Engine
This is Rietveld 408576698