Skip to content

Commit e0aac83

Browse files
committed
make the python module emit event data
1 parent 1e6e197 commit e0aac83

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

src/python.rs

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
use pyo3::prelude::*;
2+
use pyo3::types::PyDict;
23
use pyo3::wrap_pyfunction;
34

45
#[pyfunction]
@@ -7,12 +8,19 @@ fn to_html(_py: Python, buffer: String) -> PyResult<String> {
78
}
89

910
#[pyfunction]
10-
fn get_offsets(_py: Python, buffer: String) -> PyResult<()> {
11+
fn get_offsets(_py: Python, buffer: String) -> PyResult<&PyDict> {
1112
let events = super::get_offsets(buffer).unwrap();
13+
let pyevents = PyDict::new(_py);
14+
let mut i: u32 = 1;
1215
for event in events.iter() {
13-
eprintln!("DEBUG={:#?}", event);
16+
let event_dict = PyDict::new(_py);
17+
event_dict.set_item("group", event.group.as_str()).unwrap();
18+
event_dict.set_item("start", event.first).unwrap();
19+
event_dict.set_item("end", event.last).unwrap();
20+
pyevents.set_item(i, event_dict)?;
21+
i += 1;
1422
}
15-
Ok(())
23+
Ok(pyevents)
1624
}
1725

1826
#[pymodule]

0 commit comments

Comments
 (0)