Make the new color imports work

Wensleydale · Feb 12, 2012 · fc6741f · fc6741f
1 parent 8eb5220
commit fc6741f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/build/rustic b/build/rustic
@@ -10,7 +10,10 @@ import urllib.request
 
 # Now set up colorama, accounting for cargo installation limitations
 sys.path.insert(0, '../')
-from build import init, Fore, Style
+try:
+    from build import init, Fore, Style
+except ImportError:
+    from bin import init, Fore, Style
 init()
 term_cyan = Fore.CYAN + Style.BRIGHT
 term_green = Fore.GREEN + Style.BRIGHT