Add color
This commit is contained in:
parent
315cfac640
commit
3773599248
41
handler.py
41
handler.py
@ -1,2 +1,41 @@
|
||||
import binascii
|
||||
import itertools
|
||||
|
||||
RED = "\033[31m"
|
||||
GREEN = "\033[32m"
|
||||
YELLOW = "\033[33m"
|
||||
BLUE = "\033[34m"
|
||||
RESET = "\033[0m"
|
||||
|
||||
|
||||
class Diff:
|
||||
def __init__(self, c1, c2):
|
||||
self.state = ""
|
||||
self.c1 = c1
|
||||
self.c2 = c2
|
||||
|
||||
def consume(self, s: str):
|
||||
for p, c in itertools.zip_longest(self.state, s):
|
||||
if c is None:
|
||||
break
|
||||
if p == c:
|
||||
print(f"{self.c1}{c}{RESET}", end="")
|
||||
else:
|
||||
print(f"{self.c2}{c}{RESET}", end="")
|
||||
self.state = s
|
||||
print()
|
||||
|
||||
|
||||
d_c = Diff(GREEN, RED)
|
||||
d_s = Diff(BLUE, YELLOW)
|
||||
|
||||
|
||||
def handle(tag: str, data: bytes):
|
||||
print(tag, repr(data))
|
||||
msg = data.decode().strip()
|
||||
if msg.startswith("0c") or msg.startswith("0a"):
|
||||
# ping, ignore
|
||||
return
|
||||
if tag == "server":
|
||||
d_s.consume(msg)
|
||||
else:
|
||||
d_c.consume(msg)
|
||||
|
Loading…
Reference in New Issue
Block a user