util: also support classes when parsing a type

4ed07470 · Bryton Lacquement · 053c0ef5 · 4ed07470
Commit 4ed07470 authored Aug 05, 2020 by Bryton Lacquement 🚪
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

my2to3/util.py my2to3/util.py +2 -2

No files found.
--- a/my2to3/util.py
+++ b/my2to3/util.py
@@ -6,9 +6,9 @@ from lib2to3.pygram import python_symbols as syms
 import re
-type_match = re.compile(r"^<type '(.*)'>$").match
+type_match = re.compile(r"^<(type|class) '(.*)'>$").match
 def parse_type(type_):
-  return type_match(str(type_)).group(1)
+  return type_match(str(type_)).group(2)
 def data2types(data):