automation.py

import time
import re
from playwright.sync_api import Page, expect, Error, TimeoutError


def generate_temp_mail(page: Page) -> str:
    """
    Retrieves a temporary email address from the temp mail provider.
    Waits for the email generation button to appear before extracting the text.
    """
    # Wait until the button containing the email address appears
    email_button = page.locator("//button[p[contains(text(),'@')]]")
    email_button.wait_for(timeout=10000)

    # Extract the email address text from the <p> tag inside the button
    email = email_button.locator("p").inner_text()

    # Ensure the extracted email is valid and not empty
    while not email.strip():
        time.sleep(0.5)
        email = email_button.locator("p").inner_text()

    return email


def refresh_until_email_appears(page: Page, sender_email: str):
    """
    Continuously checks for an email from the specified sender.
    Clicks the 'Refresh' button if the email is not yet visible.
    """
    while True:
        try:
            # Check if an email button containing the sender's address exists
            email_btn = page.locator("button").filter(has_text=sender_email).first
            if email_btn.is_visible():
                print(f"Email from {sender_email} found!")
                break  # Stop clicking refresh once the email is found
        except (Error, TimeoutError) as e:
            pass

        # If the email is not found, attempt to click the 'Refresh' button
        try:
            refresh_button = page.locator("button", has_text="Refresh").first
            if refresh_button.is_visible():
                refresh_button.click()
                print("Clicked Refresh button")
            else:
                pass
        except (Error, TimeoutError) as e:
            print(f"Error clicking refresh: {e}")

        time.sleep(1)


def click_email_when_appears(page: Page, sender_email: str):
    """
    Waits for the email from the sender to appear and clicks it to open the message.
    """
    while True:
        try:
            # Locate the button for the specific email and click it
            email_button = page.locator("button").filter(has_text=sender_email).first
            if email_button.is_visible():
                email_button.click()
                print(f"Clicked email from {sender_email}")
                break  # Exit loop after successful click
        except (Error, TimeoutError) as e:
            pass

        # If the email isn't found yet, try refreshing the inbox
        try:
            refresh_button = page.locator("button", has_text="Refresh").first
            if refresh_button.is_visible():
                refresh_button.click()
                print("Clicked Refresh button (in click_email loop)")
        except (Error, TimeoutError) as e:
            pass

        time.sleep(2)  # Wait briefly to allow the inbox to reload


def extract_url_from_message(page: Page, email_div_title: str) -> str:
    """
    Extracts the IPTV playlist URL from the email content.
    Searches for a div containing the specified title text.
    """
    while True:
        # Attempt to find the div that contains the playlist information.
        # It may be plain text or nested within a bold <b> tag.

        playlist_div = None

        # Option 1: Search for div directly containing the text
        loc1 = page.locator(f"//div[contains(text(),'{email_div_title}')]")

        # Option 2: Search for div containing a bold tag with the text
        loc2 = page.locator(f"//div[b[contains(text(),'{email_div_title}')]]")

        if loc1.is_visible():
            playlist_div = loc1
        elif loc2.is_visible():
            playlist_div = loc2

        if playlist_div:
            # Retrieve the full text content of the identified div
            full_text = playlist_div.inner_text()
            # Use regex to find the HTTP URL ending in .m3u
            match = re.search(r"http://.*?/tv\.m3u", full_text)
            if match:
                playlist_url = match.group(0)
                print("Playlist URL:", playlist_url)
                return playlist_url

        # If the URL hasn't been found, wait and try again
        time.sleep(1)


def register_iptv_account(page: Page, email: str):
    # Fill in the registration form with the temporary email
    page.locator("input[name='email']").fill(email)

    # Handle Google reCAPTCHA
    # Locate the reCAPTCHA iframe
    frame = page.frame_locator("iframe[src*='recaptcha']").first

    # Click the "I'm not a robot" checkbox inside the iframe
    checkbox = frame.locator("#recaptcha-anchor")
    checkbox.click()

    # Wait for captcha resolution
    # Wait until the checkbox indicates it is checked (captcha solved)
    expect(checkbox).to_have_class(re.compile(r"recaptcha-checkbox-checked"), timeout=600000)

    # Submit the registration
    # Click the registration button on the main page
    page.locator("#regBtn").click()