generate the uuid on the model

This commit is contained in:
c0de 2022-12-09 21:34:32 -06:00
parent ad70870d8e
commit 7577e3afee
2 changed files with 4 additions and 5 deletions

View File

@ -9,6 +9,7 @@
""" """
import os import os
import uuid
import datetime import datetime
from peewee import ( from peewee import (
@ -55,7 +56,7 @@ class PlayerModel(BaseModel):
class GameModel(BaseModel): class GameModel(BaseModel):
"""Games that are ran""" """Games that are ran"""
game_id = UUIDField(primary_key=True) game_id = UUIDField(primary_key=True, default=uuid.uuid4)
server_id = IntegerField() server_id = IntegerField()
pitch_value = IntegerField(null=True) pitch_value = IntegerField(null=True)
@ -66,7 +67,7 @@ class GameModel(BaseModel):
class GuessModel(BaseModel): class GuessModel(BaseModel):
"""Guesses for a particular game""" """Guesses for a particular game"""
guess_id = UUIDField(primary_key=True) guess_id = UUIDField(primary_key=True, default=uuid.uuid4)
player = ForeignKeyField(PlayerModel, backref="guesses") player = ForeignKeyField(PlayerModel, backref="guesses")
game = ForeignKeyField(GameModel, backref="guesses") game = ForeignKeyField(GameModel, backref="guesses")

View File

@ -4,8 +4,6 @@
# pylint: disable=missing-module-docstring # pylint: disable=missing-module-docstring
import uuid
from database.models import GameModel as Game from database.models import GameModel as Game
from game.base import BaseGameManager from game.base import BaseGameManager
@ -28,6 +26,6 @@ class NewGameManager(BaseGameManager):
self.is_running = True self.is_running = True
# game.pitch_value is unknown at the start of the game # game.pitch_value is unknown at the start of the game
self.game = Game.create(game_id=uuid.uuid4(), server_id=self.message.channel.id) self.game = Game.create(server_id=self.message.channel.id)
await self.message.channel.send("Send me your guesses with !guess <number>") await self.message.channel.send("Send me your guesses with !guess <number>")