utils.py

"""
Utility functions for the Akamai Property and Hostname Manager
"""

import dns.resolver

def resolve_hostname(hostname):
    """
    Resolve hostname to IP addresses using DNS
    Args:
        hostname: Hostname to resolve
    Returns:
        List of IP addresses
    """
    try:
        # Create a new resolver
        resolver = dns.resolver.Resolver()
        # Get A records
        answers = resolver.resolve(hostname, 'A')
        return [str(rdata) for rdata in answers]
    except Exception as e:
        return [f"Resolution failed: {str(e)}"]

def get_staging_hostname(prod_hostname):
    """
    Convert production hostname to staging hostname
    Args:
        prod_hostname: Production edgekey hostname
    Returns:
        Staging hostname
    """
    if '.edgekey.net' in prod_hostname:
        return prod_hostname.replace('.edgekey.net', '.edgekey-staging.net')
    return prod_hostname

def generate_html_table(headers, data):
    """
    Generate HTML table from headers and data
    Args:
        headers: List of table headers
        data: List of rows or dictionaries containing the table data
    Returns:
        HTML table as string
    """
    html_table = "<table><tr>"
    
    # Add headers
    for header in headers:
        html_table += f"<th>{header}</th>"
    html_table += "</tr>"
    
    # Add rows
    for row in data:
        html_table += "<tr>"
        # Handle both list and dictionary data formats
        if isinstance(row, dict):
            for header in headers:
                html_table += f"<td>{row.get(header, '')}</td>"
        else:
            for cell in row:
                html_table += f"<td>{cell}</td>"
        html_table += "</tr>"
    html_table += "</table>"
    
    return html_table

def generate_markdown_table(headers, data):
    """
    Generate Markdown table from headers and data
    Args:
        headers: List of table headers
        data: List of rows or dictionaries containing the table data
    Returns:
        Markdown table as string
    """
    md_table = "| " + " | ".join(headers) + " |\n"
    md_table += "| " + " | ".join(["---"] * len(headers)) + " |\n"
    
    for row in data:
        # Handle both list and dictionary data formats
        if isinstance(row, dict):
            values = [str(row.get(header, '')) for header in headers]
            md_row = "| " + " | ".join(values) + " |"
        else:
            md_row = "| " + " | ".join([str(cell) for cell in row]) + " |"
        md_table += md_row + "\n"
    
    return md_table