auto_refresh.py

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
from dotenv import load_dotenv
import smtplib
from email.mime.text import MIMEText
import os
import traceback


load_dotenv()

# ------ Credentials --------
USERNAME = os.getenv("USER_NAME")
PASSWORD = os.getenv("USER_PASS")
EMAIL_SENDER = os.getenv("EMAIL_SENDER")
EMAIL_PASS = os.getenv("EMAIL_PASS")
EMAIL_RECEIVER = os.getenv("EMAIL_RECEIVER")

local_driver_path = "D:/Downloads/chromedriver/win64/140.0.7339.207/chromedriver.exe"
login_url = "https://www.pythonanywhere.com/login/"


# ------- Chrome options ---------
chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument("--no-sandbox")
chrome_options.add_argument("--disable-dev-shm-usage")
chrome_options.add_argument("--window-size=1920,1080")
chrome_options.add_argument("--start-maximized")
chrome_options.add_argument("--headless")
# chrome_options.add_experimental_option("detach", True)


log_messages = []

# ------------------- LOGGING ---------------------
def log_step(step, status, detail=""):
    msg = f"{status} {step}"
    if detail:
        msg += f" → {detail}"
    log_messages.append(msg)
    print(msg)


# ------------------ EMAIL SENDER --------------------
def send_email_report(subject, body):
    try:
        msg = MIMEText(body)
        msg["Subject"] = subject
        msg["From"] = EMAIL_SENDER
        msg["To"] = EMAIL_RECEIVER

        with smtplib.SMTP("smtp.gmail.com", 587) as server:
            server.starttls()
            server.login(EMAIL_SENDER, EMAIL_PASS)
            server.send_message(msg)
        
        print("📧 Email report sent successfully.")
    except Exception as e:
        print(f"⚠️ Failed to send email report: {e}")


# --------------- WAIT FUNCTION ------------------
def wait_for_element(driver, by, locator, timeout=15):
    """
    Waits until the element is present or visible (max timeout seconds)
    Returns the WebElement once found
    """
    return WebDriverWait(driver, timeout).until(
        EC.presence_of_element_located((by, locator))
    )


# -------------------- MAIN SCRIPT ----------------------
driver = None
try:
    # Initialize ChromeDriver
    try:
        if os.path.exists(local_driver_path):
            print(f"✅ Local ChromeDriver found and working: {local_driver_path}")
            service = Service(local_driver_path)
            driver = webdriver.Chrome(service=service, options=chrome_options)
            log_step("Initialize ChromeDriver", "✅", "Local driver used")
        else:
            raise FileNotFoundError("Local chromedriver not found.")
    except Exception as e:
        log_step("Initialize ChromeDriver", "⚠️", str(e))
        driver = webdriver.Chrome(options=chrome_options)


    # step 1: open login page
    try:
        driver.get(login_url)
        log_step("Open login page", "✅")
    except Exception as e:
        log_step("Open login page", "❌", str(e))
        raise 

    # step 2: login deatails
    try:
        username_input = wait_for_element(driver, By.ID, "id_auth-username")
        username_input.send_keys(USERNAME)
        password_input = wait_for_element(driver, By.ID, "id_auth-password")
        password_input.send_keys(PASSWORD)
        password_input.send_keys(Keys.RETURN)
        log_step("Login", "✅")
    except Exception as e:
        log_step("Login", "❌", str(e))
        raise

    # step 3: go to web app page and get domain name
    try:
        webapp_link_element = wait_for_element(driver, By.ID, "id_web_app_link")
        WebDriverWait(driver, 5).until(EC.element_to_be_clickable((By.ID, "id_web_app_link")))
        webapp_link_element.click()

        log_step("Navigate to webapps page", "✅")

        # Wait until the page fully loads
        wait_for_element(driver, By.CLASS_NAME, "webapp_expiry")

        site_link_element = wait_for_element(driver, By.CLASS_NAME, "domain_name_link")
        site_url = site_link_element.get_attribute("href")
        log_step("Site URL", "✅", site_url)

    except Exception as e:
        log_step("Navigate to webapps page", "❌", str(e))
        raise

    # step 4: click "Run until 3 months from today"
    try:
        run_button = wait_for_element(driver, By.XPATH, "//input[@value='Run until 3 months from today']")
        run_button.click()
        log_step("Click run button", "✅")
    except Exception as e:
        log_step("Click run button", "❌", str(e))
        raise

    # step 5: get expiry info
    try:
        expiry_element = wait_for_element(driver, By.CLASS_NAME, "webapp_expiry")
        expiry_text = expiry_element.text.strip()
        log_step("Get expiry date", "✅", expiry_text)
    except Exception as e:
        log_step("Get expiry date", "❌", str(e))
        raise

    final_status = "✅ SUCCESS"

except Exception as e:
    final_status = "❌ FAILED"
    log_step("Unhandled error", "❌", traceback.format_exc())

finally:
    if driver:
        driver.quit()
    
    # Always send email summary
    email_subject = f"PythonAnywhere Auto-Run Report — {final_status}"
    email_body = "\n".join(log_messages)
    send_email_report(email_subject, email_body)