mirror of
https://github.com/solero/houdini.git
synced 2024-11-14 22:58:21 +00:00
64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
from Houdini.Data import db
|
|
|
|
|
|
class Room(db.Model):
|
|
__tablename__ = 'room'
|
|
|
|
ID = db.Column(db.Integer, primary_key=True)
|
|
InternalID = db.Column(db.Integer, nullable=False, unique=True,
|
|
server_default=db.text("nextval('\"room_InternalID_seq\"'::regclass)"))
|
|
Name = db.Column(db.String(50), nullable=False)
|
|
Member = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
|
|
MaxUsers = db.Column(db.SmallInteger, nullable=False, server_default=db.text("80"))
|
|
RequiredItem = db.Column(db.ForeignKey('item.ID', ondelete='CASCADE', onupdate='CASCADE'))
|
|
Game = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
|
|
Blackhole = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))
|
|
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'
|
|
|
|
ID = db.Column(db.Integer, primary_key=True, nullable=False)
|
|
RoomID = db.Column(db.ForeignKey('room.ID', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
|
|
nullable=False)
|
|
Game = db.Column(db.String(20), nullable=False)
|
|
|
|
|
|
class RoomWaddle(db.Model):
|
|
__tablename__ = 'room_waddle'
|
|
|
|
ID = db.Column(db.Integer, primary_key=True, nullable=False)
|
|
RoomID = db.Column(db.ForeignKey('room.ID', ondelete='CASCADE', onupdate='CASCADE'), primary_key=True,
|
|
nullable=False)
|
|
Seats = db.Column(db.SmallInteger, nullable=False, server_default=db.text("2"))
|
|
Game = db.Column(db.String(20), nullable=False)
|