betterproto2-compiler 0.0.3__py3-none-any.whl → 0.1.0__py3-none-any.whl

betterproto2_compiler/compile/importing.py

@@ -3,11 +3,6 @@ from __future__ import annotations
 import os
 from typing import (
     TYPE_CHECKING,
-    Dict,
-    List,
-    Set,
-    Tuple,
-    Type,
 )
 
 from ..casing import safe_snake_case
@@ -18,7 +13,7 @@ if TYPE_CHECKING:
     from ..plugin.models import PluginRequestCompiler
     from ..plugin.typing_compiler import TypingCompiler
 
-WRAPPER_TYPES: Dict[str, Type] = {
+WRAPPER_TYPES: dict[str, type] = {
     ".google.protobuf.DoubleValue": google_protobuf.DoubleValue,
     ".google.protobuf.FloatValue": google_protobuf.FloatValue,
     ".google.protobuf.Int32Value": google_protobuf.Int32Value,
@@ -31,7 +26,7 @@ WRAPPER_TYPES: Dict[str, Type] = {
 }
 
 
-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')
@@ -77,7 +72,7 @@ def get_type_reference(
     imports: set,
     source_type: str,
     typing_compiler: TypingCompiler,
-    request: "PluginRequestCompiler",
+    request: PluginRequestCompiler,
     unwrap: bool = True,
     pydantic: bool = False,
 ) -> str:
@@ -98,8 +93,8 @@ def get_type_reference(
 
     source_package, source_type = parse_source_type_name(source_type, request)
 
-    current_package: List[str] = package.split(".") if package else []
-    py_package: List[str] = source_package.split(".") if source_package else []
+    current_package: list[str] = package.split(".") if package else []
+    py_package: list[str] = source_package.split(".") if source_package else []
     py_type: str = pythonize_class_name(source_type)
 
     compiling_google_protobuf = current_package == ["google", "protobuf"]
@@ -122,7 +117,7 @@ def get_type_reference(
     return reference_cousin(current_package, imports, py_package, py_type)
 
 
-def reference_absolute(imports: Set[str], py_package: List[str], py_type: str) -> str:
+def reference_absolute(imports: set[str], py_package: list[str], py_type: str) -> str:
     """
     Returns a reference to a python type located in the root, i.e. sys.path.
     """
@@ -139,7 +134,7 @@ def reference_sibling(py_type: str) -> str:
     return f"{py_type}"
 
 
-def reference_descendent(current_package: List[str], imports: Set[str], py_package: List[str], py_type: str) -> str:
+def reference_descendent(current_package: list[str], imports: set[str], py_package: list[str], py_type: str) -> str:
     """
     Returns a reference to a python type in a package that is a descendent of the
     current package, and adds the required import that is aliased to avoid name
@@ -157,7 +152,7 @@ def reference_descendent(current_package: List[str], imports: Set[str], py_packa
         return f"{string_import}.{py_type}"
 
 
-def reference_ancestor(current_package: List[str], imports: Set[str], py_package: List[str], py_type: str) -> str:
+def reference_ancestor(current_package: list[str], imports: set[str], py_package: list[str], py_type: str) -> str:
     """
     Returns a reference to a python type in a package which is an ancestor to the
     current package, and adds the required import that is aliased (if possible) to avoid
@@ -178,7 +173,7 @@ def reference_ancestor(current_package: List[str], imports: Set[str], py_package
         return string_alias
 
 
-def reference_cousin(current_package: List[str], imports: Set[str], py_package: List[str], py_type: str) -> str:
+def reference_cousin(current_package: list[str], imports: set[str], py_package: list[str], py_type: str) -> str:
     """
     Returns a reference to a python type in a package that is not descendent, ancestor
     or sibling, and adds the required import that is aliased to avoid name conflicts.

betterproto2_compiler/lib/pydantic/google/protobuf/__init__.py

@@ -2401,13 +2401,13 @@ class Value(betterproto2_compiler.Message):
     )
     """Represents a null value."""
 
-    number_value: Optional[float] = betterproto2_compiler.double_field(2, optional=True, group="kind")
+    number_value: float | None = betterproto2_compiler.double_field(2, optional=True, group="kind")
     """Represents a double value."""
 
-    string_value: Optional[str] = betterproto2_compiler.string_field(3, optional=True, group="kind")
+    string_value: str | None = betterproto2_compiler.string_field(3, optional=True, group="kind")
     """Represents a string value."""
 
-    bool_value: Optional[bool] = betterproto2_compiler.bool_field(4, optional=True, group="kind")
+    bool_value: bool | None = betterproto2_compiler.bool_field(4, optional=True, group="kind")
     """Represents a boolean value."""
 
     struct_value: Optional["Struct"] = betterproto2_compiler.message_field(5, optional=True, group="kind")

betterproto2_compiler/lib/std/google/protobuf/__init__.py

@@ -78,7 +78,6 @@ from typing import (
     Dict,
     List,
     Mapping,
-    Optional,
 )
 
 import betterproto2
@@ -1022,7 +1021,7 @@ class FieldDescriptorProto(betterproto2.Message):
     TODO(kenton):  Base-64 encode?
     """
 
-    oneof_index: Optional[int] = betterproto2.int32_field(9, optional=True)
+    oneof_index: int | None = betterproto2.int32_field(9, optional=True)
     """
     If set, gives the index of a oneof in the containing type's oneof_decl
     list.  This field is a member of that oneof.

betterproto2_compiler/plugin/compiler.py

@@ -1,6 +1,7 @@
 import os.path
 import subprocess
 import sys
+from importlib import metadata
 
 from .module_validation import ModuleValidator
 
@@ -14,7 +15,7 @@ except ImportError as err:
         "Please ensure that you've installed betterproto as "
         '`pip install "betterproto[compiler]"` so that compiler dependencies '
         "are included."
-        "\033[0m"
+        "\033[0m",
     )
     raise SystemExit(1)
 
@@ -24,6 +25,8 @@ from .models import OutputTemplate
 def outputfile_compiler(output_file: OutputTemplate) -> str:
     templates_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "templates"))
 
+    version = metadata.version("betterproto2_compiler")
+
     env = jinja2.Environment(
         trim_blocks=True,
         lstrip_blocks=True,
@@ -35,7 +38,7 @@ def outputfile_compiler(output_file: OutputTemplate) -> str:
     header_template = env.get_template("header.py.j2")
 
     code = body_template.render(output_file=output_file)
-    code = header_template.render(output_file=output_file) + "\n" + code
+    code = header_template.render(output_file=output_file, version=version) + "\n" + code
 
     # Sort imports, delete unused ones
     code = subprocess.check_output(

betterproto2_compiler/plugin/models.py

@@ -31,18 +31,12 @@ reference to `A` to `B`'s `fields` attribute.
 
 import builtins
 import re
+from collections.abc import Iterable, Iterator
 from dataclasses import (
     dataclass,
     field,
 )
 from typing import (
-    Dict,
-    Iterable,
-    Iterator,
-    List,
-    Optional,
-    Set,
-    Type,
     Union,
 )
 
@@ -59,6 +53,7 @@ from betterproto2_compiler.lib.google.protobuf import (
     FieldDescriptorProto,
     FieldDescriptorProtoLabel,
     FieldDescriptorProtoType,
+    FieldDescriptorProtoType as FieldType,
     FileDescriptorProto,
     MethodDescriptorProto,
 )
@@ -146,7 +141,7 @@ PROTO_PACKED_TYPES = (
 
 def get_comment(
     proto_file: "FileDescriptorProto",
-    path: List[int],
+    path: list[int],
 ) -> str:
     for sci_loc in proto_file.source_code_info.location:
         if list(sci_loc.path) == path:
@@ -182,10 +177,10 @@ class ProtoContentBase:
 
     source_file: FileDescriptorProto
     typing_compiler: TypingCompiler
-    path: List[int]
+    path: list[int]
     parent: Union["betterproto2.Message", "OutputTemplate"]
 
-    __dataclass_fields__: Dict[str, object]
+    __dataclass_fields__: dict[str, object]
 
     def __post_init__(self) -> None:
         """Checks that no fake default fields were left as placeholders."""
@@ -225,10 +220,10 @@ class ProtoContentBase:
 @dataclass
 class PluginRequestCompiler:
     plugin_request_obj: CodeGeneratorRequest
-    output_packages: Dict[str, "OutputTemplate"] = field(default_factory=dict)
+    output_packages: dict[str, "OutputTemplate"] = field(default_factory=dict)
 
     @property
-    def all_messages(self) -> List["MessageCompiler"]:
+    def all_messages(self) -> list["MessageCompiler"]:
         """All of the messages in this request.
 
         Returns
@@ -250,11 +245,11 @@ class OutputTemplate:
 
     parent_request: PluginRequestCompiler
     package_proto_obj: FileDescriptorProto
-    input_files: List[str] = field(default_factory=list)
-    imports_end: Set[str] = field(default_factory=set)
-    messages: Dict[str, "MessageCompiler"] = field(default_factory=dict)
-    enums: Dict[str, "EnumDefinitionCompiler"] = field(default_factory=dict)
-    services: Dict[str, "ServiceCompiler"] = field(default_factory=dict)
+    input_files: list[str] = field(default_factory=list)
+    imports_end: set[str] = field(default_factory=set)
+    messages: dict[str, "MessageCompiler"] = field(default_factory=dict)
+    enums: dict[str, "EnumDefinitionCompiler"] = field(default_factory=dict)
+    services: dict[str, "ServiceCompiler"] = field(default_factory=dict)
     pydantic_dataclasses: bool = False
     output: bool = True
     typing_compiler: TypingCompiler = field(default_factory=DirectImportTypingCompiler)
@@ -290,9 +285,9 @@ class MessageCompiler(ProtoContentBase):
     typing_compiler: TypingCompiler
     parent: Union["MessageCompiler", OutputTemplate] = PLACEHOLDER
     proto_obj: DescriptorProto = PLACEHOLDER
-    path: List[int] = PLACEHOLDER
-    fields: List[Union["FieldCompiler", "MessageCompiler"]] = field(default_factory=list)
-    builtins_types: Set[str] = field(default_factory=set)
+    path: list[int] = PLACEHOLDER
+    fields: list[Union["FieldCompiler", "MessageCompiler"]] = field(default_factory=list)
+    builtins_types: set[str] = field(default_factory=set)
 
     def __post_init__(self) -> None:
         # Add message to output file
@@ -328,11 +323,9 @@ class MessageCompiler(ProtoContentBase):
     @property
     def has_message_field(self) -> bool:
         return any(
-            (
-                field.proto_obj.type in PROTO_MESSAGE_TYPES
-                for field in self.fields
-                if isinstance(field.proto_obj, FieldDescriptorProto)
-            )
+            field.proto_obj.type in PROTO_MESSAGE_TYPES
+            for field in self.fields
+            if isinstance(field.proto_obj, FieldDescriptorProto)
         )
 
 
@@ -347,7 +340,7 @@ def is_map(proto_field_obj: FieldDescriptorProto, parent_message: DescriptorProt
         map_entry = f"{proto_field_obj.name.replace('_', '').lower()}entry"
         if message_type == map_entry:
             for nested in parent_message.nested_type:  # parent message
-                if nested.name.replace("_", "").lower() == map_entry and nested.options.map_entry:
+                if nested.name.replace("_", "").lower() == map_entry and nested.options and nested.options.map_entry:
                     return True
     return False
 
@@ -374,8 +367,8 @@ def is_oneof(proto_field_obj: FieldDescriptorProto) -> bool:
 class FieldCompiler(ProtoContentBase):
     source_file: FileDescriptorProto
     typing_compiler: TypingCompiler
-    path: List[int] = PLACEHOLDER
-    builtins_types: Set[str] = field(default_factory=set)
+    path: list[int] = PLACEHOLDER
+    builtins_types: set[str] = field(default_factory=set)
 
     parent: MessageCompiler = PLACEHOLDER
     proto_obj: FieldDescriptorProto = PLACEHOLDER
@@ -390,13 +383,16 @@ class FieldCompiler(ProtoContentBase):
         """Construct string representation of this field as a field."""
         name = f"{self.py_name}"
         field_args = ", ".join(([""] + self.betterproto_field_args) if self.betterproto_field_args else [])
-        betterproto_field_type = f"betterproto2.{self.field_type}_field({self.proto_obj.number}{field_args})"
+
+        betterproto_field_type = (
+            f"betterproto2.field({self.proto_obj.number}, betterproto2.{str(self.field_type)}{field_args})"
+        )
         if self.py_name in dir(builtins):
             self.parent.builtins_types.add(self.py_name)
         return f'{name}: "{self.annotation}" = {betterproto_field_type}'
 
     @property
-    def betterproto_field_args(self) -> List[str]:
+    def betterproto_field_args(self) -> list[str]:
         args = []
         if self.field_wraps:
             args.append(f"wraps={self.field_wraps}")
@@ -404,9 +400,9 @@ class FieldCompiler(ProtoContentBase):
             args.append("optional=True")
         if self.repeated:
             args.append("repeated=True")
-        if self.field_type == "enum":
+        if self.field_type == FieldType.TYPE_ENUM:
             t = self.py_type
-            args.append(f"enum_default_value=lambda: {t}.try_value(0)")
+            args.append(f"default_factory=lambda: {t}.try_value(0)")
         return args
 
     @property
@@ -416,7 +412,7 @@ class FieldCompiler(ProtoContentBase):
         )
 
     @property
-    def field_wraps(self) -> Optional[str]:
+    def field_wraps(self) -> str | None:
         """Returns betterproto wrapped field type or None."""
         match_wrapper = re.match(r"\.google\.protobuf\.(.+)Value$", self.proto_obj.type_name)
         if match_wrapper:
@@ -428,17 +424,19 @@ class FieldCompiler(ProtoContentBase):
     @property
     def repeated(self) -> bool:
         return self.proto_obj.label == FieldDescriptorProtoLabel.LABEL_REPEATED and not is_map(
-            self.proto_obj, self.parent
+            self.proto_obj,
+            self.parent,
         )
 
     @property
     def optional(self) -> bool:
-        return self.proto_obj.proto3_optional or (self.field_type == "message" and not self.repeated)
+        # TODO not for maps
+        return self.proto_obj.proto3_optional or (self.field_type == FieldType.TYPE_MESSAGE and not self.repeated)
 
     @property
-    def field_type(self) -> str:
-        """String representation of proto field type."""
-        return FieldDescriptorProtoType(self.proto_obj.type).name.lower().replace("type_", "")
+    def field_type(self) -> FieldType:
+        # TODO it should be possible to remove constructor
+        return FieldType(self.proto_obj.type)
 
     @property
     def packed(self) -> bool:
@@ -500,7 +498,7 @@ class OneOfFieldCompiler(FieldCompiler):
         return True
 
     @property
-    def betterproto_field_args(self) -> List[str]:
+    def betterproto_field_args(self) -> list[str]:
         args = super().betterproto_field_args
         group = self.parent.proto_obj.oneof_decl[self.proto_obj.oneof_index].name
         args.append(f'group="{group}"')
@@ -509,8 +507,8 @@ class OneOfFieldCompiler(FieldCompiler):
 
 @dataclass
 class MapEntryCompiler(FieldCompiler):
-    py_k_type: Optional[Type] = None
-    py_v_type: Optional[Type] = None
+    py_k_type: type | None = None
+    py_v_type: type | None = None
     proto_k_type: str = ""
     proto_v_type: str = ""
 
@@ -547,13 +545,17 @@ class MapEntryCompiler(FieldCompiler):
 
         raise ValueError("can't find enum")
 
-    @property
-    def betterproto_field_args(self) -> List[str]:
-        return [f"betterproto2.{self.proto_k_type}", f"betterproto2.{self.proto_v_type}"]
-
-    @property
-    def field_type(self) -> str:
-        return "map"
+    def get_field_string(self) -> str:
+        """Construct string representation of this field as a field."""
+        betterproto_field_type = (
+            f"betterproto2.field({self.proto_obj.number}, "
+            "betterproto2.TYPE_MAP, "
+            f"map_types=(betterproto2.{self.proto_k_type}, "
+            f"betterproto2.{self.proto_v_type}))"
+        )
+        if self.py_name in dir(builtins):
+            self.parent.builtins_types.add(self.py_name)
+        return f'{self.py_name}: "{self.annotation}" = {betterproto_field_type}'
 
     @property
     def annotation(self) -> str:
@@ -569,7 +571,7 @@ class EnumDefinitionCompiler(MessageCompiler):
     """Representation of a proto Enum definition."""
 
     proto_obj: EnumDescriptorProto = PLACEHOLDER
-    entries: List["EnumDefinitionCompiler.EnumEntry"] = PLACEHOLDER
+    entries: list["EnumDefinitionCompiler.EnumEntry"] = PLACEHOLDER
 
     @dataclass(unsafe_hash=True)
     class EnumEntry:
@@ -597,8 +599,8 @@ class ServiceCompiler(ProtoContentBase):
     source_file: FileDescriptorProto
     parent: OutputTemplate = PLACEHOLDER
     proto_obj: DescriptorProto = PLACEHOLDER
-    path: List[int] = PLACEHOLDER
-    methods: List["ServiceMethodCompiler"] = field(default_factory=list)
+    path: list[int] = PLACEHOLDER
+    methods: list["ServiceMethodCompiler"] = field(default_factory=list)
 
     def __post_init__(self) -> None:
         # Add service to output file
@@ -619,7 +621,7 @@ class ServiceMethodCompiler(ProtoContentBase):
     source_file: FileDescriptorProto
     parent: ServiceCompiler
     proto_obj: MethodDescriptorProto
-    path: List[int] = PLACEHOLDER
+    path: list[int] = PLACEHOLDER
 
     def __post_init__(self) -> None:
         # Add method to service

betterproto2_compiler/plugin/module_validation.py

@@ -1,15 +1,10 @@
 import re
 from collections import defaultdict
+from collections.abc import Iterator
 from dataclasses import (
     dataclass,
     field,
 )
-from typing import (
-    Dict,
-    Iterator,
-    List,
-    Tuple,
-)
 
 
 @dataclass
@@ -17,7 +12,7 @@ class ModuleValidator:
     line_iterator: Iterator[str]
     line_number: int = field(init=False, default=0)
 
-    collisions: Dict[str, List[Tuple[int, str]]] = field(init=False, default_factory=lambda: defaultdict(list))
+    collisions: dict[str, list[tuple[int, str]]] = field(init=False, default_factory=lambda: defaultdict(list))
 
     def add_import(self, imp: str, number: int, full_line: str):
         """

betterproto2_compiler/plugin/parser.py

@@ -1,12 +1,6 @@
 import pathlib
 import sys
-from typing import (
-    Generator,
-    List,
-    Set,
-    Tuple,
-    Union,
-)
+from collections.abc import Generator
 
 from betterproto2_compiler.lib.google.protobuf import (
     DescriptorProto,
@@ -45,13 +39,13 @@ from .typing_compiler import (
 
 def traverse(
     proto_file: FileDescriptorProto,
-) -> Generator[Tuple[Union[EnumDescriptorProto, DescriptorProto], List[int]], None, None]:
+) -> Generator[tuple[EnumDescriptorProto | DescriptorProto, list[int]], None, None]:
     # Todo: Keep information about nested hierarchy
     def _traverse(
-        path: List[int],
-        items: Union[List[EnumDescriptorProto], List[DescriptorProto]],
+        path: list[int],
+        items: list[EnumDescriptorProto] | list[DescriptorProto],
         prefix: str = "",
-    ) -> Generator[Tuple[Union[EnumDescriptorProto, DescriptorProto], List[int]], None, None]:
+    ) -> Generator[tuple[EnumDescriptorProto | DescriptorProto, list[int]], None, None]:
         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
@@ -82,7 +76,8 @@ def generate_code(request: CodeGeneratorRequest) -> CodeGeneratorResponse:
         if output_package_name not in request_data.output_packages:
             # Create a new output if there is no output for this package
             request_data.output_packages[output_package_name] = OutputTemplate(
-                parent_request=request_data, package_proto_obj=proto_file
+                parent_request=request_data,
+                package_proto_obj=proto_file,
             )
         # Add this input file to the output corresponding to this package
         request_data.output_packages[output_package_name].input_files.append(proto_file)
@@ -144,7 +139,7 @@ def generate_code(request: CodeGeneratorRequest) -> CodeGeneratorResponse:
             service.ready()
 
     # Generate output files
-    output_paths: Set[pathlib.Path] = set()
+    output_paths: set[pathlib.Path] = set()
     for output_package_name, output_package in request_data.output_packages.items():
         if not output_package.output:
             continue
@@ -158,7 +153,7 @@ def generate_code(request: CodeGeneratorRequest) -> CodeGeneratorResponse:
                 name=str(output_path),
                 # Render and then format the output file
                 content=outputfile_compiler(output_file=output_package),
-            )
+            ),
         )
 
     # Make each output directory a package with __init__ file
@@ -183,7 +178,7 @@ def _make_one_of_field_compiler(
     source_file: "FileDescriptorProto",
     parent: MessageCompiler,
     proto_obj: "FieldDescriptorProto",
-    path: List[int],
+    path: list[int],
 ) -> FieldCompiler:
     return OneOfFieldCompiler(
         source_file=source_file,
@@ -196,7 +191,7 @@ def _make_one_of_field_compiler(
 
 def read_protobuf_type(
     item: DescriptorProto,
-    path: List[int],
+    path: list[int],
     source_file: "FileDescriptorProto",
     output_package: OutputTemplate,
 ) -> None:

betterproto2_compiler/plugin/typing_compiler.py

@@ -1,15 +1,11 @@
 import abc
+import builtins
 from collections import defaultdict
+from collections.abc import Iterator
 from dataclasses import (
     dataclass,
     field,
 )
-from typing import (
-    Dict,
-    Iterator,
-    Optional,
-    Set,
-)
 
 
 class TypingCompiler(metaclass=abc.ABCMeta):
@@ -42,7 +38,7 @@ class TypingCompiler(metaclass=abc.ABCMeta):
         raise NotImplementedError
 
     @abc.abstractmethod
-    def imports(self) -> Dict[str, Optional[Set[str]]]:
+    def imports(self) -> builtins.dict[str, set[str] | None]:
         """
         Returns either the direct import as a key with none as value, or a set of
         values to import from the key.
@@ -63,7 +59,7 @@ class TypingCompiler(metaclass=abc.ABCMeta):
 
 @dataclass
 class DirectImportTypingCompiler(TypingCompiler):
-    _imports: Dict[str, Set[str]] = field(default_factory=lambda: defaultdict(set))
+    _imports: dict[str, set[str]] = field(default_factory=lambda: defaultdict(set))
 
     def optional(self, type_: str) -> str:
         self._imports["typing"].add("Optional")
@@ -93,7 +89,7 @@ class DirectImportTypingCompiler(TypingCompiler):
         self._imports["typing"].add("AsyncIterator")
         return f"AsyncIterator[{type_}]"
 
-    def imports(self) -> Dict[str, Optional[Set[str]]]:
+    def imports(self) -> builtins.dict[str, set[str] | None]:
         return {k: v if v else None for k, v in self._imports.items()}
 
 
@@ -129,7 +125,7 @@ class TypingImportTypingCompiler(TypingCompiler):
         self._imported = True
         return f"typing.AsyncIterator[{type_}]"
 
-    def imports(self) -> Dict[str, Optional[Set[str]]]:
+    def imports(self) -> builtins.dict[str, set[str] | None]:
         if self._imported:
             return {"typing": None}
         return {}
@@ -137,7 +133,7 @@ class TypingImportTypingCompiler(TypingCompiler):
 
 @dataclass
 class NoTyping310TypingCompiler(TypingCompiler):
-    _imports: Dict[str, Set[str]] = field(default_factory=lambda: defaultdict(set))
+    _imports: dict[str, set[str]] = field(default_factory=lambda: defaultdict(set))
 
     def optional(self, type_: str) -> str:
         return f"{type_} | None"
@@ -163,5 +159,5 @@ class NoTyping310TypingCompiler(TypingCompiler):
         self._imports["collections.abc"].add("AsyncIterator")
         return f"AsyncIterator[{type_}]"
 
-    def imports(self) -> Dict[str, Optional[Set[str]]]:
+    def imports(self) -> builtins.dict[str, set[str] | None]:
         return {k: v if v else None for k, v in self._imports.items()}

betterproto2_compiler/templates/header.py.j2

@@ -48,3 +48,5 @@ if TYPE_CHECKING:
     import grpclib.server
     from betterproto2.grpc.grpclib_client import MetadataLike
     from grpclib.metadata import Deadline
+
+betterproto2.check_compiler_version("{{ version }}")

{betterproto2_compiler-0.0.3.dist-info → betterproto2_compiler-0.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: betterproto2_compiler
-Version: 0.0.3
+Version: 0.1.0
 Summary: Compiler for betterproto2
 Home-page: https://github.com/betterproto/python-betterproto2-compiler
 License: MIT

{betterproto2_compiler-0.0.3.dist-info → betterproto2_compiler-0.1.0.dist-info}/RECORD

@@ -1,41 +1,34 @@
 betterproto2_compiler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-betterproto2_compiler/_types.py,sha256=nIsUxcId43N1Gu8EqdeuflR9iUZB1JWu4JTGQV9NeUI,294
 betterproto2_compiler/casing.py,sha256=bMdI4W0hfYh6kV-DQIqFEjSfGYEqUtPciAzP64z5HLQ,3587
 betterproto2_compiler/compile/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-betterproto2_compiler/compile/importing.py,sha256=UNBvayHZmvvwQMIEZ0Zzk0KsqIGqGSbPKhv_DUg2qV8,7442
+betterproto2_compiler/compile/importing.py,sha256=MN1pzFISG96wd6Djsym8q9yfsknBC5logvPrW6qnWyc,7388
 betterproto2_compiler/compile/naming.py,sha256=zf0VOmNojzyv33upOGelGxjZTEDE8JULEEED5_3inHg,562
-betterproto2_compiler/enum.py,sha256=LcILQf1BEjnszouUCtPwifJAR_8u2tf9U9hfwP4vXTc,5396
-betterproto2_compiler/grpc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-betterproto2_compiler/grpc/grpclib_client.py,sha256=6yKUqLFEfiMUPT85g81ajvVI4bvH4kBULZa7pIL18Y8,5275
-betterproto2_compiler/grpc/grpclib_server.py,sha256=Tv3NIGPrxdA48_THUl3jut_IekQgkecX8NPfjvF0kdg,872
-betterproto2_compiler/grpc/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-betterproto2_compiler/grpc/util/async_channel.py,sha256=4sfqoHtS_-qU1GFc0LatnV_dLsmVrGEm74WOJ9RpV1Y,6778
 betterproto2_compiler/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 betterproto2_compiler/lib/google/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 betterproto2_compiler/lib/google/protobuf/__init__.py,sha256=RvtT0CjNfAegDv42ITEvaIVrGpEh2fXvGNyA7bd99oo,60
 betterproto2_compiler/lib/google/protobuf/compiler/__init__.py,sha256=1EvLU05Ck-rwv2Y_jW4ylJlCgqd7VpSHeFgbBFWfAmg,69
 betterproto2_compiler/lib/pydantic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 betterproto2_compiler/lib/pydantic/google/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-betterproto2_compiler/lib/pydantic/google/protobuf/__init__.py,sha256=eAvwcNussnueselEJwX0KcmGbhqz4oTux7e1ugCV6KE,96981
+betterproto2_compiler/lib/pydantic/google/protobuf/__init__.py,sha256=RgGj0p9-KkaO-CFGV_CLTXm1yhPf-Vm02cUkF7z39SI,96972
 betterproto2_compiler/lib/pydantic/google/protobuf/compiler/__init__.py,sha256=uI4F0DytRxbrwXqiRKOJD0nJFewlQE0Lj_3WJdiGoDU,9217
 betterproto2_compiler/lib/std/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 betterproto2_compiler/lib/std/google/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-betterproto2_compiler/lib/std/google/protobuf/__init__.py,sha256=s5Tkt3684xnq2VL0cHZXytN2ja4D73C6ZNf7APEYdcI,80454
+betterproto2_compiler/lib/std/google/protobuf/__init__.py,sha256=bfxwQzYXJ-ImFktGBN5_8HY9JgASJ8BbBg9fKeLTVQ0,80437
 betterproto2_compiler/lib/std/google/protobuf/compiler/__init__.py,sha256=z63o4vkjnDWB7yZpYnx-T0FZCbmSs5gBzfKrNP_qf8c,8646
 betterproto2_compiler/plugin/__init__.py,sha256=L3pW0b4CvkM5x53x_sYt1kYiSFPO0_vaeH6EQPq9FAM,43
 betterproto2_compiler/plugin/__main__.py,sha256=vBQ82334kX06ImDbFlPFgiBRiLIinwNk3z8Khs6hd74,31
-betterproto2_compiler/plugin/compiler.py,sha256=J_0WvuOVXuIINogNTOtU9Kyhzbu3NDrKh7ojbMjSjJk,2032
+betterproto2_compiler/plugin/compiler.py,sha256=jICLI4-5rAOkWQI1v5j7JqIvoao-ZM9szMuq0OBRteA,2138
 betterproto2_compiler/plugin/main.py,sha256=Q9PmcJqXuYYFe51l7AqHVzJrHqi2LWCUu80CZSQOOwk,1469
-betterproto2_compiler/plugin/models.py,sha256=Ljo08MGmZ68q7bFTI7532KrqyjjJhHMe15a-tvxAGyI,24288
-betterproto2_compiler/plugin/module_validation.py,sha256=vye8PjsZFs1Ikh0yNLQXuy12EdM0em0Bflgx7xYrMhk,4853
-betterproto2_compiler/plugin/parser.py,sha256=PFFlK7Di7BF7_tCzkIWUeaRqDfVKJQc9YjSVKeHVXWM,9651
+betterproto2_compiler/plugin/models.py,sha256=7l1P8-ijdatENy-ERO5twHeMpYxLG3YEqhYM9Slm5TY,24655
+betterproto2_compiler/plugin/module_validation.py,sha256=RdPFwdmkbD6NKADaHC5eaPix_pz-yGxHvYJj8Ev48fA,4822
+betterproto2_compiler/plugin/parser.py,sha256=i_HDW0O4ieIatbq1hckOj0kM9MCfdzuib0fumSJqIA4,9610
 betterproto2_compiler/plugin/plugin.bat,sha256=lfLT1WguAXqyerLLsRL6BfHA0RqUE6QG79v-1BYVSpI,48
-betterproto2_compiler/plugin/typing_compiler.py,sha256=gMrKsrA_xFoy33tbm4VjiktXyGAARFGHPL6iVmUiPLU,4866
+betterproto2_compiler/plugin/typing_compiler.py,sha256=IK6m4ggHXK7HL98Ed_WjvQ_yeWfIpf_fIBZ9SA8UcyM,4873
 betterproto2_compiler/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-betterproto2_compiler/templates/header.py.j2,sha256=H3R2v5MiCebp5p1zMKCDjsVlnOLklqrwO24SUtjfdN0,1410
+betterproto2_compiler/templates/header.py.j2,sha256=nxqsengMcM_IRqQYNVntPQ0gUFUPCy_1P1mcoLvbDos,1464
 betterproto2_compiler/templates/template.py.j2,sha256=icyiNdSTJRgyD20e_lTgTAvSjgnSFSn4t1L1-yZnkEM,8712
-betterproto2_compiler-0.0.3.dist-info/LICENSE.md,sha256=Pgl2pReU-2yw2miGeQ55UFlyzqAZ_EpYVyZ2nWjwRv4,1121
-betterproto2_compiler-0.0.3.dist-info/METADATA,sha256=-h0sesZ9kJlHnXU645Jmgbo-DA2BMVqfTZu7b8pCbiQ,1163
-betterproto2_compiler-0.0.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-betterproto2_compiler-0.0.3.dist-info/entry_points.txt,sha256=DE80wLfBwKlvu82d9pAYzEo7Cp22WNqwU7WJZq6JAWk,83
-betterproto2_compiler-0.0.3.dist-info/RECORD,,
+betterproto2_compiler-0.1.0.dist-info/LICENSE.md,sha256=Pgl2pReU-2yw2miGeQ55UFlyzqAZ_EpYVyZ2nWjwRv4,1121
+betterproto2_compiler-0.1.0.dist-info/METADATA,sha256=sJiWJ2eOTCN0ARCo0VmNcMlCPgEFli0tfsnM2vYM-Gc,1163
+betterproto2_compiler-0.1.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+betterproto2_compiler-0.1.0.dist-info/entry_points.txt,sha256=re3Qg8lLljbVobeeKH2f1FVQZ114wfZkGv3zCZTD8Ok,84
+betterproto2_compiler-0.1.0.dist-info/RECORD,,

betterproto2_compiler-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+protoc-gen-python_betterproto2=betterproto2_compiler.plugin:main
+

betterproto2_compiler/_types.py

@@ -1,13 +0,0 @@
-from typing import (
-    TYPE_CHECKING,
-    TypeVar,
-)
-
-if TYPE_CHECKING:
-    from grpclib._typing import IProtoMessage
-
-    from . import Message
-
-# Bound type variable to allow methods to return `self` of subclasses
-T = TypeVar("T", bound="Message")
-ST = TypeVar("ST", bound="IProtoMessage")

betterproto2_compiler/enum.py

@@ -1,180 +0,0 @@
-from __future__ import annotations
-
-from enum import (
-    EnumMeta,
-    IntEnum,
-)
-from types import MappingProxyType
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Dict,
-    Optional,
-    Tuple,
-)
-
-if TYPE_CHECKING:
-    from collections.abc import (
-        Generator,
-        Mapping,
-    )
-
-    from typing_extensions import (
-        Never,
-        Self,
-    )
-
-
-def _is_descriptor(obj: object) -> bool:
-    return hasattr(obj, "__get__") or hasattr(obj, "__set__") or hasattr(obj, "__delete__")
-
-
-class EnumType(EnumMeta if TYPE_CHECKING else type):
-    _value_map_: Mapping[int, Enum]
-    _member_map_: Mapping[str, Enum]
-
-    def __new__(mcs, name: str, bases: Tuple[type, ...], namespace: Dict[str, Any]) -> Self:
-        value_map = {}
-        member_map = {}
-
-        new_mcs = type(
-            f"{name}Type",
-            tuple(
-                dict.fromkeys([base.__class__ for base in bases if base.__class__ is not type] + [EnumType, type])
-            ),  # reorder the bases so EnumType and type are last to avoid conflicts
-            {"_value_map_": value_map, "_member_map_": member_map},
-        )
-
-        members = {
-            name: value for name, value in namespace.items() if not _is_descriptor(value) and not name.startswith("__")
-        }
-
-        cls = type.__new__(
-            new_mcs,
-            name,
-            bases,
-            {key: value for key, value in namespace.items() if key not in members},
-        )
-        # this allows us to disallow member access from other members as
-        # members become proper class variables
-
-        for name, value in members.items():
-            member = value_map.get(value)
-            if member is None:
-                member = cls.__new__(cls, name=name, value=value)  # type: ignore
-                value_map[value] = member
-            member_map[name] = member
-            type.__setattr__(new_mcs, name, member)
-
-        return cls
-
-    if not TYPE_CHECKING:
-
-        def __call__(cls, value: int) -> Enum:
-            try:
-                return cls._value_map_[value]
-            except (KeyError, TypeError):
-                raise ValueError(f"{value!r} is not a valid {cls.__name__}") from None
-
-        def __iter__(cls) -> Generator[Enum, None, None]:
-            yield from cls._member_map_.values()
-
-        def __reversed__(cls) -> Generator[Enum, None, None]:
-            yield from reversed(cls._member_map_.values())
-
-        def __getitem__(cls, key: str) -> Enum:
-            return cls._member_map_[key]
-
-        @property
-        def __members__(cls) -> MappingProxyType[str, Enum]:
-            return MappingProxyType(cls._member_map_)
-
-    def __repr__(cls) -> str:
-        return f"<enum {cls.__name__!r}>"
-
-    def __len__(cls) -> int:
-        return len(cls._member_map_)
-
-    def __setattr__(cls, name: str, value: Any) -> Never:
-        raise AttributeError(f"{cls.__name__}: cannot reassign Enum members.")
-
-    def __delattr__(cls, name: str) -> Never:
-        raise AttributeError(f"{cls.__name__}: cannot delete Enum members.")
-
-    def __contains__(cls, member: object) -> bool:
-        return isinstance(member, cls) and member.name in cls._member_map_
-
-
-class Enum(IntEnum if TYPE_CHECKING else int, metaclass=EnumType):
-    """
-    The base class for protobuf enumerations, all generated enumerations will
-    inherit from this. Emulates `enum.IntEnum`.
-    """
-
-    name: Optional[str]
-    value: int
-
-    if not TYPE_CHECKING:
-
-        def __new__(cls, *, name: Optional[str], value: int) -> Self:
-            self = super().__new__(cls, value)
-            super().__setattr__(self, "name", name)
-            super().__setattr__(self, "value", value)
-            return self
-
-    def __str__(self) -> str:
-        return self.name or "None"
-
-    def __repr__(self) -> str:
-        return f"{self.__class__.__name__}.{self.name}"
-
-    def __setattr__(self, key: str, value: Any) -> Never:
-        raise AttributeError(f"{self.__class__.__name__} Cannot reassign a member's attributes.")
-
-    def __delattr__(self, item: Any) -> Never:
-        raise AttributeError(f"{self.__class__.__name__} Cannot delete a member's attributes.")
-
-    def __copy__(self) -> Self:
-        return self
-
-    def __deepcopy__(self, memo: Any) -> Self:
-        return self
-
-    @classmethod
-    def try_value(cls, value: int = 0) -> Self:
-        """Return the value which corresponds to the value.
-
-        Parameters
-        -----------
-        value: :class:`int`
-            The value of the enum member to get.
-
-        Returns
-        -------
-        :class:`Enum`
-            The corresponding member or a new instance of the enum if
-            ``value`` isn't actually a member.
-        """
-        try:
-            return cls._value_map_[value]
-        except (KeyError, TypeError):
-            return cls.__new__(cls, name=None, value=value)
-
-    @classmethod
-    def from_string(cls, name: str) -> Self:
-        """Return the value which corresponds to the string name.
-
-        Parameters
-        -----------
-        name: :class:`str`
-            The name of the enum member to get.
-
-        Raises
-        -------
-        :exc:`ValueError`
-            The member was not found in the Enum.
-        """
-        try:
-            return cls._member_map_[name]
-        except KeyError as e:
-            raise ValueError(f"Unknown value {name} for enum {cls.__name__}") from e

betterproto2_compiler/grpc/grpclib_client.py

@@ -1,172 +0,0 @@
-import asyncio
-from abc import ABC
-from typing import (
-    TYPE_CHECKING,
-    AsyncIterable,
-    AsyncIterator,
-    Collection,
-    Iterable,
-    Mapping,
-    Optional,
-    Tuple,
-    Type,
-    Union,
-)
-
-import grpclib.const
-
-if TYPE_CHECKING:
-    from grpclib.client import Channel
-    from grpclib.metadata import Deadline
-
-    from .._types import (
-        IProtoMessage,
-        T,
-    )
-
-
-Value = Union[str, bytes]
-MetadataLike = Union[Mapping[str, Value], Collection[Tuple[str, Value]]]
-MessageSource = Union[Iterable["IProtoMessage"], AsyncIterable["IProtoMessage"]]
-
-
-class ServiceStub(ABC):
-    """
-    Base class for async gRPC clients.
-    """
-
-    def __init__(
-        self,
-        channel: "Channel",
-        *,
-        timeout: Optional[float] = None,
-        deadline: Optional["Deadline"] = None,
-        metadata: Optional[MetadataLike] = None,
-    ) -> None:
-        self.channel = channel
-        self.timeout = timeout
-        self.deadline = deadline
-        self.metadata = metadata
-
-    def __resolve_request_kwargs(
-        self,
-        timeout: Optional[float],
-        deadline: Optional["Deadline"],
-        metadata: Optional[MetadataLike],
-    ):
-        return {
-            "timeout": self.timeout if timeout is None else timeout,
-            "deadline": self.deadline if deadline is None else deadline,
-            "metadata": self.metadata if metadata is None else metadata,
-        }
-
-    async def _unary_unary(
-        self,
-        route: str,
-        request: "IProtoMessage",
-        response_type: Type["T"],
-        *,
-        timeout: Optional[float] = None,
-        deadline: Optional["Deadline"] = None,
-        metadata: Optional[MetadataLike] = None,
-    ) -> "T":
-        """Make a unary request and return the response."""
-        async with self.channel.request(
-            route,
-            grpclib.const.Cardinality.UNARY_UNARY,
-            type(request),
-            response_type,
-            **self.__resolve_request_kwargs(timeout, deadline, metadata),
-        ) as stream:
-            await stream.send_message(request, end=True)
-            response = await stream.recv_message()
-        assert response is not None
-        return response
-
-    async def _unary_stream(
-        self,
-        route: str,
-        request: "IProtoMessage",
-        response_type: Type["T"],
-        *,
-        timeout: Optional[float] = None,
-        deadline: Optional["Deadline"] = None,
-        metadata: Optional[MetadataLike] = None,
-    ) -> AsyncIterator["T"]:
-        """Make a unary request and return the stream response iterator."""
-        async with self.channel.request(
-            route,
-            grpclib.const.Cardinality.UNARY_STREAM,
-            type(request),
-            response_type,
-            **self.__resolve_request_kwargs(timeout, deadline, metadata),
-        ) as stream:
-            await stream.send_message(request, end=True)
-            async for message in stream:
-                yield message
-
-    async def _stream_unary(
-        self,
-        route: str,
-        request_iterator: MessageSource,
-        request_type: Type["IProtoMessage"],
-        response_type: Type["T"],
-        *,
-        timeout: Optional[float] = None,
-        deadline: Optional["Deadline"] = None,
-        metadata: Optional[MetadataLike] = None,
-    ) -> "T":
-        """Make a stream request and return the response."""
-        async with self.channel.request(
-            route,
-            grpclib.const.Cardinality.STREAM_UNARY,
-            request_type,
-            response_type,
-            **self.__resolve_request_kwargs(timeout, deadline, metadata),
-        ) as stream:
-            await stream.send_request()
-            await self._send_messages(stream, request_iterator)
-            response = await stream.recv_message()
-        assert response is not None
-        return response
-
-    async def _stream_stream(
-        self,
-        route: str,
-        request_iterator: MessageSource,
-        request_type: Type["IProtoMessage"],
-        response_type: Type["T"],
-        *,
-        timeout: Optional[float] = None,
-        deadline: Optional["Deadline"] = None,
-        metadata: Optional[MetadataLike] = None,
-    ) -> AsyncIterator["T"]:
-        """
-        Make a stream request and return an AsyncIterator to iterate over response
-        messages.
-        """
-        async with self.channel.request(
-            route,
-            grpclib.const.Cardinality.STREAM_STREAM,
-            request_type,
-            response_type,
-            **self.__resolve_request_kwargs(timeout, deadline, metadata),
-        ) as stream:
-            await stream.send_request()
-            sending_task = asyncio.ensure_future(self._send_messages(stream, request_iterator))
-            try:
-                async for response in stream:
-                    yield response
-            except:
-                sending_task.cancel()
-                raise
-
-    @staticmethod
-    async def _send_messages(stream, messages: MessageSource):
-        if isinstance(messages, AsyncIterable):
-            async for message in messages:
-                await stream.send_message(message)
-        else:
-            for message in messages:
-                await stream.send_message(message)
-        await stream.end()

betterproto2_compiler/grpc/grpclib_server.py

@@ -1,32 +0,0 @@
-from abc import ABC
-from collections.abc import AsyncIterable
-from typing import (
-    Any,
-    Callable,
-)
-
-import grpclib
-import grpclib.server
-
-
-class ServiceBase(ABC):
-    """
-    Base class for async gRPC servers.
-    """
-
-    async def _call_rpc_handler_server_stream(
-        self,
-        handler: Callable,
-        stream: grpclib.server.Stream,
-        request: Any,
-    ) -> None:
-        response_iter = handler(request)
-        # check if response is actually an AsyncIterator
-        # this might be false if the method just returns without
-        # yielding at least once
-        # in that case, we just interpret it as an empty iterator
-        if isinstance(response_iter, AsyncIterable):
-            async for response_message in response_iter:
-                await stream.send_message(response_message)
-        else:
-            response_iter.close()

betterproto2_compiler/grpc/util/async_channel.py

@@ -1,190 +0,0 @@
-import asyncio
-from typing import (
-    AsyncIterable,
-    AsyncIterator,
-    Iterable,
-    Optional,
-    TypeVar,
-    Union,
-)
-
-T = TypeVar("T")
-
-
-class ChannelClosed(Exception):
-    """
-    An exception raised on an attempt to send through a closed channel
-    """
-
-
-class ChannelDone(Exception):
-    """
-    An exception raised on an attempt to send receive from a channel that is both closed
-    and empty.
-    """
-
-
-class AsyncChannel(AsyncIterable[T]):
-    """
-    A buffered async channel for sending items between coroutines with FIFO ordering.
-
-    This makes decoupled bidirectional steaming gRPC requests easy if used like:
-
-    .. code-block:: python
-        client = GeneratedStub(grpclib_chan)
-        request_channel = await AsyncChannel()
-        # We can start be sending all the requests we already have
-        await request_channel.send_from([RequestObject(...), RequestObject(...)])
-        async for response in client.rpc_call(request_channel):
-            # The response iterator will remain active until the connection is closed
-            ...
-            # More items can be sent at any time
-            await request_channel.send(RequestObject(...))
-            ...
-            # The channel must be closed to complete the gRPC connection
-            request_channel.close()
-
-    Items can be sent through the channel by either:
-    - providing an iterable to the send_from method
-    - passing them to the send method one at a time
-
-    Items can be received from the channel by either:
-    - iterating over the channel with a for loop to get all items
-    - calling the receive method to get one item at a time
-
-    If the channel is empty then receivers will wait until either an item appears or the
-    channel is closed.
-
-    Once the channel is closed then subsequent attempt to send through the channel will
-    fail with a ChannelClosed exception.
-
-    When th channel is closed and empty then it is done, and further attempts to receive
-    from it will fail with a ChannelDone exception
-
-    If multiple coroutines receive from the channel concurrently, each item sent will be
-    received by only one of the receivers.
-
-    :param source:
-        An optional iterable will items that should be sent through the channel
-        immediately.
-    :param buffer_limit:
-        Limit the number of items that can be buffered in the channel, A value less than
-        1 implies no limit. If the channel is full then attempts to send more items will
-        result in the sender waiting until an item is received from the channel.
-    :param close:
-        If set to True then the channel will automatically close after exhausting source
-        or immediately if no source is provided.
-    """
-
-    def __init__(self, *, buffer_limit: int = 0, close: bool = False):
-        self._queue: asyncio.Queue[T] = asyncio.Queue(buffer_limit)
-        self._closed = False
-        self._waiting_receivers: int = 0
-        # Track whether flush has been invoked so it can only happen once
-        self._flushed = False
-
-    def __aiter__(self) -> AsyncIterator[T]:
-        return self
-
-    async def __anext__(self) -> T:
-        if self.done():
-            raise StopAsyncIteration
-        self._waiting_receivers += 1
-        try:
-            result = await self._queue.get()
-            if result is self.__flush:
-                raise StopAsyncIteration
-            return result
-        finally:
-            self._waiting_receivers -= 1
-            self._queue.task_done()
-
-    def closed(self) -> bool:
-        """
-        Returns True if this channel is closed and no-longer accepting new items
-        """
-        return self._closed
-
-    def done(self) -> bool:
-        """
-        Check if this channel is done.
-
-        :return: True if this channel is closed and and has been drained of items in
-        which case any further attempts to receive an item from this channel will raise
-        a ChannelDone exception.
-        """
-        # After close the channel is not yet done until there is at least one waiting
-        # receiver per enqueued item.
-        return self._closed and self._queue.qsize() <= self._waiting_receivers
-
-    async def send_from(self, source: Union[Iterable[T], AsyncIterable[T]], close: bool = False) -> "AsyncChannel[T]":
-        """
-        Iterates the given [Async]Iterable and sends all the resulting items.
-        If close is set to True then subsequent send calls will be rejected with a
-        ChannelClosed exception.
-        :param source: an iterable of items to send
-        :param close:
-            if True then the channel will be closed after the source has been exhausted
-
-        """
-        if self._closed:
-            raise ChannelClosed("Cannot send through a closed channel")
-        if isinstance(source, AsyncIterable):
-            async for item in source:
-                await self._queue.put(item)
-        else:
-            for item in source:
-                await self._queue.put(item)
-        if close:
-            # Complete the closing process
-            self.close()
-        return self
-
-    async def send(self, item: T) -> "AsyncChannel[T]":
-        """
-        Send a single item over this channel.
-        :param item: The item to send
-        """
-        if self._closed:
-            raise ChannelClosed("Cannot send through a closed channel")
-        await self._queue.put(item)
-        return self
-
-    async def receive(self) -> Optional[T]:
-        """
-        Returns the next item from this channel when it becomes available,
-        or None if the channel is closed before another item is sent.
-        :return: An item from the channel
-        """
-        if self.done():
-            raise ChannelDone("Cannot receive from a closed channel")
-        self._waiting_receivers += 1
-        try:
-            result = await self._queue.get()
-            if result is self.__flush:
-                return None
-            return result
-        finally:
-            self._waiting_receivers -= 1
-            self._queue.task_done()
-
-    def close(self):
-        """
-        Close this channel to new items
-        """
-        self._closed = True
-        asyncio.ensure_future(self._flush_queue())
-
-    async def _flush_queue(self):
-        """
-        To be called after the channel is closed. Pushes a number of self.__flush
-        objects to the queue to ensure no waiting consumers get deadlocked.
-        """
-        if not self._flushed:
-            self._flushed = True
-            deadlocked_receivers = max(0, self._waiting_receivers - self._queue.qsize())
-            for _ in range(deadlocked_receivers):
-                await self._queue.put(self.__flush)
-
-    # A special signal object for flushing the queue when the channel is closed
-    __flush = object()

betterproto2_compiler-0.0.3.dist-info/entry_points.txt

@@ -1,3 +0,0 @@
-[console_scripts]
-protoc-gen-python_betterproto=betterproto2_compiler.plugin:main
-