detail_pages_redshift.py

import mako.template
import mako.lookup
import mako.exceptions
import io
import json
import datetime
import os
import csv
import bisect
import yaml
import re


def initial_prices(i, instance_type):
    try:
        od = i["Pricing"]["us-east-1"]["ondemand"]
    except:
        # If prices are not available for us-east-1 it means this is a custom instance of some kind
        return ["'N/A'", "'N/A'", "'N/A'"]

    try:
        _1yr = i["Pricing"]["us-east-1"]["_1yr"]["Standard.partialUpfront"]
        _3yr = i["Pricing"]["us-east-1"]["_3yr"]["Standard.partialUpfront"]
    except:
        # If we can't get a reservation, likely a previous generation
        _1yr = "'N/A'"
        _3yr = "'N/A'"

    return [od, _1yr, _3yr]


def description(id, defaults):
    name = id["Amazon"][1]["value"]
    family_category = id["Amazon"][2]["value"].lower()
    cpus = id["Compute"][0]["value"]
    memory = id["Compute"][1]["value"]

    return "The {} instance is in the {} family and has {} vCPUs and {} GiB of memory with a starting price of ${} per hour.".format(
        name, family_category, cpus, memory, defaults[0]
    )


def unavailable_instances(itype, instance_details):
    data_file = "meta/regions_aws.yaml"

    denylist = []
    with open(data_file, "r") as f:
        aws_regions = yaml.safe_load(f)
        instance_regions = instance_details["Pricing"].keys()

        # If there is no price for a region and os, then it is unavailable
        for r in aws_regions:
            if r not in instance_regions:
                denylist.append([aws_regions[r], r, "All", "*"])

    return denylist


def assemble_the_families(instances):
    # Build 2 lists - one where we can lookup what family an instance belongs to
    # and another where we can get the family and see what the members are
    instance_fam_map = {}
    families = {}
    variant_families = {}

    for i in instances:
        name = i["instance_type"]
        itype = name.split(".")[0]
        suffix = "".join(name.split(".")[2:])
        variant = itype[0:2]

        if variant not in variant_families:
            variant_families[variant] = [[itype, name]]
        else:
            dupe = 0
            for v, _ in variant_families[variant]:
                if v == itype:
                    dupe = 1
            if not dupe:
                variant_families[variant].append([itype, name])

        member = {"name": name, "cpus": int(i["vcpu"]), "memory": float(i["memory"])}
        if itype not in instance_fam_map:
            instance_fam_map[itype] = [member]
        else:
            instance_fam_map[itype].append(member)

        # The second list, where we will get the family from knowing the instance
        families[name] = itype

    # Order the families by number of cpus so they display this way on the webpage
    for f, ilist in instance_fam_map.items():
        ilist.sort(key=lambda x: x["cpus"])
        instance_fam_map[f] = ilist

    return instance_fam_map, families, variant_families


def prices(pricing):
    display_prices = {}

    for region, p in pricing.items():
        display_prices[region] = {}

        # Doing a lot of work to deal with prices having up to 6 places
        # after the decimal, as well as prices not existing for all regions
        # and operating systems.
        try:
            display_prices[region]["ondemand"] = p["ondemand"]
        except KeyError:
            display_prices[region]["ondemand"] = "N/A"

        # In the next 2 blocks, we need to split out the list of 1 year,
        # 3 year, upfront, partial, and no upfront RI prices into 2 sets
        # of prices: _1yr (all, partial, no) and _3yr (all, partial, no)
        # These are then rendered into the 2 bottom pricing dropdowns
        try:
            reserved = {}
            for k, v in p["reserved"].items():
                if "Term1" in k:
                    key = k[7:]
                    reserved[key] = v
            display_prices[region]["_1yr"] = reserved
        except KeyError:
            display_prices[region]["_1yr"] = "N/A"

        try:
            reserved = {}
            for k, v in p["reserved"].items():
                if "Term3" in k:
                    key = k[7:]
                    reserved[key] = v
            display_prices[region]["_3yr"] = reserved
        except KeyError:
            display_prices[region]["_3yr"] = "N/A"

    return display_prices


