""" Gmail Verification - Handhabt die Verifizierungsprozesse """ import logging import time import random from typing import Dict, Optional from playwright.sync_api import Page from social_networks.gmail import gmail_selectors as selectors from social_networks.gmail.gmail_ui_helper import GmailUIHelper from utils.email_handler import EmailHandler logger = logging.getLogger("gmail_verification") class GmailVerification: """ Handhabt die Gmail/Google Account Verifizierung """ def __init__(self, page: Page, ui_helper: GmailUIHelper, email_handler: EmailHandler = None, screenshots_dir: str = None, save_screenshots: bool = True): """ Initialisiert den Verification Handler """ self.page = page self.ui_helper = ui_helper self.email_handler = email_handler self.screenshots_dir = screenshots_dir self.save_screenshots = save_screenshots def handle_phone_verification(self, account_data: Dict[str, str]) -> Dict[str, any]: """ Handhabt die Telefonnummer-Verifizierung """ try: logger.info("Starte Telefon-Verifizierung") # Warte auf Telefonnummer-Eingabefeld if not self.ui_helper.wait_for_element(selectors.PHONE_INPUT, timeout=10000): logger.info("Telefonnummer-Eingabefeld nicht gefunden") return { "success": True, "message": "Keine Telefon-Verifizierung erforderlich" } self.ui_helper.take_screenshot("phone_verification_page") # Telefonnummer eingeben phone = account_data.get("phone", "") if not phone: logger.warning("Keine Telefonnummer vorhanden, überspringe wenn möglich") # Versuche zu überspringen if self.ui_helper.is_element_visible(selectors.SKIP_BUTTON): self.ui_helper.click_with_retry(selectors.SKIP_BUTTON) time.sleep(random.uniform(2, 3)) return {"success": True} else: return { "success": False, "error": "Telefonnummer erforderlich aber nicht vorhanden", "message": "Telefonnummer wird benötigt" } logger.info(f"Gebe Telefonnummer ein: {phone}") # Telefonnummer eingeben self.ui_helper.type_with_delay(selectors.PHONE_INPUT, phone) time.sleep(random.uniform(1, 2)) # Screenshot vor dem Absenden self.ui_helper.take_screenshot("phone_entered") # Absenden if self.ui_helper.wait_for_element(selectors.NEXT_BUTTON, timeout=5000): logger.info("Sende Telefonnummer ab") self.ui_helper.click_with_retry(selectors.NEXT_BUTTON) self.ui_helper.wait_for_loading_to_finish() time.sleep(random.uniform(3, 5)) # Warte auf SMS-Code Eingabefeld if self.ui_helper.wait_for_element(selectors.SMS_CODE_INPUT, timeout=15000): logger.info("SMS-Code Eingabefeld gefunden") self.ui_helper.take_screenshot("sms_code_page") # Hier würde normalerweise der SMS-Code abgerufen werden sms_code = self._get_sms_code(phone) if not sms_code: logger.error("Kein SMS-Code erhalten") return { "success": False, "error": "Kein SMS-Code erhalten", "message": "SMS-Verifizierung fehlgeschlagen" } # SMS-Code eingeben logger.info(f"Gebe SMS-Code ein: {sms_code}") self.ui_helper.type_with_delay(selectors.SMS_CODE_INPUT, sms_code) time.sleep(random.uniform(1, 2)) # Code bestätigen if self.ui_helper.wait_for_element(selectors.VERIFY_BUTTON, timeout=5000): logger.info("Bestätige SMS-Code") self.ui_helper.click_with_retry(selectors.VERIFY_BUTTON) else: # Versuche alternativen Button self.ui_helper.click_with_retry(selectors.NEXT_BUTTON) self.ui_helper.wait_for_loading_to_finish() time.sleep(random.uniform(3, 5)) # Prüfe auf Erfolg if self._check_verification_success(): logger.info("Telefon-Verifizierung erfolgreich") self.ui_helper.take_screenshot("verification_success") return { "success": True, "message": "Verifizierung erfolgreich" } else: error_msg = self._get_verification_error() logger.error(f"Verifizierung fehlgeschlagen: {error_msg}") return { "success": False, "error": error_msg, "message": f"Verifizierung fehlgeschlagen: {error_msg}" } else: logger.info("Kein SMS-Code erforderlich") return { "success": True, "message": "Telefon-Verifizierung abgeschlossen" } except Exception as e: logger.error(f"Fehler bei der Telefon-Verifizierung: {e}") self.ui_helper.take_screenshot("verification_error") return { "success": False, "error": str(e), "message": f"Verifizierung fehlgeschlagen: {str(e)}" } def handle_captcha(self) -> Dict[str, any]: """ Handhabt Captcha-Herausforderungen """ try: logger.info("Prüfe auf Captcha") if self.ui_helper.is_element_visible(selectors.CAPTCHA_CONTAINER): logger.warning("Captcha erkannt - manuelle Lösung erforderlich") self.ui_helper.take_screenshot("captcha_detected") # TODO: Implementiere Captcha-Lösung return { "success": False, "error": "Captcha erkannt", "message": "Manuelle Captcha-Lösung erforderlich" } return { "success": True, "message": "Kein Captcha vorhanden" } except Exception as e: logger.error(f"Fehler bei der Captcha-Prüfung: {e}") return { "success": False, "error": str(e) } def _get_sms_code(self, phone: str) -> Optional[str]: """ Ruft den SMS-Code ab TODO: Implementierung für echte SMS-Code Abfrage """ logger.warning("SMS-Code Abruf noch nicht implementiert - verwende Platzhalter") # In einer echten Implementierung würde hier der SMS-Code # von einem SMS-Service abgerufen werden return "123456" # Platzhalter def _check_verification_success(self) -> bool: """ Prüft ob die Verifizierung erfolgreich war """ try: # Prüfe ob wir weitergeleitet wurden current_url = self.page.url if any(indicator in current_url for indicator in ["myaccount", "mail.google", "youtube"]): return True # Prüfe ob SMS-Code Feld noch sichtbar ist if self.ui_helper.is_element_visible(selectors.SMS_CODE_INPUT): # Prüfe auf Fehlermeldung if self.ui_helper.is_element_visible(selectors.ERROR_MESSAGE): return False # Wenn kein Fehler aber noch SMS-Code Feld, warten wir noch return False return True except Exception as e: logger.warning(f"Fehler bei der Verifizierungs-Prüfung: {e}") return False def _get_verification_error(self) -> str: """ Holt die Fehlermeldung falls vorhanden """ try: error_selectors = [ selectors.ERROR_MESSAGE, selectors.ERROR_MESSAGE_ALT, selectors.FORM_ERROR ] for selector in error_selectors: if self.ui_helper.is_element_visible(selector): error_text = self.ui_helper.get_element_text(selector) if error_text: return error_text return "Verifizierung fehlgeschlagen" except: return "Unbekannter Fehler"