/// Testing if a string is a keyword.
pub(crate) use keyword::make_module;

#[pymodule]
mod keyword {
    use crate::vm::{builtins::PyStr, PyObjectRef, VirtualMachine};
    use itertools::Itertools;
    use rustpython_parser::lexer;

    #[pyfunction]
    fn iskeyword(s: PyObjectRef) -> bool {
        if let Some(s) = s.payload::<PyStr>() {
            lexer::KEYWORDS.contains_key(s.as_str())
        } else {
            false
        }
    }

    #[pyattr]
    fn kwlist(vm: &VirtualMachine) -> Vec<PyObjectRef> {
        lexer::KEYWORDS
            .keys()
            .sorted()
            .map(|&k| vm.ctx.new_str(k).into())
            .collect()
    }
}