rawfile-localpv/volume_schema.py
2021-02-26 16:10:10 +03:30

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