diff --git a/GhostBallBot/database/models.py b/GhostBallBot/database/models.py index 284085a..390e693 100644 --- a/GhostBallBot/database/models.py +++ b/GhostBallBot/database/models.py @@ -75,8 +75,9 @@ class GuessModel(BaseModel): difference = IntegerField(null=True) date_guessed = DateTimeField(null=True) + def create_models(): """Create database tables""" with database: - database.create_tables([GameModel, GuessModel]) + database.create_tables([GameModel, GuessModel, PlayerModel()])