From 48745f2ae93ae875ba57e45cd331ba1665145d21 Mon Sep 17 00:00:00 2001 From: Yuyao Huang Date: Tue, 23 Apr 2024 21:19:11 +0800 Subject: [PATCH] AugAssign() --- trace_commentor/handlers/__init__.py | 2 +- trace_commentor/handlers/statements.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/trace_commentor/handlers/__init__.py b/trace_commentor/handlers/__init__.py index e5acb47..c63b467 100644 --- a/trace_commentor/handlers/__init__.py +++ b/trace_commentor/handlers/__init__.py @@ -1,5 +1,5 @@ from .definitions import FunctionDef, Return, Lambda -from .statements import Pass, Assign, AnnAssign +from .statements import Pass, Assign, AnnAssign, AugAssign from .expressions import Expr, UnaryOp, BinOp, Call, Compare, Attribute, Subscript, Slice, keyword, IfExp from .literals import Constant, Tuple, List, Set, Dict, FormattedValue, JoinedStr from .variables import Name diff --git a/trace_commentor/handlers/statements.py b/trace_commentor/handlers/statements.py index 164c88d..b23c304 100644 --- a/trace_commentor/handlers/statements.py +++ b/trace_commentor/handlers/statements.py @@ -19,3 +19,8 @@ def AnnAssign(self, cmtor): if type(self.value) not in flags.ASSIGN_SILENT: cmtor.append_comment(f"----------") cmtor.process(self.target) + + +def AugAssign(self, cmtor): + AnnAssign(self, cmtor) +