mirror of https://github.com/Aidaho12/haproxy-wi
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from playhouse.migrate import *
|
|
from app.modules.db.db_model import connect
|
|
|
|
migrator = connect(get_migrator=1)
|
|
|
|
|
|
def up():
|
|
"""Apply the migration."""
|
|
# This migration renames columns in the backups table
|
|
try:
|
|
migrate(
|
|
migrator.rename_column('backups', 'cred', 'cred_id'),
|
|
migrator.rename_column('backups', 'backup_type', 'type'),
|
|
)
|
|
except Exception as e:
|
|
if e.args[0] == 'no such column: "cred"' or str(e) == '(1060, no such column: "cred")':
|
|
print("Columns already renamed")
|
|
elif e.args[0] == "'bool' object has no attribute 'sql'":
|
|
print("Columns already renamed")
|
|
else:
|
|
raise e
|
|
|
|
|
|
def down():
|
|
"""Roll back the migration."""
|
|
# This migration renames columns back to their original names
|
|
try:
|
|
migrate(
|
|
migrator.rename_column('backups', 'cred_id', 'cred'),
|
|
migrator.rename_column('backups', 'type', 'backup_type'),
|
|
)
|
|
except Exception as e:
|
|
print(f"Error rolling back migration: {str(e)}")
|
|
raise e
|