Support for python 3.10 match statement (#10191)

python · Jan 18, 2022 · 9b63751 · 9b63751
1 parent 1aa9cf9
commit 9b63751
Show file tree

Hide file tree

Showing 39 changed files with 3,620 additions and 133 deletions.
diff --git a/mypy/checker.py b/mypy/checker.py
diff --git a/mypy/checkexpr.py b/mypy/checkexpr.py
@@ -3077,7 +3077,11 @@ def nonliteral_tuple_index_helper(self, left_type: TupleType, index: Expression)
             else:
                 return union
 
-    def visit_typeddict_index_expr(self, td_type: TypedDictType, index: Expression) -> Type:
+    def visit_typeddict_index_expr(self, td_type: TypedDictType,
+                                   index: Expression,
+                                   local_errors: Optional[MessageBuilder] = None
+                                   ) -> Type:
+        local_errors = local_errors or self.msg
         if isinstance(index, (StrExpr, UnicodeExpr)):
             key_names = [index.value]
         else:
@@ -3097,14 +3101,14 @@ def visit_typeddict_index_expr(self, td_type: TypedDictType, index: Expression)
                         and key_type.fallback.type.fullname != 'builtins.bytes'):
                     key_names.append(key_type.value)
                 else:
-                    self.msg.typeddict_key_must_be_string_literal(td_type, index)
+                    local_errors.typeddict_key_must_be_string_literal(td_type, index)
                     return AnyType(TypeOfAny.from_error)
 
         value_types = []
         for key_name in key_names:
             value_type = td_type.items.get(key_name)
             if value_type is None:
-                self.msg.typeddict_key_not_found(td_type, key_name, index)
+                local_errors.typeddict_key_not_found(td_type, key_name, index)
                 return AnyType(TypeOfAny.from_error)
             else:
                 value_types.append(value_type)