Skip to content

Commit

Permalink
Basic Python 3 compat
Browse files Browse the repository at this point in the history
  • Loading branch information
jonnor committed Nov 13, 2017
1 parent 24c05b2 commit b68b1d9
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 5 deletions.
2 changes: 1 addition & 1 deletion msgflo/__init__.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@

from msgflo import *
from .msgflo import *
13 changes: 9 additions & 4 deletions msgflo/msgflo.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,13 @@
#!/usr/bin/env python

import sys, os, json, random, urlparse
import sys, os, json, random
sys.path.append(os.path.abspath("."))

try:
from urllib.parse import urlparse
except ImportError:
from urlparse import urlparse

from optparse import OptionParser

import logging
Expand Down Expand Up @@ -71,7 +76,7 @@ def nack(self, msg):
class Engine(object):
def __init__(self, broker, discovery_period=None):
self.broker_url = broker
self.broker_info = urlparse.urlparse(self.broker_url)
self.broker_info = urlparse(self.broker_url)
self.discovery_period = discovery_period if discovery_period else DEFAULT_DISCOVERY_PERIOD

def done_callback(self, done_cb):
Expand Down Expand Up @@ -338,7 +343,7 @@ def run(participant, broker=None, done_cb=None, iips={}):
broker = os.environ.get('MSGFLO_BROKER', 'amqp://localhost')

engine = None
broker_info = urlparse.urlparse(broker)
broker_info = urlparse(broker)
if broker_info.scheme == 'amqp':
engine = AmqpEngine(broker)
elif broker_info.scheme == 'mqtt':
Expand Down Expand Up @@ -379,6 +384,6 @@ def main(Participant, role=None):
waiter = gevent.event.AsyncResult()
engine = run(participant, done_cb=waiter.set, iips=config.iips)
anon_url = "%s://%s" % (engine.broker_info.scheme, engine.broker_info.hostname)
print "%s(%s) running on %s" % (d['role'], d['component'], anon_url)
print("%s(%s) running on %s" % (d['role'], d['component'], anon_url))
sys.stdout.flush()
waiter.wait()

0 comments on commit b68b1d9

Please sign in to comment.