Add --no-hostname option

Author: jwodder

CHANGELOG.md

@@ -2,6 +2,7 @@ v0.7.0 (in development)
 -----------------------
 - The branch name/`HEAD` description is now truncated if it gets too long.
 - Add `--theme` option for selecting between dark and light themes
+- Add `--no-hostname` option
 - Drop support for Python 3.8
 
 v0.6.0 (2024-05-12)

README.rst

@@ -113,6 +113,8 @@ Options
                 If running ``git status`` takes longer than the given number of
                 seconds (default: 3), disable the Git integration
 
+--no-hostname   Do not include the local hostname in the prompt string
+
 -T THEME, --theme THEME
                 Select the theme to use for coloring prompt elements.  The
                 available themes are ``dark`` (the default, for use with light

src/jwodder_ps1/__main__.py

@@ -43,6 +43,11 @@ def main() -> None:
             "  [default: 3]"
         ),
     )
+    parser.add_argument(
+        "--no-hostname",
+        action="store_true",
+        help="Do not show the local hostname",
+    )
     parser.add_argument(
         "-T",
         "--theme",
@@ -76,7 +81,9 @@ def main() -> None:
         else:
             s = ""
     else:
-        s = PromptInfo.get(git=show_git, git_timeout=args.git_timeout).display(paint)
+        s = PromptInfo.get(git=show_git, git_timeout=args.git_timeout).display(
+            paint, hostname=not args.no_hostname
+        )
     print(s)
 
 

src/jwodder_ps1/info.py

@@ -95,7 +95,7 @@ def get(cls, git: bool = True, git_timeout: float = 3) -> PromptInfo:
             git=gs,
         )
 
-    def display(self, paint: Painter) -> str:
+    def display(self, paint: Painter, hostname: bool = True) -> str:
         """
         Construct & return a complete prompt string for the current environment
         """
@@ -121,11 +121,11 @@ def display(self, paint: Painter) -> str:
         if self.venv_prompt is not None:
             ps1 += paint(f"({self.venv_prompt}) ", SC.VENV)
 
-        # Show the current hostname:
-        ps1 += paint(self.hostname, SC.HOST)
-
-        # Separator:
-        ps1 += ":"
+        if hostname:
+            # Show the current hostname:
+            ps1 += paint(self.hostname, SC.HOST)
+            # Separator:
+            ps1 += ":"
 
         # Show the path to the current working directory:
         ps1 += paint(self.cwdstr, SC.CWD)

test/test_info.py

@@ -107,3 +107,17 @@ def test_display_full_git_prompt_info_ansi_light() -> None:
         "\x1B[34m~/work\x1B[m"
         "@\x1B[32mmain\x1B[m$ "
     )
+
+
+def test_display_prompt_info_ansi_no_hostname() -> None:
+    info = PromptInfo(
+        mail=False,
+        debian_chroot=None,
+        conda_prompt_modifier=None,
+        venv_prompt=None,
+        hostname="firefly",
+        cwdstr="~/work",
+        git=None,
+    )
+    paint = Painter(ANSIStyler(), DARK_THEME)
+    assert info.display(paint, hostname=False) == "\x1B[96m~/work\x1B[m$ "