updates for generating python types

Signed-off-by: Jessie Frazelle <[email protected]>

Author: jessfraz

Cargo.lock

@@ -1,12 +1,12 @@
 [workspace]
 resolver = "2"
 members = [
-    "bumper",
-    "modeling-cmds",
-    "modeling-cmds-macros",
-    "modeling-cmds-macros-impl",
-    "modeling-session",
-    "unit-conversion-derive",
+  "bumper",
+  "modeling-cmds",
+  "modeling-cmds-macros",
+  "modeling-cmds-macros-impl",
+  "modeling-session",
+  "unit-conversion-derive",
 ]
 
 [workspace.lints.rust]

Cargo.toml

@@ -23,7 +23,7 @@ convert_client_crate = ["dep:kittycad"]
 websocket = ["dep:serde_json"]
 webrtc = ["dep:webrtc"]
 unstable_exhaustive = []
-python = ["dep:pyo3"]
+python = ["dep:pyo3", "dep:pyo3-stub-gen"]
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
@@ -42,6 +42,7 @@ measurements = "0.11.0"
 parse-display = "0.9.1"
 parse-display-derive = "0.9.0"
 pyo3 = { version = "0.25.1", optional = true }
+pyo3-stub-gen = { version = "0.12.2", optional = true }
 schemars = { version = "0.8.22", features = [
   "bigdecimal04",
   "chrono",
@@ -54,10 +55,10 @@ serde_json = { version = "1.0.139", optional = true }
 slog = { version = "2.7.0", optional = true }
 tabled = { version = "0.18", optional = true }
 ts-rs = { version = "11.0.1", optional = true, features = [
-	"chrono-impl",
-    "uuid-impl",
-    "no-serde-warnings",
-    "serde-json-impl",
+  "chrono-impl",
+  "uuid-impl",
+  "no-serde-warnings",
+  "serde-json-impl",
 ] }
 uuid = { version = "1.16.0", features = ["serde", "v4", "js"] }
 webrtc = { version = "0.12", optional = true }
@@ -67,4 +68,3 @@ bson = "2.14.0"
 
 [lints]
 workspace = true
-

modeling-cmds/Cargo.toml

@@ -12,6 +12,7 @@ use serde::{Deserialize, Serialize};
 #[display(style = "snake_case")]
 #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
 #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
+#[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass_enum)]
 pub enum Axis {
     /// 'Y' axis.
     Y = 1,
@@ -25,6 +26,7 @@ pub enum Axis {
 #[display(style = "snake_case")]
 #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
 #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
+#[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass_enum)]
 pub enum Direction {
     /// Increasing numbers.
     Positive = 1,
@@ -48,6 +50,7 @@ impl std::ops::Mul for Direction {
 #[display("({axis}, {direction})")]
 #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
 #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
+#[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
 pub struct AxisDirectionPair {
     /// Axis specifier.
     pub axis: Axis,
@@ -67,6 +70,7 @@ pub struct AxisDirectionPair {
 #[display("forward: {forward}, up: {up}")]
 #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
 #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
+#[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
 pub struct System {
     /// Axis the front face of a model looks along.
     pub forward: AxisDirectionPair,

modeling-cmds/src/coord.rs

@@ -12,6 +12,7 @@ pub mod export {
     #[serde(rename = "DxfStorage", rename_all = "snake_case")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass_enum)]
     pub enum Storage {
         /// Plaintext encoding.
         ///
@@ -29,7 +30,7 @@ pub mod export {
     #[serde(rename = "DxfExportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     pub struct Options {
         /// Export storage.
         pub storage: Storage,

modeling-cmds/src/format/dxf.rs

@@ -11,7 +11,7 @@ pub mod import {
     #[serde(rename = "FbxImportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     pub struct Options {}
 
     #[cfg(feature = "python")]
@@ -33,7 +33,7 @@ pub mod export {
     #[derive(Clone, Debug, Default, Eq, Hash, PartialEq, Serialize, Deserialize, JsonSchema)]
     #[serde(rename = "FbxExportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
     pub struct Options {
         /// Specifies which kind of FBX will be exported.
@@ -77,6 +77,7 @@ pub mod export {
     #[serde(rename = "FbxStorage", rename_all = "snake_case")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass_enum)]
     pub enum Storage {
         /// ASCII FBX encoding.
         Ascii,

modeling-cmds/src/format/fbx.rs

@@ -12,7 +12,7 @@ pub mod import {
     #[serde(rename = "GltfImportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     pub struct Options {}
 
     #[cfg(feature = "python")]
@@ -34,7 +34,7 @@ pub mod export {
     #[display("storage: {storage}, presentation: {presentation}")]
     #[serde(rename = "GltfExportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
     pub struct Options {
         /// Specifies which kind of glTF 2.0 will be exported.
@@ -61,6 +61,7 @@ pub mod export {
     #[serde(rename = "GltfStorage", rename_all = "snake_case")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass_enum)]
     pub enum Storage {
         /// Binary glTF 2.0.
         ///

modeling-cmds/src/format/gltf.rs

@@ -79,7 +79,11 @@ pub type InputFormat = InputFormat3d;
 #[derive(Clone, Debug, Eq, Hash, PartialEq, Serialize, Deserialize, JsonSchema, Display, FromStr)]
 #[serde(tag = "type", rename_all = "snake_case")]
 #[display(style = "snake_case")]
-#[cfg_attr(feature = "python", pyo3::pyclass)]
+#[cfg_attr(
+    feature = "python",
+    pyo3::pyclass,
+    pyo3_stub_gen::derive::gen_stub_pyclass_complex_enum
+)]
 #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
 #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
 pub enum InputFormat3d {

modeling-cmds/src/format/mod.rs

@@ -14,7 +14,7 @@ pub mod import {
     #[serde(rename = "ObjImportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     pub struct Options {
         /// Co-ordinate system of input data.
         ///
@@ -61,7 +61,7 @@ pub mod export {
     #[display("coords: {coords}, units: {units}")]
     #[serde(rename = "ObjExportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
     pub struct Options {
         /// Co-ordinate system of output data.

modeling-cmds/src/format/obj.rs

@@ -14,7 +14,7 @@ pub mod import {
     #[serde(rename = "PlyImportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     pub struct Options {
         /// Co-ordinate system of input data.
         ///
@@ -62,7 +62,7 @@ pub mod export {
     #[display("coords: {coords}, selection: {selection}, storage: {storage}, units: {units}")]
     #[serde(rename = "PlyExportOptions")]
     #[cfg_attr(feature = "ts-rs", derive(ts_rs::TS))]
-    #[cfg_attr(feature = "python", pyo3::pyclass)]
+    #[cfg_attr(feature = "python", pyo3::pyclass, pyo3_stub_gen::derive::gen_stub_pyclass)]
     #[cfg_attr(feature = "ts-rs", ts(export_to = "ModelingCmd.ts"))]
     pub struct Options {
         /// Co-ordinate system of output data.