Add build script.

This commit is contained in:
luxick
2018-02-24 14:06:59 +01:00
parent af57942ad8
commit f1160ddb55
12 changed files with 868 additions and 347 deletions

View File

@@ -58,13 +58,23 @@ class Death(Model):
info = CharField(null=True)
player = ForeignKeyField(Player)
enemy = ForeignKeyField(Enemy)
penalty = ForeignKeyField(Drink)
episode = ForeignKeyField(Episode, backref='deaths')
class Meta:
database = connection
class Penalty(Model):
id = AutoField()
size = DecimalField()
ForeignKeyField(Drink)
ForeignKeyField(Player, backref='penalties')
ForeignKeyField(Death, backref='penalties')
class Meta:
database = connection
class Victory(Model):
id = AutoField()
info = CharField(null=True)
@@ -77,8 +87,11 @@ class Victory(Model):
def create_tables():
models = [Season, Episode, Player, Drink, Enemy, Death, Victory, Episode.players.get_through_model()]
models = [Season, Episode, Player, Drink, Enemy, Death, Victory, Penalty, Episode.players.get_through_model()]
for model in models:
model.create_table()
def drop_tables():
models = [Season, Episode, Player, Drink, Enemy, Death, Victory, Penalty, Episode.players.get_through_model()]
connection.drop_tables(models)