BaseballBot/GhostBallBot/database/models.py

#!/usr/bin/env python3
# Copyright 2022 - c0de <c0de@c0de.dev>
# Licensed under the MIT License (https://opensource.org/licenses/MIT)

import os
import datetime

from peewee import *

# User can provide path to database, or it will be put next to main.py
DATABASE = os.environ.get('database_path', os.getcwd() + '/ghostball.db')
database = SqliteDatabase(DATABASE)

class BaseModel(Model):
    """All of our models will inherit this class
       and use the same database connection"""

    class Meta:
        database = database

class GameModel(BaseModel):

    game_id = UUIDField(primary_key=True)
    server_id = IntegerField()

    pitch_value = IntegerField(null=True)
    date_created = DateTimeField(default=datetime.datetime.now)
    date_ended = DateTimeField(null=True)

class GuessModel(BaseModel):

    player_id = IntegerField(primary_key=True)
    game_id   = ForeignKeyField(GameModel, backref="guesses")

    player_name = CharField()
    guess = IntegerField()
    difference = IntegerField(null=True)
    date_guessed = DateTimeField(null=True)

    # TODO: Add unique constraint for player_id and game_id
    # ie: one guess per player allowed per game


def create_models():
    with database:
        database.create_tables([GameModel, GuessModel])