Update extract.py

extract.py

@@ -3,7 +3,9 @@ from selenium.common.exceptions import WebDriverException
 from PIL import Image
 from io import BytesIO
 
-import time
+import time, requests
+
+from bs4 import BeautifulSoup
 
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
@@ -121,6 +123,66 @@ def scrape_vehicle(driver):
     return data_kendaraan, total_tagihan, rincians_pkb, rincians_swd
 
 
+def get_vehicle_info_bs4(nopol: str):
+    content = requests.get("https://www.jambisamsat.net/infopkb.php", params={"no_polisi":"BH 1337 NN"}).content
+    soup = BeautifulSoup(content, "html.parser")
+    
+    # === 1. Data Kendaraan ===
+    data_kendaraan = {}
+    table = soup.find("table")
+    if table:
+        for row in table.find_all("tr"):
+            cols = row.find_all("td")
+            if len(cols) >= 3:
+                key = cols[0].get_text(strip=True).lower().replace(".", "").replace(" ", "_")
+                val = cols[2].get_text(strip=True)
+                data_kendaraan[key] = val
+    
+    # === 2. Total Tagihan (div.row outside det_pkb and det_swd) ===
+    all_rows = soup.find_all("div", class_="row")
+    total_tagihan = []
+    for row in all_rows:
+        if row.find_parent(id="det_pkb") or row.find_parent(id="det_swd"):
+            continue  # skip rows inside det_pkb or det_swd
+        ps = row.find_all("p")
+        if len(ps) >= 3 and all(keyword in ps[i].text for i, keyword in enumerate(["Pokok", "Denda", "Total"])):
+            total_tagihan.append({
+                "pokok": ps[0].get_text(strip=True),
+                "denda": ps[1].get_text(strip=True),
+                "total": ps[2].get_text(strip=True)
+            })
+    
+    # === 3. Rincian PKB ===
+    rincians_pkb = []
+    pkb_div = soup.find("div", id="det_pkb")
+    if pkb_div:
+        rows = pkb_div.find_all("div", class_="row")[1:]  # skip header
+        for row in rows:
+            cols = row.find_all("p")
+            if len(cols) >= 3:
+                rincians_pkb.append({
+                    "pokok": cols[0].get_text(strip=True),
+                    "denda": cols[1].get_text(strip=True),
+                    "total": cols[2].get_text(strip=True)
+                })
+    
+    # === 4. Rincian SWDKLLJ ===
+    rincians_swd = []
+    swd_div = soup.find("div", id="det_swd")
+    if swd_div:
+        rows = swd_div.find_all("div", class_="row")[1:]  # skip header
+        for row in rows:
+            cols = row.find_all("p")
+            if len(cols) >= 3:
+                rincians_swd.append({
+                    "pokok": cols[0].get_text(strip=True),
+                    "denda": cols[1].get_text(strip=True),
+                    "total": cols[2].get_text(strip=True)
+                })
+    return data_kendaraan, total_tagihan, rincians_pkb, rincians_swd
+
+
+
 def get_vehicle_info(driver, plate_number: str):
     # options = webdriver.ChromeOptions()
     # options.add_argument("--headless")