Reorganize core persistence crate Python wranglers

nautechsystems · Oct 21, 2023 · 05a1cc9 · 05a1cc9
1 parent 546b64b
commit 05a1cc9
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 7 deletions.
diff --git a/nautilus_core/persistence/src/lib.rs b/nautilus_core/persistence/src/lib.rs
@@ -16,7 +16,6 @@
 pub mod arrow;
 pub mod backend;
 mod kmerge_batch;
-pub mod wranglers;
 
 #[cfg(feature = "python")]
 pub mod python;
diff --git a/nautilus_core/persistence/src/python/mod.rs b/nautilus_core/persistence/src/python/mod.rs
@@ -16,6 +16,7 @@
 use pyo3::prelude::*;
 
 mod backend;
+mod wranglers;
 
 /// Loaded as nautilus_pyo3.persistence
 #[pymodule]
@@ -24,9 +25,9 @@ pub fn persistence(_: Python<'_>, m: &PyModule) -> PyResult<()> {
     m.add_class::<crate::backend::session::DataBackendSession>()?;
     m.add_class::<crate::backend::session::DataQueryResult>()?;
     m.add_class::<backend::transformer::DataTransformer>()?;
-    m.add_class::<crate::wranglers::bar::BarDataWrangler>()?;
-    m.add_class::<crate::wranglers::delta::OrderBookDeltaDataWrangler>()?;
-    m.add_class::<crate::wranglers::quote::QuoteTickDataWrangler>()?;
-    m.add_class::<crate::wranglers::trade::TradeTickDataWrangler>()?;
+    m.add_class::<wranglers::bar::BarDataWrangler>()?;
+    m.add_class::<wranglers::delta::OrderBookDeltaDataWrangler>()?;
+    m.add_class::<wranglers::quote::QuoteTickDataWrangler>()?;
+    m.add_class::<wranglers::trade::TradeTickDataWrangler>()?;
     Ok(())
 }
diff --git a/...lus_core/persistence/src/wranglers/bar.rs → ...e/persistence/src/python/wranglers/bar.rs b/...lus_core/persistence/src/wranglers/bar.rs → ...e/persistence/src/python/wranglers/bar.rs
diff --git a/...s_core/persistence/src/wranglers/delta.rs → ...persistence/src/python/wranglers/delta.rs b/...s_core/persistence/src/wranglers/delta.rs → ...persistence/src/python/wranglers/delta.rs
@@ -30,7 +30,6 @@ pub struct OrderBookDeltaDataWrangler {
     metadata: HashMap<String, String>,
 }
 
-#[cfg(feature = "python")]
 #[pymethods]
 impl OrderBookDeltaDataWrangler {
     #[new]

diff --git a/...lus_core/persistence/src/wranglers/mod.rs → ...e/persistence/src/python/wranglers/mod.rs b/...lus_core/persistence/src/wranglers/mod.rs → ...e/persistence/src/python/wranglers/mod.rs
diff --git a/...s_core/persistence/src/wranglers/quote.rs → ...persistence/src/python/wranglers/quote.rs b/...s_core/persistence/src/wranglers/quote.rs → ...persistence/src/python/wranglers/quote.rs
@@ -30,7 +30,6 @@ pub struct QuoteTickDataWrangler {
     metadata: HashMap<String, String>,
 }
 
-#[cfg(feature = "python")]
 #[pymethods]
 impl QuoteTickDataWrangler {
     #[new]

diff --git a/...s_core/persistence/src/wranglers/trade.rs → ...persistence/src/python/wranglers/trade.rs b/...s_core/persistence/src/wranglers/trade.rs → ...persistence/src/python/wranglers/trade.rs