Pavlin Radoslavov | da7ef61 | 2013-10-30 16:12:14 -0700 | [diff] [blame] | 1 | #! /usr/bin/env python |
| 2 | # -*- Mode: python; py-indent-offset: 4; tab-width: 8; indent-tabs-mode: t; -*- |
| 3 | |
| 4 | import pprint |
| 5 | import os |
| 6 | import sys |
| 7 | import subprocess |
| 8 | import json |
| 9 | import argparse |
| 10 | import io |
| 11 | import time |
| 12 | |
| 13 | from flask import Flask, json, Response, render_template, make_response, request |
| 14 | |
| 15 | ## Global Var ## |
| 16 | ControllerIP="127.0.0.1" |
| 17 | ControllerPort=8080 |
| 18 | |
| 19 | DEBUG=0 |
| 20 | pp = pprint.PrettyPrinter(indent=4) |
| 21 | |
| 22 | app = Flask(__name__) |
| 23 | |
| 24 | ## Worker Functions ## |
| 25 | def log_error(txt): |
| 26 | print '%s' % (txt) |
| 27 | |
| 28 | def debug(txt): |
| 29 | if DEBUG: |
| 30 | print '%s' % (txt) |
| 31 | |
Naoki Shiota | 862cc3b | 2013-12-13 15:42:50 -0800 | [diff] [blame] | 32 | # @app.route("/wm/onos/datagrid/get/map/<map-name>/json ") |
Pavlin Radoslavov | da7ef61 | 2013-10-30 16:12:14 -0700 | [diff] [blame] | 33 | # Sample output: |
| 34 | |
| 35 | def print_datagrid_map(parsedResult): |
| 36 | print '%s' % (parsedResult) |
| 37 | |
| 38 | def get_datagrid_map(map_name): |
| 39 | try: |
Naoki Shiota | 862cc3b | 2013-12-13 15:42:50 -0800 | [diff] [blame] | 40 | command = "curl -s \"http://%s:%s/wm/onos/datagrid/get/map/%s/json\"" % (ControllerIP, ControllerPort, map_name) |
Pavlin Radoslavov | da7ef61 | 2013-10-30 16:12:14 -0700 | [diff] [blame] | 41 | debug("get_datagrid_map %s" % command) |
| 42 | |
| 43 | result = os.popen(command).read() |
| 44 | debug("result %s" % result) |
| 45 | if len(result) == 0: |
| 46 | print "No Map found" |
| 47 | return; |
| 48 | |
| 49 | # TODO: For now, the string is not JSON-formatted |
| 50 | # parsedResult = json.loads(result) |
| 51 | parsedResult = result |
| 52 | debug("parsed %s" % parsedResult) |
| 53 | except: |
| 54 | log_error("Controller IF has issue") |
| 55 | exit(1) |
| 56 | |
| 57 | print_datagrid_map(parsedResult) |
| 58 | |
| 59 | |
| 60 | if __name__ == "__main__": |
| 61 | usage_msg1 = "Usage:\n" |
| 62 | usage_msg2 = "%s <map_name> : Print datagrid map with name of <map_name>\n" % (sys.argv[0]) |
| 63 | usage_msg3 = " Valid map names:\n" |
| 64 | usage_msg4 = " all : Print all maps\n" |
| 65 | usage_msg5 = " flow : Print all flows\n" |
| 66 | usage_msg6 = " flow-entry : Print all flow entries\n" |
| 67 | usage_msg7 = " topology : Print the topology\n" |
| 68 | usage_msg = usage_msg1 + usage_msg2 + usage_msg3 + usage_msg4 + usage_msg5 |
| 69 | usage_msg = usage_msg + usage_msg6 + usage_msg7 |
| 70 | |
| 71 | # app.debug = False; |
| 72 | |
| 73 | # Usage info |
| 74 | if len(sys.argv) > 1 and (sys.argv[1] == "-h" or sys.argv[1] == "--help"): |
| 75 | print(usage_msg) |
| 76 | exit(0) |
| 77 | |
| 78 | # Check arguments |
| 79 | if len(sys.argv) < 2: |
| 80 | log_error(usage_msg) |
| 81 | exit(1) |
| 82 | |
| 83 | # Do the work |
| 84 | get_datagrid_map(sys.argv[1]) |