X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/3501cefb09eb8448bd82287840c9093f10c25299..b97ec62368ac57b29a0ccd5fc68ba875418eb8cc:/src/black/__init__.py diff --git a/src/black/__init__.py b/src/black/__init__.py index f2efdec..d8b9819 100644 --- a/src/black/__init__.py +++ b/src/black/__init__.py @@ -1210,6 +1210,21 @@ def get_features_used( # noqa: C901 if argch.type in STARS: features.add(feature) + elif ( + n.type in {syms.return_stmt, syms.yield_expr} + and len(n.children) >= 2 + and n.children[1].type == syms.testlist_star_expr + and any(child.type == syms.star_expr for child in n.children[1].children) + ): + features.add(Feature.UNPACKING_ON_FLOW) + + elif ( + n.type == syms.annassign + and len(n.children) >= 4 + and n.children[3].type == syms.testlist_star_expr + ): + features.add(Feature.ANN_ASSIGN_EXTENDED_RHS) + # Python 2 only features (for its deprecation) except for integers, see above elif n.type == syms.print_stmt: features.add(Feature.PRINT_STMT)