You've already forked haproxy-wi
mirror of
https://github.com/roxy-wi/roxy-wi.git
synced 2025-12-15 11:54:05 +08:00
b1bec4ec06
Introduce migration management tools using Peewee's migrate module. Add scripts for migration creation, application, rollback, and listing. Include initial batch of database migrations for schema and data changes.
32 lines
1.1 KiB
Python
32 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 |