main.py

# ==========================================
# CodeAlpha Project 03 - Email Extractor
# Developed by: Dua Sajjad
# ==========================================

import re
import os

INPUT_FILE = "input.txt"
OUTPUT_FILE = "emails.txt"

emails = []


# ---------------- CLEAR SCREEN ---------------- #

def clear():
    os.system("cls" if os.name == "nt" else "clear")


# ---------------- PAUSE ---------------- #

def pause():
    input("\nPress Enter to continue...")


# ---------------- MENU ---------------- #

def show_menu():
    clear()

    print("=" * 55)
    print("             EMAIL EXTRACTOR TOOL")
    print("=" * 55)
    print("1. Read Input File")
    print("2. Extract Emails")
    print("3. View Extracted Emails")
    print("4. Show Total Emails")
    print("5. Save Emails")
    print("6. Exit")
    print("=" * 55)


# ---------------- READ FILE ---------------- #

def read_file():

    if not os.path.exists(INPUT_FILE):
        print(f"\n'{INPUT_FILE}' does not exist.")
        return None

    try:
        with open(INPUT_FILE, "r", encoding="utf-8") as file:
            data = file.read()

        if not data.strip():
            print("\nInput file is empty.")
            return None

        print("\nInput file loaded successfully!")
        return data

    except Exception as e:
        print(f"\nError: {e}")
        return None


# ---------------- EXTRACT EMAILS ---------------- #

def extract_emails():

    global emails

    text = read_file()

    if text is None:
        return

    pattern = r"[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}"

    found = re.findall(pattern, text)

    emails = sorted(list(set(found)))

    if emails:
        print(f"\n{len(emails)} email(s) extracted successfully!")
    else:
        print("\nNo email addresses found.")


# ---------------- VIEW EMAILS ---------------- #

def view_emails():

    if not emails:
        print("\nNo emails extracted yet.")
        return

    print("\n" + "=" * 55)
    print("           EXTRACTED EMAIL ADDRESSES")
    print("=" * 55)

    for i, email in enumerate(emails, start=1):
        print(f"{i}. {email}")


# ---------------- TOTAL EMAILS ---------------- #

def total_emails():

    print("\n" + "=" * 55)
    print(f"Total Emails Found : {len(emails)}")
    print("=" * 55)


# ---------------- SAVE EMAILS ---------------- #

def save_emails():

    if not emails:
        print("\nNothing to save.")
        return

    try:
        with open(OUTPUT_FILE, "w", encoding="utf-8") as file:

            file.write("=" * 40 + "\n")
            file.write("EMAIL EXTRACTION REPORT\n")
            file.write("=" * 40 + "\n\n")

            for email in emails:
                file.write(email + "\n")

            file.write("\n")
            file.write(f"Total Emails : {len(emails)}")

        print(f"\nEmails saved successfully to '{OUTPUT_FILE}'")

    except Exception as e:
        print(f"\nError: {e}")


# ---------------- MAIN PROGRAM ---------------- #

while True:

    show_menu()

    choice = input("Enter your choice (1-6): ")

    if choice == "1":

        read_file()
        pause()

    elif choice == "2":

        extract_emails()
        pause()

    elif choice == "3":

        view_emails()
        pause()

    elif choice == "4":

        total_emails()
        pause()

    elif choice == "5":

        save_emails()
        pause()

    elif choice == "6":

        print("\nThank you for using Email Extractor!")
        print("Have a Great Day!")
        break

    else:

        print("\nInvalid choice!")
        pause()