Skip to content

Commit 39bb43e

Browse files
committed
Remove dynamic attribute lookups in fastparse
1 parent eb144ce commit 39bb43e

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

mypy/fastparse.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -426,8 +426,8 @@ def visit(self, node: AST | None) -> Any:
426426
def set_line(self, node: N, n: AstNode) -> N:
427427
node.line = n.lineno
428428
node.column = n.col_offset
429-
node.end_line = getattr(n, "end_lineno", None)
430-
node.end_column = getattr(n, "end_col_offset", None)
429+
node.end_line = n.end_lineno
430+
node.end_column = n.end_col_offset
431431

432432
return node
433433

@@ -586,8 +586,8 @@ def set_block_lines(self, b: Block, stmts: Sequence[ast3.stmt]) -> None:
586586
first, last = stmts[0], stmts[-1]
587587
b.line = first.lineno
588588
b.column = first.col_offset
589-
b.end_line = getattr(last, "end_lineno", None)
590-
b.end_column = getattr(last, "end_col_offset", None)
589+
b.end_line = last.end_lineno
590+
b.end_column = last.end_col_offset
591591
if not b.body:
592592
return
593593
new_first = b.body[0]
@@ -1030,8 +1030,8 @@ def do_func_def(
10301030
)
10311031

10321032
# End position is always the same.
1033-
end_line = getattr(n, "end_lineno", None)
1034-
end_column = getattr(n, "end_col_offset", None)
1033+
end_line = n.end_lineno
1034+
end_column = n.end_col_offset
10351035

10361036
self.class_and_function_stack.pop()
10371037
self.class_and_function_stack.append("F")
@@ -1079,7 +1079,7 @@ def transform_args(
10791079
) -> list[Argument]:
10801080
new_args = []
10811081
names: list[ast3.arg] = []
1082-
posonlyargs = getattr(args, "posonlyargs", cast(list[ast3.arg], []))
1082+
posonlyargs = args.posonlyargs
10831083
args_args = posonlyargs + args.args
10841084
args_defaults = args.defaults
10851085
num_no_defaults = len(args_args) - len(args_defaults)
@@ -1152,8 +1152,8 @@ def make_argument(
11521152
argument.set_line(
11531153
arg.lineno,
11541154
arg.col_offset,
1155-
getattr(arg, "end_lineno", None),
1156-
getattr(arg, "end_col_offset", None),
1155+
arg.end_lineno,
1156+
arg.end_col_offset,
11571157
)
11581158
return argument
11591159

@@ -1190,8 +1190,8 @@ def visit_ClassDef(self, n: ast3.ClassDef) -> ClassDef:
11901190
if self.options.include_docstrings:
11911191
cdef.docstring = ast3.get_docstring(n, clean=False)
11921192
cdef.column = n.col_offset
1193-
cdef.end_line = getattr(n, "end_lineno", None)
1194-
cdef.end_column = getattr(n, "end_col_offset", None)
1193+
cdef.end_line = n.end_lineno
1194+
cdef.end_column = n.end_col_offset
11951195
self.class_and_function_stack.pop()
11961196
return cdef
11971197

@@ -2113,8 +2113,8 @@ def visit_Subscript(self, n: ast3.Subscript) -> Type:
21132113
column=value.column,
21142114
empty_tuple_index=empty_tuple_index,
21152115
)
2116-
result.end_column = getattr(n, "end_col_offset", None)
2117-
result.end_line = getattr(n, "end_lineno", None)
2116+
result.end_column = n.end_col_offset
2117+
result.end_line = n.end_lineno
21182118
return result
21192119
else:
21202120
return self.invalid_type(n)

0 commit comments

Comments
 (0)