From 0a863ee75ff2efa56c87902a85e1500a9a994360 Mon Sep 17 00:00:00 2001
From: Floke <floke.com@gmail.com>
Date: Fri, 4 Jul 2025 05:13:24 +0000
Subject: [PATCH] dealfront_enrichment.py aktualisiert

---
 dealfront_enrichment.py | 59 +++++++++++++++++++++++------------------
 1 file changed, 33 insertions(+), 26 deletions(-)

diff --git a/dealfront_enrichment.py b/dealfront_enrichment.py
index f7d54677..f096138a 100644
--- a/dealfront_enrichment.py
+++ b/dealfront_enrichment.py
@@ -221,47 +221,54 @@ class DealfrontScraper:
             self._save_debug_artifacts()
             return False
             
-    def extract_current_page_results(self):
+        def extract_current_page_results(self):
         """
-        Extrahiert die Firmennamen und Webseiten von der aktuellen Ergebnisseite
-        mit einem finalen, robusten XPath-Selektor für die Datenzeilen.
+        Extrahiert Firmennamen und Webseiten direkt von der Seite
+        mithilfe der präzisen, von Ihnen verifizierten CSS-Selektoren.
         """
         try:
-            logger.info("Extrahiere Ergebnisse von der aktuellen Seite...")
+            logger.info("Extrahiere Ergebnisse mit direkten CSS-Selektoren...")
             results = []
             
-            # Warten, bis die Tabelle (bzw. ihr Body) da ist
-            table_body_selector = (By.CSS_SELECTOR, "table#t-result-table tbody")
-            self.wait.until(EC.presence_of_element_located(table_body_selector))
-            time.sleep(3) # Letzte Pause, um das JS-Rendering abzuwarten
-            
-            # Finaler, robuster Selektor für die Datenzeilen:
-            # Wir suchen alle `tr`-Elemente innerhalb des `tbody`, die eine `id` haben.
-            rows_selector = (By.CSS_SELECTOR, "table#t-result-table tbody tr[id]")
-            rows = self.driver.find_elements(*rows_selector)
+            # Warten, bis das erste Element, das wir suchen, vorhanden ist.
+            # Das ist ein guter Indikator, dass die Liste geladen ist.
+            company_name_selector = ".sticky-column a.t-highlight-text"
+            self.wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, company_name_selector)))
+            time.sleep(3) # Kurze Pause, um sicherzustellen, dass alle Elemente gerendert sind.
 
-            if not rows:
-                logger.warning("Keine Ergebniszeilen (tr[id]) auf der Seite gefunden. HTML-Struktur hat sich möglicherweise geändert.")
+            # === DIREKTE EXTRAKTION MIT IHREN SELEKTOREN ===
+            company_elements = self.driver.find_elements(By.CSS_SELECTOR, company_name_selector)
+            website_elements = self.driver.find_elements(By.CSS_SELECTOR, "a.text-gray-400.t-highlight-text")
+
+            logger.info(f"{len(company_elements)} Firmennamen und {len(website_elements)} Webseiten-Elemente gefunden.")
+
+            # Wir gehen davon aus, dass die Anzahl der Elemente übereinstimmt und iterieren
+            # über die kürzere der beiden Listen, um Fehler zu vermeiden.
+            num_results = min(len(company_elements), len(website_elements))
+            if num_results == 0:
+                logger.warning("Keine Firmen oder Webseiten mit den angegebenen Selektoren gefunden.")
                 self._save_debug_artifacts()
                 return []
 
-            logger.info(f"{len(rows)} Firmen-Datenzeilen zur Verarbeitung gefunden.")
-
-            for i, row in enumerate(rows, 1):
+            for i in range(num_results):
                 try:
-                    # Wir verwenden die stabilen Selektoren, die Sie identifiziert haben.
-                    company_name = row.find_element(By.CSS_SELECTOR, ".sticky-column a.t-highlight-text").get_attribute("title").strip()
-                    website = row.find_element(By.CSS_SELECTOR, "a.text-gray-400.t-highlight-text").text.strip()
-                    results.append({'name': company_name, 'website': website})
-                except NoSuchElementException:
-                    logger.warning(f"Zeile {i}: Konnte Name oder Webseite nicht extrahieren. Überspringe.")
+                    company_name = company_elements[i].get_attribute("title").strip()
+                    website = website_elements[i].text.strip()
+                    
+                    if company_name and website:
+                        results.append({'name': company_name, 'website': website})
+                    else:
+                        logger.warning(f"Element {i+1}: Unvollständige Daten.")
+                        
+                except Exception as e:
+                    logger.warning(f"Fehler bei der Verarbeitung von Element {i+1}: {e}")
                     continue
             
-            logger.info(f"Extraktion abgeschlossen. {len(results)} Firmen gefunden.")
+            logger.info(f"Extraktion abgeschlossen. {len(results)} Firmen erfolgreich zugeordnet.")
             return results
             
         except Exception as e:
-            logger.error(f"Schwerwiegender Fehler bei der Extraktion: {type(e).__name__}", exc_info=True)
+            logger.error(f"Ein schwerwiegender Fehler ist bei der Extraktion der Ergebnisse aufgetreten: {type(e).__name__}", exc_info=True)
             self._save_debug_artifacts()
             return []