refactoring

src/HistogramFactoryAndPersister.py

@@ -0,0 +1,24 @@
+from HistogramFactory import createGlobalHistograms, createHistograms
+from HistogramPersister import saveHistograms
+
+
+def createAndSaveGlobalHistograms(symptomByBatchcodeTable):
+    dictByBatchcodeTable4Country = createGlobalHistograms(symptomByBatchcodeTable)
+    saveHistograms(dictByBatchcodeTable4Country, 'Global')
+
+
+def createAndSaveHistogramsForCountries(symptomByBatchcodeTable, countries):
+    dictByBatchcodeTable = createHistograms(symptomByBatchcodeTable)
+    for count, country in enumerate(countries, start = 1):
+        _createAndSaveHistogramsForCountry(
+            count = count,
+            numCountries = len(countries),
+            country = country,
+            dictByBatchcodeTable = dictByBatchcodeTable)
+
+
+def _createAndSaveHistogramsForCountry(count, numCountries, country, dictByBatchcodeTable):
+    # FK-TODO: use https://github.com/tqdm/tqdm
+    print(f'saving histograms for country {count}/{numCountries}: {country}')
+    dictByBatchcodeTable4Country = dictByBatchcodeTable[dictByBatchcodeTable['COUNTRY'] == country]
+    saveHistograms(dictByBatchcodeTable4Country, country)