From c23f61a200023baae5fa9d5f19d6cae4bb662267 Mon Sep 17 00:00:00 2001
From: frankknoll <Knoll_Frank@web.de>
Date: Tue, 22 Nov 2022 17:35:05 +0100
Subject: [PATCH] refactoring

---
 src/HowBadIsMyBatch.ipynb  | 5 ++++-
 src/VAERSFileDownloader.py | 5 +++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/HowBadIsMyBatch.ipynb b/src/HowBadIsMyBatch.ipynb
index 13bfc0a8733..14afaa20aa3 100644
--- a/src/HowBadIsMyBatch.ipynb
+++ b/src/HowBadIsMyBatch.ipynb
@@ -80,7 +80,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "updateVAERSFiles(needsUpdate, getWorkingDirectory())"
+    "updateVAERSFiles(\n",
+    "        needsUpdate,\n",
+    "        years = [2022],\n",
+    "        workingDirectory = getWorkingDirectory())"
    ]
   },
   {
diff --git a/src/VAERSFileDownloader.py b/src/VAERSFileDownloader.py
index 55d9edc4567..3ae21783d97 100644
--- a/src/VAERSFileDownloader.py
+++ b/src/VAERSFileDownloader.py
@@ -56,7 +56,8 @@ def downloadVAERSFileAndUnzip(file, workingDirectory):
         zipFile = downloadedFile,
         dstDir = workingDirectory + '/VAERS/')
 
-def updateVAERSFiles(needsUpdate,  workingDirectory):
+def updateVAERSFiles(needsUpdate,  years, workingDirectory):
     if needsUpdate:
-        downloadVAERSFileAndUnzip('2022VAERSData.zip', workingDirectory)
+        for year in years:
+            downloadVAERSFileAndUnzip(f'{year}VAERSData.zip', workingDirectory)
         downloadVAERSFileAndUnzip('NonDomesticVAERSData.zip', workingDirectory)
\ No newline at end of file