fix context manager api in idowner

Author: somethingelseentirely

notebooks/experiments.py

@@ -1,14 +1,15 @@
 import marimo
 
-__generated_with = "0.9.30"
+__generated_with = "0.9.32"
 app = marimo.App()
 
 
 @app.cell
 def __():
     import maturin_import_hook
     maturin_import_hook.install(settings=maturin_import_hook.MaturinSettings(release=True))
-    return (maturin_import_hook,)
+    import tribles
+    return maturin_import_hook, tribles
 
 
 @app.cell
@@ -17,12 +18,6 @@ def __():
     return (mo,)
 
 
-@app.cell
-def __():
-    import tribles
-    return (tribles,)
-
-
 @app.cell
 def __():
     import sys
@@ -97,6 +92,26 @@ def __(tribles):
     return (Value,)
 
 
+@app.cell
+def __(tribles):
+    IdOwner = tribles.IdOwner
+    return (IdOwner,)
+
+
+@app.cell
+def __(IdOwner):
+    owner = IdOwner()
+    with owner.lock() as o:
+        o.rngid()
+
+    return o, owner
+
+
+@app.cell
+def __():
+    return
+
+
 @app.cell
 def __(FR256LE, Value, fractions):
     Value.of(FR256LE, fractions.Fraction(1,2)).to(fractions.Fraction)
@@ -581,8 +596,15 @@ def __(Id):
 
 
 @app.cell
-def __(Id, bench_consume, element_count_exp, experiments, time_ns):
-    _experiment = Id.genid()
+def __(
+    Id,
+    IdOwner,
+    bench_consume,
+    element_count_exp,
+    experiments,
+    time_ns,
+):
+    _experiment = IdOwner.genid()
     bench_insert_consume_data = experiments.entity({Id: _experiment, "label": "consume"})
     for _i in range(element_count_exp):
         bench_insert_consume_data += experiments.entity(

src/lib.rs

@@ -186,8 +186,13 @@ impl PyIdOwnerGuard {
         self.0 = None;
     }
 
-    pub fn __exit__(&mut self) {
+    pub fn __enter__(slf: PyRef<Self>) -> PyRef<Self> {
+        slf
+    }
+
+    pub fn __exit__(&mut self, _exc_type: PyObject, _exc_value: PyObject, _traceback: PyObject) -> bool {
         self.release();
+        false
     }
 }
 
@@ -409,6 +414,7 @@ impl PyQuery {
 pub fn tribles_python(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<PyTribleSet>()?;
     m.add_class::<PyId>()?;
+    m.add_class::<PyIdOwner>()?;
     m.add_class::<PyValue>()?;
     m.add_class::<PyVariable>()?;
     m.add_class::<PyConstraint>()?;