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