Add Basic Database functions.

dsst/dsst_sql/core.py

@@ -0,0 +1,56 @@
+from dsst_sql import models, sql
+
+
+class DSSTCore:
+    def __init__(self):
+        # Set DB Connection
+        sql.create_tables()
+
+    @staticmethod
+    def insert_player(model: models.Player):
+        return sql.Player.create(name=model.name, hex_id=model.hex_id)
+
+    @staticmethod
+    def insert_enemy(enemy: models.Enemy):
+        return sql.Enemy.create(name=enemy.name)
+
+    @staticmethod
+    def insert_drink(model: models.Drink):
+        return sql.Drink.create(name=model.name, vol=model.vol)
+
+    @staticmethod
+    def insert_season(season: models.Season):
+        return sql.Season.create(number=season.number, game_name=season.game_name, start_date=season.start_date,
+                                 end_date=season.end_date)
+
+    @staticmethod
+    def insert_death(death: models.Death):
+        return sql.Death.create(info=death.info, player=death.player.id, enemy=death.enemy.id, penalty=death.penalty.id)
+
+    @staticmethod
+    def insert_victory(victory: models.Victory):
+        return sql.Death.create(info=victory.info, player=victory.player.id, enemy=victory.enemy.id)
+
+    @staticmethod
+    def insert_episode(season_id: int, episode: models.Episode):
+        with sql.connection.atomic():
+            # Insert Episode Row
+            new_ep = sql.Episode.create(seq_number=episode.seq_number, number=episode.number, date=episode.date,
+                                        season=season_id)
+            # Insert participating players
+            for player in episode.players:
+                sql.EpisodePlayer.insert(episode=new_ep.id, player=player.id)
+            # Insert deaths in this episode
+            if episode.deaths:
+                for death in episode.deaths:
+                    new_death = DSSTCore.insert_death(death)
+                    sql.EpisodeDeath.create(death=new_death.id, episode=new_ep.id)
+            # Insert victories in this episode
+            if episode.victories:
+                for victory in episode.victories:
+                    new_vic = DSSTCore.insert_victory(victory)
+                    sql.EpisodeVictory.create(victory=new_vic.id, episode=new_ep.id)
+
+
+if __name__ == '__main__':
+    core = DSSTCore()

dsst/dsst_sql/sql.py

@@ -0,0 +1,108 @@
+from peewee import *
+
+connection = MySQLDatabase('dsst', user='dsst', password='dsst')
+
+
+class Season(Model):
+    id = AutoField()
+    number = IntegerField()
+    game_name = CharField()
+    start_date = DateField(null=True)
+    end_date = DateField(null=True)
+
+    class Meta:
+        database = connection
+
+
+class Episode(Model):
+    id = AutoField()
+    seq_number = IntegerField()
+    number = CharField()
+    date = DateTimeField(null=True)
+    season = ForeignKeyField(Season, backref='episodes')
+
+    class Meta:
+        database = connection
+
+
+class Player(Model):
+    id = AutoField()
+    name = CharField()
+    hex_id = CharField(null=True)
+
+    class Meta:
+        database = connection
+
+
+class Drink(Model):
+    id = AutoField()
+    name = CharField()
+    vol = DecimalField()
+
+    class Meta:
+        database = connection
+
+
+class Enemy(Model):
+    id = AutoField()
+    name = CharField()
+
+    class Meta:
+        database = connection
+
+
+class Death(Model):
+    id = AutoField()
+    info = CharField(null=True)
+    player = ForeignKeyField(Player)
+    enemy = ForeignKeyField(Enemy)
+    penalty = ForeignKeyField(Drink)
+
+    class Meta:
+        database = connection
+
+
+class Victory(Model):
+    id = AutoField()
+    info = CharField(null=True)
+    player = ForeignKeyField(Player)
+    enemy = ForeignKeyField(Enemy)
+
+    class Meta:
+        database = connection
+
+
+class EpisodePlayer(Model):
+    id = AutoField()
+    episode = ForeignKeyField(Episode)
+    player = ForeignKeyField(Player)
+
+    class Meta:
+        database = connection
+
+
+class EpisodeDeath(Model):
+    id = AutoField()
+    episode = ForeignKeyField(Episode)
+    death = ForeignKeyField(Death)
+
+    class Meta:
+        database = connection
+
+
+class EpisodeVictory(Model):
+    id = AutoField()
+    episode = ForeignKeyField(Episode)
+    victory = ForeignKeyField(Victory)
+
+    class Meta:
+        database = connection
+
+
+def create_tables():
+    models = [Season, Episode, Player, Drink, Enemy, Death, Victory, EpisodePlayer, EpisodeDeath,
+              EpisodeVictory]
+    for model in models:
+        model.create_table()
+
+