From 0bb3135cb77c0a6e07ccfeb65360244923b8f9d6 Mon Sep 17 00:00:00 2001 From: "Yuyao Huang (Sam)" Date: Wed, 16 Jul 2025 21:31:32 +0800 Subject: [PATCH] add support to "Starred" expressions --- .gitignore | 2 ++ trace_commentor/handlers/__init__.py | 2 +- trace_commentor/handlers/expressions.py | 4 ++++ 3 files changed, 7 insertions(+), 1 deletion(-) diff --git a/.gitignore b/.gitignore index 43070c2..f67a085 100644 --- a/.gitignore +++ b/.gitignore @@ -2,3 +2,5 @@ __pycache__ *.log *.log.py .vscode-upload.json +dist +*.egg-info \ No newline at end of file diff --git a/trace_commentor/handlers/__init__.py b/trace_commentor/handlers/__init__.py index da9a0f7..e284069 100644 --- a/trace_commentor/handlers/__init__.py +++ b/trace_commentor/handlers/__init__.py @@ -1,6 +1,6 @@ from .definitions import FunctionDef, Return, Lambda from .statements import Pass, Assign, AnnAssign, AugAssign -from .expressions import Expr, UnaryOp, BinOp, Call, Compare, Attribute, Subscript, Slice, ExtSlice, keyword, IfExp, Index +from .expressions import Expr, UnaryOp, BinOp, Call, Compare, Attribute, Subscript, Slice, ExtSlice, Starred, keyword, IfExp, Index from .literals import Constant, Tuple, List, Set, Dict, FormattedValue, JoinedStr from .variables import Name from .control_flow import If, For, Continue, Break, With, Raise diff --git a/trace_commentor/handlers/expressions.py b/trace_commentor/handlers/expressions.py index f2e5927..7c1bb3c 100644 --- a/trace_commentor/handlers/expressions.py +++ b/trace_commentor/handlers/expressions.py @@ -60,6 +60,10 @@ def ExtSlice(self, cmtor): cmtor.process(dim) +def Starred(self, cmtor): + cmtor.process(self.value) + + def keyword(self, cmtor): cmtor.process(self.value)