180 lines
3.7 KiB
Python
180 lines
3.7 KiB
Python
import datetime
|
|
import random
|
|
import string
|
|
|
|
WEECHAT_BASE_COLORS = {
|
|
"black": "0",
|
|
"red": "1",
|
|
"green": "2",
|
|
"brown": "3",
|
|
"blue": "4",
|
|
"magenta": "5",
|
|
"cyan": "6",
|
|
"default": "7",
|
|
"gray": "8",
|
|
"lightred": "9",
|
|
"lightgreen": "10",
|
|
"yellow": "11",
|
|
"lightblue": "12",
|
|
"lightmagenta": "13",
|
|
"lightcyan": "14",
|
|
"white": "15"
|
|
}
|
|
|
|
|
|
def color(color_name):
|
|
# type: (str) -> str
|
|
# yapf: disable
|
|
escape_codes = []
|
|
reset_code = "0"
|
|
|
|
def make_fg_color(color_code):
|
|
return "38;5;{}".format(color_code)
|
|
|
|
def make_bg_color(color_code):
|
|
return "48;5;{}".format(color_code)
|
|
|
|
attributes = {
|
|
"bold": "1",
|
|
"-bold": "21",
|
|
"reverse": "27",
|
|
"-reverse": "21",
|
|
"italic": "3",
|
|
"-italic": "23",
|
|
"underline": "4",
|
|
"-underline": "24",
|
|
"reset": "0",
|
|
"resetcolor": "39"
|
|
}
|
|
|
|
short_attributes = {
|
|
"*": "1",
|
|
"!": "27",
|
|
"/": "3",
|
|
"_": "4"
|
|
}
|
|
|
|
colors = color_name.split(",", 2)
|
|
|
|
fg_color = colors.pop(0)
|
|
|
|
bg_color = colors.pop(0) if colors else ""
|
|
|
|
if fg_color in attributes:
|
|
escape_codes.append(attributes[fg_color])
|
|
else:
|
|
chars = list(fg_color)
|
|
|
|
for char in chars:
|
|
if char in short_attributes:
|
|
escape_codes.append(short_attributes[char])
|
|
elif char == "|":
|
|
reset_code = ""
|
|
else:
|
|
break
|
|
|
|
stripped_color = fg_color.lstrip("*_|/!")
|
|
|
|
if stripped_color in WEECHAT_BASE_COLORS:
|
|
escape_codes.append(
|
|
make_fg_color(WEECHAT_BASE_COLORS[stripped_color]))
|
|
|
|
elif stripped_color.isdigit():
|
|
num_color = int(stripped_color)
|
|
if 0 <= num_color < 256:
|
|
escape_codes.append(make_fg_color(stripped_color))
|
|
|
|
if bg_color in WEECHAT_BASE_COLORS:
|
|
escape_codes.append(make_bg_color(WEECHAT_BASE_COLORS[bg_color]))
|
|
else:
|
|
if bg_color.isdigit():
|
|
num_color = int(bg_color)
|
|
if 0 <= num_color < 256:
|
|
escape_codes.append(make_bg_color(bg_color))
|
|
|
|
escape_string = "\033[{}{}m".format(reset_code, ";".join(escape_codes))
|
|
|
|
return escape_string
|
|
|
|
|
|
def prefix(prefix_string):
|
|
prefix_to_symbol = {
|
|
"error": "=!=",
|
|
"network": "--",
|
|
"action": "*",
|
|
"join": "-->",
|
|
"quit": "<--"
|
|
}
|
|
|
|
if prefix_string in prefix_to_symbol:
|
|
return prefix_to_symbol[prefix]
|
|
|
|
return ""
|
|
|
|
|
|
def prnt(_, message):
|
|
print(message)
|
|
|
|
|
|
def prnt_date_tags(_, date, tags_string, data):
|
|
message = "{} {} [{}]".format(
|
|
datetime.datetime.fromtimestamp(date),
|
|
data,
|
|
tags_string
|
|
)
|
|
print(message)
|
|
|
|
|
|
def config_search_section(*_, **__):
|
|
pass
|
|
|
|
|
|
def config_new_option(*_, **__):
|
|
pass
|
|
|
|
|
|
def mkdir_home(*_, **__):
|
|
return True
|
|
|
|
|
|
def info_get(info, *_):
|
|
if info == "nick_color_name":
|
|
return random.choice(list(WEECHAT_BASE_COLORS.keys()))
|
|
|
|
return ""
|
|
|
|
|
|
def buffer_new(*_, **__):
|
|
return "".join(
|
|
random.choice(string.ascii_uppercase + string.digits) for _ in range(8)
|
|
)
|
|
|
|
|
|
def buffer_set(*_, **__):
|
|
return
|
|
|
|
|
|
def buffer_get_string(_ptr, property):
|
|
if property == "localvar_type":
|
|
return "channel"
|
|
return ""
|
|
|
|
|
|
def nicklist_add_group(*_, **__):
|
|
return
|
|
|
|
|
|
def nicklist_add_nick(*_, **__):
|
|
return
|
|
|
|
|
|
def nicklist_remove_nick(*_, **__):
|
|
return
|
|
|
|
|
|
def nicklist_search_nick(*args, **kwargs):
|
|
return buffer_new(args, kwargs)
|
|
|
|
|
|
def string_remove_color(message, _):
|
|
return message
|