forked from c0de/BaseballBot
85 lines
2.1 KiB
Python
85 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
# Copyright 2022 - c0de <c0de@c0de.dev>
|
|
# Licensed under the MIT License (https://opensource.org/licenses/MIT)
|
|
|
|
# pylint: disable=too-few-public-methods
|
|
|
|
"""
|
|
An implementation of a SQLite database
|
|
"""
|
|
|
|
import os
|
|
import uuid
|
|
import datetime
|
|
|
|
from peewee import (
|
|
SqliteDatabase,
|
|
Model,
|
|
UUIDField,
|
|
IntegerField,
|
|
CharField,
|
|
DateTimeField,
|
|
ForeignKeyField,
|
|
)
|
|
|
|
# User can provide path to database, or it will be put next to main.py
|
|
DATABASE = os.environ.get("database_path", os.getcwd() + "/baseball.db")
|
|
database = SqliteDatabase(DATABASE, pragmas={"foreign_keys": 1})
|
|
|
|
|
|
class BaseModel(Model):
|
|
"""All of our models will inherit this class
|
|
and use the same database connection"""
|
|
|
|
class Meta:
|
|
"""meta"""
|
|
|
|
database = database
|
|
|
|
|
|
class PlayerModel(BaseModel):
|
|
"""Need to keep track of total player score"""
|
|
|
|
player_id = IntegerField(primary_key=True)
|
|
player_name = CharField()
|
|
|
|
total_points = IntegerField(default=0)
|
|
date_joined = DateTimeField(default=datetime.datetime.now)
|
|
last_update = DateTimeField(null=True)
|
|
|
|
def save(self, *args, **kwargs):
|
|
"""Should set the last update everytime the record is saved"""
|
|
self.last_update = datetime.datetime.now()
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class GameModel(BaseModel):
|
|
"""Games that are ran"""
|
|
|
|
game_id = UUIDField(primary_key=True, default=uuid.uuid4)
|
|
server_id = IntegerField()
|
|
|
|
pitch_value = IntegerField(null=True)
|
|
date_created = DateTimeField(default=datetime.datetime.now)
|
|
date_ended = DateTimeField(null=True)
|
|
|
|
|
|
class GuessModel(BaseModel):
|
|
"""Guesses for a particular game"""
|
|
|
|
guess_id = UUIDField(primary_key=True, default=uuid.uuid4)
|
|
|
|
player = ForeignKeyField(PlayerModel, backref="guesses")
|
|
game = ForeignKeyField(GameModel, backref="guesses")
|
|
|
|
guess = IntegerField(default=0)
|
|
difference = IntegerField(null=True)
|
|
date_guessed = DateTimeField(null=True)
|
|
|
|
|
|
def create_models():
|
|
"""Create database tables"""
|
|
|
|
with database:
|
|
database.create_tables([GameModel, GuessModel, PlayerModel])
|