Update extract.py

extract.py

@@ -69,7 +69,8 @@ def scrape_vehicle(driver):
         all_rows = driver.find_elements(By.CSS_SELECTOR, "div.row")
         for row in all_rows:
             print("[ROW TOTAL]", row.text)
-            if not ("Pokok" in row.text or "Denda" in row.text or "Total" in row.text):
+            # if not ("POKOK" in row.text or "DENDA" in row.text or "Total" in row.text):
+            if not row.text in ["POKOK", "DENDA", "TOTAL"]:
                 cols = row.find_elements(By.TAG_NAME, "p")
                 print("[COLS TOTAL]", [x.text for x in cols])
                 if len(cols) >= 4: