From 7b6d6198815d248bf1fdd4b97fc442a22fcf35ac Mon Sep 17 00:00:00 2001
From: frankknoll <Knoll_Frank@web.de>
Date: Mon, 31 Jan 2022 10:01:11 +0100
Subject: [PATCH] adding method saveSevereEffectsBatchCodeTable()

---
 HowBadIsMyBatch.ipynb | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/HowBadIsMyBatch.ipynb b/HowBadIsMyBatch.ipynb
index 7a636cd246d..87410edc323 100644
--- a/HowBadIsMyBatch.ipynb
+++ b/HowBadIsMyBatch.ipynb
@@ -86,6 +86,13 @@
     "    return createAndFilterDataFrameFromDescrs(\n",
     "        [readVaersDescr(dataDir, \"2021\"), readVaersDescr(dataDir, \"2022\")],\n",
     "        manufacturer,\n",
+    "        dose)\n",
+    "\n",
+    "# FK-TODO: DRY with createAndFilterDataFrameFromFiles()\n",
+    "def createDataFrameSevereEffectsFromFiles(dataDir, dose):\n",
+    "    return createDataFrameSevereEffectsFromDescrs(\n",
+    "        # FK-TODO: reactivate: [readVaersDescr(dataDir, \"2021\"), readVaersDescr(dataDir, \"2022\")],\n",
+    "        [readVaersDescr(dataDir, \"2022\")],\n",
     "        dose)"
    ]
   },
@@ -475,6 +482,29 @@
     "saveBatchCodeTable(\"PFIZER\\BIONTECH\", \"results/pfizer.xlsx\")\n",
     "saveBatchCodeTable(\"JANSSEN\", \"results/janssen.xlsx\")"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "bc56831d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def saveSevereEffectsBatchCodeTable(excelFile):\n",
+    "    severeEffectsBatchCodeTable = createSevereEffectsBatchCodeTable(createDataFrameSevereEffectsFromFiles(\"VAERS\", '1'))\n",
+    "    display('severeEffectsBatchCodeTable', severeEffectsBatchCodeTable)\n",
+    "    severeEffectsBatchCodeTable.to_excel(excelFile)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ace3fed9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "saveSevereEffectsBatchCodeTable('results/severeEffects.xlsx')"
+   ]
   }
  ],
  "metadata": {