Skip to content

Commit c95d8b7

Browse files
author
yangzhentao
committed
support python 3
1 parent be3cf4d commit c95d8b7

File tree

4 files changed

+18
-8
lines changed

4 files changed

+18
-8
lines changed

stompy/frame.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,12 @@
22
import random
33
from pprint import pformat
44
from errno import EAGAIN, EWOULDBLOCK
5-
from Queue import Queue
6-
from Queue import Empty as QueueEmpty
5+
if sys.version_info[0] < 3:
6+
from Queue import Queue
7+
from Queue import Empty as QueueEmpty
8+
else:
9+
from queue import Queue
10+
from queue import Empty as QueueEmpty
711

812

913
class UnknownBrokerResponseError(Exception):
@@ -268,7 +272,7 @@ def _getline(self, nb=False):
268272
partial = self.sock.recv(1)
269273
if not partial or partial == '':
270274
raise UnknownBrokerResponseError('empty reply')
271-
except socket.error, exc:
275+
except socket.error as exc:
272276
if exc[0] == EAGAIN or exc[0] == EWOULDBLOCK:
273277
if not buffer or buffer == '\n':
274278
raise UnknownBrokerResponseError('empty reply')

stompy/simple.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
from stompy.stomp import Stomp
2-
from Queue import Empty
2+
if sys.version_info[0] < 3:
3+
from Queue import Queue
4+
else:
5+
from queue import Queue
36
from uuid import uuid4
47

58

stompy/stomp.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,9 @@ def connect(self, username=None, password=None, clientid=None):
4646
try:
4747
self.sock.connect((self.host, self.port))
4848
self.frame.connect(self.sock, username=username, password=password, clientid=clientid)
49-
except socket.timeout, exc:
49+
except socket.timeout as exc:
5050
raise self.ConnectionTimeoutError(*exc.args)
51-
except socket.error, exc:
51+
except socket.error as exc:
5252
raise self.ConnectionError(*exc.args)
5353
self.connected = True
5454

@@ -63,7 +63,7 @@ def disconnect(self, conf=None):
6363
try:
6464
self.sock.shutdown(0)
6565
self.sock.close()
66-
except socket.error, exc:
66+
except socket.error as exc:
6767
# likely wasn't connected
6868
pass
6969
self.connected = False

tests/test_frame.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,10 @@
66
from stompy import frame
77
from stompy.frame import Frame, IntermediateMessageQueue, \
88
UnknownBrokerResponseError, BrokerErrorResponse
9-
from Queue import Empty as QueueEmpty
9+
if sys.version_info[0] < 3:
10+
from Queue import Empty as QueueEmpty
11+
else:
12+
from queue import Empty as QueueEmpty
1013

1114

1215
class WhenSettingUp(DingusTestCase(Frame)):

0 commit comments

Comments
 (0)