Colours!

fedicurses.py

@@ -2,17 +2,22 @@ import curses
 from html.parser import HTMLParser
 from mastodon import Mastodon
 
+def initColours():
+                       # 0: White on black
+    curses.init_pair(1, 4, 0) # 1: Blue on black
+    curses.init_pair(2, 1, 0) # 2: Red on black
+    curses.init_pair(3, 6, 0) # 2: Cyan on black
+
 def newLine(ncwin, lines=1):
     cy, cx = ncwin.getyx()
     ncwin.move(cy + lines, 0)
 
 def printPost(win, post, parser):
+    win.addstr(post["account"]["acct"], curses.A_BOLD)
     if (post["reblog"] is not None):
-        win.addstr("{} boosted {}:".format(post["account"]["acct"],
-                                           post["reblog"]["account"]["acct"]),
-                   curses.A_BOLD)
-    else:
-        win.addstr("{}:".format(post["account"]["acct"]), curses.A_BOLD)
+        win.addstr(" boosted ")
+        win.addstr(post["reblog"]["account"]["acct"], curses.A_BOLD)
+    win.addstr(":")
     newLine(win)
     if (post["spoiler_text"] != ""):
         win.addstr(post["spoiler_text"], curses.A_UNDERLINE)
@@ -55,10 +60,14 @@ class PostParser(HTMLParser):
         self.defncatt = defncatt
         self.curatt = defncatt
         self.openp = False
+        self.colour = 0
+        self.colstack = []
 
     def handle_starttag(self, tag, attrs):
         if tag == 'p':
             self.curatt = self.defncatt
+            self.colstack = []
+            self.colour = 0
             # newLine(self.win)
         elif tag == 'strong' or tag == 'b':
             self.curatt = self.curatt ^ curses.A_REVERSE
@@ -66,6 +75,12 @@ class PostParser(HTMLParser):
             self.curatt = self.curatt ^ curses.A_BOLD
         elif tag == 'br':
             newLine(self.win)
+        elif tag == 'a':
+            self.colstack.append(self.colour)
+            self.colour = 1
+        elif tag == 'code':
+            self.colstack.append(self.colour)
+            self.colour = 3
 
     def handle_endtag(self, tag):
         if tag == 'p':
@@ -75,10 +90,14 @@ class PostParser(HTMLParser):
             self.curatt = self.curatt ^ curses.A_REVERSE
         elif tag == 'em' or tag == 'i':
             self.curatt = self.curatt ^ curses.A_BOLD
+        elif tag == 'a':
+            self.colour = self.colstack.pop()
+        elif tag == 'code':
+            self.colour = self.colstack.pop()
 
     def handle_data(self, data):
         try:
-            self.win.addstr(data, self.curatt)
+            self.win.addstr(data, self.curatt | curses.color_pair(self.colour))
         except curses.error:
             pass
         self.openp = True
@@ -88,6 +107,7 @@ class PostParser(HTMLParser):
 
 
 def main(stdscr):
+    initColours()
     mastodon = Mastodon(
             access_token = 'd100.club_usercred.secret',
             api_base_url = 'd100.club'