diff options
Diffstat (limited to 'subcmds/start.py')
| -rw-r--r-- | subcmds/start.py | 68 |
1 files changed, 43 insertions, 25 deletions
diff --git a/subcmds/start.py b/subcmds/start.py index 6ec0b2ce..2addaf2e 100644 --- a/subcmds/start.py +++ b/subcmds/start.py | |||
| @@ -1,5 +1,3 @@ | |||
| 1 | # -*- coding:utf-8 -*- | ||
| 2 | # | ||
| 3 | # Copyright (C) 2008 The Android Open Source Project | 1 | # Copyright (C) 2008 The Android Open Source Project |
| 4 | # | 2 | # |
| 5 | # Licensed under the Apache License, Version 2.0 (the "License"); | 3 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| @@ -14,19 +12,20 @@ | |||
| 14 | # See the License for the specific language governing permissions and | 12 | # See the License for the specific language governing permissions and |
| 15 | # limitations under the License. | 13 | # limitations under the License. |
| 16 | 14 | ||
| 17 | from __future__ import print_function | 15 | import functools |
| 18 | import os | 16 | import os |
| 19 | import sys | 17 | import sys |
| 20 | 18 | ||
| 21 | from command import Command | 19 | from command import Command, DEFAULT_LOCAL_JOBS |
| 22 | from git_config import IsImmutable | 20 | from git_config import IsImmutable |
| 23 | from git_command import git | 21 | from git_command import git |
| 24 | import gitc_utils | 22 | import gitc_utils |
| 25 | from progress import Progress | 23 | from progress import Progress |
| 26 | from project import SyncBuffer | 24 | from project import SyncBuffer |
| 27 | 25 | ||
| 26 | |||
| 28 | class Start(Command): | 27 | class Start(Command): |
| 29 | common = True | 28 | COMMON = True |
| 30 | helpSummary = "Start a new branch for development" | 29 | helpSummary = "Start a new branch for development" |
| 31 | helpUsage = """ | 30 | helpUsage = """ |
| 32 | %prog <newbranchname> [--all | <project>...] | 31 | %prog <newbranchname> [--all | <project>...] |
| @@ -35,6 +34,7 @@ class Start(Command): | |||
| 35 | '%prog' begins a new branch of development, starting from the | 34 | '%prog' begins a new branch of development, starting from the |
| 36 | revision specified in the manifest. | 35 | revision specified in the manifest. |
| 37 | """ | 36 | """ |
| 37 | PARALLEL_JOBS = DEFAULT_LOCAL_JOBS | ||
| 38 | 38 | ||
| 39 | def _Options(self, p): | 39 | def _Options(self, p): |
| 40 | p.add_option('--all', | 40 | p.add_option('--all', |
| @@ -42,7 +42,8 @@ revision specified in the manifest. | |||
| 42 | help='begin branch in all projects') | 42 | help='begin branch in all projects') |
| 43 | p.add_option('-r', '--rev', '--revision', dest='revision', | 43 | p.add_option('-r', '--rev', '--revision', dest='revision', |
| 44 | help='point branch at this revision instead of upstream') | 44 | help='point branch at this revision instead of upstream') |
| 45 | p.add_option('--head', dest='revision', action='store_const', const='HEAD', | 45 | p.add_option('--head', '--HEAD', |
| 46 | dest='revision', action='store_const', const='HEAD', | ||
| 46 | help='abbreviation for --rev HEAD') | 47 | help='abbreviation for --rev HEAD') |
| 47 | 48 | ||
| 48 | def ValidateOptions(self, opt, args): | 49 | def ValidateOptions(self, opt, args): |
| @@ -53,6 +54,26 @@ revision specified in the manifest. | |||
| 53 | if not git.check_ref_format('heads/%s' % nb): | 54 | if not git.check_ref_format('heads/%s' % nb): |
| 54 | self.OptionParser.error("'%s' is not a valid name" % nb) | 55 | self.OptionParser.error("'%s' is not a valid name" % nb) |
| 55 | 56 | ||
| 57 | def _ExecuteOne(self, revision, nb, project): | ||
| 58 | """Start one project.""" | ||
| 59 | # If the current revision is immutable, such as a SHA1, a tag or | ||
| 60 | # a change, then we can't push back to it. Substitute with | ||
| 61 | # dest_branch, if defined; or with manifest default revision instead. | ||
| 62 | branch_merge = '' | ||
| 63 | if IsImmutable(project.revisionExpr): | ||
| 64 | if project.dest_branch: | ||
| 65 | branch_merge = project.dest_branch | ||
| 66 | else: | ||
| 67 | branch_merge = self.manifest.default.revisionExpr | ||
| 68 | |||
| 69 | try: | ||
| 70 | ret = project.StartBranch( | ||
| 71 | nb, branch_merge=branch_merge, revision=revision) | ||
| 72 | except Exception as e: | ||
| 73 | print('error: unable to checkout %s: %s' % (project.name, e), file=sys.stderr) | ||
| 74 | ret = False | ||
| 75 | return (ret, project) | ||
| 76 | |||
| 56 | def Execute(self, opt, args): | 77 | def Execute(self, opt, args): |
| 57 | nb = args[0] | 78 | nb = args[0] |
| 58 | err = [] | 79 | err = [] |
| @@ -60,7 +81,7 @@ revision specified in the manifest. | |||
| 60 | if not opt.all: | 81 | if not opt.all: |
| 61 | projects = args[1:] | 82 | projects = args[1:] |
| 62 | if len(projects) < 1: | 83 | if len(projects) < 1: |
| 63 | projects = ['.',] # start it in the local project by default | 84 | projects = ['.'] # start it in the local project by default |
| 64 | 85 | ||
| 65 | all_projects = self.GetProjects(projects, | 86 | all_projects = self.GetProjects(projects, |
| 66 | missing_ok=bool(self.gitc_manifest)) | 87 | missing_ok=bool(self.gitc_manifest)) |
| @@ -84,11 +105,8 @@ revision specified in the manifest. | |||
| 84 | if not os.path.exists(os.getcwd()): | 105 | if not os.path.exists(os.getcwd()): |
| 85 | os.chdir(self.manifest.topdir) | 106 | os.chdir(self.manifest.topdir) |
| 86 | 107 | ||
| 87 | pm = Progress('Starting %s' % nb, len(all_projects)) | 108 | pm = Progress('Syncing %s' % nb, len(all_projects), quiet=opt.quiet) |
| 88 | for project in all_projects: | 109 | for project in all_projects: |
| 89 | pm.update() | ||
| 90 | |||
| 91 | if self.gitc_manifest: | ||
| 92 | gitc_project = self.gitc_manifest.paths[project.relpath] | 110 | gitc_project = self.gitc_manifest.paths[project.relpath] |
| 93 | # Sync projects that have not been opened. | 111 | # Sync projects that have not been opened. |
| 94 | if not gitc_project.already_synced: | 112 | if not gitc_project.already_synced: |
| @@ -101,21 +119,21 @@ revision specified in the manifest. | |||
| 101 | sync_buf = SyncBuffer(self.manifest.manifestProject.config) | 119 | sync_buf = SyncBuffer(self.manifest.manifestProject.config) |
| 102 | project.Sync_LocalHalf(sync_buf) | 120 | project.Sync_LocalHalf(sync_buf) |
| 103 | project.revisionId = gitc_project.old_revision | 121 | project.revisionId = gitc_project.old_revision |
| 122 | pm.update() | ||
| 123 | pm.end() | ||
| 104 | 124 | ||
| 105 | # If the current revision is immutable, such as a SHA1, a tag or | 125 | def _ProcessResults(_pool, pm, results): |
| 106 | # a change, then we can't push back to it. Substitute with | 126 | for (result, project) in results: |
| 107 | # dest_branch, if defined; or with manifest default revision instead. | 127 | if not result: |
| 108 | branch_merge = '' | 128 | err.append(project) |
| 109 | if IsImmutable(project.revisionExpr): | 129 | pm.update() |
| 110 | if project.dest_branch: | ||
| 111 | branch_merge = project.dest_branch | ||
| 112 | else: | ||
| 113 | branch_merge = self.manifest.default.revisionExpr | ||
| 114 | 130 | ||
| 115 | if not project.StartBranch( | 131 | self.ExecuteInParallel( |
| 116 | nb, branch_merge=branch_merge, revision=opt.revision): | 132 | opt.jobs, |
| 117 | err.append(project) | 133 | functools.partial(self._ExecuteOne, opt.revision, nb), |
| 118 | pm.end() | 134 | all_projects, |
| 135 | callback=_ProcessResults, | ||
| 136 | output=Progress('Starting %s' % (nb,), len(all_projects), quiet=opt.quiet)) | ||
| 119 | 137 | ||
| 120 | if err: | 138 | if err: |
| 121 | for p in err: | 139 | for p in err: |
