From da498c46c3c3aedd8be443bb4a33cfdf3967cb52 Mon Sep 17 00:00:00 2001 From: Hykilpikonna Date: Mon, 27 Dec 2021 11:42:39 -0500 Subject: [PATCH] [+] Colors --- hypy_utils/__init__.py | 51 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) diff --git a/hypy_utils/__init__.py b/hypy_utils/__init__.py index 5becc17..3fa8dc3 100644 --- a/hypy_utils/__init__.py +++ b/hypy_utils/__init__.py @@ -1 +1,52 @@ __version__ = "1.0.0" + + +def ansi_rgb(r: int, g: int, b: int, foreground: bool = True) -> str: + """ + Convert rgb color into ANSI escape code format + + :param r: + :param g: + :param b: + :param foreground: Whether the color applies to forground + :return: Escape code + """ + c = '38' if foreground else '48' + return f'\033[{c};2;{r};{g};{b}m' + + +def color(msg: str) -> str: + """ + Replace extended minecraft color codes in string + + :param msg: Message with minecraft color codes + :return: Message with escape codes + """ + replacements = ["&0/\033[0;30m", "&1/\033[0;34m", "&2/\033[0;32m", "&3/\033[0;36m", "&4/\033[0;31m", "&5/\033[0;35m", "&6/\033[0;33m", "&7/\033[0;37m", "&8/\033[1;30m", "&9/\033[1;34m", "&a/\033[1;32m", "&b/\033[1;36m", "&c/\033[1;31m", "&d/\033[1;35m", "&e/\033[1;33m", "&f/\033[1;37m", "&r/\033[0m", "&n/\n"] + for r in replacements: + msg = msg.replace(r[:2], r[3:]) + + while '&gf(' in msg or '&gb(' in msg: + i = msg.index('&gf(') if '&gf(' in msg else msg.index('&gb(') + end = msg.index(')', i) + code = msg[i + 4:end] + fore = msg[i + 2] == 'f' + + if code.startswith('#'): + rgb = tuple(int(code.lstrip('#')[i:i+2], 16) for i in (0, 2, 4)) + else: + code = code.replace(',', ' ').replace(';', ' ').replace(' ', ' ') + rgb = tuple(int(c) for c in code.split(' ')) + + msg = msg[:i] + ansi_rgb(*rgb, foreground=fore) + msg[end + 1:] + + return msg + + +def printc(msg: str): + """ + Print with color + + :param msg: Message with minecraft color codes + """ + print(color(msg + '&r'))