Mirrors/tg
2
0
Fork 0
mirror of https://github.com/paul-nameless/tg synced 2024-11-25 13:20:25 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add colors module to patch

Browse source
This commit is contained in:
Paul Nameless 2020-05-05 14:30:28 +08:00
parent a4d02b2b89
commit 94c4179245
2 changed files with 36 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
colors.py Normal file
View file

@ -0,0 +1,24 @@
import curses
@curses.wrapper
def main(win):
def print_all_colors(attr):
for color in range(-1, curses.COLORS):
try:
curses.init_pair(color, color, 0)
except curses.error:
pass
else:
win.addstr(str(color) + ' ', curses.color_pair(color) | attr)
curses.start_color()
try:
curses.use_default_colors()
except curses.error:
pass
win.addstr("available colors: %d\n\n" % curses.COLORS)
print_all_colors(0)
win.addstr("\n\n")
print_all_colors(curses.A_BOLD)
win.refresh()
win.getch()

26
tg/views/__init__.py
View file

@ -160,16 +160,18 @@ class ChatView:
# if len(msg) < self.w:
# msg += ' ' * (self.w - len(msg) - 1)
colors = [(cyan, -1), (blue, -1), (white, -1), (magenta, -1)]
mode = normal
msg_color = get_color(white, -1)
unread_color = get_color(magenta, -1)
attrs = [get_color(cyan, -1), get_color(blue, -1), msg_color]
if i == current:
mode = reverse
attrs = [attr | reverse for attr in attrs]
msg_color |= reverse
unread_color |= reverse
offset = 0
j = 0
# for color, e in zip(colors, msg.split(' ', maxsplit=3)):
for color, e in zip(colors, [" " + date, title]):
attr = mode | get_color(*color)
for attr, e in zip(attrs, [" " + date, title]):
if offset > self.w:
break
j += 1
@ -181,8 +183,7 @@ class ChatView:
if offset >= self.w:
continue
attr = get_color(*colors[-2]) | mode
attr = msg_color
msg = last_msg[: self.w - offset - 1]
# msg = msg[:self.w-1]
@ -192,8 +193,7 @@ class ChatView:
self.win.addstr(i, offset, msg, attr)
if unread:
attr = get_color(*colors[-1]) | mode
attr = unread_color
unread = " " + str(unread) + " "
self.win.addstr(i, self.w - len(unread) - 1, unread, attr)
@ -247,17 +247,15 @@ class MsgView:
# log.warning('Reched end of lines')
break
colors = [(cyan, -1), (blue, -1), (white, -1)]
mode = normal
attrs = [get_color(cyan, -1), get_color(blue, -1), get_color(white, -1)]
if i == current:
mode = reverse
attrs = [attr | reverse for attr in attrs]
offset = 0
j = 0
for color, e in zip(colors, [" " + dt, user_id, msg]):
for attr, e in zip(attrs, [" " + dt, user_id, msg]):
if not e.strip():
continue
attr = get_color(*color) | mode
j += 1
if j < 4:
e = e + " "