X-Git-Url: https://git.armaanb.net/?a=blobdiff_plain;f=lightcards%2Fdeck.py;h=06ed135b21b24ade39f78bd3c1d830d507fb585f;hb=da0b2a5411a041e29a0574d3aeb3f0da311ff5c1;hp=d8ee725856b19e8b5bcdfe626edf690ee4a0b578;hpb=df1f0b77c2104505501b9e1b342f6268cf108c8d;p=lightcards.git

diff --git a/lightcards/deck.py b/lightcards/deck.py
index d8ee725..06ed135 100644
--- a/lightcards/deck.py
+++ b/lightcards/deck.py
@@ -1,23 +1,40 @@
 # Classes pertaining to the card deck
 # Armaan Bhojwani 2021
 
-class Status():
-    index = 0
-    side = 0
 
-    def forward(self, stack):
-        if not self.index == len(stack) - 1:
-            self.index += 1
+class Card:
+    """Card extends the list class, and adds ability to star them."""
 
-    def back(self):
-        if not self.index < 1:
-            self.index -= 1
+    def __init__(self, inp):
+        self.starred = False
+        self.side = 0
+        self.front = ""
+        self.back = ""
+        if len(inp) >= 1:
+            self.front = inp[0]
+        if len(inp) >= 2:
+            self.back = inp[1]
 
-    def flip(self):
-        if self.side == 0:
-            self.side = 1
+    def unStar(self):
+        self.starred = False
+
+    def star(self):
+        self.starred = True
+
+    def toggleStar(self):
+        if self.starred:
+            self.starred = False
         else:
-            self.side = 0
+            self.starred = True
+
+    def getStar(self):
+        return self.starred
+
+    def printStar(self):
+        if self.starred:
+            return "★ Starred ★"
+        else:
+            return "Not starred"
 
     def setSide(self, inp):
         self.side = inp
@@ -25,5 +42,38 @@ class Status():
     def getSide(self):
         return self.side
 
+    def get(self):
+        if self.side == 0:
+            return self.front
+        else:
+            return self.back
+
+    def getFront(self):
+        return self.front
+
+    def flip(self):
+        if self.side == 0:
+            self.side = 1
+        else:
+            self.side = 0
+
+
+class Status:
+    """The status class keeps track of where in the deck the user is"""
+
+    def __init__(self):
+        self.index = 0
+
+    def forward(self, stack):
+        if self.index != len(stack):
+            self.index += 1
+
+    def back(self):
+        if not self.index < 1:
+            self.index -= 1
+
+    def setIdx(self, inp):
+        self.index = inp
+
     def getIdx(self):
         return self.index