Make this code path more safe to prevent stack trace/regression.

pypa · Sep 29, 2024 · fa08ff5 · fa08ff5
1 parent 7b8ba87
commit fa08ff5
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pipenv/exceptions.py b/pipenv/exceptions.py
@@ -12,9 +12,13 @@
 
 def unstyle(text: str) -> str:
  """Remove all styles from the given text."""
- styled_text = Text.from_markup(text)
- stripped_text = styled_text.strip_styles()
- return stripped_text.to_plain_text()
+ try:
+ styled_text = Text.from_markup(text)
+ stripped_text = styled_text.strip_styles()
+ return stripped_text.plain
+ except AttributeError:
+ # Fallback if the expected methods are not available
+ return str(text)
 
 
 KnownException = namedtuple(