101
101
)
102
102
from mypyc .primitives .exc_ops import (
103
103
error_catch_op ,
104
+ error_clear_op ,
104
105
exc_matches_op ,
105
106
get_exc_info_op ,
106
107
get_exc_value_op ,
107
108
keep_propagating_op ,
108
- err_occurred_op ,
109
109
no_err_occurred_op ,
110
110
propagate_if_error_op ,
111
111
raise_exception_op ,
112
112
reraise_exception_op ,
113
113
restore_exc_info_op ,
114
- error_clear_op ,
115
114
)
116
- from mypyc .primitives .generic_ops import iter_op , next_op , next_raw_op , py_delattr_op
115
+ from mypyc .primitives .generic_ops import iter_op , next_raw_op , py_delattr_op
117
116
from mypyc .primitives .misc_ops import (
118
117
check_stop_op ,
119
118
coro_op ,
120
- debug_print_op ,
121
119
import_from_many_op ,
122
120
import_many_op ,
123
121
import_op ,
@@ -993,7 +991,10 @@ def _transform_with_contextmanager(
993
991
994
992
# mgrv = ctx.__wrapped__(*args, **kwargs)
995
993
wrapped_call = mypy .nodes .CallExpr (
996
- mypy .nodes .MemberExpr (expr .callee , "__wrapped__" ), expr .args , expr .arg_kinds , expr .arg_names
994
+ mypy .nodes .MemberExpr (expr .callee , "__wrapped__" ),
995
+ expr .args ,
996
+ expr .arg_kinds ,
997
+ expr .arg_names ,
997
998
)
998
999
gen = builder .accept (wrapped_call )
999
1000
@@ -1007,7 +1008,9 @@ def _transform_with_contextmanager(
1007
1008
builder .add (Branch (mgr_target , runtime_block , main_block , Branch .IS_ERROR ))
1008
1009
1009
1010
builder .activate_block (runtime_block )
1010
- builder .add (RaiseStandardError (RaiseStandardError .RUNTIME_ERROR , "generator didn't yield" , line ))
1011
+ builder .add (
1012
+ RaiseStandardError (RaiseStandardError .RUNTIME_ERROR , "generator didn't yield" , line )
1013
+ )
1011
1014
builder .add (Unreachable ())
1012
1015
1013
1016
builder .activate_block (main_block )
@@ -1046,7 +1049,11 @@ def except_body() -> None:
1046
1049
builder .builder .pop_error_handler ()
1047
1050
1048
1051
builder .activate_block (no_error_block )
1049
- builder .add (RaiseStandardError (RaiseStandardError .RUNTIME_ERROR , "generator didn't stop after throw()" , line ))
1052
+ builder .add (
1053
+ RaiseStandardError (
1054
+ RaiseStandardError .RUNTIME_ERROR , "generator didn't stop after throw()" , line
1055
+ )
1056
+ )
1050
1057
builder .add (Unreachable ())
1051
1058
1052
1059
builder .activate_block (error_block )
@@ -1085,20 +1092,16 @@ def else_body() -> None:
1085
1092
builder .activate_block (close_block )
1086
1093
# TODO: this isn't exactly the right order
1087
1094
builder .py_call (builder .py_get_attr (gen , "close" , line ), [], line )
1088
- builder .add (RaiseStandardError (RaiseStandardError .RUNTIME_ERROR , "generator didn't stop" , line ))
1095
+ builder .add (
1096
+ RaiseStandardError (RaiseStandardError .RUNTIME_ERROR , "generator didn't stop" , line )
1097
+ )
1089
1098
builder .add (Unreachable ())
1090
1099
1091
1100
builder .activate_block (stop_block )
1092
1101
# TODO: should check for StopIteration
1093
1102
builder .call_c (error_clear_op , [], - 1 )
1094
1103
1095
- transform_try_except (
1096
- builder ,
1097
- try_body ,
1098
- handlers ,
1099
- else_body ,
1100
- line ,
1101
- )
1104
+ transform_try_except (builder , try_body , handlers , else_body , line )
1102
1105
1103
1106
1104
1107
def transform_with_stmt (builder : IRBuilder , o : WithStmt ) -> None :
0 commit comments