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(