Skip to content

Commit a714a5b

Browse files
committed
Fix: Filter noise and refine regex in AST reference extraction
1 parent 3b5c95f commit a714a5b

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

src/pyob/entrance.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -535,11 +535,12 @@ def update_ledger_for_file(self, rel_path: str, code: str):
535535
elif isinstance(n.func, ast.Attribute):
536536
potential_refs.add(n.func.attr)
537537
elif isinstance(n, ast.Name) and isinstance(n.ctx, ast.Load):
538-
potential_refs.add(n.id)
538+
if n.id not in ["self", "None", "True", "False"]:
539+
potential_refs.add(n.id)
539540
except Exception as e:
540541
logger.warning(f"Failed to parse Python AST for {rel_path}: {e}")
541542
potential_refs.update(
542-
re.findall(r"\b[a-zA-Z_][a-zA-Z0-9_]{3,}\b", code)
543+
re.findall(r"\b[a-zA-Z_][a-zA-Z0-9_]{4,}\b", code)
543544
)
544545
elif ext in [".js", ".ts"]:
545546
defs = re.findall(

0 commit comments

Comments
 (0)