Small cleaning (#175)

Author: AdrienVannson

betterproto2/src/betterproto2/__init__.py

@@ -1301,24 +1301,6 @@ def _validate_field_groups(cls, values):
 Message.__annotations__ = {}  # HACK to avoid typing.get_type_hints breaking :)
 
 
-# The Rust codec is not available for now
-# try:
-#     import betterproto2_rust_codec  # pyright: ignore[reportMissingImports]
-# except ModuleNotFoundError:
-#     pass
-# else:
-
-#     def parse_patched(self, data: bytes) -> Message:
-#         betterproto2_rust_codec.deserialize(self, data)
-#         return self
-
-#     def bytes_patched(self) -> bytes:
-#         return betterproto2_rust_codec.serialize(self)
-
-#     Message.parse = parse_patched
-#     Message.__bytes__ = bytes_patched
-
-
 def which_one_of(message: Message, group_name: str) -> tuple[str, Any | None]:
     """
     Return the name and value of a message's one-of field group.

betterproto2_compiler/src/betterproto2_compiler/plugin/compiler.py

@@ -3,23 +3,10 @@
 import sys
 from importlib import metadata
 
-from .module_validation import ModuleValidator
-
-try:
-    # betterproto[compiler] specific dependencies
-    import jinja2
-except ImportError as err:
-    print(
-        "\033[31m"
-        f"Unable to import `{err.name}` from betterproto plugin! "
-        "Please ensure that you've installed betterproto as "
-        '`pip install "betterproto[compiler]"` so that compiler dependencies '
-        "are included."
-        "\033[0m",
-    )
-    raise SystemExit(1)
+import jinja2
 
 from .models import OutputTemplate
+from .module_validation import ModuleValidator
 
 
 def outputfile_compiler(output_file: OutputTemplate) -> str:

betterproto2_compiler/src/betterproto2_compiler/plugin/models.py

@@ -27,10 +27,7 @@
 import builtins
 import inspect
 from collections.abc import Iterator
-from dataclasses import (
-    dataclass,
-    field,
-)
+from dataclasses import dataclass, field
 
 from betterproto2 import unwrap