Multi-proc local ursula fleet script

examples/run_demo_ursulas.py

@@ -0,0 +1,74 @@
+"""
+This file is part of nucypher.
+
+nucypher is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+nucypher is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with nucypher.  If not, see <https://www.gnu.org/licenses/>.
+"""
+
+
+# WARNING This is not a mining script!
+# you will not perform any re-encryptions, and you will not get paid.
+# It might be (but might not be) useful for determining whether you have
+# the proper dependencies and configuration to run an actual mining node.
+
+
+import os
+
+from twisted.internet import protocol
+from twisted.internet import reactor
+from twisted.logger import globalLogPublisher
+
+from nucypher.utilities.logging import SimpleObserver
+from nucypher.utilities.sandbox.constants import MOCK_URSULA_STARTING_PORT, select_test_port
+
+
+def spin_up_federated_ursulas(quantity: int = 2):
+
+    globalLogPublisher.addObserver(SimpleObserver())
+
+    starting_port = select_test_port()
+    ports = map(str, range(starting_port, starting_port + quantity))
+    ursulas, ursula_processes = set(), list()
+    for index, port in enumerate(ports):
+
+        executable = 'nucypher'
+        args = ['nucypher', 'ursula', 'run',
+                '--federated-only', '--rest-port', port,
+                '--dev', '--debug']
+
+        if index != 0:    # Skip first iteration
+            args.extend(['--teacher-uri', 'https://127.0.0.1:{}'.format(int(port)-1)])
+
+        env = {'PATH': os.environ['PATH'],
+               'NUCYPHER_SENTRY_LOGS': '0',
+               'NUCYPHER_FILE_LOGS': '0',
+               'LC_ALL': 'C.UTF-8',
+               'LANG': 'C.UTF-8'}
+
+        childFDs = {0: 0, 1: 1, 2: 2}
+
+        class UrsulaProcessProtocol(protocol.Protocol):
+
+            def __init__(self, command):
+                self.command = command
+
+        processProtocol = UrsulaProcessProtocol(command=args)
+        p = reactor.spawnProcess(processProtocol, executable, args, env=env, childFDs=childFDs)
+        ursula_processes.append(p)
+
+    reactor.run()
+
+
+if __name__ == "__main__":
+    spin_up_federated_ursulas()
+