Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 11:30:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add full typehints support

Browse Source
This commit is contained in:
blueloveTH 2023-05-27 21:47:52 +08:00
parent db283d4ba6
commit 492c3fda00
2 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/compiler.h
View File

@ -814,6 +814,8 @@ __SUBSCR_END:
default: {
advance(-1); // do revert since we have pre-called advance() at the beginning
EXPR_TUPLE();
// eat variable's type hint
if(match(TK(":"))) consume_type_hints();
if(!try_compile_assignment()){
ctx()->emit_expr();
if((mode()==CELL_MODE || mode()==REPL_MODE) && name_scope()==NAME_GLOBAL){
@ -827,6 +829,10 @@ __SUBSCR_END:
}
}
void consume_type_hints(){
EXPR();
ctx()->s_expr.pop();
}
void compile_class(){
consume(TK("@id"));
@ -876,7 +882,7 @@ __SUBSCR_END:
}
// eat type hints
if(enable_type_hints && match(TK(":"))) consume(TK("@id"));
if(enable_type_hints && match(TK(":"))) consume_type_hints();
if(state == 0 && curr().type == TK("=")) state = 2;
int index = ctx()->add_varname(name);
switch (state)
@ -917,9 +923,7 @@ __SUBSCR_END:
_compile_f_args(decl, true);
consume(TK(")"));
}
if(match(TK("->"))){
if(!match(TK("None"))) consume(TK("@id"));
}
if(match(TK("->"))) consume_type_hints();
compile_block_body();
pop_context();

13
tests/22_typehints.py
View File

@ -42,3 +42,16 @@ def i(x, y: int, *args):
def j(x, y: int, *args: str) -> int:
return x + y + len(args)
x: int = 1
y: 'str' = '2'
x: 'list[int]' = [1, 2, 3]
y: 'list[str]' = ['1', '2', '3']
def g(x: 'list[int]', y: 'list[str]') -> 'list[int]':
return x + y
def z(x: float):
x: int = 1
y: 'str' = '2'