codeflash-ai/codeflash
1
0
Fork 0
Code Issues 18 Pull requests 33 Projects Releases Packages Wiki Activity Actions 4.8k

small refactoring

Browse source 
Signed-off-by: ali <mohammed18200118@gmail.com>
This commit is contained in:
ali 2025-08-22 16:12:55 +03:00
parent c5b1d30319
commit e60608e4bf
No known key found for this signature in database
GPG key ID: 44F9B42770617B9B
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
codeflash/code_utils/code_extractor.py
View file

@ -165,10 +165,12 @@ class GlobalAssignmentTransformer(cst.CSTTransformer):
# Add a blank line after the last assignment if needed
after_index = insert_index + len(assignment_lines)
if after_index < len(new_statements):
next_statement = new_statements[after_index]
if not next_statement.leading_lines or not isinstance(next_statement.leading_lines[-1], cst.EmptyLine):
new_statements[after_index] = next_statement.with_changes(
leading_lines=[cst.EmptyLine(), *next_statement.leading_lines]
next_stmt = new_statements[after_index]
# If there's no empty line, add one
has_empty = any(isinstance(line, cst.EmptyLine) for line in next_stmt.leading_lines)
if not has_empty:
new_statements[after_index] = next_stmt.with_changes(
leading_lines=[cst.EmptyLine(), *next_stmt.leading_lines]
)
return updated_node.with_changes(body=new_statements)