#![allow(non_snake_case)]
pub(crate) use _wmi::module_def;
#[pymodule]
mod _wmi {
use crate::builtins::PyStrRef;
use crate::convert::ToPyException;
use crate::{PyResult, VirtualMachine};
use rustpython_host_env::wmi as host_wmi;
#[pyfunction]
fn exec_query(query: PyStrRef, vm: &VirtualMachine) -> PyResult<String> {
let query_str = query.expect_str();
if !query_str
.get(..7)
.is_some_and(|s| s.eq_ignore_ascii_case("select "))
{
return Err(vm.new_value_error("only SELECT queries are supported"));
}
host_wmi::exec_query(query_str).map_err(|err| err.to_pyexception(vm))
}
}