Add object.{__lt__, __le__, __gt__, __gt__}

janczer · janczer · commit bfb122ed2da0 · 2019-02-09T22:28:49.000+01:00
diff --git a/vm/src/obj/objobject.rs b/vm/src/obj/objobject.rs
@@ -44,6 +44,62 @@ fn object_ne(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     objbool::not(vm, &eq)
 }
 
+fn object_lt(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(
+        vm,
+        args,
+        required = [(i, Some(vm.ctx.object())), (i2, None)]
+    );
+
+    Err(vm.new_type_error(format!(
+        "'<' not supported between instances of {} and {}",
+        i.borrow(),
+        i2.borrow()
+    )))
+}
+
+fn object_le(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(
+        vm,
+        args,
+        required = [(i, Some(vm.ctx.object())), (i2, None)]
+    );
+
+    Err(vm.new_type_error(format!(
+        "'<=' not supported between instances of {} and {}",
+        i.borrow(),
+        i2.borrow()
+    )))
+}
+
+fn object_gt(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(
+        vm,
+        args,
+        required = [(i, Some(vm.ctx.object())), (i2, None)]
+    );
+
+    Err(vm.new_type_error(format!(
+        "'>' not supported between instances of {} and {}",
+        i.borrow(),
+        i2.borrow()
+    )))
+}
+
+fn object_ge(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    arg_check!(
+        vm,
+        args,
+        required = [(i, Some(vm.ctx.object())), (i2, None)]
+    );
+
+    Err(vm.new_type_error(format!(
+        "'>=' not supported between instances of {} and {}",
+        i.borrow(),
+        i2.borrow()
+    )))
+}
+
 fn object_hash(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     arg_check!(vm, args, required = [(_zelf, Some(vm.ctx.object()))]);
 
@@ -92,6 +148,10 @@ pub fn init(context: &PyContext) {
     context.set_attr(&object, "__init__", context.new_rustfunc(object_init));
     context.set_attr(&object, "__eq__", context.new_rustfunc(object_eq));
     context.set_attr(&object, "__ne__", context.new_rustfunc(object_ne));
+    context.set_attr(&object, "__lt__", context.new_rustfunc(object_lt));
+    context.set_attr(&object, "__le__", context.new_rustfunc(object_le));
+    context.set_attr(&object, "__gt__", context.new_rustfunc(object_gt));
+    context.set_attr(&object, "__ge__", context.new_rustfunc(object_ge));
     context.set_attr(&object, "__delattr__", context.new_rustfunc(object_delattr));
     context.set_attr(
         &object,
