Run pytest and email summary in Python

import smtplib
import subprocess
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart


def run_tests():
    """Run pytest and capture the summary output."""
    result = subprocess.run(
        ["pytest", "-q"],
        capture_output=True,
        text=True
    )
    return result.stdout + result.stderr


def summarize_tests(test_output):
    """Extract the test summary line from pytest output."""
    for line in test_output.splitlines():
        if "passed" in line or "failed" in line or "error" in line:
            if "==" in line and ("passed" in line or "failed" in line or "error" in line):
                return line.strip()
    return "No summary found"


def send_email_summary(subject, body):
    """Mock email sending by printing the message content."""
    msg = MIMEMultipart()
    msg["From"] = "sender@example.com"
    msg["To"] = "team@example.com"
    msg["Subject"] = subject
    msg.attach(MIMEText(body, "plain"))
    print("=== EMAIL MESSAGE (mock) ===")
    print(str(msg))
    print("=== END EMAIL ===")


if __name__ == "__main__":
    test_output = run_tests()
    summary = summarize_tests(test_output)
    send_email_summary(f"Test Results: {summary}", f"Pytest summary:\n{summary}")