Merge branch 'master' of github.com:rayhogenson/bridge

1 files changed, 12 insertions, 3 deletions

diff --git a/bridge.py b/bridge.py
index 76f3043..7271db8 100644
--- a/bridge.py
+++ b/bridge.py
@@ -1,11 +1,17 @@
-#!/usr/bin/python
+#!/usr/bin/python2
 """Simple demo of oscillating colors."""
 from __future__ import division, unicode_literals
 import color
 import constants
+from DummyRig import DummyRig
 import fish_render
-import lumiversepython
 import time
+try:
+    import lumiversepython
+except ImportError:
+    have_lumiversepython = False
+else:
+    have_lumiversepython = True
 
 def get_all_lights(rig):
     """Return a list of all panels."""
@@ -28,7 +34,10 @@ def demo(lights, rig):
 
 def main():
     """Run show."""
-    rig = lumiversepython.Rig('/home/teacher/Lumiverse/PBridge.rig.json')
+    if have_lumiversepython:
+        rig = lumiversepython.Rig('/home/teacher/Lumiverse/PBridge.rig.json')
+    else:
+        rig = DummyRig(200, 2)
     rig.init()
     lights = get_all_lights(rig)
     demo(lights, rig)