python: expose jail iteration

bindings/python/jail/__init__.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 from __future__ import absolute_import
 
-from ._jail import RunningJail, StoppedJail
+from ._jail import RunningJail, StoppedJail, Jls
 
-__all__ = ['RunningJail', 'StoppedJail']
+__all__ = ['RunningJail', 'StoppedJail', 'Jls']

bindings/python/src/jls.rs

@@ -0,0 +1,38 @@
+use pyo3::prelude::*;
+use pyo3::{PyIterProtocol, PyObjectWithToken};
+
+use jail as native;
+use running::RunningJail;
+
+#[pyclass]
+pub struct Jls {
+    token: PyToken,
+    it: native::RunningJailIter,
+}
+
+#[pymethods]
+impl Jls {
+    #[new]
+    fn __new__(obj: &PyRawObject) -> PyResult<()> {
+        obj.init(|token| Jls {
+            it: native::RunningJail::all(),
+            token,
+        })
+    }
+}
+
+#[pyproto]
+impl PyIterProtocol for Jls {
+    fn __iter__(&mut self) -> PyResult<PyObject> {
+        self.it = native::RunningJail::all();
+        Ok(self.into())
+    }
+
+    fn __next__(&mut self) -> PyResult<Option<Py<RunningJail>>> {
+        match self.it.next() {
+            None => Ok(None),
+            Some(next) => Ok(Some(self.py().init(|token| RunningJail::create(token, next))?)
+                )
+        }
+    }
+}

bindings/python/src/lib.rs

@@ -11,9 +11,12 @@ mod error;
 mod param;
 mod running;
 mod stopped;
+mod jls;
+
 use child::Child;
 use running::RunningJail;
 use stopped::StoppedJail;
+use jls::Jls;
 
 #[pyclass]
 struct JailError {
@@ -33,6 +36,7 @@ fn jail_modinit(_py: Python, m: &PyModule) -> PyResult<()> {
     m.add_class::<RunningJail>()?;
     m.add_class::<StoppedJail>()?;
     m.add_class::<Child>()?;
+    m.add_class::<Jls>()?;
 
     Ok(())
 }