[WORK IN PROGRESS] Add support to manage multiple switches

.dockerignore

@@ -1 +1,2 @@
 Dockerfile
+venv/

.gitignore

@@ -1,3 +1,4 @@
 *~
 *.pyc
 .coverage
+venv/

README.md

@@ -272,22 +272,24 @@ import p4runtime_sh.shell as sh
 
 # you can omit the config argument if the switch is already configured with the
 # correct P4 dataplane.
-sh.setup(
+client = sh.setup(
     device_id=1,
     grpc_addr='localhost:50051',
     election_id=(0, 1), # (high, low)
-    config=sh.FwdPipeConfig('config/p4info.pb.txt', 'config/device_config.bin')
+    config=sh.FwdPipeConfig('config/p4info.pb.txt', 'config/device_config.bin'),
+    # This enables controlling multiple switches
+    set_global_client=False
 )
 
 # see p4runtime_sh/test.py for more examples
-te = sh.TableEntry('<table_name>')(action='<action_name>')
+te = sh.TableEntry('<table_name>', client=client)(action='<action_name>')
 te.match['<name>'] = '<value>'
 te.action['<name>'] = '<value>'
 te.insert()
 
 # ...
 
-sh.teardown()
+sh.teardown(client)
 ```
 
 Note that at the moment the P4Runtime client object is a global variable, which

p4runtime_sh/p4runtime.py

@@ -26,6 +26,8 @@
 from p4.v1 import p4runtime_pb2
 from p4.v1 import p4runtime_pb2_grpc
 
+from p4runtime_sh.context import Context
+
 
 class P4RuntimeErrorFormatException(Exception):
     def __init__(self, message):
@@ -137,6 +139,7 @@ def handle(*args, **kwargs):
 
 class P4RuntimeClient:
     def __init__(self, device_id, grpc_addr, election_id):
+        self.context = Context()
         self.device_id = device_id
         self.election_id = election_id
         logging.debug("Connecting to device {} at {}".format(device_id, grpc_addr))