Add back the Rust codec (#57)

* Add back the Rust codec

* Execute the tests with the Rust codec

* Install before downloading tests

Author: AdrienVannson

.github/workflows/ci.yml

@@ -62,4 +62,12 @@ jobs:
 
       - name: Execute test suite
         shell: bash
-        run: poetry run python -m pytest tests/
+        run: poetry run poe test
+
+      - name: Install dependencies with the Rust codec
+        shell: bash
+        run: poetry install -E rust-codec
+
+      - name: Execute test suite with the Rust codec
+        shell: bash
+        run: poetry run poe test

poetry.lock

@@ -16,7 +16,7 @@ python = "^3.10"
 grpclib = "^0.4.1"
 python-dateutil = "^2.8"
 typing-extensions = "^4.7.1"
-betterproto-rust-codec = { version = "0.1.1", optional = true }
+betterproto2-rust-codec = { version = "^0.1.2", optional = true }
 
 [tool.poetry.group.dev.dependencies]
 asv = "^0.4.2"
@@ -43,7 +43,7 @@ cachelib = "^0.10.2"
 protoc-gen-python_betterproto = "betterproto2.plugin:main"
 
 [tool.poetry.extras]
-rust-codec = ["betterproto-rust-codec"]
+rust-codec = ["betterproto2-rust-codec"]
 
 [tool.ruff]
 extend-exclude = ["tests/output_*", "src/betterproto2/lib"]

pyproject.toml

@@ -1343,6 +1343,23 @@ def _validate_field_groups(cls, values):
 Message.__annotations__ = {}  # HACK to avoid typing.get_type_hints breaking :)
 
 
+try:
+    import betterproto2_rust_codec
+except ModuleNotFoundError:
+    pass
+else:
+
+    def parse_patched(self, data: bytes) -> T:
+        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, Optional[Any]]:
     """
     Return the name and value of a message's one-of field group.