Automatic management of parentheses in elif, too

This commit is contained in:
Łukasz Langa 2018-05-08 21:01:50 -07:00
parent c9b73b99ac
commit 1b51e2c078

View File

@ -1375,7 +1375,9 @@ def __attrs_post_init__(self) -> None:
v = self.visit_stmt v = self.visit_stmt
Ø: Set[str] = set() Ø: Set[str] = set()
self.visit_assert_stmt = partial(v, keywords={"assert"}, parens={"assert", ","}) self.visit_assert_stmt = partial(v, keywords={"assert"}, parens={"assert", ","})
self.visit_if_stmt = partial(v, keywords={"if", "else", "elif"}, parens={"if"}) self.visit_if_stmt = partial(
v, keywords={"if", "else", "elif"}, parens={"if", "elif"}
)
self.visit_while_stmt = partial(v, keywords={"while", "else"}, parens={"while"}) self.visit_while_stmt = partial(v, keywords={"while", "else"}, parens={"while"})
self.visit_for_stmt = partial(v, keywords={"for", "else"}, parens={"for", "in"}) self.visit_for_stmt = partial(v, keywords={"for", "else"}, parens={"for", "in"})
self.visit_try_stmt = partial( self.visit_try_stmt = partial(