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)