add missing dir in GenericAlias

Signed-off-by: snowapril <[email protected]>

Author: Snowapril

vm/src/builtins/genericalias.rs

@@ -1,12 +1,24 @@
 use crate::{
-    builtins::{PyStr, PyTuple, PyTupleRef, PyTypeRef},
+    builtins::{PyList, PyStr, PyTuple, PyTupleRef, PyTypeRef},
     common::hash,
+    function::IntoPyObject,
     slots::{Hashable, SlotConstructor},
     IdProtocol, PyClassImpl, PyContext, PyObjectRef, PyRef, PyResult, PyValue, TryFromObject,
     TypeProtocol, VirtualMachine,
 };
 use std::fmt;
 
+static ATTR_EXCEPTIONS: [&str; 8] = [
+    "__origin__",
+    "__args__",
+    "__parameters__",
+    "__mro_entries__",
+    "__reduce_ex__", // needed so we don't look up object.__reduce_ex__
+    "__reduce__",
+    "__copy__",
+    "__deepcopy__",
+];
+
 #[pyclass(module = "types", name = "GenericAlias")]
 pub struct PyGenericAlias {
     origin: PyTypeRef,
@@ -111,6 +123,17 @@ impl PyGenericAlias {
     fn origin(&self) -> PyObjectRef {
         self.origin.as_object().clone()
     }
+
+    #[pymethod(magic)]
+    fn dir(&self, vm: &VirtualMachine) -> PyResult<PyList> {
+        let dir = vm.dir(Some(self.origin()))?;
+        for exc in ATTR_EXCEPTIONS.iter() {
+            if !dir.contains((*exc).into_pyobject(vm), vm)? {
+                dir.append((*exc).into_pyobject(vm));
+            }
+        }
+        Ok(dir)
+    }
 }
 
 fn is_typevar(obj: PyObjectRef) -> bool {

vm/src/builtins/list.rs

@@ -265,7 +265,7 @@ impl PyList {
     }
 
     #[pymethod(magic)]
-    fn contains(&self, needle: PyObjectRef, vm: &VirtualMachine) -> PyResult<bool> {
+    pub fn contains(&self, needle: PyObjectRef, vm: &VirtualMachine) -> PyResult<bool> {
         // TODO: to_vec() cause copy which leads to cost O(N). It need to be improved.
         let elements = self.borrow_vec().to_vec();
         for elem in elements.iter() {