[modules/git] Add some status icons
Add icons for: * local modifications * local delete * local add
This commit is contained in:
parent
212966945c
commit
5738aac1ce
3 changed files with 30 additions and 2 deletions
|
@ -22,20 +22,40 @@ class Module(bumblebee.engine.Module):
|
|||
super(Module, self).__init__(engine, config,
|
||||
bumblebee.output.Widget(full_text=self.gitinfo)
|
||||
)
|
||||
self._fmt = self.parameter("format", "{branch} - {directory}")
|
||||
self._engine = engine
|
||||
self._fmt = self.parameter("format", "{branch} {flags}")
|
||||
|
||||
def hidden(self):
|
||||
return False # TODO
|
||||
|
||||
def gitinfo(self, widget):
|
||||
info = ""
|
||||
directory = None
|
||||
data = {
|
||||
"branch": "n/a",
|
||||
"directory": "n/a",
|
||||
"flags": {},
|
||||
}
|
||||
try:
|
||||
directory = bumblebee.util.execute("xcwd").strip()
|
||||
directory = self._get_git_root(directory)
|
||||
repo = pygit2.Repository(directory)
|
||||
|
||||
for filepath, flags in repo.status().items():
|
||||
if flags == pygit2.GIT_STATUS_WT_NEW or \
|
||||
flags == pygit2.GIT_STATUS_INDEX_NEW:
|
||||
data["flags"]["new"] = True
|
||||
if flags == pygit2.GIT_STATUS_WT_DELETED or \
|
||||
flags == pygit2.GIT_STATUS_INDEX_DELETED:
|
||||
data["flags"]["deleted"] = True
|
||||
if flags == pygit2.GIT_STATUS_WT_MODIFIED or \
|
||||
flags == pygit2.GIT_STATUS_INDEX_MODIFIED:
|
||||
data["flags"]["modified"] = True
|
||||
|
||||
data["branch"] = repo.head.shorthand
|
||||
data["directory"] = directory
|
||||
data["flags"] = " ".join([self._engine._theme.symbol(widget, name, name[0]) for name in data["flags"].keys()])
|
||||
|
||||
except Exception as e:
|
||||
return e
|
||||
|
||||
|
|
|
@ -115,6 +115,9 @@ class Theme(object):
|
|||
pre = self._get(widget, "prefix", None)
|
||||
return u"{}{}{}".format(padding, pre, padding) if pre else default
|
||||
|
||||
def symbol(self, widget, name, default=None):
|
||||
return self._get(widget, name, default)
|
||||
|
||||
def suffix(self, widget, default=None):
|
||||
"""Return the theme suffix for a widget's full text"""
|
||||
padding = self._get(widget, "padding", "")
|
||||
|
|
|
@ -184,5 +184,10 @@
|
|||
"copying": {
|
||||
"prefix": ""
|
||||
}
|
||||
},
|
||||
"git": {
|
||||
"added": "",
|
||||
"modified": "",
|
||||
"deleted": ""
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue