Fixed auto-role add #151

kdb-bot/src/bot_data/migration/auto_role_fix1_migration.py

@@ -0,0 +1,29 @@
+from bot_core.logging.database_logger import DatabaseLogger
+from bot_data.abc.migration_abc import MigrationABC
+from bot_data.db_context import DBContext
+
+
+class AutoRoleFix1Migration(MigrationABC):
+    name = '0.3.0_AutoRoleMigration'
+
+    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"""
+            ALTER TABLE AutoRoles ADD DiscordChannelId BIGINT NOT NULL AFTER ServerId;
+            """)
+        )
+
+    def downgrade(self):
+        self._cursor.execute(
+            str(f"""
+            ALTER TABLE AutoRoles DROP COLUMN DiscordChannelId;
+            """)
+        )

kdb-bot/src/bot_data/model/auto_role.py

@@ -6,9 +6,10 @@ from cpl_core.database import TableABC
 
 class AutoRole(TableABC):
 
-    def __init__(self, server_id: int, dc_message_id: int, created_at: datetime=None, modified_at: datetime=None, id=0):
+    def __init__(self, server_id: int, dc_channel_id: int, dc_message_id: int, created_at: datetime=None, modified_at: datetime=None, id=0):
         self._auto_role_id = id
         self._server_id = server_id
+        self._discord_channel_id = dc_channel_id
         self._discord_message_id = dc_message_id
         
         TableABC.__init__(self)
@@ -23,6 +24,10 @@ class AutoRole(TableABC):
     def server_id(self) -> int:
         return self._server_id
 
+    @property
+    def discord_channel_id(self) -> int:
+        return self._discord_channel_id
+
     @property
     def discord_message_id(self) -> int:
         return self._discord_message_id
@@ -58,9 +63,10 @@ class AutoRole(TableABC):
     def insert_string(self) -> str:
         return str(f"""
             INSERT INTO `AutoRoles` (
-                `ServerId`, `DiscordMessageId`, `CreatedAt`, `LastModifiedAt`
+                `ServerId`, `DiscordChannelId`, `DiscordMessageId`, `CreatedAt`, `LastModifiedAt`
             ) VALUES (
                 {self._server_id},
+                {self._discord_channel_id},
                 {self._discord_message_id},
                 '{self._created_at}',
                 '{self._modified_at}'
@@ -72,6 +78,7 @@ class AutoRole(TableABC):
         return str(f"""
             UPDATE `AutoRoles`
             SET `ServerId` = {self._server_id},
+            `DiscordChannelId` = {self._discord_channel_id},
             `DiscordMessageId` = {self._discord_message_id},
             `LastModifiedAt` = '{self._modified_at}'
             WHERE `AutoRoleId` = {self._auto_role_id};

kdb-bot/src/bot_data/service/auto_role_repository_service.py

@@ -27,6 +27,7 @@ class AutoRoleRepositoryService(AutoRoleRepositoryABC):
                 result[2],
                 result[3],
                 result[4],
+                result[5],
                 id=result[0]
             ))
 
@@ -40,6 +41,7 @@ class AutoRoleRepositoryService(AutoRoleRepositoryABC):
             result[2],
             result[3],
             result[4],
+            result[5],
             id=result[0]
         )
 
@@ -56,6 +58,7 @@ class AutoRoleRepositoryService(AutoRoleRepositoryABC):
             result[2],
             result[3],
             result[4],
+            result[5],
             id=result[0]
         )
 
@@ -69,6 +72,7 @@ class AutoRoleRepositoryService(AutoRoleRepositoryABC):
                 result[2],
                 result[3],
                 result[4],
+                result[5],
                 id=result[0]
             ))
 
@@ -82,6 +86,7 @@ class AutoRoleRepositoryService(AutoRoleRepositoryABC):
             result[2],
             result[3],
             result[4],
+            result[5],
             id=result[0]
         )
 
@@ -98,6 +103,7 @@ class AutoRoleRepositoryService(AutoRoleRepositoryABC):
             result[2],
             result[3],
             result[4],
+            result[5],
             id=result[0]
         )