blob: 0a49c601de224e3cb41818c7098a025f1c1f9830 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options as ChromeOptions
from selenium import webdriver
from dotenv import load_dotenv
import time
import os
load_dotenv()
class SiteScraper:
def __init__(self, headless: bool = False, wait_time = 5):
"""
Initialize the SiteScraper with the path to ChromeDriver
:param chrome_driver_path: Path to the ChromeDriver executable
:param headless: Run the browser in headless mode if True
"""
self.wait_time = wait_time
try:
self.service = Service(os.environ.get("CHROME_DRIVER_PATH"))
self.chrome_options = ChromeOptions()
if headless:
self.chrome_options.add_argument("--headless")
self.chrome_options.add_argument("--disable-gpu")
self.chrome_options.add_argument("--disable-dev-shm-usage")
self.chrome_options.add_argument("--window-size=1920,1080")
self.chrome_options.add_argument("--no-sandbox")
self.driver = webdriver.Chrome(service=self.service, options=self.chrome_options)
except FileNotFoundError:
print("The ChromeDriver executable was not found. Is it installed and accessible in PATH?")
quit()
except Exception as e:
print(f"An unknown error occurred: {e}")
quit()
def get_page_source(self, url) -> str:
"""
Get the page source of the given URL
:param url: The URL of the page to scrape
:param wait_time: The time to wait for the page to load (for JavaScript)
"""
try:
self.driver.get(url)
except Exception as e:
print(f"An error occurred while trying to get the page source: {e}")
return ""
if self.wait_time > 0:
time.sleep(self.wait_time)
return self.driver.page_source
def close(self):
"""
Close the WebDriver
"""
self.driver.quit()
self.service.stop()
print("WebDriver closed successfully")
|
