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 ###############################################################################
32 class Myexception(Exception):
33 def __init__(self, faultCode, faultString):
34 self.faultCode = faultCode
35 self.faultString = faultString
36 self.args = (faultCode, faultString)
39 def __init__(self, sock=None):
41 self.sock = socket.socket(
42 socket.AF_INET, socket.SOCK_STREAM)
45 self.sock.settimeout(120)
46 def connect(self, host, port=False):
48 protocol, buf = host.split('//')
49 host, port = buf.split(':')
50 self.sock.connect((host, int(port)))
52 self.sock.shutdown(socket.SHUT_RDWR)
54 def mysend(self, msg, exception=False, traceback=None):
55 msg = cPickle.dumps([msg,traceback])
57 self.sock.send('%8d' % size)
58 self.sock.send(exception and "1" or "0")
60 while totalsent < size:
61 sent = self.sock.send(msg[totalsent:])
63 raise RuntimeError, "socket connection broken"
64 totalsent = totalsent + sent
68 chunk = self.sock.recv(8 - len(buf))
70 raise RuntimeError, "socket connection broken"
73 buf = self.sock.recv(1)
79 while len(msg) < size:
80 chunk = self.sock.recv(size-len(msg))
82 raise RuntimeError, "socket connection broken"
84 res = cPickle.loads(msg)
85 if isinstance(res[0],Exception):
87 raise Myexception(str(res[0]), str(res[1]))