mirror of
https://github.com/correl/SleekXMPP.git
synced 2024-11-27 19:19:54 +00:00
Updated xmlstream.filesocket.
This commit is contained in:
parent
a2c515bc97
commit
6677df39f2
2 changed files with 26 additions and 12 deletions
|
@ -5,21 +5,36 @@
|
|||
|
||||
See the file LICENSE for copying permission.
|
||||
"""
|
||||
|
||||
from socket import _fileobject
|
||||
import socket
|
||||
|
||||
class filesocket(_fileobject):
|
||||
class FileSocket(_fileobject):
|
||||
|
||||
"""
|
||||
Create a file object wrapper for a socket to work around
|
||||
issues present in Python 2.6 when using sockets as file objects.
|
||||
|
||||
The parser for xml.etree.cElementTree requires a file, but we will
|
||||
be reading from the XMPP connection socket instead.
|
||||
"""
|
||||
|
||||
def read(self, size=4096):
|
||||
"""Read data from the socket as if it were a file."""
|
||||
data = self._sock.recv(size)
|
||||
if data is not None:
|
||||
return data
|
||||
|
||||
class Socket26(socket._socketobject):
|
||||
|
||||
"""
|
||||
A custom socket implementation that uses our own FileSocket class
|
||||
to work around issues in Python 2.6 when using sockets as files.
|
||||
"""
|
||||
|
||||
def makefile(self, mode='r', bufsize=-1):
|
||||
"""makefile([mode[, bufsize]]) -> file object
|
||||
Return a regular file object corresponding to the socket. The mode
|
||||
and bufsize arguments are as for the built-in open() function."""
|
||||
return filesocket(self._sock, mode, bufsize)
|
||||
return FileSocket(self._sock, mode, bufsize)
|
||||
|
||||
|
|
|
@ -139,8 +139,7 @@ class XMLStream(object):
|
|||
self.socket = ssl.wrap_socket(self.socket, ssl_version=ssl.PROTOCOL_TLSv1, do_handshake_on_connect=False)
|
||||
self.socket.do_handshake()
|
||||
if sys.version_info < (3,0):
|
||||
from . filesocket import filesocket
|
||||
self.filesocket = filesocket(self.socket)
|
||||
self.filesocket = filesocket.FileSocket(self.socket)
|
||||
else:
|
||||
self.filesocket = self.socket.makefile('rb', 0)
|
||||
return True
|
||||
|
|
Loading…
Reference in a new issue