import logging import os from datetime import datetime import pandas as pd from country_codes import get_county_code from utils import sha1sum, write_timestamp_file, download_nijz_xslx_file logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__file__) covid_data_path = os.getenv('COVID_DATA_PATH') assert covid_data_path, 'COVID_DATA_PATH env variable must be set. (The location of the COVID-DATA folder)' download_nijz_xslx_file(download_folder=os.path.join(covid_data_path, 'EPI'), search_for='tedenski_prikaz_okuzeni') download_nijz_xslx_file(download_folder=os.path.join(covid_data_path, 'EPI'), search_for='tedenski_prikaz_umrli') SOURCE_FILE_INFECTED = max( glob.glob( os.path.join(covid_data_path, 'EPI') + '/tedenski_prikaz_okuzeni*.xlsx')) # take latest logger.info(f'SOURCE_FILE okuzeni: {SOURCE_FILE_INFECTED}') SOURCE_FILE_DECEASED = max( glob.glob( os.path.join(covid_data_path, 'EPI') + '/tedenski_prikaz_umrli*.xlsx')) # take latest logger.info(f'SOURCE_FILE umrli: {SOURCE_FILE_DECEASED}') CSV_FOLDER = os.path.join(os.path.dirname(__file__), '../csv')
import logging import os from datetime import datetime, timedelta import pandas as pd from utils import sha1sum, write_timestamp_file, download_nijz_xslx_file logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__file__) covid_data_path = os.getenv('COVID_DATA_PATH') assert covid_data_path, 'COVID_DATA_PATH env variable must be set. (The location of the COVID-DATA folder)' download_nijz_xslx_file(download_folder=os.path.join(covid_data_path, 'EPI'), search_for='dnevni_prikazi') SOURCE_FILE = max(glob.glob(os.path.join(covid_data_path, 'EPI') + '/dnevni_prikazi*.xlsx')) # take latest logger.info(f'SOURCE_FILE: {SOURCE_FILE}') CSV_FOLDER = os.path.join(os.path.dirname(__file__), '../csv') def export_dataframe_to_csv(name: str, dataframe): filename = os.path.join(CSV_FOLDER, f'{name}.csv') old_hash = sha1sum(filename) dataframe.replace({0: None}).astype('Int64').to_csv(filename, line_terminator='\r\n') write_timestamp_file(filename=filename, old_hash=old_hash) municipalities = {} with open(file=os.path.join(CSV_FOLDER, 'dict-municipality.csv'), encoding='utf-8') as f: