refactoring

This commit is contained in:
frankknoll
2022-05-13 20:43:53 +02:00
parent f01cad8c5d
commit 8382945e18
2 changed files with 9 additions and 5 deletions

1
.gitignore vendored
View File

@@ -10,3 +10,4 @@ src/results/
docs/data/*.xlsx docs/data/*.xlsx
docs/data/*.html docs/data/*.html
.env .env
src/captchaImage.jpeg

View File

@@ -169,7 +169,7 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"import time\n", "import time\n",
"import os.path\n", "import os\n",
"\n", "\n",
"def waitUntilDownloadHasFinished(file):\n", "def waitUntilDownloadHasFinished(file):\n",
" while not os.path.exists(file):\n", " while not os.path.exists(file):\n",
@@ -183,6 +183,8 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"import os\n",
"\n",
"def getTextInCaptchaImage(captchaImageFile):\n", "def getTextInCaptchaImage(captchaImageFile):\n",
" baseDir = \"~/AndroidStudioProjects/TextRecognizer\"\n", " baseDir = \"~/AndroidStudioProjects/TextRecognizer\"\n",
" ! cp $captchaImageFile $baseDir/app/src/main/assets/captchas/captcha_image.jpeg\n", " ! cp $captchaImageFile $baseDir/app/src/main/assets/captchas/captcha_image.jpeg\n",
@@ -197,10 +199,9 @@
" driver.find_element(By.ID, \"verificationCode\").send_keys(textInCaptchaImage)\n", " driver.find_element(By.ID, \"verificationCode\").send_keys(textInCaptchaImage)\n",
" driver.find_element(By.CSS_SELECTOR, '[name=\"downloadbut\"]').click()\n", " driver.find_element(By.CSS_SELECTOR, '[name=\"downloadbut\"]').click()\n",
"\n", "\n",
"# FK-TODO: file is part of absoluteFile, so remove file\n", "def downloadFile(absoluteFile, driver, maxTries):\n",
"def downloadFile(file, absoluteFile, driver, maxTries):\n",
" def _downloadFile():\n", " def _downloadFile():\n",
" driver.get('https://vaers.hhs.gov/eSubDownload/index.jsp?fn=' + file)\n", " driver.get('https://vaers.hhs.gov/eSubDownload/index.jsp?fn=' + os.path.basename(absoluteFile))\n",
" solveCaptchaAndStartFileDownload(driver, 'captchaImage.jpeg')\n", " solveCaptchaAndStartFileDownload(driver, 'captchaImage.jpeg')\n",
"\n", "\n",
" numTries = 1\n", " numTries = 1\n",
@@ -218,7 +219,6 @@
"def downloadVAERSFile(file, downloadDir):\n", "def downloadVAERSFile(file, downloadDir):\n",
" driver = getWebDriver(downloadDir, isHeadless = False)\n", " driver = getWebDriver(downloadDir, isHeadless = False)\n",
" downloadedFile = downloadFile(\n", " downloadedFile = downloadFile(\n",
" file = file,\n",
" absoluteFile = downloadDir + \"/\" + file,\n", " absoluteFile = downloadDir + \"/\" + file,\n",
" driver = driver,\n", " driver = driver,\n",
" maxTries = None)\n", " maxTries = None)\n",
@@ -1203,6 +1203,9 @@
} }
], ],
"metadata": { "metadata": {
"interpreter": {
"hash": "31f2aee4e71d21fbe5cf8b01ff0e069b9275f58929596ceb00d14d90e3e16cd6"
},
"kernelspec": { "kernelspec": {
"display_name": "Python 3.10.4 64-bit", "display_name": "Python 3.10.4 64-bit",
"language": "python", "language": "python",