From cdadd3c0c9073b3677b0d11dcd2a014e578f3ba1 Mon Sep 17 00:00:00 2001 From: Igor Chubin Date: Sun, 5 Aug 2018 18:56:42 +0000 Subject: [PATCH] minor fixes after refactoring (#79) --- lib/colorize_internal.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/lib/colorize_internal.py b/lib/colorize_internal.py index 2e7dd41..2d709ac 100644 --- a/lib/colorize_internal.py +++ b/lib/colorize_internal.py @@ -31,17 +31,19 @@ PALETTES = { def _reverse_palette(code): return { 1 : Fore.BLACK + _back_color(code), - 2 : Style.DIM - } + 2 : Style.DIM + } def _back_color(code): - if code == 0 or code.lower()=="white": + if code == 0 or (isinstance(code, str) and code.lower() == "white"): return Back.WHITE - if code == 1 or code.lower()=="cyan": + if code == 1 or (isinstance(code, str) and code.lower() == "cyan"): return Back.CYAN - if code == 2 or code.lower()=="red": + if code == 2 or (isinstance(code, str) and code.lower() == "red"): return Back.RED + return Back.WHITE + def colorize_internal(text, palette_number=1): """ Colorize `text`, use `palette` @@ -58,12 +60,12 @@ def colorize_internal(text, palette_number=1): stripped = text.lstrip('0123456789') return (text, stripped, factor) - def _extract_color_number(text,stripped,factor = 1): + def _extract_color_number(text, stripped, factor=1): return int(text[:len(text)-len(stripped)])*factor def _colorize_curlies_block(text): text, stripped, factor = _process_text(text) - color_number = _extract_color_number(text,stripped,factor) + color_number = _extract_color_number(text, stripped, factor) if stripped.startswith('='): stripped = stripped[1:]