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) +