refactoring
This commit is contained in:
13
src/HistogramPersister.py
Normal file
13
src/HistogramPersister.py
Normal file
@@ -0,0 +1,13 @@
|
||||
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}'))
|
||||
Reference in New Issue
Block a user