def main():

    # Get and log the config from the Env variable
    config = json.loads(os.environ["CUSTOM_SCRIPT_CONFIG"])
    print(config)

    # Get PC info from the config dict
    pc_info = config.get("tdaas_pc")
    pc_external_ip = pc_info.get("ips")[0][0]
    pc_internal_ip = pc_info.get("ips")[0][1]
    pc_password = pc_info.get("prism_password")

    try:

        # Get marketplace items
        mp_items = body_via_v3_post(pc_external_ip, "calm_marketplace_items",
                                    pc_password, None).json

        # Loop through our items
        for item in mp_items["entities"]:

            # We only care about those PENDING
            if item["status"]["app_state"] == "PENDING":

                # Get and modify the body
                body = body_via_v3_get(
                    pc_external_ip,
                    "calm_marketplace_items",
                    pc_password,
                    item["metadata"]["uuid"],
                ).json
                del body["status"]
                body["spec"]["resources"]["app_state"] = "ACCEPTED"

                # Update the item
                resp = update_via_v3_put(
                    pc_external_ip,
                    "calm_marketplace_items",
                    pc_password,
                    item["metadata"]["uuid"],
                    body,
                )

                # Log appropriately based on response
                if resp.code == 200 or resp.code == 202:
                    print(
                        f"{item['status']['name']} bp approved successfully.")
                else:
                    raise Exception(
                        f"{item['status']['name']} bp approved failed with:\n"
                        + f"Resp: {resp}\n" + f"Error Code: {resp.code}\n" +
                        f"Error Message: {resp.message}")

    except Exception as ex:
        print(traceback.format_exc())
        sys.exit(1)
예제 #2
0
def main():

    # Get and log the config from the Env variable
    config = json.loads(os.environ["CUSTOM_SCRIPT_CONFIG"])
    INFO(config)

    # Get PC info from the config dict
    pc_info = config.get("tdaas_pc")
    pc_external_ip = pc_info.get("ips")[0][0]
    pc_internal_ip = pc_info.get("ips")[0][1]
    pc_password = pc_info.get("prism_password")

    try:

        # Convert our spec to dict
        apps_spec = file_to_dict("specs/calm_bp_publish.json")

        # Get our non-default projects before the loop
        projects_payload = {"filter": "name!=default"}
        projects_resp = body_via_v3_post(pc_external_ip, "projects",
                                         pc_password, projects_payload)

        # Loop through our to-be-published apps
        for app in apps_spec["entities"]:

            # Construct payload and make call
            mp_payload = {"filter": f"name=={app['mp_name']}"}
            mp_post = body_via_v3_post(pc_external_ip,
                                       "calm_marketplace_items", pc_password,
                                       mp_payload)

            # Loop through our response to find matching version
            for mp_item in mp_post.json["entities"]:
                if (mp_item["status"]["app_state"] == "ACCEPTED"
                        and mp_item["status"]["version"] == app["bp_version"]
                        and mp_item["status"]["app_source"]
                        == app["app_source"]):

                    # Make a GET with our UUID
                    mp_get = body_via_v3_get(
                        pc_external_ip,
                        "calm_marketplace_items",
                        pc_password,
                        mp_item["metadata"]["uuid"],
                    )

                    # Modify the response body
                    mp_body = mp_get.json
                    del mp_body["status"]
                    mp_body["spec"]["resources"]["app_state"] = "PUBLISHED"
                    for project in projects_resp.json["entities"]:
                        mp_body["spec"]["resources"][
                            "project_reference_list"].append(
                                project["metadata"]["project_reference"])

                    # Publish the blueprint
                    pub_resp = update_via_v3_put(
                        pc_external_ip,
                        "calm_marketplace_items",
                        pc_password,
                        mp_body["metadata"]["uuid"],
                        mp_body,
                    )

                    # Log appropriately based on response
                    if pub_resp.code == 200 or pub_resp.code == 202:
                        INFO(
                            f"{mp_body['spec']['name']} MP item published successfully."
                        )
                    else:
                        raise Exception(
                            f"{mp_body['spec']['name']} MP App Publish failed with:\n"
                            + f"Resp: {resp}\n" +
                            f"Error Code: {pub_resp.code}\n" +
                            f"Error Message: {pub_resp.message}")

    except Exception as ex:
        ERROR(traceback.format_exc())
예제 #3
0
def main():

    # Get and log the config from the Env variable
    config = json.loads(os.environ["CUSTOM_SCRIPT_CONFIG"])
    INFO(config)

    # Get PC info from the config dict
    pc_info = config.get("tdaas_pc")
    pc_external_ip = pc_info.get("ips")[0][0]
    pc_internal_ip = pc_info.get("ips")[0][1]
    pc_password = pc_info.get("prism_password")

    try:

        # Read in the spec files and conver to dicts
        subnet_spec = file_to_dict("specs/calm_subnet.spec")
        INFO(f"subnet_spec: {subnet_spec}")
        secret_spec = file_to_dict("specs/calm_secrets.spec")
        INFO(f"secret_spec: {secret_spec}")

        # Get our subnet and image info from the infra
        subnet_info = get_subnet_info(pc_external_ip, pc_password,
                                      subnet_spec["vlan"])
        image_info = body_via_v3_post(pc_external_ip, "images", pc_password,
                                      None).json

        # Get a list of DRAFT blueprints
        payload = {"filter": "state==DRAFT"}
        draft_resp = body_via_v3_post(pc_external_ip, "blueprints",
                                      pc_password, payload).json

        # Loop through the blueprints to modify
        for bp in draft_resp["entities"]:

            # Get the body of our blueprint
            bp_body = body_via_v3_get(pc_external_ip, "blueprints",
                                      pc_password, bp["metadata"]["uuid"]).json

            # Remove unneeded status
            del bp_body["status"]

            # Configure secrets
            for secret in bp_body["spec"]["resources"]\
                                 ["credential_definition_list"]:
                secret["secret"]["attrs"]["is_secret_modified"] = True
                # Find a matching type/username from our secret_spec
                for ss in secret_spec["entities"]:
                    if secret["type"] == ss["type"] and\
                       secret["username"] == ss["username"]:
                        secret["secret"]["value"] = ss["secret"]
                print(json.dumps(secret, sort_keys=True, indent=4))

            # Configure NICs and Images
            for substrate in bp_body["spec"]["resources"][
                    "substrate_definition_list"]:
                if substrate["type"] == "AHV_VM":
                    for nic in substrate["create_spec"]["resources"][
                            "nic_list"]:
                        nic["subnet_reference"]["uuid"] = subnet_info["uuid"]
                        nic["subnet_reference"]["name"] = subnet_info["name"]
                        print(json.dumps(nic, sort_keys=True, indent=4))
                    for disk in substrate["create_spec"]["resources"][
                            "disk_list"]:
                        if disk["data_source_reference"] is not None and\
                           disk["data_source_reference"]["kind"] == "image":
                            for image in image_info["entities"]:
                                if image["status"]["name"] == disk[
                                        "data_source_reference"]["name"]:
                                    disk["data_source_reference"][
                                        "uuid"] = image["metadata"]["uuid"]
                                    print(
                                        json.dumps(disk,
                                                   sort_keys=True,
                                                   indent=4))

            # Update our blueprint
            resp = update_via_v3_put(pc_external_ip, "blueprints", pc_password,
                                     bp["metadata"]["uuid"], bp_body)

            # Log appropriately based on response
            if (resp.code == 200 or resp.code == 202):
                INFO(
                    f"{bp['metadata']['name']} blueprint updated successfully."
                )
            else:
                raise Exception(f"{bp['metadata']['name']} blueprint update" +
                                f" failed with:\n" +
                                f"Error Code: {resp.code}\n" +
                                f"Error Message: {resp.message}")

    except Exception as ex:
        INFO(ex)
