def GetRepositoryRoot():
"""Get the root directory (i.e. the top level directory) of this
repository. This returns None if this isn't a git repository."""
cmd = "git rev-parse --show-toplevel"
status = shell.CaptureShell(cmd)
if len(status[0]) > 0: return status[0].splitlines()[0]
return None
def BranchUntracked():
"""Check if there are untracked files in the current branch."""
status = shell.CaptureShell("git status --porcelain")
lines = status[0].splitlines()
untracked = 0
for line in lines:
if line[0] == "?": untracked += 1
return untracked
def BranchModified():
"""Check if there are modified files in the current branch."""
status = shell.CaptureShell("git status --porcelain")
lines = status[0].splitlines()
modified = 0
for line in lines:
if line[0] == "M" or line[1] == "M": modified += 1
return modified
def GetRemotes(glob=None):
"""Get the list of remotes for this repository. If glob is
provided, then only remote repositories that match the shell glob
are returned."""
cmd = "git rev-parse --symbolic --remotes"
if glob != None: cmd += "=" + glob
status = shell.CaptureShell(cmd)
if len(status[0]) > 0: return status[0].splitlines()
return []
def BranchAhead(fetch=True):
"""Check if the current HEAD is ahead of the remote. This parses
the git status command to determine if the current HEAD is up to
date. It returns an integer that is 0 if the branch is up to
date, negative if it is behind, and positive if it is ahead. If
fetch is false, then this will not try to synchronize with the
remote."""
# First fetch from the remote so we have all of the information
if fetch: shell.Shell("git fetch --all")
status = shell.CaptureShell("git status -b -s")
lines = status[0].splitlines()
match = re.search(r"\[ahead ([0-9])*\]",lines[0])
if match != None: return int(match.group(1))
match = re.search(r"\[behind ([0-9])*\]",lines[0])
if match != None: return int(match.group(1))
return 0
def BranchName():
"""Get the name of the current branch."""
status = shell.CaptureShell("git symbolic-ref HEAD");
if len(status[0]) > 0: return os.path.basename(status[0].splitlines()[0])
return "detached"