docker version of heartbeat demo

examples/heartbeat_demo/alicia.py

@@ -1,4 +1,5 @@
 import datetime
+import sys
 import json
 import os
 import shutil
@@ -23,7 +24,15 @@ globalLogPublisher.addObserver(SimpleObserver())
 
 TEMP_ALICE_DIR = os.path.join('/', 'tmp', 'heartbeat-demo-alice')
 
-SEEDNODE_URL = 'localhost:11500'
+
+# if your ursulas are NOT running on your current host,
+# run like this: python finnegans-wake-demo.py 172.28.1.3:11500
+# otherwise the default will be fine.
+
+try:
+    SEEDNODE_URI = sys.argv[1]
+except IndexError:
+    SEEDNODE_URI = "localhost:11500"
 
 POLICY_FILENAME = "policy-metadata.json"
 
@@ -42,7 +51,7 @@ passphrase = "TEST_ALICIA_INSECURE_DEVELOPMENT_PASSWORD"
 
 shutil.rmtree(TEMP_ALICE_DIR, ignore_errors=True)
 
-ursula = Ursula.from_seed_and_stake_info(seed_uri=SEEDNODE_URL,
+ursula = Ursula.from_seed_and_stake_info(seed_uri=SEEDNODE_URI,
                                          federated_only=True,
                                          minimum_stake=0)
 

examples/heartbeat_demo/doctor.py

@@ -1,5 +1,6 @@
 import json
 import os
+import sys
 import shutil
 import msgpack
 import maya
@@ -25,7 +26,11 @@ globalLogPublisher.addObserver(SimpleObserver())
 # Boring setup stuff #
 ######################
 
-SEEDNODE_URL = 'localhost:11501'
+try:
+    SEEDNODE_URI = sys.argv[1]
+except IndexError:
+    SEEDNODE_URI = "localhost:11500"
+
 
 # TODO: path joins?
 TEMP_DOCTOR_DIR = "{}/doctor-files".format(os.path.dirname(os.path.abspath(__file__)))
@@ -33,7 +38,7 @@ TEMP_DOCTOR_DIR = "{}/doctor-files".format(os.path.dirname(os.path.abspath(__fil
 # Remove previous demo files and create new ones
 shutil.rmtree(TEMP_DOCTOR_DIR, ignore_errors=True)
 
-ursula = Ursula.from_seed_and_stake_info(seed_uri=SEEDNODE_URL,
+ursula = Ursula.from_seed_and_stake_info(seed_uri=SEEDNODE_URI,
                                          federated_only=True,
                                          minimum_stake=0)
 

scripts/demos/run_heartbeat_demo_docker.sh

@@ -0,0 +1,23 @@
+#!/usr/bin/env bash
+
+set -e
+echo "Starting Up Heartbeat Demo Test..."
+
+COMPOSE_FILE="${0%/*}/../../dev/docker/8-federated-ursulas.yml"
+DEMO_DIR="/code/examples/heartbeat_demo/"
+
+# run some ursulas
+docker-compose -f $COMPOSE_FILE up -d
+echo "Wait for Ursula learning to occur"
+sleep 5
+
+# Run Alicia
+echo "Starting Alicia..."
+docker-compose -f $COMPOSE_FILE run -w $DEMO_DIR nucypher-dev python3 alicia.py 172.28.1.3:11500
+
+# Run Dr. Bob
+echo "Starting Bob..."
+docker-compose -f $COMPOSE_FILE run -w $DEMO_DIR nucypher-dev python3 doctor.py 172.28.1.3:11500
+
+# tear it down
+docker-compose -f $COMPOSE_FILE stop