diff --git a/src/intensivstationen/Chart.py b/src/intensivstationen/Chart.py index 0ec26fd2197..ae5ef09c89a 100644 --- a/src/intensivstationen/Chart.py +++ b/src/intensivstationen/Chart.py @@ -1,9 +1,17 @@ from decouple import AutoConfig from Datawrapper import Datawrapper +from ColumnsAdder import ColumnsAdder +from KreiseReader import readKreise +from MedianOfFreeBedsByKreisTableFactory import MedianOfFreeBedsByKreisTableFactory -def createMedianOfFreeBedsByKreisChart(medianOfFreeBedsByKreisTable, chartTitle): - config = AutoConfig(search_path='../..') - dataWrapper = Datawrapper(config('DATAWRAPPER_API_TOKEN')) + +def createMedianOfFreeBedsByKreisChart(timeSeries, chartTitle): + dataWrapper = Datawrapper(AutoConfig(search_path='../..')('DATAWRAPPER_API_TOKEN')) dataWrapper.setChartTitle(chartTitle) - dataWrapper.uploadChartData(medianOfFreeBedsByKreisTable) + dataWrapper.uploadChartData(data = _createMedianOfFreeBedsByKreisTableForChoroplethMap(timeSeries)) dataWrapper.publishChart() + +def _createMedianOfFreeBedsByKreisTableForChoroplethMap(timeSeries): + medianOfFreeBedsByKreisTableFactory = MedianOfFreeBedsByKreisTableFactory(timeSeries) + medianOfFreeBedsByKreisTable = medianOfFreeBedsByKreisTableFactory.createMedianOfFreeBedsByKreisTable('gemeindeschluessel').reset_index() + return ColumnsAdder(readKreise()).addKreisAndEinwohnerzahlColumns(medianOfFreeBedsByKreisTable) diff --git a/src/intensivstationen/Intensivstationen.ipynb b/src/intensivstationen/Intensivstationen.ipynb index 74e259318b9..941883a393a 100644 --- a/src/intensivstationen/Intensivstationen.ipynb +++ b/src/intensivstationen/Intensivstationen.ipynb @@ -83,7 +83,8 @@ "metadata": {}, "outputs": [], "source": [ - "timeSeries = readTimeseries(download = needsUpdate)" + "timeSeries = readTimeseries(download = needsUpdate)\n", + "timeSeries" ] }, { @@ -119,23 +120,10 @@ "metadata": {}, "outputs": [], "source": [ - "medianOfFreeBedsByKreisTableFactory = MedianOfFreeBedsByKreisTableFactory(timeSeries)\n", - "medianOfFreeBedsByKreisTable = medianOfFreeBedsByKreisTableFactory.createMedianOfFreeBedsByKreisTable('Kreis')\n", - "medianOfFreeBedsByKreisTable.reset_index().to_json('../../docs/data/intensivstationen/medianOfFreeBedsByKreisTable.json', orient = \"records\")" - ] - }, - { - "cell_type": "code", - "execution_count": null, - "id": "396cb4f8", - "metadata": {}, - "outputs": [], - "source": [ - "def createMedianOfFreeBedsByKreisTableForChoroplethMap(medianOfFreeBedsByKreisTableFactory):\n", - " medianOfFreeBedsByKreisTable = medianOfFreeBedsByKreisTableFactory.createMedianOfFreeBedsByKreisTable('gemeindeschluessel').reset_index()\n", - " return ColumnsAdder(readKreise()).addKreisAndEinwohnerzahlColumns(medianOfFreeBedsByKreisTable)\n", - " \n", - "medianOfFreeBedsByKreisTable = createMedianOfFreeBedsByKreisTableForChoroplethMap(medianOfFreeBedsByKreisTableFactory)" + "medianOfFreeBedsByKreisTable = MedianOfFreeBedsByKreisTableFactory(timeSeries).createMedianOfFreeBedsByKreisTable('Kreis')\n", + "medianOfFreeBedsByKreisTable.reset_index().to_json(\n", + " '../../docs/data/intensivstationen/medianOfFreeBedsByKreisTable.json',\n", + " orient = \"records\")" ] }, { @@ -160,7 +148,7 @@ "outputs": [], "source": [ "createMedianOfFreeBedsByKreisChart(\n", - " medianOfFreeBedsByKreisTable,\n", + " timeSeries,\n", " chartTitle = getChartTitle(\n", " dateStart=timeSeries['date'].min(),\n", " dateEnd=timeSeries['date'].max()))"