[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, super(Module, self).__init__(engine, config,
bumblebee.output.Widget(full_text=self.gitinfo) 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): def gitinfo(self, widget):
info = "" info = ""
directory = None directory = None
data = { data = {
"branch": "n/a", "branch": "n/a",
"directory": "n/a",
"flags": {},
} }
try: try:
directory = bumblebee.util.execute("xcwd").strip() directory = bumblebee.util.execute("xcwd").strip()
directory = self._get_git_root(directory) directory = self._get_git_root(directory)
repo = pygit2.Repository(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["branch"] = repo.head.shorthand
data["directory"] = directory data["directory"] = directory
data["flags"] = " ".join([self._engine._theme.symbol(widget, name, name[0]) for name in data["flags"].keys()])
except Exception as e: except Exception as e:
return e return e

View file

@ -115,6 +115,9 @@ class Theme(object):
pre = self._get(widget, "prefix", None) pre = self._get(widget, "prefix", None)
return u"{}{}{}".format(padding, pre, padding) if pre else default 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): def suffix(self, widget, default=None):
"""Return the theme suffix for a widget's full text""" """Return the theme suffix for a widget's full text"""
padding = self._get(widget, "padding", "") padding = self._get(widget, "padding", "")

View file

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