[modules/git] Add some status icons

Add icons for:
* local modifications
* local delete
* local add
This commit is contained in:
Tobias Witek 2019-01-20 14:43:27 +01:00
parent 212966945c
commit 5738aac1ce
3 changed files with 30 additions and 2 deletions

View file

@ -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

View file

@ -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", "")

View file

@ -184,5 +184,10 @@
"copying": {
"prefix": ""
}
},
"git": {
"added": "",
"modified": "",
"deleted": ""
}
}