diff options
Diffstat (limited to 'src/pyrecwrap.jl')
| -rw-r--r-- | src/pyrecwrap.jl | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/src/pyrecwrap.jl b/src/pyrecwrap.jl index 9cfc5e4..d2c2b9f 100644 --- a/src/pyrecwrap.jl +++ b/src/pyrecwrap.jl @@ -11,8 +11,11 @@ # PyCall. _pyrecwrap_cache = Dict{PyObject,Module}() +_pyx_module_skiplist = Set{ASCIIString}(["math", "os", "sys", "ast", "bytearray", "random", "errno", + "shutil", "zlib", "copy", "base64", "array", "importlib", "glob", "email", "xml", + "binascii", "encodings", "io", "tarfile", "threading", "itertools", "logging", "collections"]) -function pyrecwrap(o::PyObject, mname::Symbol=:__anon__) +function pyrecwrap(o::PyObject, mname::Symbol=:__anon__; skiplist=_pyx_module_skiplist) members = convert(Vector{Tuple{AbstractString,PyObject}}, pycall(PyCall.inspect["getmembers"], PyObject, o)) if PyCall.pyversion >= v"3" @@ -25,6 +28,7 @@ function pyrecwrap(o::PyObject, mname::Symbol=:__anon__) filter!(m -> !(m[1] == PyCall.inspect), members) end filter!(m -> !(m[1] in PyCall.reserved), members) + filter!(m -> !(m[1] in skiplist), members) m = Module(mname, false) # Preload module cache with this (so far empty) module _pyrecwrap_cache[o] = m @@ -37,7 +41,7 @@ function pyrecwrap(o::PyObject, mname::Symbol=:__anon__) end # Before recursing, check if we've seen this python module before if !haskey(_pyrecwrap_cache, mo) - _pyrecwrap_cache[mo] = pyrecwrap(mo) + _pyrecwrap_cache[mo] = pyrecwrap(mo, skiplist=skiplist) end mm = _pyrecwrap_cache[mo] push!(consts, Expr(:const, Expr(:(=), symbol(ms), mm))) |