Updated Examples to support python3

Author: kkumara3

examples/cli.py

@@ -1,4 +1,4 @@
-""" 
+"""
 GRPC Client with command line options
 
 Usage:
@@ -26,14 +26,13 @@
 import sys
 sys.path.insert(0, '../')
 from lib.cisco_grpc_client import CiscoGRPCClient
-
 from docopt import docopt
 
 def main():
 
     __version__ = 'GRPC_Client 1.0'
     arguments = docopt(__doc__, version=__version__)
-    
+
     IP = arguments['<router_IP>']
     TCP_PORT = int(arguments['<port>'])
     user = arguments['<user>']
@@ -51,13 +50,13 @@ def main():
         else:
             path = 'Error'
             print(
-                'get-oper argument must include --file option and json file to filter yang operational namespace' 
+                'get-oper argument must include --file option and json file to filter yang operational namespace'
                 )
         try:
             err, result = client.getoper(path)
             if err:
-                print err
-            print result
+                print(err)
+            print(result)
         except AbortionError:
             print(
                 'Unable to connect to local box, check your gRPC destination.'
@@ -69,14 +68,14 @@ def main():
             file = arguments['--file']
             path = open(file).read()
 #            path = open('json/' + file).read()
-        else:    
-            path = "" 
+        else:
+            path = ""
 
         try:
             err, result = client.getconfig(path)
             if err:
-                print err
-            print result
+                print(err)
+            print(result)
         except AbortionError:
             print(
                 'Unable to connect to local box, check your gRPC destination.'
@@ -95,7 +94,7 @@ def main():
         try:
             err = client.mergeconfig(path)
             if err:
-                print err
+                print(err)
             #print result
         except AbortionError:
             print(
@@ -115,7 +114,7 @@ def main():
         try:
             err = client.replaceconfig(path)
             if err:
-                print err
+                print(err)
         except AbortionError:
             print(
                 'Unable to connect to local box, check your gRPC destination.'

examples/grpc_cfg.py

@@ -20,8 +20,8 @@ def get(self):
         try:
             err, result = self.client.getconfig(path)
             if err:
-                print err
-            print json.dumps(json.loads(result))
+                print(err)
+            print(json.dumps(json.loads(result)))
         except AbortionError:
             print(
                 'Unable to connect to local box, check your gRPC destination.'
@@ -33,7 +33,7 @@ def replace(self):
             response = self.client.replaceconfig(path)
             if response.errors:
                 err = json.loads(response.errors)
-                print err
+                print(err)
         except AbortionError:
             print(
                 'Unable to connect to local box, check your gRPC destination.'
@@ -45,7 +45,7 @@ def merge(self):
             response = self.client.mergeconfig(path)
             if response.errors:
                 err = json.loads(response.errors)
-                print err
+                print(err)
         except AbortionError:
             print(
                 'Unable to connect to local box, check your gRPC destination.'
@@ -57,7 +57,7 @@ def delete(self):
             response = self.client.deleteconfig(path)
             if response.errors:
                 err = json.loads(response.errors)
-                print err
+                print(err)
         except AbortionError:
             print(
                 'Unable to connect to local box, check your gRPC destination.'

examples/grpc_example.py

@@ -22,8 +22,8 @@ def main():
     try:
         err, result = client.getconfig(path)
         if err:
-            print err
-        print json.dumps(json.loads(result))
+            print(err)
+        print(json.dumps(json.loads(result)))
     except AbortionError:
         print(
             'Unable to connect to local box, check your gRPC destination.'

examples/route_policy.py

@@ -6,7 +6,10 @@
 4. It needs to use templating to create new policies and apply to upload them in the end.
 """
 import json
-from collections import OrderedDict
+try:
+    from collections import OrderedDict
+except ImportError:
+    from ordereddict import OrderedDict
 import sys
 sys.path.insert(0, '../')
 from lib.cisco_grpc_client import CiscoGRPCClient
@@ -36,7 +39,7 @@ def get_policies(self):
         path = '{"openconfig-routing-policy:routing-policy": [null]}'
         err, result = self.client.getconfig(path)
         if err:
-            print err
+            print(err)
         policies = json.loads(result, object_pairs_hook=OrderedDict)
         return policies
 
@@ -45,9 +48,9 @@ def list_policies(self):
         """
         bgp_policies = self.get_policies()
         policy_definitions = bgp_policies['openconfig-routing-policy:routing-policy']['policy-definitions']['policy-definition']
-        print '\nPolicy Names:\n'
+        print('\nPolicy Names:\n')
         for policy in policy_definitions:
-            print 'Name: %s' % policy['name']
+            print('Name: %s' % policy['name'])
 
     def detail_policy(self, policy_name):
         """Prints the full json of a policy in Openconfig and returns it
@@ -60,7 +63,7 @@ def detail_policy(self, policy_name):
         policy_definitions = bgp_policies['openconfig-routing-policy:routing-policy']['policy-definitions']['policy-definition']
         for policy in policy_definitions:
             if policy_name == policy['name']:
-                print json.dumps(policy, indent=4, separators=(',', ': '))
+                print(json.dumps(policy, indent=4, separators=(',', ': ')))
                 inner = json.dumps(policy)
         template = '{"openconfig-routing-policy:routing-policy": {"policy-definitions": {"policy-definition": [%s]}}}' % inner
         return json.dumps(json.loads(template, object_pairs_hook=OrderedDict), indent=4, separators=(',', ': '))
@@ -73,7 +76,7 @@ def get_neighbors(self):
         path = '{"openconfig-bgp:bgp": [null]}'
         err, result = self.client.getconfig(path)
         if err:
-            print err
+            print(err)
         bgp = json.loads(result, object_pairs_hook=OrderedDict)
         return bgp
 
@@ -82,9 +85,9 @@ def list_neighbors(self):
         """
         bgp = self.get_neighbors()
         bgp_neighbors = bgp['openconfig-bgp:bgp']['neighbors']['neighbor']
-        print "\nNeighbor's\n"
+        print("\nNeighbor's\n")
         for neighbor in bgp_neighbors:
-            print 'Neighbor: %s AS: %s' % (neighbor['neighbor-address'], neighbor['config']['peer-as'])
+            print('Neighbor: %s AS: %s' % (neighbor['neighbor-address'], neighbor['config']['peer-as']))
 
     def detail_neighbor(self, neighbor_address):
         """Prints the full json of a neighbor in Openconfig format
@@ -95,7 +98,7 @@ def detail_neighbor(self, neighbor_address):
         bgp_neighbors = bgp['openconfig-bgp:bgp']['neighbors']['neighbor']
         for neighbor in bgp_neighbors:
             if neighbor_address == neighbor['neighbor-address']:
-                print json.dumps(neighbor, indent=4, separators=(',', ': '))
+                print(json.dumps(neighbor, indent=4, separators=(',', ': ')))
                 inner = json.dumps(neighbor)
         template = '{"openconfig-bgp:bgp": {"neighbors": {"neighbor" :  [%s]}}}' % inner
         return json.dumps(json.loads(template, object_pairs_hook=OrderedDict), indent=4, separators=(',', ': '))
@@ -158,12 +161,12 @@ def neighbor_policy(self, neighbor_address, policy, direction):
                     # Add the removed neighbors to list.
                     updated_neighbors.append((neighbor['neighbor-address'], ip_type, curr_policy))
         updated_neighbors = json.dumps(updated_neighbors)
-        print updated_neighbors
+        print(updated_neighbors)
         bgp_config = json.dumps(bgp)
         err = self.merge_config(bgp_config)
         if not err:
-            print err
-        print '\nNew Neighbor Detail:\n'
+            print(err)
+        print('\nNew Neighbor Detail:\n')
         self.detail_neighbor(neighbor_address)
 
     def multiple_policies(self, policies, neighbor):
@@ -186,7 +189,7 @@ def multiple_policies(self, policies, neighbor):
             json_policy = json.loads(json_policy, object_pairs_hook=OrderedDict)
             conditions.append(json_policy)
         multi_policy = json.dumps(shell)
-        print self.merge_config(multi_policy)
+        print(self.merge_config(multi_policy))
         return policy_name
 
 def main():
@@ -197,19 +200,19 @@ def main():
     route = RoutePolicy('localhost', 57777, 'vagrant', 'vagrant')
     # List  Policies from Router
     route.list_policies()
-    print '\nnext function\n'
+    print('\nnext function\n')
     # Get the full json object of the policy, allows for manipulation
     policy = route.detail_policy('TEST')
-    print '\nnext function\n'
+    print('\nnext function\n')
     # List BGP neighbors
     route.list_neighbors()
-    print '\nnext function\n'
+    print('\nnext function\n')
     # Get the full json object of the BGP neighbor, allows for manipulation
     route.detail_neighbor('11.1.1.2')
-    print '\nnext function\n'
+    print('\nnext function\n')
     # Apply multiple policies to a single neighbor interface
     route.neighbor_policy('11.1.1.2', ['TEST', 'SEND-MED-IGP'], 'export-policy')
-    print '\nnext function\n'
+    print('\nnext function\n')
     # Store policy in json file to modify
     policy_file = open('policy.json', 'w')
     policy_file.write(policy)

examples/telemetry_collector.py

@@ -22,7 +22,7 @@ def print_connectivity(connectivity):
     # unmarshal is an optional argument, default is unmarshal = True
     # If unmarshal is false, out is in gpb k/v
 	recv_count += 1
-	print json.dumps(segment, indent=4, separators=(',', ': '))
-	print 'End Telemetry Segment'
-	print str(recv_count) + ' Segments Received'
+	print(json.dumps(segment, indent=4, separators=(',', ': ')))
+	print('End Telemetry Segment')
+	print(str(recv_count) + ' Segments Received')
 

lib/cisco_grpc_client.py

@@ -1,3 +1,4 @@
+#!bin/env
 # Copyright 2016 Cisco Systems All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
@@ -19,6 +20,7 @@
 # License for the specific language governing permissions and limitations under
 # the License.
 
+""" Libraries to connect XR gRPC server """
 
 import grpc
 from . import ems_grpc_pb2
@@ -58,7 +60,7 @@ def __init__(self, host, port, timeout, user, password, creds=None, options=None
             self._port = port
             self._channel = implementations.insecure_channel(self._host, self._port)
         self._stub = ems_grpc_pb2.beta_create_gRPCConfigOper_stub(self._channel)
-        self._timeout = int(timeout)
+        self._timeout = float(timeout)
         self._metadata = [('username', user), ('password', password)]
 
     def __repr__(self):