Python - Add Encoding repr + improve example

This commit is contained in:
Anthony MOI
2019-12-10 15:18:07 -05:00
parent 99773d9ce4
commit 6c294c60b0
4 changed files with 37 additions and 26 deletions

View File

@ -1,8 +1,9 @@
extern crate tokenizers as tk;
use pyo3::prelude::*;
use pyo3::PyObjectProtocol;
#[pyclass]
#[pyclass(dict)]
#[repr(transparent)]
pub struct Encoding {
encoding: tk::tokenizer::Encoding,
@ -14,6 +15,16 @@ impl Encoding {
}
}
#[pyproto]
impl PyObjectProtocol for Encoding {
fn __repr__(&self) -> PyResult<String> {
Ok(format!(
"Encoding {{ original: '{}', ... }}",
self.encoding.get_original()
))
}
}
#[pymethods]
impl Encoding {
#[getter]