bin-utils/show_status

171 lines
5.2 KiB
Plaintext
Raw Permalink Normal View History

#!/usr/bin/env python
# @desc Tired of having to go into each sub dir to find out whether or
# not you did a git commit? Tire no more, just use this!
#
# @author Mike Pearce <mike@mikepearce.net>
# @since 18/05/2010
# Grab some libraries
import sys
import os
import glob
import commands
from optparse import OptionParser
# Setup some stuff
dirname = './'
gitted = False
mini = True
class bcolors:
HEADER = '\033[95m'
OKBLUE = '\033[94m'
OKGREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'
def disable(self):
self.HEADER = ''
self.OKBLUE = ''
self.OKGREEN = ''
self.WARNING = ''
self.FAIL = ''
self.ENDC = ''
parser = OptionParser(description="\
Show Status is awesome. If you tell it a directory to look in, it'll scan \
through all the sub dirs looking for a .git directory. When it finds one \
it'll look to see if there are any changes and let you know. \
It can also push and pull to/from a remote location (like github.com) \
(but only if there are no changes.) \
Contact mike@mikepearce.net for any support.")
parser.add_option("-d", "--dir",
dest = "dirname",
action = "store",
help = "The directory to parse sub dirs from",
default = os.path.abspath("./")+"/"
)
parser.add_option("-v", "--verbose",
action = "store_true",
dest = "verbose",
default = False,
help = "Show the full detail of git status"
)
parser.add_option("-r", "--remote",
action = "store",
dest = "remote",
default = "",
help = "Set the remote name (remotename:branchname)"
)
parser.add_option("--push",
action = "store_true",
dest = "push",
default = False,
help = "Do a 'git push' if you've set a remote with -r it will push to there"
)
parser.add_option("-p", "--pull",
action = "store_true",
dest = "pull",
default = False,
help = "Do a 'git pull' if you've set a remote with -r it will pull from there"
)
# Now, parse the args
(options, args) = parser.parse_args()
#-------------------
def show_error(error="Undefined Error!"):
#-------------------
"""Writes an error to stderr"""
sys.stderr.write(error)
sys.exit(1)
#-------------------
# Now, onto the main event!
#-------------------
if __name__ == "__main__":
os.system('clear')
sys.stdout.write('-- Starting git status...\n')
os.environ['LANGUAGE'] = 'en_US:en';
os.environ['LANG'] = 'en_US.UTF-8';
sys.stdout.write('Scanning sub directories of %s\n' %options.dirname)
# See whats here
for infile in glob.glob( os.path.join(options.dirname, '*') ):
#is there a .git file
if os.path.exists( os.path.join(infile, ".git") ):
#Yay, we found one!
gitted = True
# OK, contains a .git file. Let's descend into it
# and ask git for a status
out = commands.getoutput('cd '+ infile + '; git status')
# Mini?
if False == options.verbose:
j = out.find('On branch');
k = out.find('\n', j);
branch = out[j+10:k];
branchColor = bcolors.WARNING;
if branch == 'master':
branchColor = bcolors.OKGREEN
branch = "[ " + branchColor + branch.ljust(15) + bcolors.ENDC + " ]"
if -1 != out.find('nothing'):
result = bcolors.OKGREEN + "No Changes" + bcolors.ENDC
# Pull from the remote
if False != options.pull:
push = commands.getoutput(
'cd '+ infile +
'; git pull '+
' '.join(options.remote.split(":"))
)
result = result + " (Pulled) \n" + push
# Push to the remote
if False != options.push:
push = commands.getoutput(
'cd '+ infile +
'; git push '+
' '.join(options.remote.split(":"))
)
result = result + " (Pushed) \n" + push
else:
result = bcolors.FAIL + "Changes" + bcolors.ENDC
# Write to screen
sys.stdout.write("--" + bcolors.OKBLUE + infile.ljust(55) + bcolors.ENDC + branch + " : " + result +"\n")
else:
#Print some repo details
sys.stdout.write("\n---------------- "+ infile +" -----------------\n")
sys.stdout.write(out)
sys.stdout.write("\n---------------- "+ infile +" -----------------\n")
# Come out of the dir and into the next
commands.getoutput('cd ../')
if False == gitted:
show_error("Error: None of those sub directories had a .git file.\n")
sys.stdout.write("Done\n")