"""
Altering MalwareScan.type in order to add ability to support
'malware-response' scan type
"""
import peewee as pw


def migrate(migrator, database, fake=False, **kwargs):
    MalwareScan = migrator.orm["malware_scans"]

    MalwareScan.update(type="realtime").where(
        MalwareScan.type == "inotify"
    ).execute()

    migrator.change_fields(
        MalwareScan, path=pw.CharField(null=True, default="")
    )
    migrator.change_fields(
        MalwareScan,
        type=pw.CharField(
            null=False,
            constraints=[
                pw.Check(
                    "type in ('on-demand', 'realtime', 'malware-response')"
                )
            ],
        ),
    )


def rollback(migrator, database, fake=False, **kwargs):
    """Write your rollback migrations here."""
    pass