Fix pickle implementation

[lightcards.git] / lightcards / deck.py

diff --git a/lightcards/deck.py b/lightcards/deck.py
index eac8b69c9de59bf5ec4b3ace6d651d3a16db230f..110c5b4fdcee15ab50a27c5eca2ea52dfa3a0ee7 100644 (file)
--- a/lightcards/deck.py
+++ b/lightcards/deck.py
@@ -2,12 +2,21 @@
 # Armaan Bhojwani 2021
 
 
-class Card(list):
-    """Card extends the list class, and adds ability to star them."""
+class Card:
+    """Class containing the card information."""
 
     def __init__(self, inp):
-        super().__init__(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 __str__(self):
+        return f"{self.front}, {self.back}"
 
     def unStar(self):
         self.starred = False
@@ -30,36 +39,44 @@ class Card(list):
         else:
             return "Not starred"
 
+    def setSide(self, inp):
+        self.side = inp
+
+    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"""
+    """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):
+        if self.index != len(stack):
             self.index += 1
 
     def back(self):
         if not self.index < 1:
             self.index -= 1
 
-    def flip(self):
-        if self.side == 0:
-            self.side = 1
-        else:
-            self.side = 0
-
-    def setSide(self, inp):
-        self.side = inp
-
     def setIdx(self, inp):
         self.index = inp
 
-    def getSide(self):
-        return self.side
-
     def getIdx(self):
         return self.index