sh-edraft.de/sh_discord_bot
4
1
Fork 1
Code Issues 21 Pull Requests Actions Packages Projects 3 Releases 27 Wiki Activity

Added logic to handle down and upgrades #428

Browse Source
This commit is contained in:
Sven Heidemann 2023-11-10 13:19:23 +01:00
parent 1f2fbc362f
commit 4628f31993
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
bot/tools/migration_to_sql/mock/migration_service.py
View File

@ -27,10 +27,17 @@ class MigrationService:
migration_id = migration.__name__ migration_id = migration.__name__
try: try:
migration_as_service: MigrationABC = self._services.get_service(migration) migration_as_service: MigrationABC = self._services.get_service(migration)
self._db.set_migration(migration_as_service.name) # save upgrade scripts
self._db.set_migration(migration_as_service.name, True)
migration_as_service.upgrade() migration_as_service.upgrade()
self._cursor.execute(MigrationHistory(migration_id).insert_string) self._cursor.execute(MigrationHistory(migration_id).insert_string)
self._db.save_changes() self._db.save_changes()
# save downgrade scripts
self._db.set_migration(migration_as_service.name)
migration_as_service.downgrade()
self._cursor.execute(MigrationHistory(migration_id).insert_string)
self._db.save_changes()
except Exception as e: except Exception as e:
Console.error(e) Console.error(e)

4
bot/tools/migration_to_sql/mock/mock_db_context.py
View File

@ -21,8 +21,8 @@ class MockDBContext(DatabaseContextABC):
cursor.execute = self.execute cursor.execute = self.execute
return cursor return cursor
def set_migration(self, name: str): def set_migration(self, name: str, upgrade: bool = False):
self._migration_name = name.split("_")[1].replace("Migration", "") self._migration_name = f'{name.split("_")[1].replace("Migration", "")}_{"up" if upgrade else "down"}'
self._migration_version = name.split("_")[0] self._migration_version = name.split("_")[0]
self._migration_script = "" self._migration_script = ""