matrix/weechat-matrix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests: Implement mock weechat module and add some color tests.

Browse source
This commit is contained in:
poljar (Damir Jelić) 2018-03-02 22:58:43 +01:00
parent 404f785f21
commit fc6a6e9da9
5 changed files with 165 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

93
matrix/_weechat.py Normal file
View file

@ -0,0 +1,93 @@
def color(color_name):
# type: (str) -> str
# yapf: disable
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"
}
escape_codes = []
reset_code = "0"
def make_fg_color(color):
return "38;5;{}".format(color)
def make_bg_color(color):
return "48;5;{}".format(color)
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.isdecimal():
num_color = int(stripped_color)
if num_color >= 0 and 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.isdecimal():
num_color = int(bg_color)
if num_color >= 0 and num_color < 256:
escape_codes.append(make_bg_color(bg_color))
escape_string = "\033[{}{}m".format(reset_code, ";".join(escape_codes))
return escape_string

66
matrix/colors.py
View file

@ -542,23 +542,26 @@ def colour_find_rgb(r, g, b):
def color_html_to_weechat(color):
# type: (str) -> str
# yapf: disable
first_16 = {
(0, 0, 0): "black", # 0
(128, 0, 0): "red", # 1
(0, 128, 0): "green", # 2
(128, 128, 0): "brown", # 3
(0, 0, 128): "blue", # 4
(128, 0, 128): "magenta", # 5
(0, 128, 128): "cyan", # 6
(192, 192, 192): "default", # 7
(128, 128, 128): "gray", # 8
(255, 0, 0): "lightred", # 9
(0, 255, 0): "lightgreen", # 11
(255, 255, 0): "yellow", # 12
(0, 0, 255): "lightblue", # 13
(255, 0, 255): "lightmagenta", # 14
(0, 255, 255): "lightcyan", # 15
(0, 0, 0): "black", # 0
(128, 0, 0): "red", # 1
(0, 128, 0): "green", # 2
(128, 128, 0): "brown", # 3
(0, 0, 128): "blue", # 4
(128, 0, 128): "magenta", # 5
(0, 128, 128): "cyan", # 6
(192, 192, 192): "default", # 7
(128, 128, 128): "gray", # 8
(255, 0, 0): "lightred", # 9
(0, 255, 0): "lightgreen", # 10
(255, 255, 0): "yellow", # 11
(0, 0, 255): "lightblue", # 12
(255, 0, 255): "lightmagenta", # 13
(0, 255, 255): "lightcyan", # 14
(255, 255, 255): "white", # 15
}
# yapf: enable
try:
rgb_color = webcolors.html5_parse_legacy_color(color)
@ -573,24 +576,25 @@ def color_html_to_weechat(color):
def color_weechat_to_html(color):
# type: (str) -> str
# yapf: disable
first_16 = {
"black": "black", # 0
"red": "maroon", # 1
"green": "green", # 2
"brown": "olive", # 3
"blue": "navy", # 4
"magenta": "purple", # 5
"cyan": "teal", # 6
"default": "silver", # 7
"gray": "grey", # 8
"lightred": "red", # 9
"lightgreen": "lime", # 11
"yellow": "yellow", # 12
"lightblue": "fuchsia", # 13
"lightmagenta": "aqua", # 14
"lightcyan": "white", # 15
"black": "black", # 0
"red": "maroon", # 1
"green": "green", # 2
"brown": "olive", # 3
"blue": "navy", # 4
"magenta": "purple", # 5
"cyan": "teal", # 6
"default": "silver", # 7
"gray": "gray", # 8
"lightred": "red", # 9
"lightgreen": "lime", # 11
"yellow": "yellow", # 12
"lightblue": "blue", # 13
"lightmagenta": "fuchsia", # 14
"lightcyan": "aqua", # 15
"white": "white", # 15
}
#yapf: disable
hex_colors = {
"0": "#000000",
"1": "#800000",

9
matrix/globals.py
View file

@ -21,9 +21,12 @@ import sys
from matrix.utf import WeechatWrapper
from matrix.plugin_options import PluginOptions
import weechat
W = weechat if sys.hexversion >= 0x3000000 else WeechatWrapper(weechat)
try:
import weechat
W = weechat if sys.hexversion >= 0x3000000 else WeechatWrapper(weechat)
except ImportError:
import matrix._weechat as weechat
W = weechat
OPTIONS = PluginOptions() # type: PluginOptions
SERVERS = dict() # type: Dict[str, MatrixServer]