Merge pull request #70 from polarkac/show_version

[rainbowstream.git] / rainbowstream / rainbow.py

diff --git a/rainbowstream/rainbow.py b/rainbowstream/rainbow.py
index 2b8a456571c676bbd477b1dee2d4a16ddf41171f..284a6282971d54ca320c76968830f530b43f8e04 100644 (file)
--- a/rainbowstream/rainbow.py
+++ b/rainbowstream/rainbow.py
@@ -178,6 +178,8 @@ def upgrade_center():
             notice += light_magenta('You should upgrade with ')
             notice += light_green('pip install -U rainbowstream')
             printNicely(notice)
+        else:
+            printNicely(yellow('You are running latest ' + current + ' version.'))
     except:
         pass
 
@@ -1778,7 +1780,8 @@ cmdset = [
     'p',
     'r',
     'c',
-    'q'
+    'q',
+    'version',
 ]
 
 # Handle function set
@@ -1824,7 +1827,8 @@ funcset = [
     pause,
     replay,
     clear,
-    quit
+    quit,
+    upgrade_center,
 ]
 
 
@@ -1903,6 +1907,7 @@ def listen():
             [],  # reconnect
             [],  # clear
             [],  # quit
+            [],  # version
         ]
     ))
     init_interactive_shell(d)