예제 #4
0
def main(launch):

    # Get and log the config from the Env variable
    config = json.loads(os.environ["CUSTOM_SCRIPT_CONFIG"])
    print(config)

    # Get PC info from the config dict
    pc_info = config.get("tdaas_pc")
    pc_external_ip = pc_info.get("ips")[0][0]
    pc_internal_ip = pc_info.get("ips")[0][1]
    pc_password = pc_info.get("prism_password")

    try:

        # Read in the spec files and convert to dicts
        launch_spec = file_to_dict(f"specs/{launch}")
        print(f"launch_spec: {launch_spec}")

        # Loop through the blueprints to launch
        for launch in launch_spec["entities"]:

            # Get our blueprint uuid
            payload = {"filter": f"name=={launch['bp_name']}"}
            bp = body_via_v3_post(
                pc_external_ip, "blueprints", pc_password, payload
            ).json
            if bp["metadata"]["total_matches"] != 1:
                raise Exception(
                    str(bp["metadata"]["total_matches"])
                    + " blueprints found, when 1 should"
                    + " have been found."
                )
            else:
                bp_uuid = bp["entities"][0]["metadata"]["uuid"]

            # Get our runtime editables
            editables = body_via_v3_get(
                pc_external_ip,
                "blueprints",
                pc_password,
                bp_uuid + "/runtime_editables",
            ).json
            for profile in editables["resources"]:
                if profile["app_profile_reference"]["name"] == launch["profile_name"]:
                    profile_ref = profile["app_profile_reference"]
                    run_editables = profile["runtime_editables"]
            print(f"{launch['bp_name']} profile_ref: {profile_ref}")
            print(f"{launch['bp_name']} run_editables: {run_editables}")

            # Determine if this blueprint has an app dependency
            if "dependencies" in launch and len(launch["dependencies"]) > 0:
                # Get a list of running apps
                apps = body_via_v3_post(pc_external_ip, "apps", pc_password, None).json
                # Cycle through the apps
                for app in apps["entities"]:
                    # Cycle through our launch dependencies
                    for depend in launch["dependencies"]:
                        # Find the matching app name
                        if app["status"]["name"] == depend["app_name"]:
                            # Get app body
                            app_body = body_via_v3_get(
                                pc_external_ip,
                                "apps",
                                pc_password,
                                app["metadata"]["uuid"],
                            ).json
                            # Loop through our dependency values
                            for value in depend["values"]:
                                # Get value from our body+key combo
                                app_val = recursive_dict_lookup(app_body, value["keys"])
                                # Set our app_val to the appropriate variable value
                                for launch_var in launch["variables"]:
                                    if value["name"] == launch_var["name"]:
                                        launch_var["value"] = app_val
                print(f'{launch["bp_name"]} vars after depend: {launch["variables"]}')

            # Set our runtime variables
            if "variable_list" in run_editables:
                for run_edit_var in run_editables["variable_list"]:
                    for launch_var in launch["variables"]:
                        if run_edit_var["name"] == launch_var["name"]:
                            run_edit_var["value"]["value"] = launch_var["value"]
                print(f"{launch['bp_name']} run_editables: {run_editables}")

            # Create our payload and launch our app
            payload = {
                "spec": {
                    "app_name": launch["app_name"],
                    "app_description": launch["app_description"],
                    "app_profile_reference": profile_ref,
                    "runtime_editables": run_editables,
                }
            }
            resp = create_via_v3_post(
                pc_external_ip,
                "blueprints/" + bp_uuid + "/simple_launch",
                pc_password,
                payload,
            )

            # Log appropriately based on response
            if resp.code == 200 or resp.code == 202:
                print(f"{launch['app_name']} app launched successfully.")
            else:
                raise Exception(
                    f"{launch['app_name']} app launch failed with:\n"
                    + f"Resp: {resp}\n"
                    + f"Error Code: {resp.code}\n"
                    + f"Error Message: {resp.message}"
                )

            time.sleep(2)

    except Exception as ex:
        print(traceback.format_exc())
        sys.exit(1)
