Clean up lightcards.py

[lightcards.git] / lightcards / deck.py

diff --git a/lightcards/deck.py b/lightcards/deck.py
index 2136ecc589bdc0f843a2db68399b64385e1934b9..b0822a9705704ad055fe2876186e62be7562d076 100644 (file)
--- a/lightcards/deck.py
+++ b/lightcards/deck.py
@@ -2,7 +2,16 @@
 # Armaan Bhojwani 2021
 
 class Card(list):
-    starred = False
+    """Card extends the list class, and adds ability to star them."""
+    def __init__(self, inp):
+        super().__init__(inp)
+        self.starred = False
+
+    def unStar(self):
+        self.starred = False
+
+    def star(self):
+        self.starred = True
 
     def toggleStar(self):
         if self.starred:
@@ -11,6 +20,9 @@ class Card(list):
             self.starred = True
 
     def getStar(self):
+        return self.starred
+
+    def printStar(self):
         if self.starred:
             return "★ Starred ★"
         else:
@@ -18,11 +30,13 @@ class Card(list):
 
 
 class Status():
-    index = 0
-    side = 0
+    """The status class keeps track of where in the deck the user is"""
+    def __init__(self):
+        self.index = 0
+        self.side = 0
 
     def forward(self, stack):
-        if not self.index == len(stack) - 1:
+        if not self.index == len(stack):
             self.index += 1
 
     def back(self):
@@ -38,6 +52,9 @@ class Status():
     def setSide(self, inp):
         self.side = inp
 
+    def setIdx(self, inp):
+        self.index = inp
+
     def getSide(self):
         return self.side