105 lines
2.3 KiB
Python
105 lines
2.3 KiB
Python
"""
|
|
This module contains the ORM class definitions for peewee.
|
|
To access the database import this module an run queries on the classes
|
|
Example:
|
|
from sql import Episode
|
|
query = Episode.select().where(Episode.name == 'MyName')
|
|
"""
|
|
import sys
|
|
import datetime
|
|
|
|
try:
|
|
from peewee import *
|
|
except ImportError:
|
|
print('peewee package not installed')
|
|
sys.exit(1)
|
|
|
|
db = MySQLDatabase(None)
|
|
|
|
|
|
class Season(Model):
|
|
id = AutoField()
|
|
number = IntegerField()
|
|
game_name = CharField()
|
|
start_date = DateField(null=True)
|
|
end_date = DateField(null=True)
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
class Player(Model):
|
|
id = AutoField()
|
|
name = CharField()
|
|
hex_id = CharField(null=True)
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
class Episode(Model):
|
|
id = AutoField()
|
|
seq_number = IntegerField()
|
|
number = CharField()
|
|
name = CharField(null=True)
|
|
date = DateField(null=True)
|
|
season = ForeignKeyField(Season, backref='episodes', on_delete='CASCADE')
|
|
players = ManyToManyField(Player, backref='episodes')
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
class Drink(Model):
|
|
id = AutoField()
|
|
name = CharField()
|
|
vol = DecimalField()
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
class Enemy(Model):
|
|
id = AutoField()
|
|
name = CharField()
|
|
boss = BooleanField()
|
|
season = ForeignKeyField(Season, backref='enemies', on_delete='CASCADE')
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
class Death(Model):
|
|
id = AutoField()
|
|
info = CharField(null=True)
|
|
time = TimeField(default=datetime.time(0, 0))
|
|
player = ForeignKeyField(Player)
|
|
enemy = ForeignKeyField(Enemy)
|
|
episode = ForeignKeyField(Episode, backref='deaths', on_delete='CASCADE')
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
class Penalty(Model):
|
|
id = AutoField()
|
|
size = DecimalField()
|
|
drink = ForeignKeyField(Drink)
|
|
player = ForeignKeyField(Player, backref='penalties', on_delete='CASCADE')
|
|
death = ForeignKeyField(Death, backref='penalties', on_delete='CASCADE')
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
|
|
class Victory(Model):
|
|
id = AutoField()
|
|
info = CharField(null=True)
|
|
time = TimeField(default=datetime.time(0, 0))
|
|
player = ForeignKeyField(Player)
|
|
enemy = ForeignKeyField(Enemy)
|
|
episode = ForeignKeyField(Episode, backref='victories', on_delete='CASCADE')
|
|
|
|
class Meta:
|
|
database = db
|