Files
HowBadIsMyBatch/src/HistogramPersister.py
frankknoll f12519d1b8 refactoring
2023-02-12 20:54:17 +01:00

14 lines
711 B
Python

from MultiIndexValuesProvider import MultiIndexValuesProvider
from HistogramDescriptionPersister import HistogramDescriptionPersister
import shutil
def saveHistograms(dictByBatchcodeTable4Country, country):
batchcodes = MultiIndexValuesProvider.getValues(dictByBatchcodeTable4Country.index)
batchcodes = {batchcode for batchcode in batchcodes if batchcode != 'nan'}
directory = f'../docs/data/histograms/{country}'
shutil.rmtree(directory, ignore_errors = True)
HistogramDescriptionPersister(directory).saveHistogramDescriptionsForBatchcodes(
batchcodes,
dictByBatchcodeTable4Country,
progress = lambda count, size, batchcode: print(f'{count}/{size}: {batchcode}'))