flyteidl2 2.0.0-alpha4

Rust bindings and utilities for FlyteIDL protobufs.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

// mod test_foo;

// mod serde;

// mod serde_impl;

use pyo3;
use pyo3::prelude::*;
// use crate::HeartbeatResponse;
// include!("flyteidl.common.rs");
// include!("flyteidl.workflow.rs");
// include!("flyteidl.workflow.tonic.rs");
// inculde!("flyteidl.logs.dataplane.rs");
// include!("flyteidl.core.rs");
// include!("google.rpc.rs");
// include!("validate.rs");
use pyo3::pymodule;

// use crate::*;
// Re-export all generated protobuf modules
pub mod flyteidl {

    pub mod common {
        include!("flyteidl.common.rs");
        //         pub mod serde {
        //             include!("flyteidl.common.serde.rs");
        //         }
    }
    pub mod workflow {
        include!("flyteidl.workflow.rs");
        //         pub mod serde {
        //             include!("flyteidl.workflow.serde.rs");
        //         }
        //         pub mod tonic {
        //             include!("flyteidl.workflow.tonic.rs");
        //         }
    }

    pub mod logs {
        pub mod dataplane {
            include!("flyteidl.logs.dataplane.rs");
            //             pub mod serde {
            //                 include!("flyteidl.logs.dataplane.serde.rs");
            //             }
        }
    }

    pub mod core {
        include!("flyteidl.core.rs");
        //         pub mod serde {
        //             include!("flyteidl.core.serde.rs");
        //         }
    }
}

// use pyo3_prost::pyclass_for_prost_struct;
pub mod google {
    pub mod rpc {
        include!("google.rpc.rs");
        //         pub mod serde {
        //             include!("google.rpc.serde.rs");
        //         }
    }
    pub mod protobuf {
        include!(concat!(env!("OUT_DIR"), "/google.protobuf.rs"));
        include!(concat!(env!("OUT_DIR"), "/google.protobuf.serde.rs"));
    }
}

pub mod validate {
    include!("validate.rs");
    //     pub mod serde {
    //         include!("validate.serde.rs");
    //     }
}

// Include the generated Box<T> implementations
include!(concat!(env!("OUT_DIR"), "/boxed_impls.rs"));
// pub mod serde {
//     include!(concat!(env!("OUT_DIR"), "/serde_impls.rs"));
// }
pub mod pymodules {
    include!(concat!(env!("OUT_DIR"), "/pymodules.rs"));
}

include!(concat!(env!("OUT_DIR"), "/serde_impls.rs"));
// include!("serde_impl.rs");

//
// // Re-export commonly used types at the root level for convenience
// pub use flyteidl::common::*;
// pub use flyteidl::workflow::*;
// pub use flyteidl::core::*;
// pub use google::rpc::*;
// pub use validate::*;
// pub use crate::*;

// #[pymodule]
// fn pb_rust(_py: Python, m: &PyModule) -> PyResult<()> {
//     m.add_submodule(flyteidl::workflow::make_module(_py)?)?;
//     // ... all submodules ...
//     Ok(())
// }