covid19.py

from platypush.event.hook import hook
from platypush.message.event.covid19 import Covid19UpdateEvent
from platypush.utils import run


@hook(Covid19UpdateEvent)
def on_covid19_update(event, **context):
    db_engine = 'postgresql+pg8000://user:password@db_ip_address/covid19'
    run('db.insert', table='tmp_covid19_data', engine=db_engine, records=[
        {
            'country': event.args['country'],
            'confirmed': event.args['confirmed'] or 0,
            'deaths': event.args['deaths'] or 0,
            'recovered': event.args['recovered'] or 0,
            'report_date': event.args['update_time'],
        }
    ])