Update extract.py

extract.py

@@ -142,16 +142,16 @@ def get_vehicle_info_bs4(nopol: str):
     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
+        if row.find_parent(id="det_pkb") or row.find_parent(id="det_swd"):
+            continue 
         if not ("POKOK" in row.text or "DENDA" in row.text or "TOTAL" in row.text):
             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)
-            })
+            if len(ps) >= 4:
+                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 = []