Add logo to status bar.

dsst/dsst_server/func_read.py

@@ -0,0 +1,47 @@
+from dsst_server.data_access import sql, sql_func, mapping
+from common import models
+from playhouse import shortcuts
+
+
+class ReadFunctions:
+    @staticmethod
+    def load_db_meta(*_):
+        return sql.db.database
+
+    @staticmethod
+    def load_seasons(*_):
+        return [mapping.db_to_season(season) for season in sql.Season.select()]
+
+    @staticmethod
+    def load_seasons_all(*_):
+        return [shortcuts.model_to_dict(season, backrefs=True, max_depth=2) for season in sql.Season.select()]
+
+    @staticmethod
+    def load_episodes(season_id, *_):
+        if not season_id:
+            raise Exception('Exception: Missing argument (season_id)')
+        return [mapping.db_to_episode(ep) for ep in sql.Season.get(sql.Season.id == season_id).episodes]
+
+    @staticmethod
+    def load_players(*_):
+        return [mapping.db_to_player(player) for player in sql.Player.select()]
+
+    @staticmethod
+    def load_drinks(*_):
+        return [mapping.db_to_drink(drink) for drink in sql.Drink.select()]
+
+    @staticmethod
+    def load_season_stats(season_id, *_):
+        season = sql.Season.get(sql.Season.id == season_id)
+        players = sql_func.players_for_season(season_id)
+        model = models.SeasonStats()
+        model.player_kd = [(player.name,
+                            sql_func.get_player_victories_for_season(season_id, player.id),
+                            sql_func.get_player_deaths_for_season(season_id, player.id))
+                           for player in players]
+        model.enemies = [(enemy.name,
+                          sql_func.enemy_attempts(enemy.id),
+                          sql.Victory.select().where(sql.Victory.enemy == enemy.id).exists(),
+                          enemy.boss)
+                         for enemy in season.enemies]
+        return model