From b5080d8c04c4f41acb2351e2f8db863a74726763 Mon Sep 17 00:00:00 2001
From: Sven Heidemann <sven.heidemann@sh-edraft.de>
Date: Wed, 9 Nov 2022 18:22:27 +0100
Subject: [PATCH] Added stats migration #46

---
 .../src/bot/startup_migration_extension.py    |  2 ++
 .../src/bot_data/migration/stats_migration.py | 36 +++++++++++++++++++
 2 files changed, 38 insertions(+)
 create mode 100644 kdb-bot/src/bot_data/migration/stats_migration.py

diff --git a/kdb-bot/src/bot/startup_migration_extension.py b/kdb-bot/src/bot/startup_migration_extension.py
index a5d75b6dca..724db39a71 100644
--- a/kdb-bot/src/bot/startup_migration_extension.py
+++ b/kdb-bot/src/bot/startup_migration_extension.py
@@ -8,6 +8,7 @@ from bot_data.migration.api_migration import ApiMigration
 from bot_data.migration.auto_role_migration import AutoRoleMigration
 from bot_data.migration.initial_migration import InitialMigration
 from bot_data.migration.level_migration import LevelMigration
+from bot_data.migration.stats_migration import StatsMigration
 from bot_data.service.migration_service import MigrationService
 
 
@@ -25,3 +26,4 @@ class StartupMigrationExtension(StartupExtensionABC):
         services.add_transient(MigrationABC, AutoRoleMigration)  # 03.10.2022 #54 - 0.2.2
         services.add_transient(MigrationABC, ApiMigration)  # 15.10.2022 #70 - 0.3.0
         services.add_transient(MigrationABC, LevelMigration)  # 06.11.2022 #25 - 0.3.0
+        services.add_transient(MigrationABC, StatsMigration)  # 09.11.2022 #46 - 0.3.0
diff --git a/kdb-bot/src/bot_data/migration/stats_migration.py b/kdb-bot/src/bot_data/migration/stats_migration.py
new file mode 100644
index 0000000000..92ba17abd3
--- /dev/null
+++ b/kdb-bot/src/bot_data/migration/stats_migration.py
@@ -0,0 +1,36 @@
+from bot_core.logging.database_logger import DatabaseLogger
+from bot_data.abc.migration_abc import MigrationABC
+from bot_data.db_context import DBContext
+
+
+class StatsMigration(MigrationABC):
+    name = '0.3_StatsMigration'
+
+    def __init__(self, logger: DatabaseLogger, db: DBContext):
+        MigrationABC.__init__(self)
+        self._logger = logger
+        self._db = db
+        self._cursor = db.cursor
+
+    def upgrade(self):
+        self._logger.debug(__name__, 'Running upgrade')
+
+        self._cursor.execute(
+            str(f"""
+            CREATE TABLE IF NOT EXISTS `Statistics` (
+                `Id` BIGINT NOT NULL AUTO_INCREMENT,
+                `Name` VARCHAR(255) NOT NULL,
+                `Description` VARCHAR(255) NOT NULL,
+                `Code` LONGTEXT NOT NULL,
+                `ServerId` BIGINT,
+                `CreatedAt` DATETIME(6),
+                `LastModifiedAt` DATETIME(6),
+                PRIMARY KEY(`Id`),
+                FOREIGN KEY (`ServerId`) REFERENCES `Servers`(`ServerId`)
+            );
+            """)
+        )
+
+    def downgrade(self):
+        self._cursor.execute('DROP TABLE `Statistics`;')
+