Commit 600d3302 authored by Stefan Behnel's avatar Stefan Behnel

fix #454: type of first parameter in classmethods

parent cf7d7053
...@@ -598,11 +598,13 @@ class CArgDeclNode(Node): ...@@ -598,11 +598,13 @@ class CArgDeclNode(Node):
# default ExprNode or None # default ExprNode or None
# default_value PyObjectConst constant for default value # default_value PyObjectConst constant for default value
# is_self_arg boolean Is the "self" arg of an extension type method # is_self_arg boolean Is the "self" arg of an extension type method
# is_type_arg boolean Is the "class" arg of an extension type classmethod
# is_kw_only boolean Is a keyword-only argument # is_kw_only boolean Is a keyword-only argument
child_attrs = ["base_type", "declarator", "default"] child_attrs = ["base_type", "declarator", "default"]
is_self_arg = 0 is_self_arg = 0
is_type_arg = 0
is_generic = 1 is_generic = 1
type = None type = None
name_declarator = None name_declarator = None
...@@ -669,6 +671,7 @@ class CSimpleBaseTypeNode(CBaseTypeNode): ...@@ -669,6 +671,7 @@ class CSimpleBaseTypeNode(CBaseTypeNode):
# longness integer # longness integer
# complex boolean # complex boolean
# is_self_arg boolean Is self argument of C method # is_self_arg boolean Is self argument of C method
# is_type_arg boolean Is type argument of class method
child_attrs = [] child_attrs = []
arg_name = None # in case the argument name was interpreted as a type arg_name = None # in case the argument name was interpreted as a type
...@@ -1710,6 +1713,21 @@ class DefNode(FuncDefNode): ...@@ -1710,6 +1713,21 @@ class DefNode(FuncDefNode):
self.create_local_scope(env) self.create_local_scope(env)
def analyse_signature(self, env): def analyse_signature(self, env):
self.is_classmethod = self.is_staticmethod = False
if self.decorators:
for decorator in self.decorators:
func = decorator.decorator
if func.is_name:
self.is_classmethod |= func.name == 'classmethod'
self.is_staticmethod |= func.name == 'staticmethod'
if self.is_classmethod and env.lookup_here('classmethod'):
# classmethod() was overridden - not much we can do here ...
self.is_classmethod = False
if self.is_staticmethod and env.lookup_here('staticmethod'):
# staticmethod() was overridden - not much we can do here ...
self.is_staticmethod = False
any_type_tests_needed = 0 any_type_tests_needed = 0
if self.entry.is_special: if self.entry.is_special:
self.entry.trivial_signature = len(self.args) == 1 and not (self.star_arg or self.starstar_arg) self.entry.trivial_signature = len(self.args) == 1 and not (self.star_arg or self.starstar_arg)
...@@ -1733,7 +1751,11 @@ class DefNode(FuncDefNode): ...@@ -1733,7 +1751,11 @@ class DefNode(FuncDefNode):
if i < len(self.args): if i < len(self.args):
arg = self.args[i] arg = self.args[i]
arg.is_generic = 0 arg.is_generic = 0
if sig.is_self_arg(i): if sig.is_self_arg(i) and not self.is_staticmethod:
if self.is_classmethod:
arg.is_type_arg = 1
arg.hdr_type = arg.type = Builtin.type_type
else:
arg.is_self_arg = 1 arg.is_self_arg = 1
arg.hdr_type = arg.type = env.parent_type arg.hdr_type = arg.type = env.parent_type
arg.needs_conversion = 0 arg.needs_conversion = 0
......
...@@ -6,5 +6,4 @@ class_attribute_init_values_T18 ...@@ -6,5 +6,4 @@ class_attribute_init_values_T18
numpy_ValueError_T172 numpy_ValueError_T172
unsignedbehaviour_T184 unsignedbehaviour_T184
missing_baseclass_in_predecl_T262 missing_baseclass_in_predecl_T262
tp_new_T454
cfunc_call_tuple_args_T408 cfunc_call_tuple_args_T408
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment