Merge pull request #24 from benbariteau/bariteau/fix_pypy3_support

fix pypy matching for change in pypy3
Yelp · Mar 9, 2021 · e02303a · e02303a
2 parents c355a0b + 808bb3d
commit e02303a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/virtualenv_tools.py b/virtualenv_tools.py
@@ -29,7 +29,7 @@
     'activate.xsh',
 ]
 _pybin_match = re.compile(r'^python\d+\.\d+$')
-_pypy_match = re.compile(r'^\d+.\d+$')
+_pypy_match = re.compile(r'^\d+(.\d+)?$')
 _activation_path_re = re.compile(
     r'^(?:set -gx |setenv |)VIRTUAL_ENV[ =][\'"](.*?)[\'"]\s*$',
 )
@@ -294,7 +294,7 @@ def _get_original_state(path):
         raise NotAVirtualenvError(
             path,
             'directory',
-            os.path.join(base_lib_dir, '#.#' if is_pypy else 'python#.#'),
+            os.path.join(base_lib_dir, '#(.#)?' if is_pypy else 'python#.#'),
         )
     lib_dir, = lib_dirs