def check_method_type(functype: FunctionLike, itype: Instance, is_classmethod: bool, context: Context, msg: MessageBuilder) -> None: for item in functype.items(): if not item.arg_types or item.arg_kinds[0] not in (ARG_POS, ARG_STAR): # No positional first (self) argument (*args is okay). msg.invalid_method_type(item, context) elif not is_classmethod: # Check that self argument has type 'Any' or valid instance type. selfarg = item.arg_types[0] # If this is a method of a tuple class, correct for the fact that # we passed to typ.fallback in analyze_member_access. See #1432. if isinstance(selfarg, TupleType): selfarg = selfarg.fallback if not subtypes.is_subtype(selfarg, itype): msg.invalid_method_type(item, context) else: # Check that cls argument has type 'Any' or valid class type. # (This is sufficient for the current treatment of @classmethod, # but probably needs to be revisited when we implement Type[C] # or advanced variants of it like Type[<args>, C].) clsarg = item.arg_types[0] if isinstance(clsarg, CallableType) and clsarg.is_type_obj(): if not subtypes.is_equivalent(clsarg.ret_type, itype): msg.invalid_class_method_type(item, context) else: if not subtypes.is_equivalent(clsarg, AnyType()): msg.invalid_class_method_type(item, context)
def check_method_type( functype: FunctionLike, itype: Instance, is_classmethod: bool, context: Context, msg: MessageBuilder ) -> None: for item in functype.items(): if not item.arg_types or item.arg_kinds[0] not in (ARG_POS, ARG_STAR): # No positional first (self) argument (*args is okay). msg.invalid_method_type(item, context) elif not is_classmethod: # Check that self argument has type 'Any' or valid instance type. selfarg = item.arg_types[0] # If this is a method of a tuple class, correct for the fact that # we passed to typ.fallback in analyze_member_access. See #1432. if isinstance(selfarg, TupleType): selfarg = selfarg.fallback if not subtypes.is_equivalent(selfarg, itype): msg.invalid_method_type(item, context) else: # Check that cls argument has type 'Any' or valid class type. # (This is sufficient for the current treatment of @classmethod, # but probably needs to be revisited when we implement Type[C] # or advanced variants of it like Type[<args>, C].) clsarg = item.arg_types[0] if isinstance(clsarg, CallableType) and clsarg.is_type_obj(): if not subtypes.is_equivalent(clsarg.ret_type, itype): msg.invalid_class_method_type(item, context) else: if not subtypes.is_equivalent(clsarg, AnyType()): msg.invalid_class_method_type(item, context)
def check_method_type(functype: FunctionLike, itype: Instance, context: Context, msg: MessageBuilder) -> None: for item in functype.items(): if not item.arg_types or item.arg_kinds[0] != ARG_POS: # No positional first (self) argument. msg.invalid_method_type(item, context) else: # Check that self argument has type 'Any' or valid instance type. selfarg = item.arg_types[0] if not subtypes.is_equivalent(selfarg, itype): msg.invalid_method_type(item, context)