diff --git a/rebound/rebound.py b/rebound/rebound.py index 850a018..ff6ed67 100644 --- a/rebound/rebound.py +++ b/rebound/rebound.py @@ -800,6 +800,15 @@ def _stylize_question(self, title, desc, stats): ## Helper Functions ## +# May not be necessary but I found that my environment needed this to be set up +# I don't know if this is needed for other OS +def windows_color(): + """Checks OS and sets Windows to utilize colored text in cmd.""" + opSys = sys.platform + + if opSys == "win32": + os.system('color') # Returns 1 when ran in python cmd but it works + def confirm(question): """Prompts a given question and handles user input.""" @@ -830,6 +839,7 @@ def print_help(): def main(): + windows_color() if len(sys.argv) == 1 or sys.argv[1].lower() == "-h" or sys.argv[1].lower() == "--help": print_help() elif sys.argv[1].lower() == "-q" or sys.argv[1].lower() == "--query":