mirror of
https://github.com/solero/houdini.git
synced 2024-11-21 21:17:21 +00:00
Add new functionality to room instances
This commit is contained in:
parent
1684e66035
commit
3e576ba602
@ -16,6 +16,33 @@ class Room(db.Model):
|
||||
Spawn = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
|
||||
StampGroup = db.Column(db.ForeignKey('stamp_group.ID', ondelete='CASCADE', onupdate='CASCADE'))
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
super().__init__(*args, **kwargs)
|
||||
self.penguins = []
|
||||
|
||||
def add_penguin(self, p):
|
||||
if p.room:
|
||||
p.room.remove_penguin(p)
|
||||
self.penguins.append(p)
|
||||
|
||||
p.room = self
|
||||
|
||||
p.send_xt('jr', self.get_string)
|
||||
self.send_xt('ap', p.server.penguin_string_compiler.compile(p))
|
||||
|
||||
def remove_penguin(self, p):
|
||||
self.send_xt('rp', p.data.ID)
|
||||
|
||||
self.penguins.remove(p)
|
||||
p.room = None
|
||||
|
||||
def get_string(self):
|
||||
return '%'.join([p.server.penguin_string_compiler.compile(p) for p in self.penguins])
|
||||
|
||||
def send_xt(self, *data):
|
||||
for penguin in self.penguins:
|
||||
penguin.send_xt(*data)
|
||||
|
||||
|
||||
class RoomTable(db.Model):
|
||||
__tablename__ = 'room_table'
|
||||
|
Loading…
Reference in New Issue
Block a user