Add Python SpacesStyle

Author: knutwannheden

rewrite/rewrite/__init__.py

@@ -11,6 +11,7 @@
 from .visitor import Cursor, TreeVisitor
 from .parser import *
 from .result import *
+from .style import *
 
 __all__ = [
     'Checksum',
@@ -36,5 +37,8 @@
     'ParserBuilder',
     'ParseExceptionResult',
     'Result',
-    'SourceFile'
-]
+    'SourceFile',
+
+    # Style
+    'Style',
+]

rewrite/rewrite/python/__init__.py

@@ -4,6 +4,7 @@
 from .parser import *
 from .visitor import *
 from .markers import *
+from .style import *
 from .format import *
 
 __all__ = [
@@ -53,6 +54,10 @@
     'VariableScope',
     'YieldFrom',
 
+    # Style
+    'PythonStyle',
+    'SpacesStyle',
+
     # Formatter
     'AutoFormat',
 ]

rewrite/rewrite/python/style.py

@@ -0,0 +1,55 @@
+from dataclasses import dataclass
+
+from ..style import Style
+
+
+class PythonStyle(Style):
+    pass
+
+
+@dataclass(frozen=True)
+class SpacesStyle(PythonStyle):
+    @dataclass(frozen=True)
+    class BeforeParentheses:
+        method_call_parentheses: bool
+        method_parentheses: bool
+        left_bracket: bool
+
+    @dataclass(frozen=True)
+    class AroundOperators:
+        assignment: bool
+        equality: bool
+        relational: bool
+        bitwise: bool
+        additive: bool
+        multiplicative: bool
+        shift: bool
+        power: bool
+        eq_in_named_parameter: bool
+        eq_in_keyword_argument: bool
+
+    @dataclass(frozen=True)
+    class Within:
+        brackets: bool
+        method_parentheses: bool
+        empty_method_parentheses: bool
+        method_call_parentheses: bool
+        empty_method_call_parentheses: bool
+        braces: bool
+
+    @dataclass(frozen=True)
+    class Other:
+        before_comma: bool
+        after_comma: bool
+        before_semicolon: bool
+        before_for_semicolon: bool
+        before_colon: bool
+        after_colon: bool
+        before_backslash: bool
+        before_hash: bool
+        after_hash: bool
+
+    beforeParentheses: BeforeParentheses
+    aroundOperators: AroundOperators
+    within: Within
+    other: Other

rewrite/rewrite/style.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+from typing import Protocol
+
+
+class Style(Protocol):
+    def merge(self, lower_precedence: Style) -> Style:
+        ...
+
+    def apply_defaults(self) -> Style:
+        return self