gtg

untitled/model.py

@@ -2,10 +2,11 @@ import time
 
 
 class Cat:
-    def __init__(self, name, traits, fullness=100, last_updated=None):
+    def __init__(self, name, traits, fullness=100, happiness=100, last_updated=None):
         self.name = name
         self.traits = traits
         self.fullness = fullness
+        self.happiness = happiness
         self.last_updated = last_updated if last_updated is not None else time.time()
 
     def to_dict(self):
@@ -13,12 +14,19 @@ class Cat:
             "name": self.name,
             "traits": self.traits,
             "fullness": self.fullness,
+            "happiness": self.happiness,
             "last_updated": self.last_updated,
         }
 
     @staticmethod
     def from_dict(data):
-        return Cat(data["name"], data["traits"], data["fullness"], data["last_updated"])
+        return Cat(
+            data["name"],
+            data["traits"],
+            data["fullness"],
+            data["happiness"],
+            data["last_updated"],
+        )
 
 
 class Save: