summaryrefslogtreecommitdiffstats
path: root/verigraph/service/src/tests/j-verigraph-generator/utility.py
diff options
context:
space:
mode:
authorZhipeng (Howard) Huang <huangzhipeng@huawei.com>2017-02-27 13:08:54 +0000
committerGerrit Code Review <gerrit@opnfv.org>2017-02-27 13:08:54 +0000
commit69e33063b3703ae4529b556b63b3c4cc239c3d9a (patch)
treeb30d14715b34f2e4e06928c555c940ddb3365534 /verigraph/service/src/tests/j-verigraph-generator/utility.py
parent280310852bb5e1c7639dedf7ce268e1e66157a51 (diff)
parent53d83244c1bf36af86e90ce5fe758a369f73563e (diff)
Merge "Add verigraph code base"
Diffstat (limited to 'verigraph/service/src/tests/j-verigraph-generator/utility.py')
-rw-r--r--verigraph/service/src/tests/j-verigraph-generator/utility.py257
1 files changed, 257 insertions, 0 deletions
diff --git a/verigraph/service/src/tests/j-verigraph-generator/utility.py b/verigraph/service/src/tests/j-verigraph-generator/utility.py
new file mode 100644
index 0000000..47d0180
--- /dev/null
+++ b/verigraph/service/src/tests/j-verigraph-generator/utility.py
@@ -0,0 +1,257 @@
+#!/usr/bin/python
+
+##############################################################################
+# Copyright (c) 2017 Politecnico di Torino and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+import json
+import contextlib
+import sys
+import os
+import subprocess
+from pprint import pprint
+
+#manages output easily (can either write to file or to stdout)
+@contextlib.contextmanager
+def smart_open(filename=None):
+ if filename and filename != '-':
+ fh = open(filename, 'w')
+ else:
+ fh = sys.stdout
+ try:
+ yield fh
+ finally:
+ if fh is not sys.stdout:
+ fh.close()
+
+def check_input_is_int(text):
+ while True:
+ data = raw_input(text)
+ try:
+ int_value = int(data)
+ except ValueError:
+ print "Please enter a valid number!"
+ continue
+ return int_value
+
+#parses a json file into a unicode dictionary
+def parse_json_file(filename):
+ with open(filename) as json_file:
+ return json.load(json_file)
+
+#returns an ascii dictionary from a unicode one
+def convert_unicode_to_ascii(input):
+ if isinstance(input, dict):
+ return {convert_unicode_to_ascii(key): convert_unicode_to_ascii(value) for key, value in input.iteritems()}
+ elif isinstance(input, list):
+ return [convert_unicode_to_ascii(element) for element in input]
+ elif isinstance(input, unicode):
+ return input.encode('utf-8')
+ else:
+ return input
+
+#parses a chains file
+def parse_chains(chains_file):
+ chains_json = convert_unicode_to_ascii(parse_json_file(chains_file))
+
+ chains = {}
+
+ for chn in chains_json["chains"]:
+ try:
+ chains[chn["id"]] = {}
+ #initiatlize the config dictionary for each node
+ for node in chn["nodes"]:
+ chains[chn["id"]][node["name"]] = {}
+ except:
+ raise KeyError("Chains file is not valid!")
+
+ for chn in chains_json["chains"]:
+ try:
+ #set chn values ---> chn(name, (field, value))
+ for node in chn["nodes"]:
+ for key, value in node.items():
+ #name key is redundant in map
+ if key != "name":
+ chains[chn["id"]][node["name"]][key] = value
+ except:
+ raise KeyError("Chains file is not valid!")
+ return chains
+
+def check_chains_integrity(filename):
+ print "Checking input file..."
+ try:
+ chains = convert_unicode_to_ascii(parse_json_file(filename))
+ print "File correctly parsed"
+ if isinstance(chains["chains"], list) == False:
+ print "Child of chains is not a list!"
+ return False
+ for chain in chains["chains"]:
+ print "Chain found, checking its fields..."
+ print "Checking chain id field... "
+ chain["id"]
+ print "OK!"
+ print "Checking chain flowspace field... "
+ chain["flowspace"]
+ print "OK!"
+ if isinstance(chain["nodes"], list) == False:
+ print "Chain #" + chain["id"] + " does not have a list of nodes!"
+ return False
+ for node in chain["nodes"]:
+ print "Node found, checking its fields..."
+ print "Checking node name... "
+ node["name"]
+ print "OK!"
+ print "Checking node functional_type field... "
+ node["functional_type"]
+ print "OK!"
+ print "Checking node address field... "
+ node["address"]
+ print "OK!"
+ except (KeyboardInterrupt, SystemExit):
+ raise
+ except:
+ print "One or more required fields are missing!"
+ return False
+ print filename + " validated successfully!"
+ return True
+
+def check_config_integrity(filename):
+ print "Checking input file..."
+ try:
+ config = convert_unicode_to_ascii(parse_json_file(filename))
+ pprint(config)
+ print "File correctly parsed"
+ if isinstance(config["nodes"], list) == False:
+ print "Child of nodes is not a list!"
+ return False
+ for node in config["nodes"]:
+ print "Node found, checking its fields..."
+ print "Checking id field... "
+ node["id"]
+ print "OK!"
+ print "Checking description field... "
+ node["description"]
+ print "OK!"
+ print "Checking configuration field... "
+ node["configuration"]
+ print "OK!"
+ if isinstance(node["configuration"], list) == False:
+ print "Checking if node configuration is a list..."
+ print "Node with id " + node["id"] + " does not have a configuration list!"
+ return False
+ for c in node["configuration"]:
+ print "Checking if node configuration element is a string or a dictionary..."
+ if (isinstance(c, str) == False and isinstance(c, dict) == False):
+ print "At least one element of node with id " + node["id"] + " has an invalid configuration (it is neither a string or a map)"
+ return False
+ except (KeyboardInterrupt, SystemExit):
+ raise
+ except:
+ print "One or more required fields are missing!"
+ return False
+ print filename + " validated successfully!"
+ return True
+
+def check_routing_integrity(filename):
+ print "Checking input file..."
+ try:
+ routing = convert_unicode_to_ascii(parse_json_file(filename))
+ print "File correctly parsed"
+ if isinstance(routing["routing_table"], list) == False:
+ print "Child of routing_table is not a list!"
+ return False
+ for node in routing["routing_table"]:
+ if isinstance(node, dict) == False:
+ print "Child of routing_table is not a map!"
+ return False
+ for n, rt in node.items():
+ if isinstance(rt, list) == False:
+ print "Routing table of element " + n + " is not a list!"
+ return False
+ for entry in rt:
+ if isinstance(entry, dict) == False:
+ print "Invalid entry for node " + n + " (not a map)!"
+ return False
+ except (KeyboardInterrupt, SystemExit):
+ raise
+ except:
+ print "One or more required fields are missing!"
+ return False
+ return True
+
+#prints every node for each input chain
+def print_chains(chains):
+ for chain in chains["chains"]:
+ print "CHAIN #" + str(chain["id"])
+ for node in chain["nodes"]:
+ print "Name: " + str(node["name"])
+ print "Functional type: " + str(node["functional_type"])
+ print "Address: " + str(node["address"])
+ print "-----------------------------------"
+ print ""
+
+#prints every node's configuration
+def print_configuration(configuration):
+ print "NODES CONFIGURATION"
+ for node in configuration["nodes"]:
+ print "Name: " + str(node["id"])
+ print "Description: " + str(node["description"])
+ print "Configuration: "
+ pprint(node["configuration"])
+ print "-----------------------------------"
+ print ""
+
+#print every node's routing table
+def print_routing_table(routing):
+ print "ROUTING"
+ for table in routing["routing_table"]:
+ for node,rt in table.items():
+ print "Name: " + str(node)
+ pprint(rt)
+ print "-----------------------------------"
+ print ""
+
+#returns a list of tuple [(k1, v1), (k2, v2)] from a list of maps like [{k1 : v1},{k2 : v2}]
+def formatted_list_from_list_of_maps(maps):
+ l = []
+ for map in maps:
+ if isinstance(map, dict):
+ for k, v in map.items():
+ #l.append("(ctx." + str(k) + ", ctx." + str(v) + ")")
+ l.append(str(k))
+ l.append(str(v))
+ else:
+ #l.append("ctx." + map)
+ l.append(map)
+ return l
+
+def list_directories(dir):
+ #output = subprocess.call(["ls", "-d", "*/"])
+ output = subprocess.call(["find", dir, "-type", "d"])
+
+ #TREE VERSION
+ #find = subprocess.Popen(["find", ".", "-type", "d"], stdout=subprocess.PIPE)
+ #output = subprocess.check_output(["sed", "-e", "s/[^-][^\/]*\// |/g", "-e", "s/|\([^ ]\)/|-\1/"], stdin=find.stdout)
+ #find.wait()
+
+# ps = subprocess.Popen(('ps', '-A'), stdout=subprocess.PIPE)
+# output = subprocess.check_output(('grep', 'process_name'), stdin=ps.stdout)
+# ps.wait()
+ return output
+
+def list_files(dir):
+ output = subprocess.call(["find", dir, "-type", "f"])
+ return output
+
+def search_node_in_chains(n):
+ found = []
+ for chain in chains["chains"]:
+ for node in chain["nodes"]:
+ if node["name"] == n:
+ found.append(node)
+ return found \ No newline at end of file