Update extract.py

extract.py

@@ -82,7 +82,7 @@ def scrape_vehicle(driver):
     except Exception as e:
         print("Gagal parsing total tagihan:", e)
     
-    # ==== 3. Ambil det_pkb ====
+
     rincians_pkb = []
     try:
         pkb_rows = driver.find_elements(By.CSS_SELECTOR, "#det_pkb .row")[1:]  # skip header
@@ -99,7 +99,7 @@ def scrape_vehicle(driver):
     except Exception as e:
         print("Gagal parsing det_pkb:", e)
     
-    # ==== 4. Ambil det_swd ====
+
     rincians_swd = []
     try:
         swd_rows = driver.find_elements(By.CSS_SELECTOR, "#det_swd .row")[1:]  # skip header
@@ -154,6 +154,13 @@ def get_vehicle_info(plate_number: str):
         
         scroll_height = driver.execute_script("return document.body.scrollHeight")
         driver.set_window_size(1920, scroll_height + 200)  # force full-page height
+
+        button = wait.until(EC.element_to_be_clickable((By.ID, "show_det_pkb")))
+        button.click()
+        
+        button = wait.until(EC.element_to_be_clickable((By.ID, "show_det_swd")))
+        button.click()
+        
         time.sleep(1)
 
         data_kendaraan, rincian = scrape_vehicle(driver)