def load_service_attributes():
    special_attrs = [
        "pricing",
    ]
    data_file = "meta/service_attributes_redshift.csv"

    display_map = {}
    with open(data_file, "r") as f:
        reader = csv.reader(f)

        for i, row in enumerate(reader):
            cloud_key = row[0]
            if i == 0:
                # Skip the header
                continue
            elif cloud_key in special_attrs:
                category = "Coming Soon"
            else:
                category = row[2]

            display_map[cloud_key] = {
                "cloud_key": cloud_key,
                "display_name": row[1],
                "category": category,
                "order": row[3],
                "style": row[4],
                "regex": row[5],
                "value": None,
                "variant_family": row[1][0:2],
            }

    return display_map


def format_attribute(display):

    if display["regex"]:
        # Use a regex extract the value to display
        toparse = str(display["value"])
        regex = str(display["regex"])
        match = re.search(regex, toparse)
        if match:
            display["value"] = match.group()
        # else:
        #     print("No match found for {} with regex {}".format(toparse, regex))

    if display["style"]:
        # Make boolean values have fancy CSS
        v = str(display["value"]).lower()
        if display["cloud_key"] == "currentGeneration" and v == "yes":
            display["style"] = "value value-current"
            display["value"] = "current"
        elif v == "false" or v == "0" or v == "none":
            display["style"] = "value value-false"
        elif v == "true" or v == "1" or v == "yes":
            display["style"] = "value value-true"
        elif display["cloud_key"] == "currentGeneration" and v == "no":
            display["style"] = "value value-previous"
            display["value"] = "previous"
        elif display["style"] == "bytes":
            display["value"] = round(int(v) / 1048576)

    return display


def map_cache_attributes(i, imap):
    # Transform keys (instance attributes like vCPUs) and values from instances.json
    # into human readable names and nicely formatted values
    categories = [
        "Compute",
        "Networking",
        "Storage",
        "Amazon",
        "Not Shown",
    ]

    # Nested attributes in instances.json that we handle differently
    special_attributes = [
        "pricing",
    ]

    instance_details = {}
    for c in categories:
        instance_details[c] = []

    # For up to date display names, inspect meta/service_attributes_cache.csv
    for attr_name, attr_val in i.items():
        try:
            if attr_name not in special_attributes:
                # This is one row on a detail page
                display = imap[attr_name]
                display["value"] = attr_val
                instance_details[display["category"]].append(format_attribute(display))

        except KeyError:
            print(
                "An instances.json attribute {} does not appear in meta/service_attributes_redshift.csv and cannot be formatted".format(
                    attr_name
                )
            )

    # Sort the instance attributes in each category alphabetically,
    # another general-purpose option could be to sort by value data type
    for c in categories:
        instance_details[c].sort(key=lambda x: int(x["order"]))

    return instance_details


def build_detail_pages_redshift(instances, destination_file):
    subdir = os.path.join("www", "aws", "redshift")

    ifam, fam_lookup, variants = assemble_the_families(instances)
    imap = load_service_attributes()

    lookup = mako.lookup.TemplateLookup(directories=["."])
    template = mako.template.Template(
        filename="in/instance-type-redshift.html.mako", lookup=lookup
    )

    # To add more data to a single instance page, do so inside this loop
    could_not_render = []
    sitemap = []
    for i in instances:
        instance_type = i["instance_type"]

        instance_page = os.path.join(subdir, instance_type + ".html")
        instance_details = map_cache_attributes(i, imap)
        instance_details["Pricing"] = prices(i["pricing"])
        fam = fam_lookup[instance_type]
        fam_members = ifam[fam]
        denylist = unavailable_instances(instance_type, instance_details)
        defaults = initial_prices(instance_details, instance_type)
        idescription = description(instance_details, defaults)

        print("Rendering %s to detail page %s..." % (instance_type, instance_page))
        with io.open(instance_page, "w+", encoding="utf-8") as fh:
            try:
                fh.write(
                    template.render(
                        i=instance_details,
                        family=fam_members,
                        description=idescription,
                        unavailable=denylist,
                        defaults=defaults,
                        variants=variants[instance_type[0:2]],
                    )
                )
                sitemap.append(instance_page)
            except:
                render_err = mako.exceptions.text_error_template().render()
                err = {"e": "ERROR for " + instance_type, "t": render_err}

                could_not_render.append(err)

    [print(err["e"], "{}".format(err["t"])) for err in could_not_render]
    [print(page["e"]) for page in could_not_render]

    return sitemap