41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
"""
|
|
SleekXMPP: The Sleek XMPP Library
|
|
Copyright (C) 2010 Nathanael C. Fritz
|
|
This file is part of SleekXMPP.
|
|
|
|
See the file LICENSE for copying permission.
|
|
"""
|
|
|
|
from socket import _fileobject
|
|
import socket
|
|
|
|
|
|
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)
|