scripts/affinity.py

   1 #!/usr/local/bin/python
   2
   3 import httplib2
   4 import json
   5 import sys
   6
   7 #
   8 # Configure an affinity link and set its waypoint address.
   9 #
  10
  11 def rest_method(url, verb):
  12     global h
  13
  14     print "REST call " + url
  15     resp, content = h.request(url, verb)
  16
  17     print content
  18     print "return code %d" % (resp.status)
  19     print "done"
  20     return content
  21
  22
  23 def list_all_hosts():
  24
  25     print "list active hosts"
  26     put_url = 'http://localhost:8080/controller/nb/v2/hosttracker/default/hosts/active'
  27     content = rest_method(put_url, "GET")
  28     hostCfg = json.loads(content)
  29     for host in hostCfg['hostConfig']:
  30         print host
  31
  32     print "list inactive hosts"
  33     put_url = 'http://localhost:8080/controller/nb/v2/hosttracker/default/hosts/inactive'
  34     rest_method(put_url, "GET")
  35     content = rest_method(put_url, "GET")
  36     hostCfg = json.loads(content)
  37     for host in hostCfg['hostConfig']:
  38         print host
  39
  40
  41 def waypoint_init():
  42     # Create two affinity groups
  43
  44     print "create web servers group"
  45     put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/create/group/webservers'
  46     rest_method(put_url, "PUT")
  47
  48     print "create external addresses"
  49     put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/create/group/clients'
  50     rest_method(put_url, "PUT")
  51
  52     print "create link inflows"
  53     put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/create/link/inflows/from/clients/to/webservers'
  54     rest_method(put_url, "PUT")
  55
  56 #    print "add ip to webservers"
  57 #    put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/group/webservers/add/ip/10.0.0.1'
  58 #    rest_method(put_url, "PUT")
  59
  60     print "add subnet to webservers"
  61     put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/group/webservers/addsubnet/ipprefix/10.0.0.1/mask/31'
  62     rest_method(put_url, "PUT")
  63
  64 #    print "add ip to webservers"
  65 #    put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/group/webservers/add/ip/10.0.0.2'
  66 #    rest_method(put_url, "PUT")
  67
  68     print "add ip to external"
  69     put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/group/clients/add/ip/10.0.0.3'
  70     rest_method(put_url, "PUT")
  71
  72
  73 def get_all_affinity_groups():
  74     print "get all affinity groups"
  75     get_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/affinity-groups'
  76     rest_method(get_url, "GET")
  77
  78 # Tbd
  79 def get_all_affinity_links():
  80     print "get all affinity groups"
  81     get_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/affinity-links'
  82     rest_method(get_url, "GET")
  83
  84 def get_affinity_group(groupname):
  85     get_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/group/' + groupname
  86     rest_method(get_url, "GET")
  87
  88 def set_waypoint_address():
  89     wp = "10.0.0.2"
  90     put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/link/inflows/setwaypoint/' + wp
  91     rest_method(put_url, "PUT")
  92
  93 def enable_waypoint():
  94     put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/link/inflows/enable/'
  95     rest_method(put_url, "PUT")
  96
  97 def disable_waypoint():
  98     put_url = 'http://localhost:8080/affinity/nb/v2/affinity/default/link/inflows/disable/'
  99     rest_method(put_url, "PUT")
 100
 101 # Add waypoint IP to an affinity link.
 102 def main():
 103     global h
 104
 105     waypoint_init()
 106
 107     get_affinity_group('webservers')
 108     get_affinity_group('clients')
 109
 110     get_all_affinity_groups()
 111     list_all_hosts()
 112
 113     set_waypoint_address()
 114     enable_waypoint()
 115     disable_waypoint()
 116
 117 #if __name__ == "__main__":
 118 #    main()
 119
 120 h = httplib2.Http(".cache")
 121 h.add_credentials('admin', 'admin')
 122
 123
 124
 125