add value classes

teodevgroup · Jan 29, 2024 · 26c7dcc · 26c7dcc
1 parent bb2be5d
commit 26c7dcc
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/lib.rs b/src/lib.rs
@@ -26,6 +26,12 @@ use crate::model::property::property::Property;
 use crate::handler::group::HandlerGroup;
 use crate::request::HandlerMatch;
 use crate::request::ctx::RequestCtx;
+use crate::object::value::ObjectId;
+use crate::object::value::OptionVariant;
+use crate::object::value::File;
+use crate::object::value::EnumVariant;
+use crate::object::value::Range;
+use crate::object::pipeline::Pipeline;
 
 #[pymodule]
 fn teo(py: Python, m: &PyModule) -> PyResult<()> {
@@ -73,5 +79,11 @@ def teo_wrap_async(callable):
     m.add_class::<Property>()?;
     m.add_class::<HandlerMatch>()?;
     m.add_class::<RequestCtx>()?;
+    m.add_class::<ObjectId>()?;
+    m.add_class::<File>()?;
+    m.add_class::<Range>()?;
+    m.add_class::<OptionVariant>()?;
+    m.add_class::<EnumVariant>()?;
+    m.add_class::<Pipeline>()?;
     Ok(())
 }
diff --git a/teo/__init__.py b/teo/__init__.py
@@ -1,4 +1,4 @@
-from .teo import App, Namespace, Model, Field, Relation, Property, Enum, EnumMember, Response, Request, ReadOnlyHeaderMap, ReadWriteHeaderMap, HandlerMatch, HandlerGroup, RequestCtx
+from .teo import App, Namespace, Model, Field, Relation, Property, Enum, EnumMember, Response, Request, ReadOnlyHeaderMap, ReadWriteHeaderMap, HandlerMatch, HandlerGroup, RequestCtx, ObjectId, Range
 from signal import signal, SIGINT
 from sys import exit