New player quest database models

houdini/data/penguin.py

@@ -4,7 +4,8 @@ from houdini.data import db
 
 from houdini.data.permission import PermissionCrumbsCollection
 from houdini.data.item import ItemCrumbsCollection
-from houdini.data.igloo import IglooCrumbsCollection, FlooringCrumbsCollection, LocationCrumbsCollection
+from houdini.data.igloo import IglooCrumbsCollection, FurnitureCrumbsCollection, FlooringCrumbsCollection, \
+    LocationCrumbsCollection
 from houdini.data.stamp import StampCrumbsCollection
 from houdini.data.ninja import CardCrumbsCollection
 from houdini.data.mail import PostcardCrumbsCollection
@@ -60,6 +61,11 @@ class Penguin(db.Model):
     rainbow_adoptability = db.Column(db.SmallInteger, nullable=False, server_default=db.text("0"))
     has_dug = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
     nuggets = db.Column(db.SmallInteger, nullable=False, server_default=db.text("0"))
+    opened_playercard = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
+    special_wave = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
+    special_dance = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
+    special_snowball = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
+    map_category = db.Column(db.SmallInteger, nullable=False, server_default=db.text("0"))
     status_field = db.Column(db.Integer, nullable=False, server_default=db.text("0"))
     timer_active = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
     timer_start = db.Column(db.Time, nullable=False, server_default=db.text("'00:00:00'::time without time zone"))

houdini/data/quest.py

@@ -0,0 +1,124 @@
+from houdini.data import db
+
+
+class Quest(db.Model):
+    __tablename__ = 'quest'
+
+    id = db.Column(db.Integer, primary_key=True, server_default=db.text("nextval('\"quest_id_seq\"'::regclass)"))
+    name = db.Column(db.String(50), nullable=False, unique=True)
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self._tasks = set()
+
+        self._items = set()
+        self._furniture = set()
+        self._pet = set()
+
+        self._complete = set()
+        self._in_progress = set()
+
+    @property
+    def items(self):
+        return self._items
+
+    @property
+    def furniture(self):
+        return self._furniture
+
+    @property
+    def pet(self):
+        return self._pet
+
+    @property
+    def complete(self):
+        return self._complete
+
+    @property
+    def in_progress(self):
+        return self._in_progress
+
+    @property
+    def awards(self):
+        return self._items.union(self._furniture.union(self._pet))
+
+    @property
+    def tasks(self):
+        return self._tasks
+
+    @items.setter
+    def items(self, child):
+        if isinstance(child, QuestAwardItem):
+            self._items.add(child)
+
+    @furniture.setter
+    def furniture(self, child):
+        if isinstance(child, QuestAwardFurniture):
+            self._furniture.add(child)
+
+    @pet.setter
+    def pet(self, child):
+        if isinstance(child, QuestAwardPuffleItem):
+            self._pet.add(child)
+
+    @tasks.setter
+    def tasks(self, child):
+        if isinstance(child, QuestTask):
+            self._tasks.add(child)
+
+    @complete.setter
+    def complete(self, child):
+        if isinstance(child, PenguinQuestTask):
+            if child.complete:
+                self._complete.add(child.task_id)
+            else:
+                self._in_progress.add(child.task_id)
+
+
+class QuestAwardItem(db.Model):
+    __tablename__ = 'quest_award_item'
+
+    quest_id = db.Column(db.ForeignKey('quest.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
+                         nullable=False)
+    item_id = db.Column(db.ForeignKey('item.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
+                        nullable=False)
+
+
+class QuestAwardFurniture(db.Model):
+    __tablename__ = 'quest_award_furniture'
+
+    quest_id = db.Column(db.ForeignKey('quest.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
+                         nullable=False)
+    furniture_id = db.Column(db.ForeignKey('furniture.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
+                             nullable=False)
+    quantity = db.Column(db.SmallInteger, nullable=False, server_default=db.text("1"))
+
+
+class QuestAwardPuffleItem(db.Model):
+    __tablename__ = 'quest_award_puffle_item'
+
+    quest_id = db.Column(db.ForeignKey('quest.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
+                         nullable=False)
+    puffle_item_id = db.Column(db.ForeignKey('puffle_item.id', ondelete='CASCADE', onupdate='CASCADE'),
+                               primary_key=True, nullable=False)
+    quantity = db.Column(db.SmallInteger, nullable=False, server_default=db.text("1"))
+
+
+class QuestTask(db.Model):
+    __tablename__ = 'quest_task'
+
+    id = db.Column(db.Integer, primary_key=True, server_default=db.text("nextval('\"quest_id_seq\"'::regclass)"))
+    quest_id = db.Column(db.ForeignKey('quest.id', ondelete='CASCADE', onupdate='CASCADE'), nullable=False)
+    description = db.Column(db.String(50), nullable=False)
+    room_id = db.Column(db.ForeignKey('room.id', ondelete='CASCADE', onupdate='CASCADE'))
+    data = db.Column(db.String(50))
+
+
+class PenguinQuestTask(db.Model):
+    __tablename__ = 'penguin_quest_task'
+
+    task_id = db.Column(db.ForeignKey('quest_task.id', ondelete='CASCADE', onupdate='CASCADE'), nullable=False,
+                        primary_key=True)
+    penguin_id = db.Column(db.ForeignKey('penguin.id', ondelete='CASCADE', onupdate='CASCADE'), nullable=False,
+                           primary_key=True)
+    complete = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))