#!/usr/bin/python
import os, sys
import gtk
import egg.trayicon     # egg == python-gnome2-extras
import gitutils
 
if len(sys.argv) != 2:
    exit('ERROR: required argument search path.')
 
searchPath = sys.argv[1]
trayIconImage = os.path.dirname(__file__) + '/icon.png'
 
def callback(widget, event):
    repos = gitutils.notCleanRepogitoryInfos(searchPath)
    menu = gtk.Menu()
    tooltips = gtk.Tooltips()
    tooltips.enable()
    tooltips.set_delay(100)
    for r in repos:
        menuitem_x = gtk.ImageMenuItem(r['menuItem'])
        menuitem_x.set_image(gtk.image_new_from_file('%s/%s.png' % (os.path.dirname(__file__), r['flag'])))
        tooltips.set_tip(menuitem_x, r['path'] + "\n" + r['status'])
        menu.append(menuitem_x)
    menuitem_exit = gtk.MenuItem("Exit")
    menu.append(menuitem_exit)
    menuitem_exit.connect("activate", lambda x: gtk.main_quit())
    menu.show_all()
    menu.popup(None, None, None, event.button, event.time, tray)
 
tray = egg.trayicon.TrayIcon("TrayIcon")
box = gtk.EventBox()
img = gtk.Image()
img.set_from_file(trayIconImage)
box.add(img)
tray.add(box)
tray.show_all()
 
box.connect("button-press-event", callback)
 
gtk.main()
 
# vim: set ts=4 sw=4 sts=4 expandtab fenc=utf-8:

#!/usr/bin/python
import os, sys
import subprocess
 
def findRepogitories(baseDir, acc):
    for filename in os.listdir(baseDir):
        fullPath = os.path.join(baseDir, filename)
        if os.path.isdir(fullPath) and filename == '.git':
            acc.append(baseDir)
        if os.path.isdir(fullPath):
            findRepogitories(fullPath, acc)
    return acc
 
def execCommand(cmd, path):
    p = subprocess.Popen(cmd, shell=True, cwd=path, stdin=subprocess.PIPE,
        stdout=subprocess.PIPE, stderr=subprocess.STDOUT,
        close_fds=True)
    return p.stdout.read().strip()
 
def createRepoInfo(path):
    branch = execCommand('git symbolic-ref HEAD', path)
    status = execCommand('git status', path)
    flag = 'C' if status.find('Changes not staged for commit:') > -1 else '_'
    flag += 'U' if status.find('Untracked files:') > -1 else '_'
    name = path.split('/')[-1] 
    menuItem = name 
    if branch != u'refs/heads/master':
        menuItem += ' (' + branch + ')'
    return {
        'name': name,
        'path': path,
        'branch': branch,
        'status': status,
        'flag': flag,
        'menuItem': menuItem
    }
 
def getRepogitoryInfos(repogitories):
    return map(createRepoInfo, repogitories)
 
def repogitoryInfos(path):
    repoPaths = findRepogitories(path, [])
    return getRepogitoryInfos(repoPaths)
 
def notCleanRepogitoryInfos(path):
    return filter(lambda x: x['status'].find('working directory clean') == -1, repogitoryInfos(path))
 
if __name__ == '__main__':
    path = sys.argv[1]
    print 'begin test.'
    repoPaths = findRepogitories(path, [])
    repos = getRepogitoryInfos(repoPaths)
    print repos
    print 'completed test.'
 
# vim: set ts=4 sw=4 sts=4 expandtab fenc=utf-8:

$ ./check-git-status.py ~/work

gsettings set com.canonical.Unity.Panel systray-whitelist "['JavaEmbeddedFrame', 'Mumble', 'Wine', 'Skype', 'hp-systray', 'scp-dbus-service', 'Dropbox', 'check-git-status']"