grapl_graph_descriptions_py/
lib.rs

1use pyo3::prelude::*;
2
3
4pub mod asset_node;
5pub mod file_node;
6pub mod graph;
7pub mod process_node;
8pub mod ip_address_node;
9pub mod ip_connection_node;
10pub mod ip_port_node;
11pub mod network_connection_node;
12pub mod process_outbound_connection_node;
13pub mod process_inbound_connection_node;
14
15
16#[pymodule]
17fn grapl_graph_descriptions_py(_py: Python, m: &PyModule) -> PyResult<()> {
18    m.add_class::<process_node::ProcessNode>()?;
19    m.add_class::<process_node::ProcessNodeBuilder>()?;
20
21    m.add_class::<file_node::FileNode>()?;
22    m.add_class::<file_node::FileNodeBuilder>()?;
23
24    m.add_class::<asset_node::AssetNode>()?;
25    m.add_class::<asset_node::AssetNodeBuilder>()?;
26
27    m.add_class::<ip_address_node::IpAddressNode>()?;
28    m.add_class::<ip_address_node::IpAddressNodeBuilder>()?;
29
30    m.add_class::<ip_connection_node::IpConnectionNode>()?;
31    m.add_class::<ip_connection_node::IpConnectionNodeBuilder>()?;
32
33    m.add_class::<ip_port_node::IpPortNode>()?;
34    m.add_class::<ip_port_node::IpPortNodeBuilder>()?;
35
36    m.add_class::<network_connection_node::NetworkConnectionNode>()?;
37    m.add_class::<network_connection_node::NetworkConnectionNodeBuilder>()?;
38
39    m.add_class::<process_outbound_connection_node::ProcessOutboundConnectionNode>()?;
40    m.add_class::<process_outbound_connection_node::ProcessOutboundConnectionNodeBuilder>()?;
41
42    m.add_class::<process_inbound_connection_node::ProcessInboundConnectionNode>()?;
43    m.add_class::<process_inbound_connection_node::ProcessInboundConnectionNodeBuilder>()?;
44
45    Ok(())
46}
47
48#[cfg(test)]
49mod tests {
50    #[test]
51    fn it_works() {
52        assert_eq!(2 + 2, 4);
53    }
54}
grapl_graph_descriptions_py/lib.rs

grapl_graph_descriptions_py/
lib.rs
