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

Collection classes for buddy tables

Browse Source
This commit is contained in:
Ben 2019-08-19 02:18:13 +01:00
parent 4bc7fe0dca
commit f989315ef3
1 changed files with 52 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
houdini/data/buddy.py
View File

@ -1,4 +1,4 @@
from houdini.data import db
from houdini.data import db, BaseCrumbsCollection
class BuddyList(db.Model):
@ -43,3 +43,53 @@ class CharacterBuddy(db.Model):
character_id = db.Column(db.ForeignKey('character.id', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
nullable=False)
best_buddy = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
class BuddyListCollection(BaseCrumbsCollection):
def __init__(self, inventory_id=None):
super().__init__(model=BuddyList,
key='buddy_id',
inventory_model=BuddyList,
inventory_key='penguin_id',
inventory_value='buddy_id',
inventory_id=inventory_id)
class IgnoreListCollection(BaseCrumbsCollection):
def __init__(self, inventory_id=None):
super().__init__(model=IgnoreList,
key='ignore_id',
inventory_model=IgnoreList,
inventory_key='penguin_id',
inventory_value='ignore_id',
inventory_id=inventory_id)
class BuddyRequestCollection(BaseCrumbsCollection):
def __init__(self, inventory_id=None):
super().__init__(model=BuddyRequest,
key='requester_id',
inventory_model=BuddyRequest,
inventory_key='penguin_id',
inventory_value='requester_id',
inventory_id=inventory_id)
class CharacterCrumbsCollection(BaseCrumbsCollection):
def __init__(self):
super().__init__(model=Character, key='id')
class CharacterBuddyCollection(BaseCrumbsCollection):
def __init__(self, inventory_id=None):
super().__init__(model=CharacterBuddy,
key='character_id',
inventory_model=CharacterBuddy,
inventory_key='penguin_id',
inventory_value='character_id',
inventory_id=inventory_id)