run_app.py

import asyncio
import logging
import os
import sys

from uz.interface.telegram import bot
from uz.metrics import statsd
from uz.scanner import UZScanner


logger = logging.getLogger('main')
DEFAULT_LEVEL = logging.WARNING

SCAN_DALAY_SEC = int(os.environ.get('SCAN_DALAY_SEC') or 10)


def get_log_level():
    level = logging.getLevelName(os.environ.get('LOG_LEVEL', '').upper())
    if not isinstance(level, int):
        level = DEFAULT_LEVEL
    return level


def configure_logging(level=None):
    level = level or get_log_level()
    loggers = [
        'main',
        'uz.client',
        'uz.metrics',
        'uz.scanner',
    ]
    handler = logging.StreamHandler(stream=sys.stdout)
    handler.setLevel(level)
    for i in loggers:
        log = logging.getLogger(i)
        log.setLevel(level)
        log.addHandler(handler)


def init_statsd():
    host = os.environ.get('STATSD_HOST')
    port = os.environ.get('STATSD_PORT')
    if host and port:
        statsd.host = host
        statsd.port = int(port)


if __name__ == '__main__':
    configure_logging()
    init_statsd()

    scanner = UZScanner(bot.ticket_booked_cb, SCAN_DALAY_SEC)
    bot.set_scanner(scanner)
    loop = asyncio.get_event_loop()
    loop.create_task(bot.loop())
    loop.create_task(scanner.run())
    logger.warning('Running...')
    try:
        loop.run_forever()
    except KeyboardInterrupt:
        bot.stop()
        scanner.stop()
        logger.warning('Shutting down...')
        logger.warning('Waiting for all tasks to complete...')
        pending = asyncio.Task.all_tasks()
        loop.run_until_complete(asyncio.gather(*pending))
        scanner.cleanup()
        loop.stop()