Viewing File: /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/migrations/082_add_manual_flag.py

import peewee as pw


def migrate(migrator, database, fake=False, **kwargs):
    IPList = migrator.orm["iplist"]
    migrator.add_fields(
        IPList, manual=pw.BooleanField(null=False, default=True)
    )


def rollback(migrator, database, fake=False, **kwargs):
    IPList = migrator.orm["iplist"]
    migrator.remove_fields(IPList, "manual")