added demos to the repo

[groupbasedpolicy.git] / demos / gbpsfc-env / rest-clean.py

diff --git a/demos/gbpsfc-env/rest-clean.py b/demos/gbpsfc-env/rest-clean.py
new file mode 100644 (file)
index 0000000..7481066
--- /dev/null
+++ b/demos/gbpsfc-env/rest-clean.py
@@ -0,0 +1,124 @@
+#!/usr/bin/python
+import argparse
+import requests,json
+from requests.auth import HTTPBasicAuth
+from subprocess import call
+import time
+import sys
+import os
+
+
+DEFAULT_PORT='8181'
+
+
+USERNAME='admin'
+PASSWORD='admin'
+
+
+OPER_NODES='/restconf/operational/opendaylight-inventory:nodes/'
+CONF_TENANT='/restconf/config/policy:tenants'
+
+def get(host, port, uri):
+    url='http://'+host+":"+port+uri
+    r = requests.get(url, auth=HTTPBasicAuth(USERNAME, PASSWORD))
+    return r
+
+def rest_delete(host, port, uri, debug=False):
+    '''Perform a DELETE rest operation, using the URL and data provided'''
+    url='http://'+host+":"+port+uri
+    headers = {'Content-type': 'application/yang.data+json',
+               'Accept': 'application/yang.data+json'}
+    if debug == True:
+        print "DELETE %s" % url
+    r = requests.delete(url, headers=headers, auth=HTTPBasicAuth(USERNAME, PASSWORD))
+    if debug == True:
+        print r.text
+    r.raise_for_status()
+
+def post(host, port, uri, data, debug=False):
+    '''Perform a POST rest operation, using the URL and data provided'''
+    url='http://'+host+":"+port+uri
+    headers = {'Content-type': 'application/yang.data+json',
+               'Accept': 'application/yang.data+json'}
+    if debug == True:
+        print "POST %s" % url
+        print json.dumps(data, indent=4, sort_keys=True)
+    r = requests.post(url, data=json.dumps(data), headers=headers, auth=HTTPBasicAuth(USERNAME, PASSWORD))
+    if debug == True:
+        print r.text
+    r.raise_for_status()
+
+def get_service_functions_uri():
+    return "/restconf/config/service-function:service-functions"
+
+def get_service_function_forwarders_uri():
+    return "/restconf/config/service-function-forwarder:service-function-forwarders"
+
+def get_service_function_chains_uri():
+    return "/restconf/config/service-function-chain:service-function-chains/"
+
+def get_service_function_paths_uri():
+    return "/restconf/config/service-function-path:service-function-paths/"
+
+def get_tenant_uri():
+    return "/restconf/config/policy:tenants/policy:tenant/f5c7d344-d1c7-4208-8531-2c2693657e12"
+
+def get_tunnel_uri():
+    return "/restconf/config/opendaylight-inventory:nodes"
+
+def get_endpoint_uri():
+    return "/restconf/operations/endpoint:unregister-endpoint"
+
+def get_topology_uri():
+    return "/restconf/config/network-topology:network-topology/topology/ovsdb:1"
+
+if __name__ == "__main__":
+    # Launch main menu
+
+
+    # Some sensible defaults
+    controller=os.environ.get('ODL')
+    if controller == None:
+        sys.exit("No controller set.")
+    else:
+       print "Contacting controller at %s" % controller
+
+    resp=get(controller,DEFAULT_PORT,'/restconf/operational/endpoint:endpoints')
+    resp_json=json.loads(resp.text)
+    l2_eps=resp_json['endpoints']['endpoint']
+    l3_eps=resp_json['endpoints']['endpoint-l3']
+
+    print "deleting service function paths"
+    rest_delete(controller, DEFAULT_PORT, get_service_function_paths_uri(), True)
+
+    print "deleting service function chains"
+    rest_delete(controller, DEFAULT_PORT, get_service_function_chains_uri(), True)
+
+    print "deleting service functions"
+    rest_delete(controller, DEFAULT_PORT, get_service_functions_uri(), True)
+
+    print "deleting service function forwarders"
+    rest_delete(controller, DEFAULT_PORT, get_service_function_forwarders_uri(), True)
+
+    print "deleting tunnel"
+    rest_delete(controller, DEFAULT_PORT, get_tunnel_uri(), True)
+
+    print "deleting tenant"
+    rest_delete(controller, DEFAULT_PORT, get_tenant_uri(), True)
+
+    print "unregistering L2 endpoints"
+    for endpoint in l2_eps:
+       data={ "input": { "l2": [ { "l2-context": endpoint['l2-context'] ,"mac-address": endpoint['mac-address'] } ] } }
+        post(controller, DEFAULT_PORT, get_endpoint_uri(),data,True)
+
+    print "unregistering L3 endpoints"
+    for endpoint in l3_eps:
+       data={ "input": { "l3": [ { "l3-context": endpoint['l3-context'] ,"ip-address": endpoint['ip-address'] } ] } }
+        post(controller, DEFAULT_PORT, get_endpoint_uri(),data,True)
+
+    print "topology removed check"
+    resp=get(controller, DEFAULT_PORT, get_topology_uri())
+    topology_json=json.loads(resp.text)
+    if resp.status_code == requests.codes.ok:
+        print "WARNING: Topology %s has not been removed! Removing now..." % get_topology_uri()
+        rest_delete(controller, DEFAULT_PORT, get_topology_uri(), True)