Update extract.py

extract.py

@@ -46,24 +46,72 @@ def scrape_vehicle(driver):
     except Exception as e:
         print("Gagal parsing tabel:", e)
     
-    rincians = []
+    # rincians = []
+    # try:
+    #     container = driver.find_element(By.ID, "det_pkb")
+    #     rows = container.find_elements(By.CLASS_NAME, "row")
+    #     for row in rows[1:]:  # skip header
+    #         cols = row.find_elements(By.TAG_NAME, "p")
+    #         if len(cols) >= 3:
+    #             rincian = {
+    #                 "pokok": cols[0].text.strip(),
+    #                 "denda": cols[1].text.strip(),
+    #                 "total": cols[2].text.strip(),
+    #             }
+    #             if len(cols) > 3:
+    #                 rincian["jenis"] = cols[3].text.strip().upper()
+    #             rincians.append(rincian)
+    # except Exception as e:
+    #     print("Gagal parsing det_pkb:", e)
+
+
+    rincian_total = {}
+    rincian_pkb = []
+    rincian_swdkllj = []
+    
     try:
         container = driver.find_element(By.ID, "det_pkb")
         rows = container.find_elements(By.CLASS_NAME, "row")
-        for row in rows[1:]:  # skip header
+        
+        current_section = None
+        for row in rows:
             cols = row.find_elements(By.TAG_NAME, "p")
-            if len(cols) >= 3:
+            texts = [col.text.strip() for col in cols if col.text.strip()]
+            if not texts:
+                continue
+    
+            label = texts[0].lower()
+    
+            if "total" in label:
+                current_section = "total"
+                continue
+            elif "rincian pkb" in label:
+                current_section = "pkb"
+                continue
+            elif "rincian swdkllj" in label:
+                current_section = "swdkllj"
+                continue
+    
+            if len(texts) >= 3:
                 rincian = {
-                    "pokok": cols[0].text.strip(),
-                    "denda": cols[1].text.strip(),
-                    "total": cols[2].text.strip(),
+                    "pokok": texts[0],
+                    "denda": texts[1],
+                    "total": texts[2]
                 }
-                if len(cols) > 3:
-                    rincian["jenis"] = cols[3].text.strip().upper()
-                rincians.append(rincian)
+                if len(texts) > 3:
+                    rincian["jenis"] = texts[3].upper()
+    
+                if current_section == "total":
+                    rincian_total = rincian
+                elif current_section == "pkb":
+                    rincian_pkb.append(rincian)
+                elif current_section == "swdkllj":
+                    rincian_swdkllj.append(rincian)
+    
     except Exception as e:
-        print("Gagal parsing det_pkb:", e)
+        print("Gagal parsing det_pkb:", e
 
+    rincians = [rincian_total, rincian_pkb, rincian_swdkllj]
     return data_kendaraan, rincians