user = get_user_data(msg.uid, session)
if user is None:
logging.warning(
"Couldn't find user details for message %s", msg.uid)
return
print "send_msg(sns_client, %s, %s)" % (user.contact, msg.text)
if __name__ == "__main__":
try:
CASSANDRA_HOST = os.environ['CASSANDRA_HOST']
except KeyError as err:
print(err, "Please set CASSANDRA_HOST environment variable")
raise
try:
KAFKA_HOST = os.environ['KAFKA_HOST']
except KeyError as err:
print(err, "Please set KAFKA_HOST environment variable")
raise
cluster = Cluster([CASSANDRA_HOST])
session = cluster.connect('users')
consumer = Consumer(KAFKA_HOST, '9092')
sns_client = boto3.client("sns", region_name='us-west-2')
for kafka_msg in consumer.messages('test-topic'):
kafka_message_handler(session, sns_client, kafka_msg)
