HOME

import json

from playhouse.shortcuts import model_to_dict

FALLBACK_LICENSE_FILE = "/var/imunify360/license_old.json"


def migrate(migrator, database, fake=False, **kwargs):
    if fake:
        return
    LicenseModel = migrator.orm["license"]
    lic, _ = LicenseModel.get_or_create(
        defaults={
            "status": True,
            "expiration": 0,
        }
    )
    with open(FALLBACK_LICENSE_FILE, "w") as f:
        json.dump(model_to_dict(lic), f)
    migrator.remove_model(LicenseModel)


def rollback(migrator, database, fake=False, **kwargs):
    pass