volume_schema.py

import sys

LATEST_SCHEMA_VERSION = 3  # type: int


def migrate_0_to_1(data: dict) -> dict:
    data["schema_version"] = 1
    return data


def migrate_1_to_2(data: dict) -> dict:
    data["schema_version"] = 2
    data.setdefault("fs_type", "ext4")
    return data


def migrate_2_to_3(data: dict) -> dict:
    data["schema_version"] = 3
    deleted_at = data.get("deleted_at", None)
    if deleted_at is not None:
        gc_at = deleted_at + 7 * 24 * 60 * 60
        data["gc_at"] = gc_at
    return data


def migrate_to(data: dict, version: int) -> dict:
    current = data.get("schema_version", 0)
    if current > version:
        raise Exception(
            f"Current schema version ({current}) is newer than target schema version ({version})"
        )
    for i in range(current, version):
        migrate_fn = getattr(sys.modules[__name__], f"migrate_{i}_to_{i + 1}")
        data = migrate_fn(data)
    return data