Fix support for custom venv prompt prefixes

Author: jwodder

CHANGELOG.md

@@ -1,3 +1,7 @@
+v0.7.1 (in development)
+-----------------------
+- Fixed support for custom venv prompt prefixes
+
 v0.7.0 (2024-05-13)
 -------------------
 - The branch name/`HEAD` description is now truncated if it gets too long.

src/jwodder_ps1/__init__.py

@@ -19,7 +19,7 @@
 Visit <https://github.com/jwodder/ps1.py> for more information.
 """
 
-__version__ = "0.7.0"
+__version__ = "0.7.1.dev1"
 __author__ = "John Thorvald Wodder II"
 __author_email__ = "ps1@varonathe.org"
 __license__ = "MIT"

src/jwodder_ps1/info.py

@@ -65,11 +65,11 @@ def get(cls, git: bool = True, git_timeout: float = 3) -> PromptInfo:
                     for line in fp:
                         line = line.strip()
                         if m := re.match(r"^prompt\s*=\s*", line):
-                            prompt = line[m.end() :]
-                            if re.fullmatch(r'([\x27"]).*\1', prompt):
+                            venv_prompt = line[m.end() :]
+                            if re.fullmatch(r'([\x27"]).*\1', venv_prompt):
                                 # repr-ized prompt produced by venv
                                 try:
-                                    venv_prompt = literal_eval(prompt)
+                                    venv_prompt = literal_eval(venv_prompt)
                                 except Exception:
                                     pass
                             break