SendGrid Template Versioning — Managing Template Changes and Rollbacks

import requests

SENDGRID_API_KEY = "your-api-key"
BASE_URL = "https://api.sendgrid.com/v3"
HEADERS = {
    "Authorization": f"Bearer {SENDGRID_API_KEY}",
    "Content-Type": "application/json"
}

def list_template_versions(template_id):
    response = requests.get(
        f"{BASE_URL}/templates/{template_id}/versions",
        headers=HEADERS
    )
    versions = response.json()
    for v in versions:
        status = "ACTIVE" if v.get('active') == 1 else "DRAFT"
        print(f"Version ID: {v['id']} ({status}) - v{v.get('version')}")
    return versions

def create_version(template_id, subject, html_content, name="v2"):
    response = requests.post(
        f"{BASE_URL}/templates/{template_id}/versions",
        headers=HEADERS,
        json={
            "template_id": template_id,
            "subject": subject,
            "html_content": html_content,
            "name": name,
            "active": 0  # Draft
        }
    )
    version = response.json()
    print(f"Created version: {version['id']}")
    return version

def activate_version(template_id, version_id):
    response = requests.patch(
        f"{BASE_URL}/templates/{template_id}/versions/{version_id}",
        headers=HEADERS,
        json={"active": 1}
    )
    return response.status_code == 200

def get_active_version(template_id):
    versions = list_template_versions(template_id)
    for v in versions:
        if v.get('active') == 1:
            return v
    return None

class TemplateVersionManager:
    def __init__(self, template_id):
        self.template_id = template_id
        self.api_url = f"{BASE_URL}/templates/{template_id}/versions"

    def promote_to_production(self, version_id):
        # Deactivate all other versions
        versions = requests.get(self.api_url, headers=HEADERS).json()
        for v in versions:
            if v['id'] != version_id and v.get('active') == 1:
                self.deactivate_version(v['id'])

        # Activate the target version
        return activate_version(self.template_id, version_id)

    def rollback(self):
        """Roll back to the previous active version"""
        versions = requests.get(self.api_url, headers=HEADERS).json()
        # Sort by version number, take the last active one
        active = [v for v in versions if v.get('active') == 1]
        if len(active) > 1:
            # There are multiple active versions
            sorted_versions = sorted(active, key=lambda x: x['version'], reverse=True)
            # Deactivate current, activate previous
            self.deactivate_version(sorted_versions[0]['id'])
            self.activate_version(sorted_versions[1]['id'])
            return True
        return False

    def deactivate_version(self, version_id):
        requests.patch(
            f"{self.api_url}/{version_id}",
            headers=HEADERS,
            json={"active": 0}
        )

def send_test_with_version(template_id, version_id, test_email):
    """Send a test email using a specific template version"""
    from sendgrid import SendGridAPIClient
    from sendgrid.helpers.mail import Mail

    sg = SendGridAPIClient(SENDGRID_API_KEY)

    message = Mail(
        from_email='test@dodatech.com',
        to_emails=test_email
    )
    message.template_id = template_id
    message.dynamic_template_data = {
        'userName': 'Test User',
        'companyName': 'DodaTech'
    }

    # Activate the test version temporarily
    activate_version(template_id, version_id)

    # Send the test
    response = sg.send(message)

    # Restore the original active version
    # (implement rollback logic here)

    return response.status_code == 202