diff --git a/src/intensivstationen/Intensivstationen.ipynb b/src/intensivstationen/Intensivstationen.ipynb index 3f4e5bcbdea..365758ee0c2 100644 --- a/src/intensivstationen/Intensivstationen.ipynb +++ b/src/intensivstationen/Intensivstationen.ipynb @@ -53,8 +53,9 @@ "\n", "def _getLastUpdatedOriginal():\n", " html = _getOriginalHtml()\n", - " dataFrame = _asDataFrame(html)\n", - " return dataFrame.loc['Landkreis-Daten', 'Letzte Änderung'].to_pydatetime()\n", + " lastUpdatedColumn = 'Letzte Änderung'\n", + " dataFrame = _asDataFrame(html, lastUpdatedColumn)\n", + " return dataFrame.loc['Landkreis-Daten', lastUpdatedColumn].to_pydatetime()\n", "\n", "def _getOriginalHtml():\n", " driver = webdriver.Firefox()\n", @@ -64,10 +65,10 @@ " driver.quit()\n", " return innerHTML\n", "\n", - "def _asDataFrame(html):\n", - " dataFrame = pd.read_html(html, parse_dates = ['Letzte Änderung'])[0]\n", + "def _asDataFrame(html, lastUpdatedColumn):\n", + " dataFrame = pd.read_html(html, parse_dates = [lastUpdatedColumn])[0]\n", + " dataFrame[lastUpdatedColumn] = pd.to_datetime(dataFrame[lastUpdatedColumn], format = \"%d.%m.%Y %H:%M Uhr\")\n", " dataFrame.set_index('Name', inplace = True)\n", - " dataFrame['Letzte Änderung'] = pd.to_datetime(dataFrame['Letzte Änderung'], format = \"%d.%m.%Y %H:%M Uhr\")\n", " return dataFrame\n", "\n", "needsUpdate = needsUpdate()\n",