Update generator.py

generator.py

@@ -1,5 +1,9 @@
 import asyncio
 from playwright.async_api import async_playwright
+import logging
+
+# Setup the logging 
+logging.basicConfig(level=logging.INFO, format='%(message)s')
 
 async def run():
     async with async_playwright() as p:
@@ -7,65 +11,53 @@ async def run():
         page = await browser.new_page()
 
         await page.goto('https://agogmail.com/')
-         # Click the dropdown menu button first
+        
         await page.wait_for_selector('.btn.btn-light.dropdown-toggle')
         await page.click('.btn.btn-light.dropdown-toggle')  
 
-        # Then click the change mailbox button
         await page.wait_for_selector('a.changeMailbox[data-type="1"]')
         await page.click('a.changeMailbox[data-type="1"]')
         await asyncio.sleep(3)
-        # Get active-mail value and print
+
         input_field = await page.query_selector('#active-mail')
         email = await input_field.get_attribute('value')
-        print("Active Mail Value: ", email)
+        logging.info("Active Mail Value: %s", email)
         
-        # Go to the form page
         form_page = await browser.new_page()
         await form_page.goto('https://altaregistrazione.net/')
 
-        # Get all input fields
         input_fields = await form_page.query_selector_all('input')
 
-        # Filling in the fields
-        await input_fields[0].fill(email)    # Assuming first field is the email field
-        print("Filled email: ", email)
+        await input_fields[0].fill(email)
+        logging.info("Filled email: %s", email)
 
-        await input_fields[1].fill(email)    # Assuming second field is the password field
-        print("Filled password")
+        await input_fields[1].fill(email)
+        logging.info("Filled password")
 
-        await input_fields[2].fill(email)    # Assuming third field is the confirm password field
-        print("Filled confirm password")
+        await input_fields[2].fill(email)
+        logging.info("Filled confirm password")
 
-        # Check the radio button
         radio_buttons = await form_page.query_selector_all('input[type="radio"]')
-        await radio_buttons[0].check()    # Assuming it's the first radio button you want to check
-        print("Checked radio button")
+        await radio_buttons[0].check()
+        logging.info("Checked radio button")
 
-        # Before clicking the button, make sure the page loads the dynamic content.
-        # we are waiting for the button to be visible.
         await form_page.wait_for_selector('.v-btn--block', state='visible')
 
-        # Click the "Continua" button.
         await form_page.click('.v-btn--block')
-        print("Continua Button Clicked")
+        logging.info("Continua Button Clicked")
 
-        # print a message once complete.
-        print("Form-filled\n")
+        logging.info("Form-filled")
 
         await page.bring_to_front()
         await asyncio.sleep(8)
-        await page.click('#refresh')    # click the refresh button
+        await page.click('#refresh')
 
-        # wait for an email with class 'font-bold' to appear
         await page.wait_for_function("document.querySelector('#message-list tr.font-bold')")
 
-        # Get active-mail value and print
         mail_field = await page.query_selector('#active-mail')
         mail = await mail_field.get_attribute('value')
-        print("Active Mail Value: ", mail)
+        logging.info("Active Mail Value: %s", mail)
 
-        # Get all unread mails with class 'font-bold'
         all_mails = await page.query_selector_all('#message-list tr.font-bold')
         
         for email in all_mails:
@@ -75,36 +67,30 @@ async def run():
             if "Altadefinizione" in sender_name:
                 href = await link.get_attribute('href')
                 full_url = f'https://agogmail.com/{href}'
-                print("Full URL: ", full_url)
+                logging.info("Full URL: %s", full_url)
 
-                # Open new page and go to the full URL
                 mail_page = await browser.new_page()
                 await mail_page.goto(full_url)
 
-                # Wait for the page to load 
                 await asyncio.sleep(5)
 
-                # Get the verification link
                 verify_link = await mail_page.query_selector('.button.button-primary')
                 verify_link_href = await verify_link.get_attribute('href')
 
-                print('Verification Link: ', verify_link_href)
+                logging.info('Verification Link: %s', verify_link_href)
 
-                # Open the verification link
                 await mail_page.goto(verify_link_href)
 
-                # Wait for the page to load 
                 await asyncio.sleep(5)
 
-                # Get cookies
                 cookies_list = await mail_page.context.cookies()
                 for cookie in cookies_list:
                     if cookie['name'] == 'ap_session':
-                        print('ap_session cookie: ', cookie['value'])
+                        logging.info('ap_session cookie: %s', cookie['value'])
 
                 await mail_page.close()
 
-                break  # Stop the loop once a valid email from 'Altadefinizione' is found
+                break
 
         await browser.close()