X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;ds=sidebyside;f=monitor%2FRpyc%2FNetProxy.py;fp=monitor%2FRpyc%2FNetProxy.py;h=96ed53e2f78b31e8f133e4048fc711222b6708ae;hb=334378a14103c3fd02332b6ce3767553f1fe11d2;hp=0000000000000000000000000000000000000000;hpb=486326759a86f1315d93aeaccf6e2641af2bd9d9;p=monitor.git

diff --git a/monitor/Rpyc/NetProxy.py b/monitor/Rpyc/NetProxy.py
new file mode 100644
index 0000000..96ed53e
--- /dev/null
+++ b/monitor/Rpyc/NetProxy.py
@@ -0,0 +1,119 @@
+from Lib import ImmDict
+
+
+def fully_dynamic_metaclass(name, bases, dict):
+    """
+    a meta class that enables special methods to be accessed like regular names 
+    (via __getattr__), like it used to be in old-style classes.
+    """
+
+    special_methods = [
+        "__hash__", "__len__", "__iter__", "next", "__reversed__",
+        "__add__", "__iadd__", "__radd__", "__sub__", "__isub__", "__rsub__", "__mul__",
+        "__imul__", "__rmul__", "__div__", "__idiv__", "__rdiv__", "__truediv__", 
+        "__itruediv__", "__rtruediv__",  "__floordiv__", "__ifloordiv__", "__rfloorfiv__", 
+        "__pow__", "__ipow__", "__rpow__", "__lshift__", "__ilshift__", "__rlshift__",
+        "__rshift__", "__irshift__", "__rrshift__", "__and__", "__iand__", "__rand__",
+        "__or__", "__ior__", "__ror__", "__xor__", "__ixor__", "__rxor__", "__mod__", 
+        "__imod__", "__rmod__", "__divmod__", "__idivmod__", "__rdivmod__", "__pos__", 
+        "__neg__", "__int__", "__float__", "__long__", "__oct__", "__hex__", "__coerce__",
+        "__eq__", "__ne__", "__le__", "__ge__", "__lt__", "__gt__", "__cmp__",
+    ]
+    
+    # i added '__class__' to the special attributes, but it broke some things 
+    # (like dir), so we'll have to live without it
+    special_attributes = ["__doc__", "__module__"] 
+
+    def make_method(name):
+        def caller(self, *a, **k):
+            return self.__getattr__(name)(*a, **k)
+        return caller
+
+    def make_property(name):
+        def getter(self):
+            return self.__getattr__(name)
+        def setter(self, value):
+            self.__setattr__(name, value)
+        def deller(self):
+            self.__delattr__(name)
+        return property(getter, setter, deller)
+
+    classdict = {}
+    for sn in special_methods:
+        classdict[sn] = make_method(sn)
+    classdict.update(dict)
+    for sa in special_attributes:
+        classdict[sa] = make_property(sa)
+    return type(name, bases, classdict)
+
+class NetProxy(object):
+    """NetProxy - convey local operations to the remote object. this is an abstract class"""
+    __metaclass__ = fully_dynamic_metaclass
+    
+    def __init__(self, conn, oid):
+        self.__dict__["____conn"] = conn
+        self.__dict__["____oid"] = oid
+
+    def __request__(self, handler, *args):
+        raise NotImplementedError()
+
+    def __call__(self, *args, **kwargs):
+        return self.__request__("handle_call", args, ImmDict(kwargs))
+
+    def __delattr__(self, *args):
+        return self.__request__("handle_delattr", *args)
+    def __getattr__(self, *args):
+        return self.__request__("handle_getattr", *args)
+    def __setattr__(self, *args):
+        return self.__request__("handle_setattr", *args)
+    
+    def __delitem__(self, *args):
+        return self.__request__("handle_delitem", *args)
+    def __getitem__(self, *args):
+        return self.__request__("handle_getitem", *args)
+    def __setitem__(self, *args):
+        return self.__request__("handle_setitem", *args)
+    
+    # special cases
+    def __repr__(self, *args):
+        return self.__request__("handle_repr", *args)
+    def __str__(self, *args):
+        return self.__request__("handle_str", *args)
+    def __nonzero__(self, *args):
+        return self.__request__("handle_bool", *args)
+
+class NetProxyWrapper(NetProxy):
+    """a netproxy that wraps an inner netproxy"""
+    __doc__ = NetProxy.__doc__
+
+    def __init__(self, proxy):
+        NetProxy.__init__(self, proxy.__dict__["____conn"], proxy.__dict__["____oid"])
+        # we must keep the original proxy alive
+        self.__dict__["____original_proxy"] = proxy 
+
+def _dummy_callback(*args, **kw):
+    pass
+
+class SyncNetProxy(NetProxy):
+    """the default, synchronous netproxy"""
+    __doc__ = NetProxy.__doc__
+
+    def __init__(self, conn, oid):
+        NetProxy.__init__(self, conn, oid)
+        self.__dict__["____conn"].sync_request("handle_incref", self.__dict__["____oid"])
+
+    def __del__(self):
+        try:
+            # decref'ing is done asynchronously, because we dont need to wait for the remote 
+            # object to die. moreover, we dont care if it fails, because that would mean the 
+            # connection is broken, so the remote object is already dead
+            self.__dict__["____conn"].async_request(_dummy_callback, "handle_decref", self.__dict__["____oid"])
+        except:
+            pass
+    
+    def __request__(self, handler, *args):
+        return self.__dict__["____conn"].sync_request(handler, self.__dict__["____oid"], *args)
+
+
+
+