psycodad
/
wttr.in
mirror of https://github.com/chubin/wttr.in
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix formatting of lib/fmt/png.py

Browse Source
pull/1062/head
Igor Chubin 1 month ago
parent 3d5ad7a303
commit 2ab6e7a4d5
1 changed files with 42 additions and 38 deletions
Show Stats Download Patch File Download Diff File

78
lib/fmt/png.py
Unescape View File

@ -39,17 +39,17 @@ CHAR_WIDTH = 8
CHAR_HEIGHT = 14 CHAR_HEIGHT = 14
FONT_SIZE = 13 FONT_SIZE = 13
FONT_CAT = { FONT_CAT = {
'default': "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf", "default": "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf",
'Cyrillic': "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf", "Cyrillic": "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf",
'Greek': "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf", "Greek": "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf",
'Arabic': "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf", "Arabic": "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf",
'Hebrew': "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf", "Hebrew": "/usr/share/fonts/truetype/dejavu/DejaVuSansMono.ttf",
'Han': "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc", "Han": "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc",
'Hiragana': "/usr/share/fonts/truetype/motoya-l-cedar/MTLc3m.ttf", "Hiragana": "/usr/share/fonts/truetype/motoya-l-cedar/MTLc3m.ttf",
'Katakana': "/usr/share/fonts/truetype/motoya-l-cedar/MTLc3m.ttf", "Katakana": "/usr/share/fonts/truetype/motoya-l-cedar/MTLc3m.ttf",
'Hangul': "/usr/share/fonts/truetype/lexi/LexiGulim.ttf", "Hangul": "/usr/share/fonts/truetype/lexi/LexiGulim.ttf",
'Braille': "/usr/share/fonts/truetype/ancient-scripts/Symbola_hint.ttf", "Braille": "/usr/share/fonts/truetype/ancient-scripts/Symbola_hint.ttf",
'Emoji': "/usr/share/fonts/truetype/ancient-scripts/Symbola_hint.ttf", "Emoji": "/usr/share/fonts/truetype/ancient-scripts/Symbola_hint.ttf",
} }
# #
@ -66,6 +66,7 @@ FONT_CAT = {
# * fonts-symbola (Braille/Emoji) # * fonts-symbola (Braille/Emoji)
# #
def render_ansi(text, options=None): def render_ansi(text, options=None):
"""Render `text` (terminal sequence) in a PNG file """Render `text` (terminal sequence) in a PNG file
paying attention to passed command line `options`. paying attention to passed command line `options`.
@ -85,24 +86,22 @@ def render_ansi(text, options=None):
return _gen_term(buf, graphemes, options=options) return _gen_term(buf, graphemes, options=options)
def _color_mapping(color, inverse=False): def _color_mapping(color, inverse=False):
"""Convert pyte color to PIL color """Convert pyte color to PIL color
Return: tuple of color values (R,G,B) Return: tuple of color values (R,G,B)
""" """
if color == 'default': if color == "default":
if inverse: if inverse:
return 'black' return "black"
return 'lightgray' return "lightgray"
if color in ['green', 'black', 'cyan', 'blue', 'brown']: if color in ["green", "black", "cyan", "blue", "brown"]:
return color return color
try: try:
return ( return (int(color[0:2], 16), int(color[2:4], 16), int(color[4:6], 16))
int(color[0:2], 16),
int(color[2:4], 16),
int(color[4:6], 16))
except (ValueError, IndexError): except (ValueError, IndexError):
# if we do not know this color and it can not be decoded as RGB, # if we do not know this color and it can not be decoded as RGB,
# print it and return it as it is (will be displayed as black) # print it and return it as it is (will be displayed as black)
@ -110,6 +109,7 @@ def _color_mapping(color, inverse=False):
return color return color
return color return color
def _strip_buf(buf): def _strip_buf(buf):
"""Strips empty spaces from behind and from the right side. """Strips empty spaces from behind and from the right side.
(from the right side is not yet implemented) (from the right side is not yet implemented)
@ -117,13 +117,13 @@ def _strip_buf(buf):
def empty_line(line): def empty_line(line):
"Returns True if the line consists from spaces" "Returns True if the line consists from spaces"
return all(x.data == ' ' for x in line) return all(x.data == " " for x in line)
def line_len(line): def line_len(line):
"Returns len of the line excluding spaces from the right" "Returns len of the line excluding spaces from the right"
last_pos = len(line) last_pos = len(line)
while last_pos > 0 and line[last_pos-1].data == ' ': while last_pos > 0 and line[last_pos - 1].data == " ":
last_pos -= 1 last_pos -= 1
return last_pos return last_pos
@ -141,6 +141,7 @@ def _strip_buf(buf):
return buf return buf
def _script_category(char): def _script_category(char):
"""Returns category of a Unicode character """Returns category of a Unicode character
@ -152,12 +153,13 @@ def _script_category(char):
return "Emoji" return "Emoji"
cat = unicodedata2.script_cat(char)[0] cat = unicodedata2.script_cat(char)[0]
if char == u'：': if char == "：":
return 'Han' return "Han"
if cat in ['Latin', 'Common']: if cat in ["Latin", "Common"]:
return 'default' return "default"
return cat return cat
def _load_emojilib(): def _load_emojilib():
"""Load known emojis from a directory, and return dictionary """Load known emojis from a directory, and return dictionary
of PIL Image objects correspodent to the loaded emojis. of PIL Image objects correspodent to the loaded emojis.
@ -167,10 +169,10 @@ def _load_emojilib():
emojilib = {} emojilib = {}
for filename in glob.glob("share/emoji/*.png"): for filename in glob.glob("share/emoji/*.png"):
character = os.path.basename(filename)[:-3] character = os.path.basename(filename)[:-3]
emojilib[character] = \ emojilib[character] = Image.open(filename).resize((CHAR_HEIGHT, CHAR_HEIGHT))
Image.open(filename).resize((CHAR_HEIGHT, CHAR_HEIGHT))
return emojilib return emojilib
# pylint: disable=too-many-locals,too-many-branches,too-many-statements # pylint: disable=too-many-locals,too-many-branches,too-many-statements
def _gen_term(buf, graphemes, options=None): def _gen_term(buf, graphemes, options=None):
"""Renders rendered pyte buffer `buf` and list of workaround `graphemes` """Renders rendered pyte buffer `buf` and list of workaround `graphemes`
@ -190,7 +192,7 @@ def _gen_term(buf, graphemes, options=None):
if "background" in options: if "background" in options:
bg_color = _color_mapping(options["background"], options.get("inverted_colors")) bg_color = _color_mapping(options["background"], options.get("inverted_colors"))
image = Image.new('RGB', (cols * CHAR_WIDTH, rows * CHAR_HEIGHT), color=bg_color) image = Image.new("RGB", (cols * CHAR_WIDTH, rows * CHAR_HEIGHT), color=bg_color)
buf = buf[-ROWS:] buf = buf[-ROWS:]
@ -207,11 +209,11 @@ def _gen_term(buf, graphemes, options=None):
x_pos = 0 x_pos = 0
for char in line: for char in line:
current_color = _color_mapping(char.fg, options.get("inverted_colors")) current_color = _color_mapping(char.fg, options.get("inverted_colors"))
if char.bg != 'default': if char.bg != "default":
draw.rectangle( draw.rectangle(
((x_pos, y_pos), ((x_pos, y_pos), (x_pos + CHAR_WIDTH, y_pos + CHAR_HEIGHT)),
(x_pos+CHAR_WIDTH, y_pos+CHAR_HEIGHT)), fill=_color_mapping(char.bg, options.get("inverted_colors")),
fill=_color_mapping(char.bg, options.get("inverted_colors"))) )
if char.data == "!": if char.data == "!":
try: try:
@ -226,20 +228,21 @@ def _gen_term(buf, graphemes, options=None):
cat = _script_category(data[0]) cat = _script_category(data[0])
if cat not in font: if cat not in font:
globals.log("Unknown font category: %s" % cat) globals.log("Unknown font category: %s" % cat)
if cat == 'Emoji' and emojilib.get(data): if cat == "Emoji" and emojilib.get(data):
image.paste(emojilib.get(data), (x_pos, y_pos)) image.paste(emojilib.get(data), (x_pos, y_pos))
else: else:
draw.text( draw.text(
(x_pos, y_pos), (x_pos, y_pos),
data, data,
font=font.get(cat, font.get('default')), font=font.get(cat, font.get("default")),
fill=current_color) fill=current_color,
)
x_pos += CHAR_WIDTH * constants.WEATHER_SYMBOL_WIDTH_VTE.get(data, 1) x_pos += CHAR_WIDTH * constants.WEATHER_SYMBOL_WIDTH_VTE.get(data, 1)
y_pos += CHAR_HEIGHT y_pos += CHAR_HEIGHT
if 'transparency' in options: if "transparency" in options:
transparency = options.get('transparency', '255') transparency = options.get("transparency", "255")
try: try:
transparency = int(transparency) transparency = int(transparency)
except ValueError: except ValueError:
@ -265,6 +268,7 @@ def _gen_term(buf, graphemes, options=None):
image.save(img_bytes, format="png") image.save(img_bytes, format="png")
return img_bytes.getvalue() return img_bytes.getvalue()
def _fix_graphemes(text): def _fix_graphemes(text):
""" """
Extract long graphemes sequences that can't be handled Extract long graphemes sequences that can't be handled

Write Preview
Loading…
Cancel
Save