fix pull request 2

[rainbowstream.git] / rainbowstream / draw.py

diff --git a/rainbowstream/draw.py b/rainbowstream/draw.py
index a6f3e565a6c3f8cad63d3067a54629ceab8aa3d4..27cc8746865e784f97d0bace7289831c9f734423 100644 (file)
--- a/rainbowstream/draw.py
+++ b/rainbowstream/draw.py
@@ -31,6 +31,13 @@ def init_cycle():
 g['cyc'] = init_cycle()
 
 
+def notify_cycle():
+    """
+    Notify from rainbow
+    """
+    g['cyc'] = init_cycle()
+
+
 def order_rainbow(s):
     """
     Print a string with ordered color with each character
@@ -94,7 +101,7 @@ def check_theme():
                 os.environ.get('USERPROFILE',
                 '')) + os.sep + '.rainbow_config.json'
         else:
-            config = 'rainbowstream/colorset/'+c['theme']+'.json'
+            config = os.path.dirname(__file__) + '/colorset/'+c['theme']+'.json'
         # Load new config
         data = load_config(config)
         if data: