Update extract.py

extract.py

@@ -69,8 +69,7 @@ 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 row.text in ["POKOK", "DENDA", "TOTAL"]:
+            if not ("POKOK" in row.text or "DENDA" in row.text or "TOTAL" in row.text):
                 cols = row.find_elements(By.TAG_NAME, "p")
                 print("[COLS TOTAL]", [x.text for x in cols])
                 if len(cols) >= 4: