1 ##############################################################################
3 # Copyright (c) 2004-2008 Tiny SPRL (http://tiny.be) All Rights Reserved.
7 # WARNING: This program as such is intended to be used by professional
8 # programmers who take the whole responsability of assessing all potential
9 # consequences resulting from its eventual inadequacies and bugs
10 # End users who are looking for a ready-to-use solution with commercial
11 # garantees and support are strongly adviced to contract a Free Software
14 # This program is Free Software; you can redistribute it and/or
15 # modify it under the terms of the GNU General Public License
16 # as published by the Free Software Foundation; either version 2
17 # of the License, or (at your option) any later version.
19 # This program is distributed in the hope that it will be useful,
20 # but WITHOUT ANY WARRANTY; without even the implied warranty of
21 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 # GNU General Public License for more details.
24 # You should have received a copy of the GNU General Public License
25 # along with this program; if not, write to the Free Software
26 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
27 ###############################################################################
34 class Myexception(Exception):
35 def __init__(self
, faultCode
, faultString
):
36 self
.faultCode
= faultCode
37 self
.faultString
= faultString
38 self
.args
= (faultCode
, faultString
)
41 def __init__(self
, sock
=None):
43 self
.sock
= socket
.socket(
44 socket
.AF_INET
, socket
.SOCK_STREAM
)
47 self
.sock
.settimeout(120)
48 def connect(self
, host
, port
=False):
50 protocol
, buf
= host
.split('//')
51 host
, port
= buf
.split(':')
53 host
= DNS_CACHE
[host
]
54 self
.sock
.connect((host
, int(port
)))
55 DNS_CACHE
[host
], port
= self
.sock
.getpeername()
57 self
.sock
.shutdown(socket
.SHUT_RDWR
)
59 def mysend(self
, msg
, exception
=False, traceback
=None):
60 msg
= cPickle
.dumps([msg
,traceback
])
62 self
.sock
.send('%8d' % size
)
63 self
.sock
.send(exception
and "1" or "0")
65 while totalsent
< size
:
66 sent
= self
.sock
.send(msg
[totalsent
:])
68 raise RuntimeError, "socket connection broken"
69 totalsent
= totalsent
+ sent
73 chunk
= self
.sock
.recv(8 - len(buf
))
75 raise RuntimeError, "socket connection broken"
78 buf
= self
.sock
.recv(1)
84 while len(msg
) < size
:
85 chunk
= self
.sock
.recv(size
-len(msg
))
87 raise RuntimeError, "socket connection broken"
89 res
= cPickle
.loads(msg
)
90 if isinstance(res
[0],Exception):
92 raise Myexception(str(res
[0]), str(res
[1]))