diff --git a/ica-rs/src/py/class.rs b/ica-rs/src/py/class.rs index ffdb614..6cc2c87 100644 --- a/ica-rs/src/py/class.rs +++ b/ica-rs/src/py/class.rs @@ -1,8 +1,8 @@ use pyo3::prelude::*; use rust_socketio::asynchronous::Client; use tokio::runtime::Runtime; -use tracing::{debug, info, warn}; use toml::Value as TomlValue; +use tracing::{debug, info, warn}; use crate::client::{delete_message, send_message, IcalinguaStatus}; use crate::data_struct::messages::{ @@ -220,7 +220,6 @@ impl ConfigRequestPy { pub fn py_new(path: String) -> Self { Self { path } } } - #[derive(Clone)] #[pyclass] #[pyo3(name = "ConfigData")] @@ -232,44 +231,27 @@ pub struct ConfigDataPy { impl ConfigDataPy { pub fn __getitem__(self_: PyRef<'_, Self>, key: String) -> Option> { match self_.data.get(&key) { - Some(value) => { - match value { - TomlValue::String(s) => { - let py_value = s.into_py(self_.py()); - Some(py_value) - - }, - TomlValue::Integer(i) => { - let py_value = i.into_py(self_.py()); - Some(py_value) - }, - TomlValue::Float(f) => { - let py_value = f.into_py(self_.py()); - Some(py_value) - }, - TomlValue::Boolean(b) => { - let py_value = b.into_py(self_.py()); - Some(py_value) - }, - TomlValue::Array(a) => { - let new_self = Self::new(TomlValue::Array(a.clone())); - let py_value = new_self.into_py(self_.py()); - Some(py_value) - }, - TomlValue::Table(t) => { - let new_self = Self::new(TomlValue::Table(t.clone())); - let py_value = new_self.into_py(self_.py()); - Some(py_value) - }, - _ => None, + Some(value) => match value { + TomlValue::String(s) => Some(s.into_py(self_.py())), + TomlValue::Integer(i) => Some(i.into_py(self_.py())), + TomlValue::Float(f) => Some(f.into_py(self_.py())), + TomlValue::Boolean(b) => Some(b.into_py(self_.py())), + TomlValue::Array(a) => { + let new_self = Self::new(TomlValue::Array(a.clone())); + let py_value = new_self.into_py(self_.py()); + Some(py_value) } - } + TomlValue::Table(t) => { + let new_self = Self::new(TomlValue::Table(t.clone())); + let py_value = new_self.into_py(self_.py()); + Some(py_value) + } + _ => None, + }, None => None, } } - pub fn have_key(&self, key: String) -> bool { - self.data.get(&key).is_some() - } + pub fn have_key(&self, key: String) -> bool { self.data.get(&key).is_some() } } impl ConfigDataPy {