Skip to content
This repository was archived by the owner on Jun 9, 2025. It is now read-only.

Commit 2ebf313

Browse files
committed
Fix typing error in code validation
1 parent a30710c commit 2ebf313

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/betterproto2_compiler/plugin/module_validation.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -120,20 +120,20 @@ def next(self):
120120
self.evaluate_import(line)
121121

122122
# Evaluate Classes.
123-
elif line.startswith("class "):
124-
class_name = re.search(r"class (\w+)", line).group(1)
123+
elif match := re.search(r"^class (\w+)", line):
124+
class_name = match.group(1)
125125
if class_name:
126126
self.add_import(class_name, self.line_number, line)
127127

128128
# Evaluate Functions.
129-
elif line.startswith("def "):
130-
function_name = re.search(r"def (\w+)", line).group(1)
129+
elif match := re.search(r"^def (\w+)", line):
130+
function_name = match.group(1)
131131
if function_name:
132132
self.add_import(function_name, self.line_number, line)
133133

134134
# Evaluate direct assignments.
135-
elif "=" in line:
136-
assignment = re.search(r"(\w+)\s*=", line).group(1)
135+
elif match := re.search(r"^(\w+)\s*=", line):
136+
assignment = match.group(1)
137137
if assignment:
138138
self.add_import(assignment, self.line_number, line)
139139

0 commit comments

Comments
 (0)