Mirrored_Repos/houdini
1
0
Fork 0
mirror of https://github.com/solero/houdini.git synced 2024-11-10 05:08:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Load buddy collections into penguin data model

Browse Source
This commit is contained in:
Ben 2019-08-19 02:22:17 +01:00
parent 8b33390acf
commit f263daba57
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
houdini/data/penguin.py
View File

@ -10,6 +10,8 @@ from houdini.data.stamp import StampCrumbsCollection
from houdini.data.ninja import CardCrumbsCollection from houdini.data.ninja import CardCrumbsCollection
from houdini.data.mail import PostcardCrumbsCollection from houdini.data.mail import PostcardCrumbsCollection
from houdini.data.pet import PuffleCrumbsCollection, PuffleItemCrumbsCollection from houdini.data.pet import PuffleCrumbsCollection, PuffleItemCrumbsCollection
from houdini.data.buddy import BuddyListCollection, BuddyRequestCollection, CharacterBuddyCollection, \
IgnoreListCollection
class Penguin(db.Model): class Penguin(db.Model):
@ -95,6 +97,10 @@ class Penguin(db.Model):
self.postcards = None self.postcards = None
self.puffles = None self.puffles = None
self.puffle_items = None self.puffle_items = None
self.buddies = None
self.buddy_requests = None
self.character_buddies = None
self.ignore = None
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
@ -111,6 +117,11 @@ class Penguin(db.Model):
self.puffles = await PuffleCrumbsCollection.get_collection(self.id) self.puffles = await PuffleCrumbsCollection.get_collection(self.id)
self.puffle_items = await PuffleItemCrumbsCollection.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): def safe_nickname(self, language_bitmask):
return self.nickname if self.approval & language_bitmask else "P" + str(self.id) return self.nickname if self.approval & language_bitmask else "P" + str(self.id)