def get_headers(context_data) -> list:
    """
    Arrange the headers by importance - 'name' and 'id' will appear first
    Args:
        context_data: list or dict containing the context data

    Returns: headers arrange by importance

    """
    if isinstance(context_data, dict):
        context_data = [context_data]
    headers = list(context_data[0].keys())
    if 'name' in headers:
        headers.remove('name')
        headers.insert(0, 'name')
    if 'id' in headers:
        headers.remove('id')
        headers.insert(1, 'id')
    return headers