Merge pull request #3475 from hrchu/pep604

youknowone · web-flow · commit 4f2d70f91514 · 2021-11-29T03:10:01.000+09:00
Allow writing union types as X | Y
diff --git a/vm/src/builtins/pytype.rs b/vm/src/builtins/pytype.rs
@@ -381,6 +381,15 @@ impl PyType {
     fn mro(zelf: PyRef<Self>) -> Vec<PyObjectRef> {
         zelf.iter_mro().map(|cls| cls.clone().into()).collect()
     }
+
+    #[pymethod(name = "__ror__")]
+    #[pymethod(magic)]
+    pub fn or(zelf: PyRef<Self>, other: PyObjectRef, vm: &VirtualMachine) -> PyResult {
+        let union = vm.import("typing", None, 0)?.get_attr("Union", vm)?;
+        let getitem = vm.get_method(union, "__getitem__").unwrap();
+        vm.invoke(&getitem?, ((zelf, other),))
+    }
+
     #[pyslot]
     fn slot_new(metatype: PyTypeRef, args: FuncArgs, vm: &VirtualMachine) -> PyResult {
         vm_trace!("type.__new__ {:?}", args);
