Format

Author: AdrienVannson

src/betterproto/compile/importing.py

@@ -17,8 +17,8 @@
 
 
 if TYPE_CHECKING:
-    from ..plugin.typing_compiler import TypingCompiler
     from ..plugin.models import PluginRequestCompiler
+    from ..plugin.typing_compiler import TypingCompiler
 
 WRAPPER_TYPES: Dict[str, Type] = {
     ".google.protobuf.DoubleValue": google_protobuf.DoubleValue,
@@ -33,7 +33,9 @@
 }
 
 
-def parse_source_type_name(field_type_name: str, request: "PluginRequestCompiler") -> Tuple[str, str]:
+def parse_source_type_name(
+    field_type_name: str, request: "PluginRequestCompiler"
+) -> Tuple[str, str]:
     """
     Split full source type name into package and type name.
     E.g. 'root.package.Message' -> ('root.package', 'Message')

src/betterproto/plugin/models.py

@@ -65,9 +65,7 @@
 from betterproto.lib.google.protobuf.compiler import CodeGeneratorRequest
 
 from .. import which_one_of
-from ..compile.importing import (
-    get_type_reference,
-)
+from ..compile.importing import get_type_reference
 from ..compile.naming import (
     pythonize_class_name,
     pythonize_enum_member_name,
@@ -225,7 +223,9 @@ def all_messages(self) -> List["MessageCompiler"]:
             List of all of the messages in this request.
         """
         return [
-            msg for output in self.output_packages.values() for msg in output.messages.values()
+            msg
+            for output in self.output_packages.values()
+            for msg in output.messages.values()
         ]
 
 

src/betterproto/plugin/parser.py

@@ -60,9 +60,13 @@ def _traverse(
         for i, item in enumerate(items):
             # Adjust the name since we flatten the hierarchy.
             # Todo: don't change the name, but include full name in returned tuple
-            should_rename = not isinstance(item, DescriptorProto) or not item.options.map_entry
+            should_rename = (
+                not isinstance(item, DescriptorProto) or not item.options.map_entry
+            )
 
-            item.name = next_prefix = f"{prefix}.{item.name}" if prefix and should_rename else item.name
+            item.name = next_prefix = (
+                f"{prefix}.{item.name}" if prefix and should_rename else item.name
+            )
             yield item, [*path, i]
 
             if isinstance(item, DescriptorProto):