refactoring

This commit is contained in:
frankknoll
2022-03-03 13:41:54 +01:00
parent f452682208
commit 824be3fcad

View File

@@ -226,30 +226,30 @@
"import json\n",
"\n",
"\n",
"def getAndPersistIntensiveCareBeds(timeSeries, kreis=None):\n",
" dataFrame = getIntensiveCareBeds(timeSeries, kreis)\n",
"def getAndPersistIntensiveCareBeds(timeSeries, kreis = None):\n",
" intensiveCareBeds = getIntensiveCareBeds(timeSeries, kreis)\n",
" display(kreis)\n",
" _saveDataFrameAsJson(dataFrame, _getFilename(kreis))\n",
" return dataFrame\n",
" _saveAsJson(intensiveCareBeds, _getFilename(kreis))\n",
" return intensiveCareBeds\n",
"\n",
"\n",
"def _saveDataFrameAsJson(dataFrame, file):\n",
"def _saveAsJson(intensiveCareBeds, file):\n",
" IOUtils.saveDictAsJson(\n",
" {\n",
" 'population': int(dataFrame.iloc[0]['Einwohnerzahl']),\n",
" 'data': _dataFrame2Dict(dataFrame),\n",
" 'population': int(intensiveCareBeds.iloc[0]['Einwohnerzahl']),\n",
" 'data': _intensiveCareBeds2Dict(intensiveCareBeds),\n",
" },\n",
" file)\n",
"\n",
"\n",
"def _dataFrame2Dict(dataFrame):\n",
" df = dataFrame[['date', 'betten_belegt', 'betten_frei']]\n",
"def _intensiveCareBeds2Dict(intensiveCareBeds):\n",
" df = intensiveCareBeds[['date', 'betten_belegt', 'betten_frei']]\n",
" df['date'] = df['date'].dt.strftime('%Y-%m-%d')\n",
" return df.to_dict(orient=\"records\")\n",
" return df.to_dict(orient = \"records\")\n",
"\n",
"\n",
"def _getFilename(kreis):\n",
" return '../../docs/data/intensivstationen/intensivstationen-{suffix}.json'.format(suffix=_getSuffix(kreis))\n",
" return '../../docs/data/intensivstationen/intensivstationen-{suffix}.json'.format(suffix = _getSuffix(kreis))\n",
"\n",
"\n",
"def _getSuffix(kreis):\n",