You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.7 KiB
43 lines
1.7 KiB
from sqlalchemy.ext.associationproxy import association_proxy |
|
from app import db |
|
from app.models import GameState, GamePlayer, Role |
|
|
|
class Game(db.Model): |
|
__tablename__ = 'game' |
|
id = db.Column(db.Integer, primary_key=True) |
|
name = db.Column(db.String(64), index=True, unique=True, nullable=False) |
|
state = db.Column(db.Enum(GameState), server_default=GameState(1).name, nullable=False) |
|
start_time = db.Column(db.DateTime) |
|
end_time = db.Column(db.DateTime) |
|
game_players = db.relationship( |
|
'GamePlayer', |
|
back_populates='game', |
|
cascade="save-update, merge, delete, delete-orphan") |
|
users = association_proxy('game_players', 'user', |
|
creator=lambda user: GamePlayer(user=user)) # to enable game.players.append(player) |
|
objectives = db.relationship( |
|
'Objective', |
|
lazy='select', |
|
backref=db.backref('game', lazy='joined')) |
|
notifications = db.relationship( |
|
'Notification', |
|
lazy='select', |
|
backref=db.backref('game', lazy='joined')) |
|
|
|
def last_player_locations(self): |
|
return [player.last_location(self) for player in self.users if player.locations] |
|
|
|
def bunnies(self): |
|
# pylint: disable=not-an-iterable |
|
return [gameplayer.player for gameplayer in self.game_players if gameplayer.role == Role.bunny] |
|
|
|
def last_locations(self, players): |
|
locations = [] |
|
for player in players: |
|
locations.append(player.last_location(self)) |
|
return locations |
|
|
|
def owned_by(self, player): |
|
# pylint: disable=not-an-iterable |
|
'''given player is an owner of game''' |
|
return player in [gameplayer.player for gameplayer in self.game_players if gameplayer.role == Role.owner]
|
|
|