예제 #5
0
def main():

    # Get and log the config from the Env variable
    config = json.loads(os.environ["CUSTOM_SCRIPT_CONFIG"])
    INFO(config)

    # Get PC info from the config dict
    pc_info = config.get("tdaas_pc")
    pc_external_ip = pc_info.get("ips")[0][0]
    pc_internal_ip = pc_info.get("ips")[0][1]
    pc_password = pc_info.get("prism_password")

    try:

        # Read in the spec files and conver to dicts
        publish_spec = file_to_dict("specs/calm_bp_publish.json")
        INFO(f"publish_spec: {publish_spec}")

        # Get user and icon info
        user_info = body_via_v3_get(pc_external_ip, "users", pc_password,
                                    "me").json
        icon_info = body_via_v3_post(pc_external_ip, "app_icons", pc_password,
                                     None).json

        # Loop through the blueprints to upload
        for publish in publish_spec["entities"]:

            # Create our payload and get bp uuid
            payload = {"filter": f"name=={publish['bp_name']}"}
            bp_info = body_via_v3_post(pc_external_ip, "blueprints",
                                       pc_password, payload).json
            bp_uuid = bp_info["entities"][0]["metadata"]["uuid"]

            # Get bp spec with uuid
            bp = body_via_v3_get(
                pc_external_ip,
                "blueprints",
                pc_password,
                f"{bp_uuid}/export_json?keep_secrets=true",
            ).json

            # Modify our body
            bp["spec"]["name"] = publish["mp_name"]
            bp["spec"]["description"] = publish["mp_description"]
            bp["status"]["name"] = publish["mp_name"]
            bp["status"]["description"] = publish["mp_description"]
            spec = copy.deepcopy(bp["spec"])
            status = copy.deepcopy(bp["status"])
            del bp["spec"]["resources"]
            del bp["status"]
            bp["metadata"]["kind"] = "marketplace_item"
            bp["spec"]["resources"] = {"app_attribute_list": ["FEATURED"]}
            bp["spec"]["resources"]["app_state"] = "PENDING"
            bp["spec"]["resources"]["project_reference_list"] = []
            bp["spec"]["resources"]["change_log"] = ""
            bp["spec"]["resources"]["app_source"] = "LOCAL"
            bp["spec"]["resources"]["app_group_uuid"] = str(uuid.uuid4())
            bp["spec"]["resources"]["author"] = user_info["status"]["name"]
            for icon in icon_info["entities"]:
                if icon["status"]["name"] == publish["icon_name"]:
                    bp["spec"]["resources"]["icon_reference_list"] = [{
                        "icon_reference": {
                            "kind": "file_item",
                            "uuid": icon["metadata"]["uuid"],
                        },
                        "icon_type":
                        "ICON",
                    }]
            bp["spec"]["resources"]["version"] = publish["bp_version"]
            bp["spec"]["resources"]["app_blueprint_template"] = {"spec": spec}
            bp["spec"]["resources"]["app_blueprint_template"][
                "status"] = status

            # Upload our marketplace item
            resp = create_via_v3_post(pc_external_ip, "calm_marketplace_items",
                                      pc_password, bp)

            # Log appropriately based on response
            if resp.code == 200 or resp.code == 202:
                INFO(f"{publish['mp_name']} bp published successfully.")
            else:
                raise Exception(
                    f"{publish['mp_name']} bp publish failed with:\n" +
                    f"Resp: {resp}\n" + f"Error Code: {resp.code}\n" +
                    f"Error Message: {resp.message}")

    except Exception as ex:
        ERROR(traceback.format_exc())
예제 #6
0
def main():

    # Get and log the config from the Env variable
    config = json.loads(os.environ["CUSTOM_SCRIPT_CONFIG"])
    print(config)

    # Get PC info from the config dict
    pc_info = config.get("tdaas_pc")
    pc_external_ip = pc_info.get("ips")[0][0]
    pc_internal_ip = pc_info.get("ips")[0][1]
    pc_password = pc_info.get("prism_password")

    # Get PE info from the config dict
    pe_info = config.get("tdaas_cluster")
    pe_external_ip = pe_info.get("ips")[0][0]
    pe_internal_ip = pe_info.get("ips")[0][1]
    pe_password = pe_info.get("prism_password")

    try:

        # Read in the spec files and conver to dicts
        subnet_spec = file_to_dict("specs/calm_subnet.json")
        print(f"subnet_spec: {subnet_spec}")
        secret_spec = file_to_dict("specs/calm_secrets.json")
        print(f"secret_spec: {secret_spec}")

        # Get our subnet and image info from the infra
        subnet_info = get_subnet_info(pc_external_ip, pc_password,
                                      subnet_spec["entities"][0]["vlan"])
        infra_subnet_info = get_subnet_info(pc_external_ip, pc_password,
                                            subnet_spec["entities"][1]["vlan"])
        image_info = body_via_v3_post(pc_external_ip, "images", pc_password,
                                      None).json

        # If we have Public UVMs / Proxy IPs, then get our array
        proxy_array = []
        if "proxy_vm" in config:
            proxy_array = create_proxy_array(config["proxy_vm"])

        # Get a list of DRAFT blueprints
        payload = {"filter": "state==DRAFT"}
        draft_resp = body_via_v3_post(pc_external_ip, "blueprints",
                                      pc_password, payload).json

        # Loop through the blueprints to modify
        for bp in draft_resp["entities"]:

            # Get the body of our blueprint
            bp_body = body_via_v3_get(pc_external_ip, "blueprints",
                                      pc_password, bp["metadata"]["uuid"]).json

            # Remove unneeded status
            del bp_body["status"]

            # Configure secrets
            for secret in bp_body["spec"]["resources"][
                    "credential_definition_list"]:
                secret["secret"]["attrs"]["is_secret_modified"] = True
                # Handle PE and PC Creds which are unique
                if secret["name"].lower() == "pc_creds":
                    secret["secret"]["username"] = "******"
                    secret["secret"]["value"] = pc_password
                elif secret["name"].lower() == "pe_creds":
                    secret["secret"]["username"] = "******"
                    secret["secret"]["value"] = pe_password
                # Find a matching type/username from our secret_spec
                else:
                    for ss in secret_spec["entities"]:
                        if (secret["type"] == ss["type"]
                                and secret["username"] == ss["username"]):
                            secret["secret"]["value"] = ss["secret"]
                print(f"secret: {secret}")

            # Configure NICs and Images
            for substrate in bp_body["spec"]["resources"][
                    "substrate_definition_list"]:
                if substrate["type"] == "AHV_VM":

                    # For NICs, determine if it's an infra or user VM based blueprint
                    for nic in substrate["create_spec"]["resources"][
                            "nic_list"]:
                        if "infra" in bp_body["metadata"]["name"].lower():
                            # Ensure we have a proxy_vm config
                            if len(proxy_array) > 0:
                                nic["subnet_reference"][
                                    "uuid"] = infra_subnet_info["uuid"]
                                nic["subnet_reference"][
                                    "name"] = infra_subnet_info["name"]
                                proxy_ips = proxy_array.pop()
                                nic["ip_endpoint_list"][0]["ip"] = proxy_ips[1]
                            else:
                                print(
                                    f'Blueprint "{bp_body["metadata"]["name"]}" has '
                                    +
                                    f'"infra" in the name, but there were not enough '
                                    + f" proxy IPs configured: {config}. If" +
                                    f" this application needs external access, "
                                    +
                                    f"please ensure enough proxy IPs are in your GCP "
                                    +
                                    f'cluster.  Otherwise, remove "infra" from the '
                                    + f"blueprint name.")
                        else:
                            nic["subnet_reference"]["uuid"] = subnet_info[
                                "uuid"]
                            nic["subnet_reference"]["name"] = subnet_info[
                                "name"]
                        print(json.dumps(nic, sort_keys=True, indent=4))

                    for disk in substrate["create_spec"]["resources"][
                            "disk_list"]:
                        if (disk["data_source_reference"] is not None
                                and disk["data_source_reference"]["kind"]
                                == "image"):
                            for image in image_info["entities"]:
                                if (image["status"]["name"] ==
                                        disk["data_source_reference"]["name"]):
                                    disk["data_source_reference"][
                                        "uuid"] = image["metadata"]["uuid"]
                                    print(
                                        json.dumps(disk,
                                                   sort_keys=True,
                                                   indent=4))

            # Update our blueprint
            resp = update_via_v3_put(
                pc_external_ip,
                "blueprints",
                pc_password,
                bp["metadata"]["uuid"],
                bp_body,
            )

            # Log appropriately based on response
            if resp.code == 200 or resp.code == 202:
                print(
                    f"{bp['metadata']['name']} blueprint updated successfully."
                )
            else:
                raise Exception(
                    f"{bp['metadata']['name']} blueprint update failed with:\n"
                    + f"Resp: {resp}\n" + f"Error Code: {resp.code}\n" +
                    f"Error Message: {resp.message}")

    except Exception as ex:
        print(traceback.format_exc())
        sys.exit(1)
