diff --git a/docs/advanced.md b/docs/advanced.md index e5f89139..2c14f84d 100644 --- a/docs/advanced.md +++ b/docs/advanced.md @@ -33,7 +33,7 @@ and can be edited with any plain text editor. - `linewrap` controls the width of the output. Set to `false` if you don't want to wrap long lines. - `colors` - dictionary that controls the colors used to display journal entries. It has two subkeys, which are: `date` and `title`. Current valid values are: `BLACK`, `RED`, `GREEN`, `YELLOW`, `BLUE`, `MAGENTA`, `CYAN`, and `WHITE`. `colorama.Fore` is used for colorization, and you can find the [docs here](https://github.com/tartley/colorama#colored-output). To disable colored output, set the value to `NONE`. + dictionary that controls the colors used to display journal entries. It has two subkeys, which are: `date` and `title`. Current valid values are: `BLACK`, `RED`, `GREEN`, `YELLOW`, `BLUE`, `MAGENTA`, `CYAN`, and `WHITE`. `colorama.Fore` is used for colorization, and you can find the [docs here](https://github.com/tartley/colorama#colored-output). To disable colored output, set the value to `NONE`. If you set the value of any color subkey to an invalid color, no color will be used. !!! note Although it seems intuitive to use the `#` diff --git a/jrnl/util.py b/jrnl/util.py index dae66a07..d4196f50 100644 --- a/jrnl/util.py +++ b/jrnl/util.py @@ -198,14 +198,12 @@ def get_text_from_editor(config, template=""): def colorize(string, color, bold=False): """Returns the string colored with colorama.Fore.color. If the color set by - the user is "NONE", it returns the string without any modification. Otherwise, - If the color set by the user doesn't exist in the colorama.Fore attributes, - the colorization is done with WHITE.""" - upper_color = color.upper() - if upper_color == "NONE": + the user is "NONE" or the color doesn't exist in the colorama.Fore attributes, + it returns the string without any modification.""" + color_escape = getattr(colorama.Fore, color.upper(), None) + if not color_escape: return string - color_escape = getattr(colorama.Fore, upper_color, colorama.Fore.WHITE) if not bold: return color_escape + string + colorama.Fore.RESET else: