consumer-scheduler/app.py

#!/usr/bin/env python3
import sentry_sdk
import time
import os
import logging

from redis_super_storage import RedisSuperStorage
from communicators import ConsumerCommunicator, ProducerCommunicator
from ip_watchdog import IPWatchdog

"""
Scheduler
"""

__author__ = "@kocsisr"
__copyright__ = "Copyright 2020, GoldenPogácsa Team"
__module_name__ = "app"
__version__text__ = "1"

sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10")


def get_initial_ip_list():
    ip_list = os.environ['INITIAL_SERVERS'].split(',')
    logging.debug('Initial ip list ' + ", ".join(ip_list))
    return ip_list


def main():
    # set logging preferences
    logging.basicConfig(filename='', level=logging.DEBUG)

    redis_storage = RedisSuperStorage(os.environ.get('REDIS_URL', "redis://localhost:6379/0"), 5)
    consumer_communicator = ConsumerCommunicator(redis_storage)
    producer_communicator = ProducerCommunicator(redis_storage)
    ip_watchdog = IPWatchdog(redis_storage)

    logging.info("Syncing to initial consumer list")
    for ip in get_initial_ip_list():
        logging.debug(f"Syncing to {ip}")
        consumer_communicator.targeted_snyc(ip)

    while True:
        logging.debug("Doing a sync")
        consumer_communicator.sync_all()

        ip_changed, ipaddr = ip_watchdog.ip_changed()
        if ip_changed:
            producer_communicator.push_ip_update(ipaddr)

        time.sleep(int(os.environ.get("RUN_INTERVAL", 10)))


if __name__ == "__main__":
    try:

        main()
    except KeyboardInterrupt:
        pass