houdini/houdini/data/quest.py
2019-06-21 01:54:37 +01:00

125 lines
4.0 KiB
Python

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"))