virl_packet_deploy.py

#!/usr/bin/env python3

"""Deploy a VIRL system on Packet via their API using the provided arguments.

Usage:
  virl_packet_deploy -a <API_KEY> -p <PROJECT_ID> -s <SERVER_NAME> ( -0 | -1 | -2 ) ( --nrt | --sjc | --ewr | --ams ) 
      [ ( --q | --qq | --uuid | --json ) ] [ -v VIRL_PW ] [ -u UWM_PW ] [ -g GUEST_PW ] [ -l LICENSE_FILE ] 

Arguments:
  -a <API_KEY>          An API key associated with your Packet account.
  -p <PROJECT_ID>       A Project Key associated with your Packet account.
  -s <SERVER_NAME>      The desired system / server name.

Options:
  -0                    System type 0.
  -1                    System type 1.
  -2                    System type 2.
  --nrt                 Facility NRT1.
  --sjc                 Facility SJC1.
  --ewr                 Facility EWR1.
  --ams                 Facility AMS1.
  -v VIRL_PW            VIRL account password
  -u UWM_PW             UWM account password.
  -g GUEST_PW           Guest account password.
  -l LICENSE_FILE       License file (format: id.domain.tld.pem)
  --q                   Display just the status, set exit-code
  --qq                  Display nothing, set exit-code
  --uuid                Display the UUID , set exit-code
  --json                Display the full JSON response, set exit-code

"""

import sys
import os
import requests
import json
import simplejson
from docopt import docopt
import time

# Some constants
BILLING = "hourly"
SYSTEM = "custom_ipxe"
IPXE_URL = "http://packet.virl.info"
API_URL = "https://api.packet.net/projects"
CONTENT_TYPE = "application/json"

# The API relevant status codes
associated_status = {201: "Success", 401: "Unauthorized", 422: "Unprocessable", 403: "Forbidden", 404: "Not Found"}

# Initialize some dictionaries / variables
api_data = {}
headers = {}
cloud_config = "#cloud-config\n"


# Let's get started
if __name__ == '__main__':
    arguments = docopt(__doc__, version='1.0.0rc3')

    # Form the request header
    headers.update({"content-type" : CONTENT_TYPE}) 
    headers.update({"X-Auth-Token" : arguments['-a']})

    # What is the hostname?
    api_data.update({"hostname" : arguments['-s']})


    # What system type are we deploying?
    if arguments['-0']:
        type = "baremetal_0"

    elif arguments['-1']:
        type = "baremetal_1"
      
    elif arguments['-2']:
        type = "baremetal_2"

    api_data.update({"plan" : type })


    # What is the billing_cycle?
    api_data.update({"billing_cycle" : BILLING})

    # What facility are we deploying to?
    if arguments['--nrt']:
        facility = "nrt1"
    
    elif arguments['--sjc']:
        facility = "sjc1"
    
    elif arguments['--ewr']:
        facility = "ewr1"

    elif arguments['--ams']:
        facility = "ams1"

    api_data.update({"facility" : facility })


    # What is the operating system?
    api_data.update({"operating_system" : SYSTEM})


    # What is IPXE URL?
    api_data.update({"ipxe_script_url" : IPXE_URL}) 


    # If we have a VIRL login password what is it?
    if arguments['-v']:
        cloud_config += ("virl_password: " + arguments['-v'] + "\n")


    # If we have a UWM account password what is it?
    if arguments['-u']:
        cloud_config += ("uwm_password: " + arguments['-u'] + "\n")


    # If we have a Guest account password what is it?
    if arguments['-g']:
        cloud_config += ("guest_password: " + arguments['-g'] + "\n")


    # If a key-file was provided form the salt-id and domain from the file
    if arguments['-l']:

        # Make sure the file exists
        lic = arguments['-l']
        if not os.path.isfile(lic):
            print("The VIRL license file specified does not exist.\n")
            sys.exit(1)

        # Make sure it ends with .pem
        if not lic.endswith('.pem'):
            print("The VIRL license file must end in \".pem\".\n")
            sys.exit(1)

        #Add the license_file to cloud_config
        cloud_config += ("license_file: " + arguments['-l'] + "\n")

        # Add the license PEM to cloud_config.  Read a line at a time
        # and indents so we get proper YAML multi-line data.

        cloud_config += "license_pem: |\n"

        with open(arguments['-l'], "r") as pem:
            for line in pem:
                cloud_config += ("  " + line)


    # If we have cloud-config data add it to api_data
    if arguments['-v'] or arguments['-u'] or arguments['-g'] or arguments['-l']:
        api_data.update({"userdata" : cloud_config})

    # Form our URL with the Project-ID parameter
    url = ('{0}/{1}/devices'.format(API_URL,arguments['-p']))

    # Make the call to Packet
    try:
        r = requests.post(url, headers=headers, json=api_data)
    except requests.ConnectionError:
        print ("An error occured while attempting to connect to Packet.")
        sys.exit(1)

    # Extract the UUID from the response JSON
    if r.status_code == 201:
      uuid = (r.json()["id"])

    # Show some output, or not.
    if arguments['--q']:
        print(associated_status[r.status_code])
    elif arguments['--uuid']:
        print(uuid)
    elif arguments['--json']:
        print(simplejson.dumps(r.json(), sort_keys=True, indent=4))
 
    sys.exit(r.status_code)