Add special accounting for pypy when computing the stack level for te…

…xt encoding warnings. Closes #114.
jaraco · Mar 13, 2024 · 4cceb49 · 4cceb49
1 parent 2ec3ed8
commit 4cceb49
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/114.bugfix.rst b/114.bugfix.rst
@@ -0,0 +1 @@
+Add special accounting for pypy when computing the stack level for text encoding warnings.
diff --git a/zipp/__init__.py b/zipp/__init__.py
@@ -5,6 +5,7 @@
 import contextlib
 import pathlib
 import re
+import sys
 
 from .compat.py310 import text_encoding
 from .glob import Translator
@@ -179,8 +180,10 @@ def _name_set(self):
 
 
 def _extract_text_encoding(encoding=None, *args, **kwargs):
-    # stacklevel=3 so that the caller of the caller see any warning.
-    return text_encoding(encoding, 3), args, kwargs
+    # compute stack level so that the caller of the caller sees any warning.
+    is_pypy = sys.implementation.name == 'pypy'
+    stack_level = 3 + is_pypy
+    return text_encoding(encoding, stack_level), args, kwargs
 
 
 class Path: