36 lines
930 B
Python
36 lines
930 B
Python
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
|