From da5f204dfb0e33a99bba1fa00842b253bf9947e0 Mon Sep 17 00:00:00 2001
From: Armaan Bhojwani <me@armaanb.net>
Date: Sat, 30 Jan 2021 20:25:46 -0500
Subject: [PATCH] Add ability to star cards

---
 lightcards/deck.py    | 16 ++++++++++++++++
 lightcards/display.py |  4 ++++
 lightcards/parse.py   |  4 +++-
 3 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/lightcards/deck.py b/lightcards/deck.py
index d8ee725..2136ecc 100644
--- a/lightcards/deck.py
+++ b/lightcards/deck.py
@@ -1,6 +1,22 @@
 # Classes pertaining to the card deck
 # Armaan Bhojwani 2021
 
+class Card(list):
+    starred = False
+
+    def toggleStar(self):
+        if self.starred:
+            self.starred = False
+        else:
+            self.starred = True
+
+    def getStar(self):
+        if self.starred:
+            return "★ Starred ★"
+        else:
+            return "Not starred"
+
+
 class Status():
     index = 0
     side = 0
diff --git a/lightcards/display.py b/lightcards/display.py
index 42a01f2..86740d6 100755
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -28,6 +28,7 @@ def disp_card(stdscr, stack, headers, obj):
             stdscr.insch(mlines, mcols+2, '2')
         stdscr.insch(mlines, mcols+3, '/')
         stdscr.insch(mlines, mcols+4, str(len(stack)))
+        stdscr.insstr(mlines, 1, stack[obj.getIdx()].getStar())
     except Exception:
         pass
 
@@ -53,5 +54,8 @@ def get_key(stdscr, stack, headers):
             elif key in ["j", "k", "KEY_UP", "KEY_DOWN"]:
                 idx.flip()
                 disp_card(stdscr, stack, headers, idx)
+            elif key in ["i", "/"]:
+                stack[idx.getIdx()].toggleStar()
+                disp_card(stdscr, stack, headers, idx)
         except Exception:
             pass
diff --git a/lightcards/parse.py b/lightcards/parse.py
index 457da13..6f3c25f 100755
--- a/lightcards/parse.py
+++ b/lightcards/parse.py
@@ -6,6 +6,8 @@ import sys
 from bs4 import BeautifulSoup
 import markdown
 
+from .deck import Card
+
 
 def md2html(file):
     with open(file, "r", encoding="utf-8") as input_file:
@@ -26,7 +28,7 @@ def parse_html(html):
     outp = []
 
     for x in soup.find_all("tr"):
-        outp.append([clean_text(y) for y in x.find_all("td")])
+        outp.append(Card([clean_text(y) for y in x.find_all("td")]))
 
     return ([clean_text(x) for x in soup.find_all("th")],
             clean_list(outp))
-- 
2.39.2