Source code for cowidev.vax.us_states.__main__

import os

from cowidev import PATHS

from cowidev.vax.us_states.etl import run_etl
from cowidev.vax.us_states.grapher import run_grapheriser
from cowidev.vax.us_states._parser import _parse_args


FILE_DS = PATHS.DATA_VAX_US_FILE
FILE_GRAPHER = os.path.join(PATHS.INTERNAL_GRAPHER_DIR, "COVID-19 - United States vaccinations.csv")




[docs]
def run_step(step: str):
    if step == "etl":
        run_etl(FILE_DS)
    elif step == "grapher-file":
        run_grapheriser(FILE_DS, FILE_GRAPHER)


    # elif step == "explorer-file":
    #     run_explorerizer(FILE_DS, FILE_EXPLORER)
    # elif step == "grapher-db":
    #     run_db_updater(FILE_GRAPHER)


if __name__ == "__main__":
    args = _parse_args()
    run_step(args.step)