implement changed

xontrib/pygitstatus/entrypoint.py

@@ -2,7 +2,7 @@
 
 from xonsh.built_ins import XonshSession
 
-from .prompts import (ahead, behind, branch, clean, repo_path, short_head, stash_count, tag, untracked)
+from .prompts import (ahead, behind, branch, changed, clean, repo_path, short_head, stash_count, tag, untracked)
 
 
 def _load_xontrib_(xsh: XonshSession, **_):
@@ -20,6 +20,7 @@ def _load_xontrib_(xsh: XonshSession, **_):
     prompt_fields['pygitstatus.ahead'] = ahead
     prompt_fields['pygitstatus.behind'] = behind
     prompt_fields['pygitstatus.branch'] = branch
+    prompt_fields['pygitstatus.changed'] = changed
     prompt_fields['pygitstatus.clean'] = clean
     prompt_fields['pygitstatus.repo_path'] = repo_path
     prompt_fields['pygitstatus.short_head'] = short_head

xontrib/pygitstatus/prompts.py

@@ -2,7 +2,7 @@
 import os
 
 # pylint: disable=no-name-in-module
-from pygit2 import GIT_STATUS_WT_NEW, Commit, GitError
+from pygit2 import GIT_STATUS_WT_MODIFIED, GIT_STATUS_WT_NEW, Commit, GitError
 from pygit2 import Repository as Repo
 from xonsh.prompt.base import MultiPromptField, PromptField, PromptFields
 
@@ -49,6 +49,16 @@ def branch(fld: PromptField, ctx: PromptFields):
         fld.value = repo.head.shorthand
 
 
+@PromptField.wrap(prefix="{BLUE}+", suffix="{RESET}", info="changed")
+def changed(fld: PromptField, ctx: PromptFields):
+    fld.value = ''
+    with contextlib.suppress(GitError):
+        repo = Repo('.')
+        untracked_count = len([v for k, v in repo.status().items() if v == GIT_STATUS_WT_MODIFIED])
+        if untracked_count > 0:
+            fld.value = str(untracked_count)
+
+
 @PromptField.wrap(prefix='{BOLD_GREEN}', suffix='{RESET}', symbol='✓')
 def clean(fld: PromptField, ctx: PromptFields):
 
@@ -115,8 +125,8 @@ def tag(fld: PromptField, ctx: PromptFields):
 
 @PromptField.wrap(prefix="…", info="untracked")
 def untracked(fld: PromptField, ctx: PromptFields):
+    fld.value = ''
     with contextlib.suppress(GitError):
-        fld.value = ''
         repo = Repo('.')
         untracked_count = len([v for k, v in repo.status().items() if v == GIT_STATUS_WT_NEW])
         if untracked_count > 0: