মাইগ্রেশন ও স্কিমা ইভোলিউশন

import copy

# ---------- প্রতিটি মাইগ্রেশনের apply ও rollback ফাংশন ----------
def migration_001_add_email_apply(schema):
    new_schema = copy.deepcopy(schema)
    new_schema["columns"]["email"] = "string"
    return new_schema

def migration_001_add_email_rollback(schema):
    new_schema = copy.deepcopy(schema)
    del new_schema["columns"]["email"]
    return new_schema


def migration_002_rename_name_apply(schema):
    new_schema = copy.deepcopy(schema)
    new_schema["columns"]["full_name"] = new_schema["columns"].pop("name")
    return new_schema

def migration_002_rename_name_rollback(schema):
    new_schema = copy.deepcopy(schema)
    new_schema["columns"]["name"] = new_schema["columns"].pop("full_name")
    return new_schema


def migration_003_add_created_at_apply(schema):
    new_schema = copy.deepcopy(schema)
    new_schema["columns"]["created_at"] = "datetime"
    return new_schema

def migration_003_add_created_at_rollback(schema):
    new_schema = copy.deepcopy(schema)
    del new_schema["columns"]["created_at"]
    return new_schema


# ---------- ধারাবাহিক মাইগ্রেশন তালিকা ----------
MIGRATIONS = [
    ("0001_add_email", migration_001_add_email_apply, migration_001_add_email_rollback),
    ("0002_rename_name_to_full_name", migration_002_rename_name_apply, migration_002_rename_name_rollback),
    ("0003_add_created_at", migration_003_add_created_at_apply, migration_003_add_created_at_rollback),
]

schema = {"table": "users", "columns": {"id": "integer", "name": "string"}}
current_version = 0
history = [copy.deepcopy(schema)]   # history[i] = i-নম্বর মাইগ্রেশন প্রয়োগের পরের স্কিমা (history[0] = শুরুর স্কিমা)

print(f"ভার্সন {current_version} (শুরু): {schema}")

for name, apply_fn, rollback_fn in MIGRATIONS:
    schema = apply_fn(schema)
    current_version += 1
    history.append(copy.deepcopy(schema))
    print(f"ভার্সন {current_version} ({name}): {schema}")

# ---------- শেষ মাইগ্রেশন রোলব্যাক করা ----------
last_name, last_apply, last_rollback = MIGRATIONS[-1]
schema_before_rollback = schema
prior_recorded_schema = history[-2]      # রোলব্যাকের আগেই রেকর্ড করা "আগের" স্কিমা

rolled_back_schema = last_rollback(schema_before_rollback)
current_version -= 1

print(f"\n'{last_name}' রোলব্যাক করার আগের স্কিমা: {schema_before_rollback}")
print(f"'{last_name}' রোলব্যাক করার পরের স্কিমা:  {rolled_back_schema}")
print(f"পূর্বে রেকর্ড করা স্কিমা (ভার্সন {current_version}):     {prior_recorded_schema}")

matches = rolled_back_schema == prior_recorded_schema
print(f"\nরোলব্যাক করা স্কিমা পূর্বের রেকর্ড করা স্কিমার সাথে হুবহু মিলছে: {matches}")
assert matches

print(f"রোলব্যাকের পর current_version: {current_version}")