refactoring

This commit is contained in:
frankknoll
2022-05-13 00:13:53 +02:00
parent 3dfc99f4c7
commit b057b1c836

View File

@@ -99,18 +99,26 @@
"from selenium.webdriver.chrome.options import Options\n", "from selenium.webdriver.chrome.options import Options\n",
"from selenium.webdriver.common.by import By\n", "from selenium.webdriver.common.by import By\n",
"\n", "\n",
"options = Options()\n",
"downloadDir = \"/home/frankknoll/Dokumente/Corona/projects/HowBadIsMyBatch/src/VAERS/tmp\"\n", "downloadDir = \"/home/frankknoll/Dokumente/Corona/projects/HowBadIsMyBatch/src/VAERS/tmp\"\n",
"prefs = {\"download.default_directory\" : downloadDir}\n",
"options.add_experimental_option(\"prefs\", prefs)\n",
"service = ChromeService(executable_path = ChromeDriverManager().install())\n",
"driver = webdriver.Chrome(service = service, options = options)\n",
"file2Download = '2022VAERSData.zip'\n", "file2Download = '2022VAERSData.zip'\n",
"driver.get('https://vaers.hhs.gov/eSubDownload/index.jsp?fn=' + file2Download)\n",
"captchaImage = driver.find_element(By.CSS_SELECTOR, \"img[src='captchaImage']\")\n",
"captchaImageFile = 'captchaImage.jpeg'\n", "captchaImageFile = 'captchaImage.jpeg'\n",
"\n",
"def getWebDriver(downloadDir):\n",
" options = Options()\n",
" options.add_experimental_option(\"prefs\", {\"download.default_directory\" : downloadDir})\n",
" return webdriver.Chrome(\n",
" service = ChromeService(executable_path = ChromeDriverManager().install()),\n",
" options = options)\n",
"\n",
"driver = getWebDriver(downloadDir)\n",
"driver.get('https://vaers.hhs.gov/eSubDownload/index.jsp?fn=' + file2Download)\n",
"\n",
"def saveCaptchaImageAs(captchaImageFile):\n",
" captchaImage = driver.find_element(By.CSS_SELECTOR, \"img[src='captchaImage']\")\n",
" with open(captchaImageFile, 'wb') as file:\n", " with open(captchaImageFile, 'wb') as file:\n",
" file.write(captchaImage.screenshot_as_png)" " file.write(captchaImage.screenshot_as_png)\n",
"\n",
"saveCaptchaImageAs(captchaImageFile)"
] ]
}, },
{ {
@@ -1189,8 +1197,11 @@
} }
], ],
"metadata": { "metadata": {
"interpreter": {
"hash": "31f2aee4e71d21fbe5cf8b01ff0e069b9275f58929596ceb00d14d90e3e16cd6"
},
"kernelspec": { "kernelspec": {
"display_name": "Python 3 (ipykernel)", "display_name": "Python 3.10.4 64-bit",
"language": "python", "language": "python",
"name": "python3" "name": "python3"
}, },