1 # cython: auto_cpdef=True, infer_types=True, language_level=3
6 # This should be done automatically
8 cython.declare(Nodes=object, ExprNodes=object, EncodedString=object)
14 from Cython.Compiler.Scanning import PyrexScanner, FileSourceDescriptor
18 from StringEncoding import EncodedString, BytesLiteral, _unicode, _bytes
19 from ModuleNode import ModuleNode
20 from Errors import error, warning, InternalError
21 from Cython import Utils
28 visibility = 'private'
37 def __init__(self, **kwds):
38 self.__dict__.update(kwds)
40 def __call__(self, **kwds):
43 d.update(self.__dict__)
47 def p_ident(s, message = "Expected an identifier"):
57 while s.sy == 'IDENT':
58 names.append(s.systring)
65 #------------------------------------------
69 #------------------------------------------
71 def p_binop_operator(s):
77 def p_binop_expr(s, ops, p_sub_expr):
80 op, pos = p_binop_operator(s)
82 n1 = ExprNodes.binop_node(pos, op, n1, n2)
84 if Future.division in s.context.future_directives:
85 n1.truedivision = True
87 n1.truedivision = None # unknown
90 #lambdef: 'lambda' [varargslist] ':' test
92 def p_lambdef(s, allow_conditional=True):
98 star_arg = starstar_arg = None
100 args, star_arg, starstar_arg = p_varargslist(
101 s, terminator=':', annotated=False)
103 if allow_conditional:
106 expr = p_test_nocond(s)
107 return ExprNodes.LambdaNode(
109 star_arg = star_arg, starstar_arg = starstar_arg,
112 #lambdef_nocond: 'lambda' [varargslist] ':' test_nocond
114 def p_lambdef_nocond(s):
115 return p_lambdef(s, allow_conditional=False)
117 #test: or_test ['if' or_test 'else' test] | lambdef
129 return ExprNodes.CondExprNode(pos, test=test, true_val=expr, false_val=other)
133 #test_nocond: or_test | lambdef_nocond
135 def p_test_nocond(s):
137 return p_lambdef_nocond(s)
141 #or_test: and_test ('or' and_test)*
144 return p_rassoc_binop_expr(s, ('or',), p_and_test)
146 def p_rassoc_binop_expr(s, ops, p_subexpr):
152 n2 = p_rassoc_binop_expr(s, ops, p_subexpr)
153 n1 = ExprNodes.binop_node(pos, op, n1, n2)
156 #and_test: not_test ('and' not_test)*
159 #return p_binop_expr(s, ('and',), p_not_test)
160 return p_rassoc_binop_expr(s, ('and',), p_not_test)
162 #not_test: 'not' not_test | comparison
168 return ExprNodes.NotNode(pos, operand = p_not_test(s))
170 return p_comparison(s)
172 #comparison: expr (comp_op expr)*
173 #comp_op: '<'|'>'|'=='|'>='|'<='|'<>'|'!='|'in'|'not' 'in'|'is'|'is' 'not'
176 n1 = p_starred_expr(s)
177 if s.sy in comparison_ops:
180 n2 = p_starred_expr(s)
181 n1 = ExprNodes.PrimaryCmpNode(pos,
182 operator = op, operand1 = n1, operand2 = n2)
183 if s.sy in comparison_ops:
184 n1.cascade = p_cascaded_cmp(s)
187 def p_test_or_starred_expr(s):
189 return p_starred_expr(s)
193 def p_starred_expr(s):
202 expr = ExprNodes.StarredTargetNode(pos, expr)
205 def p_cascaded_cmp(s):
208 n2 = p_starred_expr(s)
209 result = ExprNodes.CascadedCmpNode(pos,
210 operator = op, operand2 = n2)
211 if s.sy in comparison_ops:
212 result.cascade = p_cascaded_cmp(s)
235 '<', '>', '==', '>=', '<=', '<>', '!=',
239 #expr: xor_expr ('|' xor_expr)*
242 return p_binop_expr(s, ('|',), p_xor_expr)
244 #xor_expr: and_expr ('^' and_expr)*
247 return p_binop_expr(s, ('^',), p_and_expr)
249 #and_expr: shift_expr ('&' shift_expr)*
252 return p_binop_expr(s, ('&',), p_shift_expr)
254 #shift_expr: arith_expr (('<<'|'>>') arith_expr)*
257 return p_binop_expr(s, ('<<', '>>'), p_arith_expr)
259 #arith_expr: term (('+'|'-') term)*
262 return p_binop_expr(s, ('+', '-'), p_term)
264 #term: factor (('*'|'/'|'%') factor)*
267 return p_binop_expr(s, ('*', '/', '%', '//'), p_factor)
269 #factor: ('+'|'-'|'~'|'&'|typecast|sizeof) factor | power
272 # little indirection for C-ification purposes
277 if sy in ('+', '-', '~'):
281 return ExprNodes.unop_node(pos, op, p_factor(s))
286 return ExprNodes.AmpersandNode(pos, operand = arg)
289 elif sy == 'IDENT' and s.systring == "sizeof":
298 base_type = p_c_base_type(s)
299 if base_type.name is None:
300 s.error("Unknown type")
301 declarator = p_c_declarator(s, empty = 1)
308 operand = p_factor(s)
309 return ExprNodes.TypecastNode(pos,
310 base_type = base_type,
311 declarator = declarator,
313 typecheck = typecheck)
316 # s.sy == ident "sizeof"
320 # Here we decide if we are looking at an expression or type
321 # If it is actually a type, but parsable as an expression,
322 # we treat it as an expression here.
323 if looking_at_expr(s):
325 node = ExprNodes.SizeofVarNode(pos, operand = operand)
327 base_type = p_c_base_type(s)
328 declarator = p_c_declarator(s, empty = 1)
329 node = ExprNodes.SizeofTypeNode(pos,
330 base_type = base_type, declarator = declarator)
334 def p_yield_expression(s):
338 if s.sy != ')' and s.sy not in statement_terminators:
342 return ExprNodes.YieldExprNode(pos, arg=arg)
344 def p_yield_statement(s):
346 yield_expr = p_yield_expression(s)
347 return Nodes.ExprStatNode(yield_expr.pos, expr=yield_expr)
349 #power: atom trailer* ('**' factor)*
352 if s.systring == 'new' and s.peek()[0] == 'IDENT':
355 while s.sy in ('(', '[', '.'):
356 n1 = p_trailer(s, n1)
361 n1 = ExprNodes.binop_node(pos, '**', n1, n2)
365 # s.systring == 'new'.
368 cppclass = p_c_base_type(s)
369 return p_call(s, ExprNodes.NewExprNode(pos, cppclass = cppclass))
371 #trailer: '(' [arglist] ')' | '[' subscriptlist ']' | '.' NAME
373 def p_trailer(s, node1):
376 return p_call(s, node1)
378 return p_index(s, node1)
381 name = EncodedString( p_ident(s) )
382 return ExprNodes.AttributeNode(pos,
383 obj = node1, attribute = name)
385 # arglist: argument (',' argument)* [',']
386 # argument: [test '='] test # Really [keyword '='] test
388 def p_call_parse_args(s, allow_genexp = True):
396 while s.sy not in ('**', ')'):
399 s.error("only one star-arg parameter allowed",
408 s.error("Expected an identifier before '='",
410 encoded_name = EncodedString(arg.name)
411 keyword = ExprNodes.IdentifierStringNode(arg.pos, value = encoded_name)
413 keyword_args.append((keyword, arg))
416 s.error("Non-keyword arg following keyword arg",
419 s.error("Non-keyword arg following star-arg",
421 positional_args.append(arg)
427 if len(positional_args) == 1 and not star_arg:
428 positional_args = [ p_genexp(s, positional_args[0]) ]
431 starstar_arg = p_test(s)
435 return positional_args, keyword_args, star_arg, starstar_arg
437 def p_call_build_packed_args(pos, positional_args, keyword_args, star_arg):
440 if positional_args or not star_arg:
441 arg_tuple = ExprNodes.TupleNode(pos,
442 args = positional_args)
444 star_arg_tuple = ExprNodes.AsTupleNode(pos, arg = star_arg)
446 arg_tuple = ExprNodes.binop_node(pos,
447 operator = '+', operand1 = arg_tuple,
448 operand2 = star_arg_tuple)
450 arg_tuple = star_arg_tuple
452 keyword_args = [ExprNodes.DictItemNode(pos=key.pos, key=key, value=value)
453 for key, value in keyword_args]
454 keyword_dict = ExprNodes.DictNode(pos,
455 key_value_pairs = keyword_args)
456 return arg_tuple, keyword_dict
458 def p_call(s, function):
462 positional_args, keyword_args, star_arg, starstar_arg = \
465 if not (keyword_args or star_arg or starstar_arg):
466 return ExprNodes.SimpleCallNode(pos,
468 args = positional_args)
470 arg_tuple, keyword_dict = p_call_build_packed_args(
471 pos, positional_args, keyword_args, star_arg)
472 return ExprNodes.GeneralCallNode(pos,
474 positional_args = arg_tuple,
475 keyword_args = keyword_dict,
476 starstar_arg = starstar_arg)
478 #lambdef: 'lambda' [varargslist] ':' test
480 #subscriptlist: subscript (',' subscript)* [',']
482 def p_index(s, base):
486 subscripts = p_subscript_list(s)
487 if len(subscripts) == 1 and len(subscripts[0]) == 2:
488 start, stop = subscripts[0]
489 result = ExprNodes.SliceIndexNode(pos,
490 base = base, start = start, stop = stop)
492 indexes = make_slice_nodes(pos, subscripts)
493 if len(indexes) == 1:
496 index = ExprNodes.TupleNode(pos, args = indexes)
497 result = ExprNodes.IndexNode(pos,
498 base = base, index = index)
502 def p_subscript_list(s):
503 items = [p_subscript(s)]
508 items.append(p_subscript(s))
511 #subscript: '.' '.' '.' | test | [test] ':' [test] [':' [test]]
514 # Parse a subscript and return a list of
515 # 1, 2 or 3 ExprNodes, depending on how
516 # many slice elements were encountered.
518 start = p_slice_element(s, (':',))
522 stop = p_slice_element(s, (':', ',', ']'))
526 step = p_slice_element(s, (':', ',', ']'))
527 return [start, stop, step]
529 def p_slice_element(s, follow_set):
530 # Simple expression which may be missing iff
531 # it is followed by something in follow_set.
532 if s.sy not in follow_set:
537 def expect_ellipsis(s):
542 def make_slice_nodes(pos, subscripts):
543 # Convert a list of subscripts as returned
544 # by p_subscript_list into a list of ExprNodes,
545 # creating SliceNodes for elements with 2 or
548 for subscript in subscripts:
549 if len(subscript) == 1:
550 result.append(subscript[0])
552 result.append(make_slice_node(pos, *subscript))
555 def make_slice_node(pos, start, stop = None, step = None):
557 start = ExprNodes.NoneNode(pos)
559 stop = ExprNodes.NoneNode(pos)
561 step = ExprNodes.NoneNode(pos)
562 return ExprNodes.SliceNode(pos,
563 start = start, stop = stop, step = step)
565 #atom: '(' [yield_expr|testlist_comp] ')' | '[' [listmaker] ']' | '{' [dict_or_set_maker] '}' | '`' testlist '`' | NAME | NUMBER | STRING+
573 result = ExprNodes.TupleNode(pos, args = [])
574 elif s.sy == 'yield':
575 result = p_yield_expression(s)
577 result = p_testlist_comp(s)
581 return p_list_maker(s)
583 return p_dict_or_set_maker(s)
585 return p_backquote_expr(s)
588 return ExprNodes.EllipsisNode(pos)
590 return p_int_literal(s)
594 return ExprNodes.FloatNode(pos, value = value)
596 value = s.systring[:-1]
598 return ExprNodes.ImagNode(pos, value = value)
599 elif sy == 'BEGIN_STRING':
600 kind, bytes_value, unicode_value = p_cat_string_literal(s)
602 return ExprNodes.CharNode(pos, value = bytes_value)
604 return ExprNodes.UnicodeNode(pos, value = unicode_value, bytes_value = bytes_value)
606 return ExprNodes.BytesNode(pos, value = bytes_value)
608 return ExprNodes.StringNode(pos, value = bytes_value, unicode_value = unicode_value)
610 name = EncodedString( s.systring )
613 return ExprNodes.NoneNode(pos)
615 return ExprNodes.BoolNode(pos, value=True)
616 elif name == "False":
617 return ExprNodes.BoolNode(pos, value=False)
619 return ExprNodes.NullNode(pos)
621 return p_name(s, name)
623 s.error("Expected an identifier or literal")
625 def p_int_literal(s):
631 while value[-1] in u"UuLl":
632 if value[-1] in u"Ll":
637 # '3L' is ambiguous in Py2 but not in Py3. '3U' and '3LL' are
638 # illegal in Py2 Python files. All suffixes are illegal in Py3
644 if longness == 'LL' or s.context.language_level >= 3:
648 error(pos, "illegal integer literal syntax in Python source file")
650 return ExprNodes.IntNode(pos,
651 is_c_literal = is_c_literal,
658 if not s.compile_time_expr and name in s.compile_time_env:
659 value = s.compile_time_env.lookup_here(name)
661 if isinstance(value, bool):
662 return ExprNodes.BoolNode(pos, value = value)
663 elif isinstance(value, int):
664 return ExprNodes.IntNode(pos, value = rep)
665 elif isinstance(value, long):
666 return ExprNodes.IntNode(pos, value = rep, longness = "L")
667 elif isinstance(value, float):
668 return ExprNodes.FloatNode(pos, value = rep)
669 elif isinstance(value, _unicode):
670 return ExprNodes.UnicodeNode(pos, value = value)
671 elif isinstance(value, _bytes):
672 return ExprNodes.BytesNode(pos, value = value)
674 error(pos, "Invalid type for compile-time constant: %s"
675 % value.__class__.__name__)
676 return ExprNodes.NameNode(pos, name = name)
678 def p_cat_string_literal(s):
679 # A sequence of one or more adjacent string literals.
680 # Returns (kind, bytes_value, unicode_value)
681 # where kind in ('b', 'c', 'u', '')
682 kind, bytes_value, unicode_value = p_string_literal(s)
683 if kind == 'c' or s.sy != 'BEGIN_STRING':
684 return kind, bytes_value, unicode_value
685 bstrings, ustrings = [bytes_value], [unicode_value]
686 bytes_value = unicode_value = None
687 while s.sy == 'BEGIN_STRING':
689 next_kind, next_bytes_value, next_unicode_value = p_string_literal(s)
691 error(pos, "Cannot concatenate char literal with another string or char literal")
692 elif next_kind != kind:
693 error(pos, "Cannot mix string literals of different types, expected %s'', got %s''" %
696 bstrings.append(next_bytes_value)
697 ustrings.append(next_unicode_value)
698 # join and rewrap the partial literals
699 if kind in ('b', 'c', '') or kind == 'u' and None not in bstrings:
700 # Py3 enforced unicode literals are parsed as bytes/unicode combination
701 bytes_value = BytesLiteral( StringEncoding.join_bytes(bstrings) )
702 bytes_value.encoding = s.source_encoding
703 if kind in ('u', ''):
704 unicode_value = EncodedString( u''.join([ u for u in ustrings if u is not None ]) )
705 return kind, bytes_value, unicode_value
707 def p_opt_string_literal(s, required_type='u'):
708 if s.sy == 'BEGIN_STRING':
709 kind, bytes_value, unicode_value = p_string_literal(s, required_type)
710 if required_type == 'u':
712 elif required_type == 'b':
715 s.error("internal parser configuration error")
719 def check_for_non_ascii_characters(string):
725 def p_string_literal(s, kind_override=None):
726 # A single string or char literal. Returns (kind, bvalue, uvalue)
727 # where kind in ('b', 'c', 'u', ''). The 'bvalue' is the source
728 # code byte sequence of the string literal, 'uvalue' is the
729 # decoded Unicode string. Either of the two may be None depending
730 # on the 'kind' of string, only unprefixed strings have both
733 # s.sy == 'BEGIN_STRING'
736 is_python3_source = s.context.language_level >= 3
737 has_non_ASCII_literal_characters = False
738 kind = s.systring[:1].lower()
743 is_raw = s.systring[1:2].lower() == 'r'
746 if kind == '' and kind_override is None and Future.unicode_literals in s.context.future_directives:
747 chars = StringEncoding.StrLiteralBuilder(s.source_encoding)
750 if kind_override is not None and kind_override in 'ub':
753 chars = StringEncoding.UnicodeLiteralBuilder()
755 chars = StringEncoding.StrLiteralBuilder(s.source_encoding)
757 chars = StringEncoding.BytesLiteralBuilder(s.source_encoding)
762 #print "p_string_literal: sy =", sy, repr(s.systring) ###
765 if is_python3_source and not has_non_ASCII_literal_characters and check_for_non_ascii_characters(systr):
766 has_non_ASCII_literal_characters = True
770 if is_python3_source and not has_non_ASCII_literal_characters \
771 and check_for_non_ascii_characters(systr):
772 has_non_ASCII_literal_characters = True
776 chars.append_charval( int(systr[1:], 8) )
779 elif c in u"abfnrtv":
781 StringEncoding.char_from_escape_sequence(systr))
786 chars.append_charval( int(systr[2:], 16) )
788 s.error("Invalid hex escape '%s'" % systr)
790 if kind in ('u', ''):
791 if len(systr) in (6,10):
792 chrval = int(systr[2:], 16)
793 if chrval > 1114111: # sys.maxunicode:
794 s.error("Invalid unicode escape '%s'" % systr)
796 s.error("Invalid unicode escape '%s'" % systr)
798 # unicode escapes in byte strings are not unescaped
800 chars.append_uescape(chrval, systr)
802 chars.append(u'\\' + systr[1:])
803 if is_python3_source and not has_non_ASCII_literal_characters \
804 and check_for_non_ascii_characters(systr):
805 has_non_ASCII_literal_characters = True
806 elif sy == 'NEWLINE':
808 elif sy == 'END_STRING':
811 s.error("Unclosed string literal", pos = pos)
814 "Unexpected token %r:%r in string literal" %
818 bytes_value = chars.getchar()
819 if len(bytes_value) != 1:
820 error(pos, u"invalid character literal: %r" % bytes_value)
822 bytes_value, unicode_value = chars.getstrings()
823 if is_python3_source and has_non_ASCII_literal_characters:
824 # Python 3 forbids literal non-ASCII characters in byte strings
826 s.error("bytes can only contain ASCII literal characters.", pos = pos)
829 return (kind, bytes_value, unicode_value)
831 # list_display ::= "[" [listmaker] "]"
832 # listmaker ::= expression ( comp_for | ( "," expression )* [","] )
833 # comp_iter ::= comp_for | comp_if
834 # comp_for ::= "for" expression_list "in" testlist [comp_iter]
835 # comp_if ::= "if" test [comp_iter]
843 return ExprNodes.ListNode(pos, args = [])
846 target = ExprNodes.ListNode(pos, args = [])
847 append = ExprNodes.ComprehensionAppendNode(
848 pos, expr=expr, target=ExprNodes.CloneNode(target))
849 loop = p_comp_for(s, append)
851 return ExprNodes.ComprehensionNode(
852 pos, loop=loop, append=append, target=target,
853 # list comprehensions leak their loop variable in Py2
854 has_local_scope = s.context.language_level >= 3)
858 exprs = p_simple_expr_list(s, expr)
862 return ExprNodes.ListNode(pos, args = exprs)
864 def p_comp_iter(s, body):
866 return p_comp_for(s, body)
868 return p_comp_if(s, body)
870 # insert the 'append' operation into the loop
873 def p_comp_for(s, body):
877 kw = p_for_bounds(s, allow_testlist=False)
878 kw.update(dict(else_clause = None, body = p_comp_iter(s, body)))
879 return Nodes.ForStatNode(pos, **kw)
881 def p_comp_if(s, body):
885 test = p_test_nocond(s)
886 return Nodes.IfStatNode(pos,
887 if_clauses = [Nodes.IfClauseNode(pos, condition = test,
888 body = p_comp_iter(s, body))],
891 #dictmaker: test ':' test (',' test ':' test)* [',']
893 def p_dict_or_set_maker(s):
899 return ExprNodes.DictNode(pos, key_value_pairs = [])
901 if s.sy == ',' or s.sy == '}':
908 values.append( p_test(s) )
910 return ExprNodes.SetNode(pos, args=values)
913 target = ExprNodes.SetNode(pos, args=[])
914 append = ExprNodes.ComprehensionAppendNode(
915 item.pos, expr=item, target=ExprNodes.CloneNode(target))
916 loop = p_comp_for(s, append)
918 return ExprNodes.ComprehensionNode(
919 pos, loop=loop, append=append, target=target)
921 # dict literal or comprehension
927 target = ExprNodes.DictNode(pos, key_value_pairs = [])
928 append = ExprNodes.DictComprehensionAppendNode(
929 item.pos, key_expr=key, value_expr=value,
930 target=ExprNodes.CloneNode(target))
931 loop = p_comp_for(s, append)
933 return ExprNodes.ComprehensionNode(
934 pos, loop=loop, append=append, target=target)
937 items = [ExprNodes.DictItemNode(key.pos, key=key, value=value)]
946 ExprNodes.DictItemNode(key.pos, key=key, value=value))
948 return ExprNodes.DictNode(pos, key_value_pairs=items)
952 return ExprNodes.DictNode(pos, key_value_pairs = [])
954 # NOTE: no longer in Py3 :)
955 def p_backquote_expr(s):
962 args.append(p_test(s))
967 arg = ExprNodes.TupleNode(pos, args = args)
968 return ExprNodes.BackquoteNode(pos, arg = arg)
970 def p_simple_expr_list(s, expr=None):
971 exprs = expr is not None and [expr] or []
972 while s.sy not in expr_terminators:
973 exprs.append( p_test(s) )
979 def p_test_or_starred_expr_list(s, expr=None):
980 exprs = expr is not None and [expr] or []
981 while s.sy not in expr_terminators:
982 exprs.append( p_test_or_starred_expr(s) )
989 #testlist: test (',' test)* [',']
996 exprs = p_simple_expr_list(s, expr)
997 return ExprNodes.TupleNode(pos, args = exprs)
1001 # testlist_star_expr: (test|star_expr) ( comp_for | (',' (test|star_expr))* [','] )
1003 def p_testlist_star_expr(s):
1005 expr = p_test_or_starred_expr(s)
1008 exprs = p_test_or_starred_expr_list(s, expr)
1009 return ExprNodes.TupleNode(pos, args = exprs)
1013 # testlist_comp: (test|star_expr) ( comp_for | (',' (test|star_expr))* [','] )
1015 def p_testlist_comp(s):
1017 expr = p_test_or_starred_expr(s)
1020 exprs = p_test_or_starred_expr_list(s, expr)
1021 return ExprNodes.TupleNode(pos, args = exprs)
1023 return p_genexp(s, expr)
1027 def p_genexp(s, expr):
1029 loop = p_comp_for(s, Nodes.ExprStatNode(
1030 expr.pos, expr = ExprNodes.YieldExprNode(expr.pos, arg=expr)))
1031 return ExprNodes.GeneratorExpressionNode(expr.pos, loop=loop)
1033 expr_terminators = (')', ']', '}', ':', '=', 'NEWLINE')
1035 #-------------------------------------------------------
1039 #-------------------------------------------------------
1041 def p_global_statement(s):
1042 # assume s.sy == 'global'
1045 names = p_ident_list(s)
1046 return Nodes.GlobalNode(pos, names = names)
1048 def p_nonlocal_statement(s):
1051 names = p_ident_list(s)
1052 return Nodes.NonlocalNode(pos, names = names)
1054 def p_expression_or_assignment(s):
1055 expr_list = [p_testlist_star_expr(s)]
1059 expr = p_yield_expression(s)
1061 expr = p_testlist_star_expr(s)
1062 expr_list.append(expr)
1063 if len(expr_list) == 1:
1064 if re.match(r"([+*/\%^\&|-]|<<|>>|\*\*|//)=", s.sy):
1066 if not isinstance(lhs, (ExprNodes.AttributeNode, ExprNodes.IndexNode, ExprNodes.NameNode) ):
1067 error(lhs.pos, "Illegal operand for inplace operation.")
1068 operator = s.sy[:-1]
1071 rhs = p_yield_expression(s)
1074 return Nodes.InPlaceAssignmentNode(lhs.pos, operator = operator, lhs = lhs, rhs = rhs)
1076 if isinstance(expr, (ExprNodes.UnicodeNode, ExprNodes.StringNode, ExprNodes.BytesNode)):
1077 return Nodes.PassStatNode(expr.pos)
1079 return Nodes.ExprStatNode(expr.pos, expr = expr)
1082 if len(expr_list) == 2:
1083 return Nodes.SingleAssignmentNode(rhs.pos,
1084 lhs = expr_list[0], rhs = rhs)
1086 return Nodes.CascadedAssignmentNode(rhs.pos,
1087 lhs_list = expr_list[:-1], rhs = rhs)
1089 def p_print_statement(s):
1099 ends_with_comma = s.sy in ('NEWLINE', 'EOF')
1103 if s.sy not in ('NEWLINE', 'EOF'):
1104 args.append(p_test(s))
1107 if s.sy in ('NEWLINE', 'EOF'):
1110 args.append(p_test(s))
1111 arg_tuple = ExprNodes.TupleNode(pos, args = args)
1112 return Nodes.PrintStatNode(pos,
1113 arg_tuple = arg_tuple, stream = stream,
1114 append_newline = not ends_with_comma)
1116 def p_exec_statement(s):
1120 args = [ p_bit_expr(s) ]
1123 args.append(p_test(s))
1126 args.append(p_test(s))
1128 error(pos, "'exec' currently requires a target mapping (globals/locals)")
1129 return Nodes.ExecStatNode(pos, args = args)
1131 def p_del_statement(s):
1135 # FIXME: 'exprlist' in Python
1136 args = p_simple_expr_list(s)
1137 return Nodes.DelStatNode(pos, args = args)
1139 def p_pass_statement(s, with_newline = 0):
1143 s.expect_newline("Expected a newline")
1144 return Nodes.PassStatNode(pos)
1146 def p_break_statement(s):
1150 return Nodes.BreakStatNode(pos)
1152 def p_continue_statement(s):
1153 # s.sy == 'continue'
1156 return Nodes.ContinueStatNode(pos)
1158 def p_return_statement(s):
1162 if s.sy not in statement_terminators:
1163 value = p_testlist(s)
1166 return Nodes.ReturnStatNode(pos, value = value)
1168 def p_raise_statement(s):
1175 if s.sy not in statement_terminators:
1176 exc_type = p_test(s)
1179 exc_value = p_test(s)
1183 if exc_type or exc_value or exc_tb:
1184 return Nodes.RaiseStatNode(pos,
1185 exc_type = exc_type,
1186 exc_value = exc_value,
1189 return Nodes.ReraiseStatNode(pos)
1191 def p_import_statement(s):
1192 # s.sy in ('import', 'cimport')
1196 items = [p_dotted_name(s, as_allowed = 1)]
1199 items.append(p_dotted_name(s, as_allowed = 1))
1201 for pos, target_name, dotted_name, as_name in items:
1202 dotted_name = EncodedString(dotted_name)
1203 if kind == 'cimport':
1204 stat = Nodes.CImportStatNode(pos,
1205 module_name = dotted_name,
1208 if as_name and "." in dotted_name:
1209 name_list = ExprNodes.ListNode(pos, args = [
1210 ExprNodes.IdentifierStringNode(pos, value = EncodedString("*"))])
1213 stat = Nodes.SingleAssignmentNode(pos,
1214 lhs = ExprNodes.NameNode(pos,
1215 name = as_name or target_name),
1216 rhs = ExprNodes.ImportNode(pos,
1217 module_name = ExprNodes.IdentifierStringNode(
1218 pos, value = dotted_name),
1219 name_list = name_list))
1221 return Nodes.StatListNode(pos, stats = stats)
1223 def p_from_import_statement(s, first_statement = 0):
1227 (dotted_name_pos, _, dotted_name, _) = \
1228 p_dotted_name(s, as_allowed = 0)
1229 if s.sy in ('import', 'cimport'):
1233 s.error("Expected 'import' or 'cimport'")
1234 is_cimport = kind == 'cimport'
1235 is_parenthesized = False
1237 imported_names = [(s.position(), "*", None, None)]
1241 is_parenthesized = True
1243 imported_names = [p_imported_name(s, is_cimport)]
1246 if is_parenthesized and s.sy == ')':
1248 imported_names.append(p_imported_name(s, is_cimport))
1249 if is_parenthesized:
1251 dotted_name = EncodedString(dotted_name)
1252 if dotted_name == '__future__':
1253 if not first_statement:
1254 s.error("from __future__ imports must occur at the beginning of the file")
1256 for (name_pos, name, as_name, kind) in imported_names:
1257 if name == "braces":
1258 s.error("not a chance", name_pos)
1261 directive = getattr(Future, name)
1262 except AttributeError:
1263 s.error("future feature %s is not defined" % name, name_pos)
1265 s.context.future_directives.add(directive)
1266 return Nodes.PassStatNode(pos)
1267 elif kind == 'cimport':
1268 return Nodes.FromCImportStatNode(pos,
1269 module_name = dotted_name,
1270 imported_names = imported_names)
1272 imported_name_strings = []
1274 for (name_pos, name, as_name, kind) in imported_names:
1275 encoded_name = EncodedString(name)
1276 imported_name_strings.append(
1277 ExprNodes.IdentifierStringNode(name_pos, value = encoded_name))
1280 ExprNodes.NameNode(name_pos,
1281 name = as_name or name)))
1282 import_list = ExprNodes.ListNode(
1283 imported_names[0][0], args = imported_name_strings)
1284 dotted_name = EncodedString(dotted_name)
1285 return Nodes.FromImportStatNode(pos,
1286 module = ExprNodes.ImportNode(dotted_name_pos,
1287 module_name = ExprNodes.IdentifierStringNode(pos, value = dotted_name),
1288 name_list = import_list),
1291 imported_name_kinds = ('class', 'struct', 'union')
1293 def p_imported_name(s, is_cimport):
1296 if is_cimport and s.systring in imported_name_kinds:
1300 as_name = p_as_name(s)
1301 return (pos, name, as_name, kind)
1303 def p_dotted_name(s, as_allowed):
1305 target_name = p_ident(s)
1307 names = [target_name]
1310 names.append(p_ident(s))
1312 as_name = p_as_name(s)
1313 return (pos, target_name, u'.'.join(names), as_name)
1316 if s.sy == 'IDENT' and s.systring == 'as':
1322 def p_assert_statement(s):
1332 return Nodes.AssertStatNode(pos, cond = cond, value = value)
1334 statement_terminators = (';', 'NEWLINE', 'EOF')
1336 def p_if_statement(s):
1340 if_clauses = [p_if_clause(s)]
1341 while s.sy == 'elif':
1343 if_clauses.append(p_if_clause(s))
1344 else_clause = p_else_clause(s)
1345 return Nodes.IfStatNode(pos,
1346 if_clauses = if_clauses, else_clause = else_clause)
1352 return Nodes.IfClauseNode(pos,
1353 condition = test, body = body)
1355 def p_else_clause(s):
1362 def p_while_statement(s):
1368 else_clause = p_else_clause(s)
1369 return Nodes.WhileStatNode(pos,
1370 condition = test, body = body,
1371 else_clause = else_clause)
1373 def p_for_statement(s):
1377 kw = p_for_bounds(s, allow_testlist=True)
1379 else_clause = p_else_clause(s)
1380 kw.update(dict(body = body, else_clause = else_clause))
1381 return Nodes.ForStatNode(pos, **kw)
1383 def p_for_bounds(s, allow_testlist=True):
1384 target = p_for_target(s)
1387 iterator = p_for_iterator(s, allow_testlist)
1388 return dict( target = target, iterator = iterator )
1389 elif not s.in_python_file:
1392 bound1 = p_bit_expr(s)
1394 # Support shorter "for a <= x < b" syntax
1395 bound1, target = target, None
1396 rel1 = p_for_from_relation(s)
1397 name2_pos = s.position()
1399 rel2_pos = s.position()
1400 rel2 = p_for_from_relation(s)
1401 bound2 = p_bit_expr(s)
1402 step = p_for_from_step(s)
1404 target = ExprNodes.NameNode(name2_pos, name = name2)
1406 if not target.is_name:
1408 "Target of for-from statement must be a variable name")
1409 elif name2 != target.name:
1411 "Variable name in for-from range does not match target")
1412 if rel1[0] != rel2[0]:
1414 "Relation directions in for-from do not match")
1415 return dict(target = target,
1426 def p_for_from_relation(s):
1427 if s.sy in inequality_relations:
1432 s.error("Expected one of '<', '<=', '>' '>='")
1434 def p_for_from_step(s):
1437 step = p_bit_expr(s)
1442 inequality_relations = ('<', '<=', '>', '>=')
1444 def p_target(s, terminator):
1446 expr = p_starred_expr(s)
1450 while s.sy != terminator:
1451 exprs.append(p_starred_expr(s))
1455 return ExprNodes.TupleNode(pos, args = exprs)
1459 def p_for_target(s):
1460 return p_target(s, 'in')
1462 def p_for_iterator(s, allow_testlist=True):
1465 expr = p_testlist(s)
1468 return ExprNodes.IteratorNode(pos, sequence = expr)
1470 def p_try_statement(s):
1477 if s.sy in ('except', 'else'):
1478 while s.sy == 'except':
1479 except_clauses.append(p_except_clause(s))
1482 else_clause = p_suite(s)
1483 body = Nodes.TryExceptStatNode(pos,
1484 body = body, except_clauses = except_clauses,
1485 else_clause = else_clause)
1486 if s.sy != 'finally':
1488 # try-except-finally is equivalent to nested try-except/try-finally
1489 if s.sy == 'finally':
1491 finally_clause = p_suite(s)
1492 return Nodes.TryFinallyStatNode(pos,
1493 body = body, finally_clause = finally_clause)
1495 s.error("Expected 'except' or 'finally'")
1497 def p_except_clause(s):
1504 exc_type = p_test(s)
1505 # normalise into list of single exception tests
1506 if isinstance(exc_type, ExprNodes.TupleNode):
1507 exc_type = exc_type.args
1509 exc_type = [exc_type]
1510 if s.sy == ',' or (s.sy == 'IDENT' and s.systring == 'as'):
1512 exc_value = p_test(s)
1513 elif s.sy == 'IDENT' and s.systring == 'as':
1514 # Py3 syntax requires a name here
1518 exc_value = ExprNodes.NameNode(pos2, name = name)
1520 return Nodes.ExceptClauseNode(pos,
1521 pattern = exc_type, target = exc_value, body = body)
1523 def p_include_statement(s, ctx):
1525 s.next() # 'include'
1526 unicode_include_file_name = p_string_literal(s, 'u')[2]
1527 s.expect_newline("Syntax error in include statement")
1528 if s.compile_time_eval:
1529 include_file_name = unicode_include_file_name
1530 include_file_path = s.context.find_include_file(include_file_name, pos)
1531 if include_file_path:
1532 s.included_files.append(include_file_name)
1533 f = Utils.open_source_file(include_file_path, mode="rU")
1534 source_desc = FileSourceDescriptor(include_file_path)
1535 s2 = PyrexScanner(f, source_desc, s, source_encoding=f.encoding, parse_comments=s.parse_comments)
1537 tree = p_statement_list(s2, ctx)
1544 return Nodes.PassStatNode(pos)
1546 def p_with_statement(s):
1548 if s.systring == 'template' and not s.in_python_file:
1549 node = p_with_template(s)
1551 node = p_with_items(s)
1554 def p_with_items(s):
1556 if not s.in_python_file and s.sy == 'IDENT' and s.systring == 'nogil':
1561 body = p_with_items(s)
1564 return Nodes.GILStatNode(pos, state = state, body = body)
1568 if s.sy == 'IDENT' and s.systring == 'as':
1570 target = p_starred_expr(s)
1573 body = p_with_items(s)
1576 return Nodes.WithStatNode(pos, manager = manager,
1577 target = target, body = body)
1579 def p_with_template(s):
1584 templates.append(s.systring)
1586 while s.systring == ',':
1588 templates.append(s.systring)
1593 s.expect_newline("Syntax error in template function declaration")
1596 body_ctx.templates = templates
1597 func_or_var = p_c_func_or_var_declaration(s, pos, body_ctx)
1601 error(pos, "Syntax error in template function declaration")
1603 def p_simple_statement(s, first_statement = 0):
1604 #print "p_simple_statement:", s.sy, s.systring ###
1605 if s.sy == 'global':
1606 node = p_global_statement(s)
1607 elif s.sy == 'nonlocal':
1608 node = p_nonlocal_statement(s)
1609 elif s.sy == 'print':
1610 node = p_print_statement(s)
1611 elif s.sy == 'exec':
1612 node = p_exec_statement(s)
1614 node = p_del_statement(s)
1615 elif s.sy == 'break':
1616 node = p_break_statement(s)
1617 elif s.sy == 'continue':
1618 node = p_continue_statement(s)
1619 elif s.sy == 'return':
1620 node = p_return_statement(s)
1621 elif s.sy == 'raise':
1622 node = p_raise_statement(s)
1623 elif s.sy in ('import', 'cimport'):
1624 node = p_import_statement(s)
1625 elif s.sy == 'from':
1626 node = p_from_import_statement(s, first_statement = first_statement)
1627 elif s.sy == 'yield':
1628 node = p_yield_statement(s)
1629 elif s.sy == 'assert':
1630 node = p_assert_statement(s)
1631 elif s.sy == 'pass':
1632 node = p_pass_statement(s)
1634 node = p_expression_or_assignment(s)
1637 def p_simple_statement_list(s, ctx, first_statement = 0):
1638 # Parse a series of simple statements on one line
1639 # separated by semicolons.
1640 stat = p_simple_statement(s, first_statement = first_statement)
1644 #print "p_simple_statement_list: maybe more to follow" ###
1646 if s.sy in ('NEWLINE', 'EOF'):
1648 stats.append(p_simple_statement(s))
1649 stat = Nodes.StatListNode(stats[0].pos, stats = stats)
1650 s.expect_newline("Syntax error in simple statement list")
1653 def p_compile_time_expr(s):
1654 old = s.compile_time_expr
1655 s.compile_time_expr = 1
1656 expr = p_testlist(s)
1657 s.compile_time_expr = old
1660 def p_DEF_statement(s):
1662 denv = s.compile_time_env
1666 expr = p_compile_time_expr(s)
1667 value = expr.compile_time_value(denv)
1668 #print "p_DEF_statement: %s = %r" % (name, value) ###
1669 denv.declare(name, value)
1671 return Nodes.PassStatNode(pos)
1673 def p_IF_statement(s, ctx):
1675 saved_eval = s.compile_time_eval
1676 current_eval = saved_eval
1677 denv = s.compile_time_env
1680 s.next() # 'IF' or 'ELIF'
1681 expr = p_compile_time_expr(s)
1682 s.compile_time_eval = current_eval and bool(expr.compile_time_value(denv))
1683 body = p_suite(s, ctx)
1684 if s.compile_time_eval:
1691 s.compile_time_eval = current_eval
1692 body = p_suite(s, ctx)
1696 result = Nodes.PassStatNode(pos)
1697 s.compile_time_eval = saved_eval
1700 def p_statement(s, ctx, first_statement = 0):
1701 cdef_flag = ctx.cdef_flag
1703 if s.sy == 'ctypedef':
1704 if ctx.level not in ('module', 'module_pxd'):
1705 s.error("ctypedef statement not allowed here")
1707 # error(s.position(), "'api' not allowed with 'ctypedef'")
1708 return p_ctypedef_statement(s, ctx)
1710 return p_DEF_statement(s)
1712 return p_IF_statement(s, ctx)
1713 elif s.sy == 'DECORATOR':
1714 if ctx.level not in ('module', 'class', 'c_class', 'function', 'property', 'module_pxd', 'c_class_pxd'):
1715 s.error('decorator not allowed here')
1717 decorators = p_decorators(s)
1718 if s.sy not in ('def', 'cdef', 'cpdef', 'class'):
1719 s.error("Decorators can only be followed by functions or classes")
1720 elif s.sy == 'pass' and cdef_flag:
1722 return p_pass_statement(s, with_newline = 1)
1728 elif s.sy == 'cpdef':
1733 if ctx.level not in ('module', 'module_pxd', 'function', 'c_class', 'c_class_pxd'):
1734 s.error('cdef statement not allowed here')
1736 node = p_cdef_statement(s, ctx(overridable = overridable))
1737 if decorators is not None:
1738 if not isinstance(node, (Nodes.CFuncDefNode, Nodes.CVarDefNode, Nodes.CClassDefNode)):
1739 s.error("Decorators can only be followed by functions or classes")
1740 node.decorators = decorators
1744 s.error("'api' not allowed with this statement")
1746 # def statements aren't allowed in pxd files, except
1747 # as part of a cdef class
1748 if ('pxd' in ctx.level) and (ctx.level != 'c_class_pxd'):
1749 s.error('def statement not allowed here')
1751 return p_def_statement(s, decorators)
1752 elif s.sy == 'class':
1753 if ctx.level not in ('module', 'function', 'class', 'other'):
1754 s.error("class definition not allowed here")
1755 return p_class_statement(s, decorators)
1756 elif s.sy == 'include':
1757 if ctx.level not in ('module', 'module_pxd'):
1758 s.error("include statement not allowed here")
1759 return p_include_statement(s, ctx)
1760 elif ctx.level == 'c_class' and s.sy == 'IDENT' and s.systring == 'property':
1761 return p_property_decl(s)
1762 elif s.sy == 'pass' and ctx.level != 'property':
1763 return p_pass_statement(s, with_newline = 1)
1765 if ctx.level in ('c_class_pxd', 'property'):
1766 s.error("Executable statement not allowed here")
1768 return p_if_statement(s)
1769 elif s.sy == 'while':
1770 return p_while_statement(s)
1772 return p_for_statement(s)
1774 return p_try_statement(s)
1775 elif s.sy == 'with':
1776 return p_with_statement(s)
1778 return p_simple_statement_list(
1779 s, ctx, first_statement = first_statement)
1781 def p_statement_list(s, ctx, first_statement = 0):
1782 # Parse a series of statements separated by newlines.
1785 while s.sy not in ('DEDENT', 'EOF'):
1786 stats.append(p_statement(s, ctx, first_statement = first_statement))
1791 return Nodes.StatListNode(pos, stats = stats)
1793 def p_suite(s, ctx = Ctx(), with_doc = 0, with_pseudo_doc = 0):
1798 if s.sy == 'NEWLINE':
1801 if with_doc or with_pseudo_doc:
1802 doc = p_doc_string(s)
1803 body = p_statement_list(s, ctx)
1807 s.error("'api' not allowed with this statement")
1808 if ctx.level in ('module', 'class', 'function', 'other'):
1809 body = p_simple_statement_list(s, ctx)
1811 body = p_pass_statement(s)
1812 s.expect_newline("Syntax error in declarations")
1818 def p_positional_and_keyword_args(s, end_sy_set, templates = None):
1820 Parses positional and keyword arguments. end_sy_set
1821 should contain any s.sy that terminate the argument list.
1822 Argument expansion (* and **) are not allowed.
1824 Returns: (positional_args, keyword_args)
1826 positional_args = []
1830 while s.sy not in end_sy_set:
1831 if s.sy == '*' or s.sy == '**':
1832 s.error('Argument expansion not allowed here.')
1835 if s.sy == 'IDENT' and s.peek()[0] == '=':
1837 s.next() # s.sy is '='
1839 if looking_at_expr(s):
1842 base_type = p_c_base_type(s, templates = templates)
1843 declarator = p_c_declarator(s, empty = 1)
1844 arg = Nodes.CComplexBaseTypeNode(base_type.pos,
1845 base_type = base_type, declarator = declarator)
1847 keyword_node = ExprNodes.IdentifierStringNode(
1848 arg.pos, value = EncodedString(ident))
1849 keyword_args.append((keyword_node, arg))
1853 if looking_at_expr(s):
1856 base_type = p_c_base_type(s, templates = templates)
1857 declarator = p_c_declarator(s, empty = 1)
1858 arg = Nodes.CComplexBaseTypeNode(base_type.pos,
1859 base_type = base_type, declarator = declarator)
1861 positional_args.append(arg)
1863 if len(keyword_args) > 0:
1864 s.error("Non-keyword arg following keyword arg",
1868 if s.sy not in end_sy_set:
1870 s.error("Unmatched %s" % " or ".join(end_sy_set))
1873 return positional_args, keyword_args
1875 def p_c_base_type(s, self_flag = 0, nonempty = 0, templates = None):
1876 # If self_flag is true, this is the base type for the
1877 # self argument of a C method of an extension type.
1879 return p_c_complex_base_type(s)
1881 return p_c_simple_base_type(s, self_flag, nonempty = nonempty, templates = templates)
1883 def p_calling_convention(s):
1884 if s.sy == 'IDENT' and s.systring in calling_convention_words:
1891 calling_convention_words = ("__stdcall", "__cdecl", "__fastcall")
1893 def p_c_complex_base_type(s):
1897 base_type = p_c_base_type(s)
1898 declarator = p_c_declarator(s, empty = 1)
1900 return Nodes.CComplexBaseTypeNode(pos,
1901 base_type = base_type, declarator = declarator)
1903 def p_c_simple_base_type(s, self_flag, nonempty, templates = None):
1904 #print "p_c_simple_base_type: self_flag =", self_flag, nonempty
1911 if not s.sy == 'IDENT':
1912 error(pos, "Expected an identifier, found '%s'" % s.sy)
1913 if looking_at_base_type(s):
1914 #print "p_c_simple_base_type: looking_at_base_type at", s.position()
1916 if s.sy == 'IDENT' and s.systring in special_basic_c_types:
1917 signed, longness = special_basic_c_types[s.systring]
1921 signed, longness = p_sign_and_longness(s)
1922 if s.sy == 'IDENT' and s.systring in basic_c_type_names:
1926 name = 'int' # long [int], short [int], long [int] complex, etc.
1927 if s.sy == 'IDENT' and s.systring == 'complex':
1930 elif looking_at_dotted_name(s):
1931 #print "p_c_simple_base_type: looking_at_type_name at", s.position()
1935 module_path.append(name)
1941 if nonempty and s.sy != 'IDENT':
1942 # Make sure this is not a declaration of a variable or function.
1945 if s.sy == '*' or s.sy == '**' or s.sy == '&':
1946 s.put_back('(', '(')
1948 s.put_back('(', '(')
1949 s.put_back('IDENT', name)
1951 elif s.sy not in ('*', '**', '[', '&'):
1952 s.put_back('IDENT', name)
1955 type_node = Nodes.CSimpleBaseTypeNode(pos,
1956 name = name, module_path = module_path,
1957 is_basic_c_type = is_basic, signed = signed,
1958 complex = complex, longness = longness,
1959 is_self_arg = self_flag, templates = templates)
1962 type_node = p_buffer_or_template(s, type_node, templates)
1967 type_node = Nodes.CNestedBaseTypeNode(pos, base_type = type_node, name = name)
1971 def p_buffer_or_template(s, base_type_node, templates):
1975 # Note that buffer_positional_options_count=1, so the only positional argument is dtype.
1976 # For templated types, all parameters are types.
1977 positional_args, keyword_args = (
1978 p_positional_and_keyword_args(s, (']',), templates)
1982 keyword_dict = ExprNodes.DictNode(pos,
1984 ExprNodes.DictItemNode(pos=key.pos, key=key, value=value)
1985 for key, value in keyword_args
1987 result = Nodes.TemplatedTypeNode(pos,
1988 positional_args = positional_args,
1989 keyword_args = keyword_dict,
1990 base_type_node = base_type_node)
1994 def looking_at_name(s):
1995 return s.sy == 'IDENT' and not s.systring in calling_convention_words
1997 def looking_at_expr(s):
1998 if s.systring in base_type_start_words:
2000 elif s.sy == 'IDENT':
2007 dotted_path.append(s.systring)
2009 saved = s.sy, s.systring
2012 elif s.sy == '*' or s.sy == '**':
2014 is_type = s.sy == ')'
2018 is_type = s.sy == '*'
2022 is_type = s.sy == ']'
2024 dotted_path.reverse()
2025 for p in dotted_path:
2026 s.put_back('IDENT', p)
2027 s.put_back('.', '.')
2028 s.put_back('IDENT', name)
2033 def looking_at_base_type(s):
2034 #print "looking_at_base_type?", s.sy, s.systring, s.position()
2035 return s.sy == 'IDENT' and s.systring in base_type_start_words
2037 def looking_at_dotted_name(s):
2041 result = s.sy == '.'
2042 s.put_back('IDENT', name)
2047 basic_c_type_names = ("void", "char", "int", "float", "double", "bint")
2049 special_basic_c_types = {
2050 # name : (signed, longness)
2051 "Py_UNICODE" : (0, 0),
2052 "Py_ssize_t" : (2, 0),
2057 sign_and_longness_words = ("short", "long", "signed", "unsigned")
2059 base_type_start_words = \
2060 basic_c_type_names + sign_and_longness_words + tuple(special_basic_c_types)
2062 def p_sign_and_longness(s):
2065 while s.sy == 'IDENT' and s.systring in sign_and_longness_words:
2066 if s.systring == 'unsigned':
2068 elif s.systring == 'signed':
2070 elif s.systring == 'short':
2072 elif s.systring == 'long':
2075 return signed, longness
2078 literal = p_opt_string_literal(s, 'u')
2079 if literal is not None:
2080 cname = EncodedString(literal)
2081 cname.encoding = s.source_encoding
2086 def p_c_declarator(s, ctx = Ctx(), empty = 0, is_type = 0, cmethod_flag = 0,
2087 assignable = 0, nonempty = 0,
2088 calling_convention_allowed = 0):
2089 # If empty is true, the declarator must be empty. If nonempty is true,
2090 # the declarator must be nonempty. Otherwise we don't care.
2091 # If cmethod_flag is true, then if this declarator declares
2092 # a function, it's a C method of an extension type.
2096 if s.sy == ')' or looking_at_name(s):
2097 base = Nodes.CNameDeclaratorNode(pos, name = EncodedString(u""), cname = None)
2098 result = p_c_func_declarator(s, pos, ctx, base, cmethod_flag)
2100 result = p_c_declarator(s, ctx, empty = empty, is_type = is_type,
2101 cmethod_flag = cmethod_flag,
2102 nonempty = nonempty,
2103 calling_convention_allowed = 1)
2106 result = p_c_simple_declarator(s, ctx, empty, is_type, cmethod_flag,
2107 assignable, nonempty)
2108 if not calling_convention_allowed and result.calling_convention and s.sy != '(':
2109 error(s.position(), "%s on something that is not a function"
2110 % result.calling_convention)
2111 while s.sy in ('[', '('):
2114 result = p_c_array_declarator(s, result)
2117 result = p_c_func_declarator(s, pos, ctx, result, cmethod_flag)
2121 def p_c_array_declarator(s, base):
2129 return Nodes.CArrayDeclaratorNode(pos, base = base, dimension = dim)
2131 def p_c_func_declarator(s, pos, ctx, base, cmethod_flag):
2132 # Opening paren has already been skipped
2133 args = p_c_arg_list(s, ctx, cmethod_flag = cmethod_flag,
2134 nonempty_declarators = 0)
2135 ellipsis = p_optional_ellipsis(s)
2138 exc_val, exc_check = p_exception_value_clause(s)
2139 with_gil = p_with_gil(s)
2140 return Nodes.CFuncDeclaratorNode(pos,
2141 base = base, args = args, has_varargs = ellipsis,
2142 exception_value = exc_val, exception_check = exc_check,
2143 nogil = nogil or ctx.nogil or with_gil, with_gil = with_gil)
2145 supported_overloaded_operators = cython.set([
2146 '+', '-', '*', '/', '%',
2147 '++', '--', '~', '|', '&', '^', '<<', '>>', ',',
2148 '==', '!=', '>=', '>', '<=', '<',
2152 def p_c_simple_declarator(s, ctx, empty, is_type, cmethod_flag,
2153 assignable, nonempty):
2155 calling_convention = p_calling_convention(s)
2158 base = p_c_declarator(s, ctx, empty = empty, is_type = is_type,
2159 cmethod_flag = cmethod_flag,
2160 assignable = assignable, nonempty = nonempty)
2161 result = Nodes.CPtrDeclaratorNode(pos,
2163 elif s.sy == '**': # scanner returns this as a single token
2165 base = p_c_declarator(s, ctx, empty = empty, is_type = is_type,
2166 cmethod_flag = cmethod_flag,
2167 assignable = assignable, nonempty = nonempty)
2168 result = Nodes.CPtrDeclaratorNode(pos,
2169 base = Nodes.CPtrDeclaratorNode(pos,
2173 base = p_c_declarator(s, ctx, empty = empty, is_type = is_type,
2174 cmethod_flag = cmethod_flag,
2175 assignable = assignable, nonempty = nonempty)
2176 result = Nodes.CReferenceDeclaratorNode(pos, base = base)
2180 name = EncodedString(s.systring)
2182 error(s.position(), "Declarator should be empty")
2184 cname = p_opt_cname(s)
2185 if name != 'operator' and s.sy == '=' and assignable:
2190 error(s.position(), "Empty declarator")
2193 if cname is None and ctx.namespace is not None and nonempty:
2194 cname = ctx.namespace + "::" + name
2195 if name == 'operator' and ctx.visibility == 'extern' and nonempty:
2197 if [1 for c in op if c in '+-*/<=>!%&|([^~,']:
2199 # Handle diphthong operators.
2206 if op in ['-', '+', '|', '&'] and s.sy == op:
2212 if op not in supported_overloaded_operators:
2213 s.error("Overloading operator '%s' not yet supported." % op)
2215 result = Nodes.CNameDeclaratorNode(pos,
2216 name = name, cname = cname, default = rhs)
2217 result.calling_convention = calling_convention
2221 if s.sy == 'IDENT' and s.systring == 'nogil':
2230 s.expect_keyword('gil')
2235 def p_exception_value_clause(s):
2238 if s.sy == 'except':
2249 exc_val = p_name(s, name)
2255 return exc_val, exc_check
2257 c_arg_list_terminators = ('*', '**', '.', ')')
2259 def p_c_arg_list(s, ctx = Ctx(), in_pyfunc = 0, cmethod_flag = 0,
2260 nonempty_declarators = 0, kw_only = 0, annotated = 1):
2261 # Comma-separated list of C argument declarations, possibly empty.
2262 # May have a trailing comma.
2264 is_self_arg = cmethod_flag
2265 while s.sy not in c_arg_list_terminators:
2266 args.append(p_c_arg_decl(s, ctx, in_pyfunc, is_self_arg,
2267 nonempty = nonempty_declarators, kw_only = kw_only,
2268 annotated = annotated))
2275 def p_optional_ellipsis(s):
2282 def p_c_arg_decl(s, ctx, in_pyfunc, cmethod_flag = 0, nonempty = 0,
2283 kw_only = 0, annotated = 1):
2285 not_none = or_none = 0
2288 if s.in_python_file:
2289 # empty type declaration
2290 base_type = Nodes.CSimpleBaseTypeNode(pos,
2291 name = None, module_path = [],
2292 is_basic_c_type = 0, signed = 0,
2293 complex = 0, longness = 0,
2294 is_self_arg = cmethod_flag, templates = None)
2296 base_type = p_c_base_type(s, cmethod_flag, nonempty = nonempty)
2297 declarator = p_c_declarator(s, ctx, nonempty = nonempty)
2298 if s.sy in ('not', 'or') and not s.in_python_file:
2301 if s.sy == 'IDENT' and s.systring == 'None':
2304 s.error("Expected 'None'")
2306 error(pos, "'%s None' only allowed in Python functions" % kind)
2307 or_none = kind == 'or'
2308 not_none = kind == 'not'
2309 if annotated and s.sy == ':':
2311 annotation = p_test(s)
2314 if 'pxd' in s.level:
2315 if s.sy not in ['*', '?']:
2316 error(pos, "default values cannot be specified in pxd files, use ? or *")
2317 default = ExprNodes.BoolNode(1)
2321 return Nodes.CArgDeclNode(pos,
2322 base_type = base_type,
2323 declarator = declarator,
2324 not_none = not_none,
2327 annotation = annotation,
2331 if s.sy == 'IDENT' and s.systring == 'api':
2337 def p_cdef_statement(s, ctx):
2339 ctx.visibility = p_visibility(s, ctx.visibility)
2340 ctx.api = ctx.api or p_api(s)
2342 if ctx.visibility not in ('private', 'public'):
2343 error(pos, "Cannot combine 'api' with '%s'" % ctx.visibility)
2344 if (ctx.visibility == 'extern') and s.sy == 'from':
2345 return p_cdef_extern_block(s, pos, ctx)
2346 elif s.sy == 'import':
2348 return p_cdef_extern_block(s, pos, ctx)
2352 error(pos, "cdef blocks cannot be declared cpdef")
2353 return p_cdef_block(s, ctx)
2356 error(pos, "cdef blocks cannot be declared cpdef")
2357 return p_cdef_block(s, ctx)
2358 elif s.sy == 'class':
2359 if ctx.level not in ('module', 'module_pxd'):
2360 error(pos, "Extension type definition not allowed here")
2362 error(pos, "Extension types cannot be declared cpdef")
2363 return p_c_class_definition(s, pos, ctx)
2364 elif s.sy == 'IDENT' and s.systring == 'cppclass':
2365 if ctx.visibility != 'extern':
2366 error(pos, "C++ classes need to be declared extern")
2367 return p_cpp_class_definition(s, pos, ctx)
2368 elif s.sy == 'IDENT' and s.systring in ("struct", "union", "enum", "packed"):
2369 if ctx.level not in ('module', 'module_pxd'):
2370 error(pos, "C struct/union/enum definition not allowed here")
2372 error(pos, "C struct/union/enum cannot be declared cpdef")
2373 if s.systring == "enum":
2374 return p_c_enum_definition(s, pos, ctx)
2376 return p_c_struct_or_union_definition(s, pos, ctx)
2378 return p_c_func_or_var_declaration(s, pos, ctx)
2380 def p_cdef_block(s, ctx):
2381 return p_suite(s, ctx(cdef_flag = 1))
2383 def p_cdef_extern_block(s, pos, ctx):
2385 error(pos, "cdef extern blocks cannot be declared cpdef")
2391 include_file = p_string_literal(s, 'u')[2]
2392 ctx = ctx(cdef_flag = 1, visibility = 'extern')
2393 if s.systring == "namespace":
2395 ctx.namespace = p_string_literal(s, 'u')[2]
2398 body = p_suite(s, ctx)
2399 return Nodes.CDefExternNode(pos,
2400 include_file = include_file,
2402 namespace = ctx.namespace)
2404 def p_c_enum_definition(s, pos, ctx):
2405 # s.sy == ident 'enum'
2410 cname = p_opt_cname(s)
2411 if cname is None and ctx.namespace is not None:
2412 cname = ctx.namespace + "::" + name
2419 if s.sy != 'NEWLINE':
2420 p_c_enum_line(s, ctx, items)
2422 s.next() # 'NEWLINE'
2424 while s.sy not in ('DEDENT', 'EOF'):
2425 p_c_enum_line(s, ctx, items)
2427 return Nodes.CEnumDefNode(
2428 pos, name = name, cname = cname, items = items,
2429 typedef_flag = ctx.typedef_flag, visibility = ctx.visibility,
2430 in_pxd = ctx.level == 'module_pxd')
2432 def p_c_enum_line(s, ctx, items):
2434 p_c_enum_item(s, ctx, items)
2437 if s.sy in ('NEWLINE', 'EOF'):
2439 p_c_enum_item(s, ctx, items)
2442 s.expect_newline("Syntax error in enum item list")
2444 def p_c_enum_item(s, ctx, items):
2447 cname = p_opt_cname(s)
2448 if cname is None and ctx.namespace is not None:
2449 cname = ctx.namespace + "::" + name
2454 items.append(Nodes.CEnumDefItemNode(pos,
2455 name = name, cname = cname, value = value))
2457 def p_c_struct_or_union_definition(s, pos, ctx):
2459 if s.systring == 'packed':
2462 if s.sy != 'IDENT' or s.systring != 'struct':
2463 s.expected('struct')
2464 # s.sy == ident 'struct' or 'union'
2468 cname = p_opt_cname(s)
2469 if cname is None and ctx.namespace is not None:
2470 cname = ctx.namespace + "::" + name
2478 while s.sy != 'DEDENT':
2481 p_c_func_or_var_declaration(s, s.position(), body_ctx))
2484 s.expect_newline("Expected a newline")
2487 s.expect_newline("Syntax error in struct or union definition")
2488 return Nodes.CStructOrUnionDefNode(pos,
2489 name = name, cname = cname, kind = kind, attributes = attributes,
2490 typedef_flag = ctx.typedef_flag, visibility = ctx.visibility,
2491 in_pxd = ctx.level == 'module_pxd', packed = packed)
2493 def p_visibility(s, prev_visibility):
2495 visibility = prev_visibility
2496 if s.sy == 'IDENT' and s.systring in ('extern', 'public', 'readonly'):
2497 visibility = s.systring
2498 if prev_visibility != 'private' and visibility != prev_visibility:
2499 s.error("Conflicting visibility options '%s' and '%s'"
2500 % (prev_visibility, visibility))
2504 def p_c_modifiers(s):
2505 if s.sy == 'IDENT' and s.systring in ('inline',):
2506 modifier = s.systring
2508 return [modifier] + p_c_modifiers(s)
2511 def p_c_func_or_var_declaration(s, pos, ctx):
2512 cmethod_flag = ctx.level in ('c_class', 'c_class_pxd')
2513 modifiers = p_c_modifiers(s)
2514 base_type = p_c_base_type(s, nonempty = 1, templates = ctx.templates)
2515 declarator = p_c_declarator(s, ctx, cmethod_flag = cmethod_flag,
2516 assignable = 1, nonempty = 1)
2517 declarator.overridable = ctx.overridable
2519 if ctx.level not in ('module', 'c_class', 'module_pxd', 'c_class_pxd') and not ctx.templates:
2520 s.error("C function definition not allowed here")
2521 doc, suite = p_suite(s, Ctx(level = 'function'), with_doc = 1)
2522 result = Nodes.CFuncDefNode(pos,
2523 visibility = ctx.visibility,
2524 base_type = base_type,
2525 declarator = declarator,
2528 modifiers = modifiers,
2530 overridable = ctx.overridable)
2533 # s.error("'api' not allowed with variable declaration")
2534 declarators = [declarator]
2537 if s.sy == 'NEWLINE':
2539 declarator = p_c_declarator(s, ctx, cmethod_flag = cmethod_flag,
2540 assignable = 1, nonempty = 1)
2541 declarators.append(declarator)
2542 s.expect_newline("Syntax error in C variable declaration")
2543 result = Nodes.CVarDefNode(pos,
2544 visibility = ctx.visibility,
2545 base_type = base_type,
2546 declarators = declarators,
2547 in_pxd = ctx.level == 'module_pxd',
2549 overridable = ctx.overridable)
2552 def p_ctypedef_statement(s, ctx):
2553 # s.sy == 'ctypedef'
2556 visibility = p_visibility(s, ctx.visibility)
2558 ctx = ctx(typedef_flag = 1, visibility = visibility)
2562 return p_c_class_definition(s, pos, ctx)
2563 elif s.sy == 'IDENT' and s.systring in ('packed', 'struct', 'union', 'enum'):
2564 if s.systring == 'enum':
2565 return p_c_enum_definition(s, pos, ctx)
2567 return p_c_struct_or_union_definition(s, pos, ctx)
2569 base_type = p_c_base_type(s, nonempty = 1)
2570 if base_type.name is None:
2571 s.error("Syntax error in ctypedef statement")
2572 declarator = p_c_declarator(s, ctx, is_type = 1, nonempty = 1)
2573 s.expect_newline("Syntax error in ctypedef statement")
2574 return Nodes.CTypeDefNode(
2575 pos, base_type = base_type,
2576 declarator = declarator, visibility = visibility,
2577 in_pxd = ctx.level == 'module_pxd')
2579 def p_decorators(s):
2581 while s.sy == 'DECORATOR':
2584 decstring = p_dotted_name(s, as_allowed=0)[2]
2585 names = decstring.split('.')
2586 decorator = ExprNodes.NameNode(pos, name=EncodedString(names[0]))
2587 for name in names[1:]:
2588 decorator = ExprNodes.AttributeNode(pos,
2589 attribute=EncodedString(name),
2592 decorator = p_call(s, decorator)
2593 decorators.append(Nodes.DecoratorNode(pos, decorator=decorator))
2594 s.expect_newline("Expected a newline after decorator")
2597 def p_def_statement(s, decorators=None):
2601 name = EncodedString( p_ident(s) )
2603 args, star_arg, starstar_arg = p_varargslist(s, terminator=')')
2606 error(pos, "Python function cannot be declared nogil")
2607 return_type_annotation = None
2610 return_type_annotation = p_test(s)
2611 doc, body = p_suite(s, Ctx(level = 'function'), with_doc = 1)
2612 return Nodes.DefNode(pos, name = name, args = args,
2613 star_arg = star_arg, starstar_arg = starstar_arg,
2614 doc = doc, body = body, decorators = decorators,
2615 return_type_annotation = return_type_annotation)
2617 def p_varargslist(s, terminator=')', annotated=1):
2618 args = p_c_arg_list(s, in_pyfunc = 1, nonempty_declarators = 1,
2619 annotated = annotated)
2625 star_arg = p_py_arg_decl(s, annotated=annotated)
2628 args.extend(p_c_arg_list(s, in_pyfunc = 1,
2629 nonempty_declarators = 1, kw_only = 1, annotated = annotated))
2630 elif s.sy != terminator:
2631 s.error("Syntax error in Python function argument list")
2634 starstar_arg = p_py_arg_decl(s, annotated=annotated)
2635 return (args, star_arg, starstar_arg)
2637 def p_py_arg_decl(s, annotated = 1):
2641 if annotated and s.sy == ':':
2643 annotation = p_test(s)
2644 return Nodes.PyArgDeclNode(pos, name = name, annotation = annotation)
2646 def p_class_statement(s, decorators):
2650 class_name = EncodedString( p_ident(s) )
2651 class_name.encoding = s.source_encoding
2656 positional_args, keyword_args, star_arg, starstar_arg = \
2657 p_call_parse_args(s, allow_genexp = False)
2658 arg_tuple, keyword_dict = p_call_build_packed_args(
2659 pos, positional_args, keyword_args, star_arg)
2660 if arg_tuple is None:
2661 # XXX: empty arg_tuple
2662 arg_tuple = ExprNodes.TupleNode(pos, args = [])
2663 doc, body = p_suite(s, Ctx(level = 'class'), with_doc = 1)
2664 return Nodes.PyClassDefNode(pos,
2667 keyword_args = keyword_dict,
2668 starstar_arg = starstar_arg,
2669 doc = doc, body = body, decorators = decorators)
2671 def p_c_class_definition(s, pos, ctx):
2675 class_name = p_ident(s)
2678 module_path.append(class_name)
2679 class_name = p_ident(s)
2680 if module_path and ctx.visibility != 'extern':
2681 error(pos, "Qualified class name only allowed for 'extern' C class")
2682 if module_path and s.sy == 'IDENT' and s.systring == 'as':
2684 as_name = p_ident(s)
2686 as_name = class_name
2687 objstruct_name = None
2689 base_class_module = None
2690 base_class_name = None
2693 base_class_path = [p_ident(s)]
2696 base_class_path.append(p_ident(s))
2698 s.error("C class may only have one base class")
2700 base_class_module = ".".join(base_class_path[:-1])
2701 base_class_name = base_class_path[-1]
2703 if ctx.visibility not in ('public', 'extern'):
2704 error(s.position(), "Name options only allowed for 'public' or 'extern' C class")
2705 objstruct_name, typeobj_name = p_c_class_options(s)
2707 if ctx.level == 'module_pxd':
2708 body_level = 'c_class_pxd'
2710 body_level = 'c_class'
2711 doc, body = p_suite(s, Ctx(level = body_level), with_doc = 1)
2713 s.expect_newline("Syntax error in C class definition")
2716 if ctx.visibility == 'extern':
2718 error(pos, "Module name required for 'extern' C class")
2720 error(pos, "Type object name specification not allowed for 'extern' C class")
2721 elif ctx.visibility == 'public':
2722 if not objstruct_name:
2723 error(pos, "Object struct name specification required for 'public' C class")
2724 if not typeobj_name:
2725 error(pos, "Type object name specification required for 'public' C class")
2726 elif ctx.visibility == 'private':
2728 error(pos, "Only 'public' C class can be declared 'api'")
2730 error(pos, "Invalid class visibility '%s'" % ctx.visibility)
2731 return Nodes.CClassDefNode(pos,
2732 visibility = ctx.visibility,
2733 typedef_flag = ctx.typedef_flag,
2735 module_name = ".".join(module_path),
2736 class_name = class_name,
2738 base_class_module = base_class_module,
2739 base_class_name = base_class_name,
2740 objstruct_name = objstruct_name,
2741 typeobj_name = typeobj_name,
2742 in_pxd = ctx.level == 'module_pxd',
2746 def p_c_class_options(s):
2747 objstruct_name = None
2753 if s.systring == 'object':
2755 objstruct_name = p_ident(s)
2756 elif s.systring == 'type':
2758 typeobj_name = p_ident(s)
2762 s.expect(']', "Expected 'object' or 'type'")
2763 return objstruct_name, typeobj_name
2765 def p_property_decl(s):
2767 s.next() # 'property'
2769 doc, body = p_suite(s, Ctx(level = 'property'), with_doc = 1)
2770 return Nodes.PropertyNode(pos, name = name, doc = doc, body = body)
2772 def p_doc_string(s):
2773 if s.sy == 'BEGIN_STRING':
2775 kind, bytes_result, unicode_result = p_cat_string_literal(s)
2777 s.expect_newline("Syntax error in doc string")
2778 if kind in ('u', ''):
2779 return unicode_result
2780 warning(pos, "Python 3 requires docstrings to be unicode strings")
2785 def p_code(s, level=None):
2786 body = p_statement_list(s, Ctx(level = level), first_statement = 1)
2788 s.error("Syntax error in statement [%s,%s]" % (
2789 repr(s.sy), repr(s.systring)))
2792 COMPILER_DIRECTIVE_COMMENT_RE = re.compile(r"^#\s*cython:\s*((\w|[.])+\s*=.*)$")
2794 def p_compiler_directive_comments(s):
2796 while s.sy == 'commentline':
2797 m = COMPILER_DIRECTIVE_COMMENT_RE.match(s.systring)
2799 directives = m.group(1).strip()
2801 result.update( Options.parse_directive_list(
2802 directives, ignore_unknown=True) )
2803 except ValueError, e:
2804 s.error(e.args[0], fatal=False)
2808 def p_module(s, pxd, full_module_name):
2811 directive_comments = p_compiler_directive_comments(s)
2812 s.parse_comments = False
2814 if 'language_level' in directive_comments:
2815 s.context.set_language_level(directive_comments['language_level'])
2817 doc = p_doc_string(s)
2819 level = 'module_pxd'
2823 body = p_statement_list(s, Ctx(level = level), first_statement = 1)
2825 s.error("Syntax error in statement [%s,%s]" % (
2826 repr(s.sy), repr(s.systring)))
2827 return ModuleNode(pos, doc = doc, body = body,
2828 full_module_name = full_module_name,
2829 directive_comments = directive_comments)
2831 def p_cpp_class_definition(s, pos, ctx):
2832 # s.sy == 'cppclass'
2835 class_name = p_ident(s)
2836 cname = p_opt_cname(s)
2837 if cname is None and ctx.namespace is not None:
2838 cname = ctx.namespace + "::" + class_name
2840 error(pos, "Qualified class name not allowed C++ class")
2843 templates = [p_ident(s)]
2846 templates.append(p_ident(s))
2852 base_classes = [p_dotted_name(s, False)[2]]
2855 base_classes.append(p_dotted_name(s, False)[2])
2860 error(s.position(), "Name options not allowed for C++ class")
2866 body_ctx = Ctx(visibility = ctx.visibility)
2867 body_ctx.templates = templates
2868 while s.sy != 'DEDENT':
2869 if s.systring == 'cppclass':
2871 p_cpp_class_definition(s, s.position(), body_ctx))
2872 elif s.sy != 'pass':
2874 p_c_func_or_var_declaration(s, s.position(), body_ctx))
2877 s.expect_newline("Expected a newline")
2881 s.expect_newline("Syntax error in C++ class definition")
2882 return Nodes.CppClassNode(pos,
2885 base_classes = base_classes,
2886 visibility = ctx.visibility,
2887 in_pxd = ctx.level == 'module_pxd',
2888 attributes = attributes,
2889 templates = templates)
2893 #----------------------------------------------
2897 #----------------------------------------------
2899 def print_parse_tree(f, node, level, key = None):
2900 from types import ListType, TupleType
2901 from Nodes import Node
2906 f.write("%s: " % key)
2909 f.write("(%s @ %s\n" % (node[0], node[1]))
2910 for i in xrange(2, len(node)):
2911 print_parse_tree(f, node[i], level+1)
2912 f.write("%s)\n" % ind)
2914 elif isinstance(node, Node):
2917 except AttributeError:
2918 tag = node.__class__.__name__
2919 f.write("%s @ %s\n" % (tag, node.pos))
2920 for name, value in node.__dict__.items():
2921 if name != 'tag' and name != 'pos':
2922 print_parse_tree(f, value, level+1, name)
2926 for i in xrange(len(node)):
2927 print_parse_tree(f, node[i], level+1)
2928 f.write("%s]\n" % ind)
2930 f.write("%s%s\n" % (ind, node))