Add support for custom cargo profiles

Author: messense

CHANGELOG.md

@@ -3,6 +3,7 @@
 ## Unreleased
 ### Added
 - Add support for `kebab-case` executable names. [#205](https://github.com/PyO3/setuptools-rust/pull/205)
+- Add support for custom cargo profiles. [#216](https://github.com/PyO3/setuptools-rust/pull/216)
 
 ## 1.1.2 (2021-12-05)
 ### Changed

setuptools_rust/build.py

@@ -221,7 +221,18 @@ def build_extension(
         # Find the shared library that cargo hopefully produced and copy
         # it into the build directory as if it were produced by build_ext.
 
-        artifacts_dir = os.path.join(target_dir, "debug" if debug else "release")
+        profile = ext.get_cargo_profile()
+        if profile:
+            # https://doc.rust-lang.org/cargo/reference/profiles.html
+            if profile in {"dev", "test"}:
+                profile_dir = "debug"
+            elif profile == "bench":
+                profile_dir = "release"
+            else:
+                profile_dir = profile
+        else:
+            profile_dir = "debug" if debug else "release"
+        artifacts_dir = os.path.join(target_dir, profile_dir)
         dylib_paths = []
 
         if ext._uses_exec_binding():
@@ -462,7 +473,9 @@ def _cargo_args(
             args.extend(["--target", target_triple])
 
         if release:
-            args.append("--release")
+            profile = ext.get_cargo_profile()
+            if not profile:
+                args.append("--release")
 
         if quiet:
             args.append("-q")

setuptools_rust/extension.py

@@ -170,6 +170,26 @@ def get_rust_version(self) -> Optional[SimpleSpec]:  # type: ignore[no-any-unimp
                 "Can not parse rust compiler version: %s", self.rust_version
             )
 
+    def get_cargo_profile(self) -> Optional[str]:
+        args = self.args or []
+        try:
+            index = args.index("--profile")
+            return args[index + 1]
+        except ValueError:
+            pass
+        except IndexError:
+            raise DistutilsSetupError("Can not parse cargo profile from %s", args)
+
+        # Handle `--profile=<profile>`
+        profile_args = [p for p in args if p.startswith("--profile=")]
+        if profile_args:
+            profile = profile_args[0].split("=", 1)[1]
+            if not profile:
+                raise DistutilsSetupError("Can not parse cargo profile from %s", args)
+            return profile
+        else:
+            return None
+
     def entry_points(self) -> List[str]:
         entry_points = []
         if self.script and self.binding == Binding.Exec: