96ed53e2f78b31e8f133e4048fc711222b6708ae
[monitor.git] / Rpyc / NetProxy.py
1 from Lib import ImmDict\r
2 \r
3 \r
4 def fully_dynamic_metaclass(name, bases, dict):\r
5     """\r
6     a meta class that enables special methods to be accessed like regular names \r
7     (via __getattr__), like it used to be in old-style classes.\r
8     """\r
9 \r
10     special_methods = [\r
11         "__hash__", "__len__", "__iter__", "next", "__reversed__",\r
12         "__add__", "__iadd__", "__radd__", "__sub__", "__isub__", "__rsub__", "__mul__",\r
13         "__imul__", "__rmul__", "__div__", "__idiv__", "__rdiv__", "__truediv__", \r
14         "__itruediv__", "__rtruediv__",  "__floordiv__", "__ifloordiv__", "__rfloorfiv__", \r
15         "__pow__", "__ipow__", "__rpow__", "__lshift__", "__ilshift__", "__rlshift__",\r
16         "__rshift__", "__irshift__", "__rrshift__", "__and__", "__iand__", "__rand__",\r
17         "__or__", "__ior__", "__ror__", "__xor__", "__ixor__", "__rxor__", "__mod__", \r
18         "__imod__", "__rmod__", "__divmod__", "__idivmod__", "__rdivmod__", "__pos__", \r
19         "__neg__", "__int__", "__float__", "__long__", "__oct__", "__hex__", "__coerce__",\r
20         "__eq__", "__ne__", "__le__", "__ge__", "__lt__", "__gt__", "__cmp__",\r
21     ]\r
22     \r
23     # i added '__class__' to the special attributes, but it broke some things \r
24     # (like dir), so we'll have to live without it\r
25     special_attributes = ["__doc__", "__module__"] \r
26 \r
27     def make_method(name):\r
28         def caller(self, *a, **k):\r
29             return self.__getattr__(name)(*a, **k)\r
30         return caller\r
31 \r
32     def make_property(name):\r
33         def getter(self):\r
34             return self.__getattr__(name)\r
35         def setter(self, value):\r
36             self.__setattr__(name, value)\r
37         def deller(self):\r
38             self.__delattr__(name)\r
39         return property(getter, setter, deller)\r
40 \r
41     classdict = {}\r
42     for sn in special_methods:\r
43         classdict[sn] = make_method(sn)\r
44     classdict.update(dict)\r
45     for sa in special_attributes:\r
46         classdict[sa] = make_property(sa)\r
47     return type(name, bases, classdict)\r
48 \r
49 class NetProxy(object):\r
50     """NetProxy - convey local operations to the remote object. this is an abstract class"""\r
51     __metaclass__ = fully_dynamic_metaclass\r
52     \r
53     def __init__(self, conn, oid):\r
54         self.__dict__["____conn"] = conn\r
55         self.__dict__["____oid"] = oid\r
56 \r
57     def __request__(self, handler, *args):\r
58         raise NotImplementedError()\r
59 \r
60     def __call__(self, *args, **kwargs):\r
61         return self.__request__("handle_call", args, ImmDict(kwargs))\r
62 \r
63     def __delattr__(self, *args):\r
64         return self.__request__("handle_delattr", *args)\r
65     def __getattr__(self, *args):\r
66         return self.__request__("handle_getattr", *args)\r
67     def __setattr__(self, *args):\r
68         return self.__request__("handle_setattr", *args)\r
69     \r
70     def __delitem__(self, *args):\r
71         return self.__request__("handle_delitem", *args)\r
72     def __getitem__(self, *args):\r
73         return self.__request__("handle_getitem", *args)\r
74     def __setitem__(self, *args):\r
75         return self.__request__("handle_setitem", *args)\r
76     \r
77     # special cases\r
78     def __repr__(self, *args):\r
79         return self.__request__("handle_repr", *args)\r
80     def __str__(self, *args):\r
81         return self.__request__("handle_str", *args)\r
82     def __nonzero__(self, *args):\r
83         return self.__request__("handle_bool", *args)\r
84 \r
85 class NetProxyWrapper(NetProxy):\r
86     """a netproxy that wraps an inner netproxy"""\r
87     __doc__ = NetProxy.__doc__\r
88 \r
89     def __init__(self, proxy):\r
90         NetProxy.__init__(self, proxy.__dict__["____conn"], proxy.__dict__["____oid"])\r
91         # we must keep the original proxy alive\r
92         self.__dict__["____original_proxy"] = proxy \r
93 \r
94 def _dummy_callback(*args, **kw):\r
95     pass\r
96 \r
97 class SyncNetProxy(NetProxy):\r
98     """the default, synchronous netproxy"""\r
99     __doc__ = NetProxy.__doc__\r
100 \r
101     def __init__(self, conn, oid):\r
102         NetProxy.__init__(self, conn, oid)\r
103         self.__dict__["____conn"].sync_request("handle_incref", self.__dict__["____oid"])\r
104 \r
105     def __del__(self):\r
106         try:\r
107             # decref'ing is done asynchronously, because we dont need to wait for the remote \r
108             # object to die. moreover, we dont care if it fails, because that would mean the \r
109             # connection is broken, so the remote object is already dead\r
110             self.__dict__["____conn"].async_request(_dummy_callback, "handle_decref", self.__dict__["____oid"])\r
111         except:\r
112             pass\r
113     \r
114     def __request__(self, handler, *args):\r
115         return self.__dict__["____conn"].sync_request(handler, self.__dict__["____oid"], *args)\r
116 \r
117 \r
118 \r
119 \r