From 4cf4eb83a10d321f3c0d33abb28a2042ff36c13b Mon Sep 17 00:00:00 2001 From: Padraic Fanning Date: Wed, 14 Apr 2021 21:00:45 -0400 Subject: [PATCH] Add index function to operator module --- vm/src/stdlib/operator.rs | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/vm/src/stdlib/operator.rs b/vm/src/stdlib/operator.rs index 71150963d..12d0f62da 100644 --- a/vm/src/stdlib/operator.rs +++ b/vm/src/stdlib/operator.rs @@ -1,3 +1,4 @@ +use crate::builtins::int::PyIntRef; use crate::builtins::pystr::PyStrRef; use crate::byteslike::PyBytesLike; use crate::common::cmp; @@ -43,10 +44,15 @@ fn _operator_compare_digest( Ok(res) } +fn _operator_index(obj: PyObjectRef, vm: &VirtualMachine) -> PyResult { + vm.to_index(&obj) +} + pub fn make_module(vm: &VirtualMachine) -> PyObjectRef { let ctx = &vm.ctx; py_module!(vm, "_operator", { "length_hint" => named_function!(ctx, _operator, length_hint), "_compare_digest" => named_function!(ctx, _operator, compare_digest), + "index" => named_function!(ctx, _operator, index), }) }