diff options
Diffstat (limited to 'Lib/SocketServer.py')
| -rw-r--r-- | Lib/SocketServer.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/Lib/SocketServer.py b/Lib/SocketServer.py index 06ed134..345a6d1 100644 --- a/Lib/SocketServer.py +++ b/Lib/SocketServer.py @@ -575,10 +575,13 @@ class DatagramRequestHandler(BaseRequestHandler): """Define self.rfile and self.wfile for datagram sockets.""" def setup(self): - import StringIO + try: + from cStringIO import StringIO + except ImportError: + from StringIO import StringIO self.packet, self.socket = self.request - self.rfile = StringIO.StringIO(self.packet) - self.wfile = StringIO.StringIO() + self.rfile = StringIO(self.packet) + self.wfile = StringIO() def finish(self): self.socket.sendto(self.wfile.getvalue(), self.client_address) |