예제 #7
0
def main():

    # Get and log the config from the Env variable
    config = json.loads(os.environ["CUSTOM_SCRIPT_CONFIG"])
    INFO(config)

    # Get PC info from the config dict
    pc_info = config.get("tdaas_pc")
    pc_external_ip = pc_info.get("ips")[0][0]
    pc_internal_ip = pc_info.get("ips")[0][1]
    pc_password = pc_info.get("prism_password")

    try:

        # Read in the spec files and convert to dicts
        launch_spec = file_to_dict("specs/calm_bp_launch.spec")
        INFO(f"launch_spec: {launch_spec}")

        # Loop through the blueprints to launch
        for launch in launch_spec["entities"]:

            # Get our blueprint uuid
            payload = {"filter": f"name=={launch['bp_name']}"}
            bp = body_via_v3_post(pc_external_ip, "blueprints", pc_password,
                                  payload).json
            if bp["metadata"]["total_matches"] != 1:
                raise Exception(
                    str(bp["metadata"]["total_matches"]) +
                    " blueprints found, when 1 should" + " have been found.")
            else:
                bp_uuid = bp["entities"][0]["metadata"]["uuid"]

            # Get our runtime editables
            editables = body_via_v3_get(pc_external_ip, "blueprints",
                                        pc_password,
                                        bp_uuid + "/runtime_editables").json
            for profile in editables["resources"]:
                if profile["app_profile_reference"]["name"] ==\
                   launch["profile_name"]:
                    profile_ref = profile["app_profile_reference"]
                    run_editables = profile["runtime_editables"]
            INFO(f"{launch['bp_name']} profile_ref: {profile_ref}")
            INFO(f"{launch['bp_name']} run_editables: {run_editables}")

            # Set our runtime variables
            if "variable_list" in run_editables:
                for run_edit_var in run_editables["variable_list"]:
                    for launch_var in launch["variables"]:
                        if run_edit_var["name"] == launch_var["name"]:
                            run_edit_var["value"]["value"] = launch_var[
                                "value"]
                INFO(f"{launch['bp_name']} run_editables: {run_editables}")

            # Create our payload and launch our app
            payload = {
                "spec": {
                    "app_name": launch["app_name"],
                    "app_description": launch["app_description"],
                    "app_profile_reference": profile_ref,
                    "runtime_editables": run_editables
                }
            }
            resp = create_via_v3_post(
                pc_external_ip, "blueprints/" + bp_uuid + "/simple_launch",
                pc_password, payload)

            # Log appropriately based on response
            if (resp.code == 200 or resp.code == 202):
                INFO(f"{launch['app_name']} app launched successfully.")
            else:
                raise Exception(f"{launch['app_name']} app launch" +
                                f" failed with:\n" +
                                f"Error Code: {resp.code}\n" +
                                f"Error Message: {resp.message}")

            time.sleep(2)

    except Exception as ex:
        INFO(ex)