|
from selenium import webdriver |
|
from selenium.common.exceptions import WebDriverException |
|
from PIL import Image |
|
from io import BytesIO |
|
|
|
from selenium.webdriver.common.by import By |
|
from selenium.webdriver.support.ui import WebDriverWait |
|
from selenium.webdriver.support import expected_conditions as EC |
|
|
|
def take_webdata(url): |
|
options = webdriver.ChromeOptions() |
|
options.add_argument('--headless') |
|
options.add_argument('--no-sandbox') |
|
options.add_argument('--disable-dev-shm-usage') |
|
|
|
try: |
|
wd = webdriver.Chrome(options=options) |
|
wd.set_window_size(1080, 720) |
|
wd.get(url) |
|
wd.implicitly_wait(5) |
|
|
|
page_title = wd.title |
|
screenshot = wd.get_screenshot_as_png() |
|
|
|
except WebDriverException as e: |
|
return Image.new('RGB', (1, 1)), page_title |
|
finally: |
|
if wd: |
|
wd.quit() |
|
|
|
return Image.open(BytesIO(screenshot)) , page_title |
|
|
|
|
|
def get_vehicle_info(plate_number: str): |
|
|
|
|
|
options = webdriver.ChromeOptions() |
|
options.add_argument("--headless") |
|
options.add_argument("--disable-gpu") |
|
options.add_argument("--no-sandbox") |
|
|
|
|
|
driver = webdriver.Chrome(options=options) |
|
|
|
try: |
|
|
|
driver.get("https://www.jambisamsat.net/infopkb.html") |
|
time.sleep(1) |
|
|
|
|
|
|
|
WebDriverWait(driver, 10).until( |
|
EC.presence_of_element_located((By.ID, "no_polisi")) |
|
) |
|
|
|
|
|
input_field = driver.find_element(By.ID, "no_polisi") |
|
input_field.clear() |
|
input_field.send_keys(plate_number) |
|
|
|
|
|
submit_button = driver.find_element(By.CSS_SELECTOR, 'button.btn.btn-primary[type="submit"]') |
|
submit_button.click() |
|
|
|
|
|
WebDriverWait(driver, 10).until( |
|
EC.url_contains("infopkb.php") |
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
driver.implicitly_wait(3) |
|
|
|
page_title = driver.title |
|
screenshot = driver.get_screenshot_as_png() |
|
|
|
return Image.open(BytesIO(screenshot)) , page_title |
|
|
|
except WebDriverException as e: |
|
return Image.new('RGB', (1, 1)), page_title |
|
|
|
finally: |
|
driver.quit() |