refill_and_predict_system/app/__main__.py

"""Main app"""

import time
import sys

from app import SCHEDULER, LOGGER, RESOURCE_NAMES, job_storage, jobs


def add_check_resources(state_id, capital_id, resource_id, do_refill, minute):
    """Add check resources job"""
    LOGGER.info('Add check for "%s", resource "%s" at "%s"', state_id, resource_id, minute)
    SCHEDULER.add_job(
        jobs.check_resources,
        'cron',
        args=[state_id, capital_id, resource_id, do_refill],
        id='{}_check_{}'.format(state_id, resource_id),
        replace_existing=True,
        minute=minute
    )

if __name__ == '__main__':
    # jobs.refill_resource(2788, 4002, 0)
    # jobs.check_resources(2788, 4002, 0, False) # VN
    # jobs.check_resources(2620, 4002, 0, False) # Zeelandiae
    # graph()
    # get_resources(4001, datetime.now(), 0)

    JOBS = job_storage.get_jobs()
    for job in JOBS:
        add_check_resources(
            job['state_id'],
            job['capital_id'],
            RESOURCE_NAMES[job['resource_type']],
            job['refill'],
            job['minutes']
        )

    SCHEDULER.add_job(
        jobs.send_telegram_update,
        'cron',
        args=[2788, '@vn_resources', 0],
        id='send_telegram_update',
        replace_existing=True,
        minute='5'
    )

    try:
        while True:
            time.sleep(100)
    except KeyboardInterrupt:
        LOGGER.info('Exiting application')
        SCHEDULER.shutdown()
        sys.exit()