factory_activity_and_growth_management/app/__main__.py

"""Main app"""

import time

from app import scheduler, session, LOGGER
from app.api import get_factories
from app.database import get_state, save_factories


def print_factories(factories):
    """Print professors"""
    for factory in factories:
        print('{:20} {:30}'.format(
            factory['id'],
            factory['name'],
        ))

def job_update_factories(state_id):
    """Update factories"""
    LOGGER.info('Run update factories for state "%s"', state_id)
    state = get_state(state_id)
    LOGGER.info('"%s": get factories', state.name)
    factories = get_factories(state.id)
    LOGGER.info('"%s": "%s" factories', state.name, len(factories))
    print_factories(factories)
    # save_factories(region.id, factories)
    LOGGER.info('"%s": done saving factories', state.name)


def add_update_factories(state_id):
    """Add jobs"""
    scheduler.add_job(
        job_update_factories,
        'cron',
        args=[state_id],
        id='factories_{}'.format(state_id),
        replace_existing=True,
        hour='2'
    )

if __name__ == '__main__':
    # jobs
    # job_update_factories(2788)

    # Verenigde Nederlanden
    add_update_factories(2788)
    # Belgium
    add_update_factories(2604)
    # De Provincien
    add_update_factories(2620)

    try:
        while True:
            time.sleep(100)
    except KeyboardInterrupt:
        print('Exiting application')
        session.close()
        exit()