From f263daba579381df7e4ed9e10faa36a08da3667b Mon Sep 17 00:00:00 2001 From: Ben Date: Mon, 19 Aug 2019 02:22:17 +0100 Subject: [PATCH] Load buddy collections into penguin data model --- houdini/data/penguin.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/houdini/data/penguin.py b/houdini/data/penguin.py index fa2da67..471a733 100644 --- a/houdini/data/penguin.py +++ b/houdini/data/penguin.py @@ -10,6 +10,8 @@ from houdini.data.stamp import StampCrumbsCollection from houdini.data.ninja import CardCrumbsCollection from houdini.data.mail import PostcardCrumbsCollection from houdini.data.pet import PuffleCrumbsCollection, PuffleItemCrumbsCollection +from houdini.data.buddy import BuddyListCollection, BuddyRequestCollection, CharacterBuddyCollection, \ + IgnoreListCollection class Penguin(db.Model): @@ -95,6 +97,10 @@ class Penguin(db.Model): self.postcards = None self.puffles = None self.puffle_items = None + self.buddies = None + self.buddy_requests = None + self.character_buddies = None + self.ignore = None super().__init__(*args, **kwargs) @@ -111,6 +117,11 @@ class Penguin(db.Model): self.puffles = await PuffleCrumbsCollection.get_collection(self.id) self.puffle_items = await PuffleItemCrumbsCollection.get_collection(self.id) + self.buddies = await BuddyListCollection.get_collection(self.id) + self.buddy_requests = await BuddyRequestCollection.get_collection(self.id) + self.character_buddies = await CharacterBuddyCollection.get_collection(self.id) + self.ignore = await IgnoreListCollection.get_collection(self.id) + def safe_nickname(self, language_bitmask): return self.nickname if self.approval & language_bitmask else "P" + str(self.id)