make AliasModule proxy attribute access methods to original module.

Author: schmir

apipkg.py

@@ -134,27 +134,27 @@ def __dict__(self):
         return dict
     __dict__ = property(__dict__)
 
-class AliasModule(ModuleType):
-    def __init__(self, name, modpath):
-        self.__name = name
-        self.__modpath = modpath
 
-    def __repr__(self):
-        return '<AliasModule %r for %r>' % (self.__name, self.__modpath)
+def AliasModule(modname, modpath):
+    mod = []
 
-    def __getattr__(self, name):
-        mod = importobj(self.__modpath, None)
-        result = getattr(mod, name)
-        setattr(self, name, result)
-        for k, v in mod.__dict__.items():
-            setattr(self, k, v)
-        return result
+    def getmod():
+        if not mod:
+            mod.append(importobj(modpath, None))
+        return mod[0]
 
-    def __dict__(self):
-        # force all the content of the module to be loaded when __dict__ is read
-        dictdescr = ModuleType.__dict__['__dict__']
-        dict = dictdescr.__get__(self)
-        if dict is not None:
-            hasattr(self, 'some')
-        return dict
-    __dict__ = property(__dict__)
+    class AliasModule(ModuleType):
+
+        def __repr__(self):
+            return '<AliasModule %r for %r>' % (modname, modpath)
+
+        def __getattribute__(self, name):
+            return getattr(getmod(), name)
+
+        def __setattr__(self, name, value):
+            setattr(getmod(), name, value)
+
+        def __delattr__(self, name):
+            delattr(getmod(), name)
+
+    return AliasModule(modname)

test_apipkg.py

@@ -409,6 +409,35 @@ def test_aliasmodule_repr():
     am.version
     assert repr(am) == r
 
+def test_aliasmodule_proxy_methods(tmpdir, monkeypatch):
+    pkgdir = tmpdir
+    pkgdir.join('aliasmodule_proxy.py').write(py.code.Source("""
+        def doit():
+            return 42
+    """))
+
+    pkgdir.join('my_aliasmodule_proxy.py').write(py.code.Source("""
+        import apipkg
+        apipkg.initpkg(__name__, dict(proxy='aliasmodule_proxy'))
+
+        def doit():
+            return 42
+    """))
+
+    monkeypatch.syspath_prepend(tmpdir)
+    import aliasmodule_proxy as orig
+    from my_aliasmodule_proxy import proxy
+
+    doit = proxy.doit
+    assert doit is orig.doit
+
+    del proxy.doit
+    py.test.raises(AttributeError, "orig.doit")
+
+    proxy.doit = doit
+    assert orig.doit is doit
+
+
 def test_initpkg_without_old_module():
     apipkg.initpkg("initpkg_without_old_module",
                    dict(modules="sys:modules"))