def __init__(self, client):
print("Setting up listener")
self.connection = PostgresConnection(host=os.environ["POSTGRES_HOST"],
dbname=os.environ["POSTGRES_DB"],
user=os.environ["POSTGRES_USER"],
password=os.environ["POSTGRES_PASS"])
print("Connected to Postgres. Now instantiating Postgres Store")
self.store = PostgresStore(self.connection)
self.client = client
def __init__(self, client):
print("Setting up listener")
self.connection = PostgresConnection(
host=os.environ["POSTGRES_HOST"],
dbname=os.environ["POSTGRES_DB"],
user=os.environ["POSTGRES_USER"],
password=os.environ["POSTGRES_PASS"],
)
print("Connected to Postgres. Now instantiating Postgres Store")
self.store = PostgresStore(self.connection)
self.client = client
class Listener:
def __init__(self, client):
print("Setting up listener")
self.connection = PostgresConnection(
host=os.environ["POSTGRES_HOST"],
dbname=os.environ["POSTGRES_DB"],
user=os.environ["POSTGRES_USER"],
password=os.environ["POSTGRES_PASS"],
)
print("Connected to Postgres. Now instantiating Postgres Store")
self.store = PostgresStore(self.connection)
self.client = client
def on_connected(self, headers, body):
print("On Connected")
self.connection.connect()
self.store.create_tables()
def on_message(self, headers, message):
m = json.loads(message.decode("utf-8"))
snapshot = False
if m["message_type"] == "snapshot":
snapshot = True
for s in m["schedule_messages"]:
self.store.save_schedule_message(s, snapshot)
for s in m["association_messages"]:
self.store.save_association_message(s, snapshot)
for d in m["deactivated_messages"]:
self.store.save_deactivated_message(d, snapshot)
for s in m["train_status_messages"]:
self.store.save_train_status_message(s, snapshot)
# Now we have finished processing, ack the message.
self.client.ack(headers)
class Listener:
def __init__(self, client):
print("Setting up listener")
self.connection = PostgresConnection(host=os.environ["POSTGRES_HOST"],
dbname=os.environ["POSTGRES_DB"],
user=os.environ["POSTGRES_USER"],
password=os.environ["POSTGRES_PASS"])
print("Connected to Postgres. Now instantiating Postgres Store")
self.store = PostgresStore(self.connection)
self.client = client
def on_connected(self, headers, body):
print("On Connected")
self.connection.connect()
self.store.create_tables()
def on_message(self, headers, message):
m = json.loads(message.decode("utf-8"))
snapshot = False
if m["message_type"] == "snapshot":
snapshot = True
for s in m["schedule_messages"]:
self.store.save_schedule_message(s, snapshot)
for s in m["association_messages"]:
self.store.save_association_message(s, snapshot)
for d in m["deactivated_messages"]:
self.store.save_deactivated_message(d, snapshot)
for s in m["train_status_messages"]:
self.store.save_train_status_message(s, snapshot)
# Now we have finished processing, ack the message.
self.client.ack(headers)
