2016-09-06 02:37:25 +00:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
|
|
import os
|
|
|
|
import subprocess
|
2016-09-14 12:54:59 +00:00
|
|
|
import xml.etree.ElementTree as ET
|
2016-09-06 02:37:25 +00:00
|
|
|
|
2016-10-10 23:14:50 +00:00
|
|
|
from common import git
|
|
|
|
from common import Colors
|
2016-09-06 02:37:25 +00:00
|
|
|
|
|
|
|
|
2016-10-10 23:14:50 +00:00
|
|
|
SCRIPTDIR = os.path.dirname(__file__)
|
2016-09-06 02:37:25 +00:00
|
|
|
|
|
|
|
|
2016-09-14 12:54:59 +00:00
|
|
|
def manifest_get_commits(manifest):
|
|
|
|
res = {}
|
|
|
|
tree = ET.parse(manifest)
|
|
|
|
root = tree.getroot()
|
|
|
|
for child in root:
|
|
|
|
if child.tag == 'project':
|
|
|
|
res[child.attrib["name"]] = child.attrib["revision"]
|
|
|
|
return res
|
|
|
|
|
|
|
|
|
2016-09-29 23:25:42 +00:00
|
|
|
def update_subprojects(manifest, no_interaction=False):
|
2016-09-14 12:54:59 +00:00
|
|
|
if manifest:
|
|
|
|
repos_commits = manifest_get_commits(manifest)
|
|
|
|
else:
|
|
|
|
repos_commits = {}
|
|
|
|
|
2016-09-06 02:37:25 +00:00
|
|
|
subprojects_dir = os.path.join(SCRIPTDIR, "subprojects")
|
|
|
|
for repo_name in os.listdir(subprojects_dir):
|
|
|
|
repo_dir = os.path.normpath(os.path.join(SCRIPTDIR, subprojects_dir, repo_name))
|
|
|
|
if not os.path.exists(os.path.join(repo_dir, '.git')):
|
|
|
|
continue
|
2016-09-29 23:25:42 +00:00
|
|
|
revision = repos_commits.get(repo_name)
|
|
|
|
if not update_repo(repo_name, repo_dir, revision, no_interaction):
|
|
|
|
return False
|
|
|
|
|
|
|
|
return True
|
2016-09-06 02:37:25 +00:00
|
|
|
|
2016-09-29 23:25:42 +00:00
|
|
|
|
|
|
|
def update_repo(repo_name, repo_dir, revision, no_interaction, recurse_i=0):
|
|
|
|
print("Updating %s..." % repo_name)
|
|
|
|
try:
|
|
|
|
if revision:
|
|
|
|
git(["fetch"], repo_dir)
|
|
|
|
git(["checkout", revision], repo_dir)
|
|
|
|
else:
|
|
|
|
git(["pull", "--rebase"], repo_dir)
|
|
|
|
except Exception as e:
|
|
|
|
out = getattr(e, "output", b"").decode()
|
|
|
|
if not no_interaction:
|
|
|
|
print("====================================="
|
|
|
|
"\n%sEntering a shell in %s to fix that"
|
|
|
|
" just `exit` once done`"
|
|
|
|
"\n=====================================" % (
|
|
|
|
out, os.getcwd()))
|
|
|
|
try:
|
|
|
|
subprocess.check_call(os.environ.get("SHELL", "/bin/sh"),
|
|
|
|
cwd=repo_dir)
|
|
|
|
except:
|
|
|
|
# Result of subshell does not really matter
|
|
|
|
pass
|
|
|
|
|
|
|
|
if recurse_i < 3:
|
|
|
|
return update_repo(repo_name, repo_dir, revision, no_interaction,
|
|
|
|
recurse_i + 1)
|
2016-09-06 02:37:25 +00:00
|
|
|
return False
|
2016-09-29 23:25:42 +00:00
|
|
|
else:
|
|
|
|
print("\nCould not rebase %s, please fix and try again."
|
|
|
|
" Error:\n\n%s %s" % (repo_dir, out, e))
|
2016-09-06 02:37:25 +00:00
|
|
|
|
2016-09-29 23:25:42 +00:00
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
|
|
commit_message = git("show", repo_dir).split("\n")
|
|
|
|
print(u" -> %s%s%s — %s" % (Colors.HEADER, commit_message[0][7:14], Colors.ENDC,
|
|
|
|
commit_message[4].strip()))
|
2016-09-06 02:37:25 +00:00
|
|
|
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
parser = argparse.ArgumentParser(prog="git-update")
|
|
|
|
|
|
|
|
parser.add_argument("--no-color",
|
|
|
|
default=False,
|
|
|
|
action='store_true',
|
|
|
|
help="Do not output ansi colors.")
|
2016-09-29 23:25:42 +00:00
|
|
|
parser.add_argument("--no-interaction",
|
|
|
|
default=False,
|
|
|
|
action='store_true',
|
|
|
|
help="Do not allow interaction with the user.")
|
2016-09-14 12:54:59 +00:00
|
|
|
parser.add_argument("--manifest",
|
|
|
|
default=None,
|
|
|
|
help="Use a android repo manifest to sync repositories"
|
|
|
|
" Note that it will let all repositories in detached state")
|
2016-09-06 02:37:25 +00:00
|
|
|
options = parser.parse_args()
|
|
|
|
if options.no_color:
|
|
|
|
Colors.disable()
|
|
|
|
|
2016-09-29 23:25:42 +00:00
|
|
|
exit(not update_subprojects(options.manifest,
|
|
|
|
options.no_interaction))
|